shell: init
This commit is contained in:
tv
parent
077a6a6991
commit
ed71d56883
210
shell.nix
Normal file
210
shell.nix
Normal file
@ -0,0 +1,210 @@
|
|||||||
|
{ nixpkgs ? import <nixpkgs> {} }: let
|
||||||
|
|
||||||
|
inherit (nixpkgs) lib pkgs;
|
||||||
|
slib = import ./lib;
|
||||||
|
spkgs = {
|
||||||
|
populate = pkgs.callPackage ./krebs/5pkgs/simple/populate {};
|
||||||
|
whatsupnix = pkgs.callPackage ./krebs/5pkgs/simple/whatsupnix {};
|
||||||
|
};
|
||||||
|
|
||||||
|
# usage: deploy system=SYSTEM [target=TARGET]
|
||||||
|
cmds.deploy = pkgs.writeScript "cmds.deploy" /* sh */ ''
|
||||||
|
#! ${pkgs.dash}/bin/dash
|
||||||
|
set -efu
|
||||||
|
|
||||||
|
command=deploy
|
||||||
|
. ${init.args}
|
||||||
|
\test -n "''${target-}" || target=$system
|
||||||
|
. ${init.env}
|
||||||
|
|
||||||
|
exec ${utils.deploy}
|
||||||
|
'';
|
||||||
|
|
||||||
|
# usage: test system=SYSTEM target=TARGET
|
||||||
|
cmds.test = pkgs.writeScript "cmds.test" /* sh */ ''
|
||||||
|
#! ${pkgs.dash}/bin/dash
|
||||||
|
set -efu
|
||||||
|
|
||||||
|
command=test
|
||||||
|
. ${init.args}
|
||||||
|
. ${init.env}
|
||||||
|
|
||||||
|
export dummy_secrets=true
|
||||||
|
exec ${utils.build} config.system.build.toplevel
|
||||||
|
'';
|
||||||
|
|
||||||
|
init.args = pkgs.writeText "init.args" /* sh */ ''
|
||||||
|
fail=
|
||||||
|
for arg; do
|
||||||
|
case $arg in
|
||||||
|
system=*) system=''${arg#*=};;
|
||||||
|
target=*) target=''${arg#*=};;
|
||||||
|
*) echo "$command: bad argument: $arg" >&2; fail=1
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
if \test -n "$fail"; then
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
unset fail
|
||||||
|
'';
|
||||||
|
|
||||||
|
init.env = pkgs.writeText "init.env" /* sh */ ''
|
||||||
|
config=''${config-$LOGNAME/1systems/$system.nix}
|
||||||
|
|
||||||
|
export config
|
||||||
|
export system
|
||||||
|
export target
|
||||||
|
|
||||||
|
export target_object="$(${init.env.parsetarget} $target)"
|
||||||
|
export target_user="$(echo $target_object | ${pkgs.jq}/bin/jq -r .user)"
|
||||||
|
export target_host="$(echo $target_object | ${pkgs.jq}/bin/jq -r .host)"
|
||||||
|
export target_port="$(echo $target_object | ${pkgs.jq}/bin/jq -r .port)"
|
||||||
|
export target_path="$(echo $target_object | ${pkgs.jq}/bin/jq -r .path)"
|
||||||
|
export target_local="$(echo $target_object | ${pkgs.jq}/bin/jq -r .local)"
|
||||||
|
|
||||||
|
export qtarget="$target_user@$target_host:$target_port$target_path"
|
||||||
|
|
||||||
|
${init.env.populate}
|
||||||
|
|
||||||
|
if \test "$target_local" != true && \test "''${DISABLE_PROXY-}" != 1; then
|
||||||
|
exec ${init.env.proxy} "$command" "$@"
|
||||||
|
fi
|
||||||
|
'' // {
|
||||||
|
parsetarget = pkgs.writeScript "init.env.parsetarget" /* sh */ ''
|
||||||
|
#! ${pkgs.dash}/bin/dash
|
||||||
|
set -efu
|
||||||
|
exec ${pkgs.jq}/bin/jq \
|
||||||
|
-enr \
|
||||||
|
--arg target "$1" \
|
||||||
|
-f ${init.env.parsetarget.jq}
|
||||||
|
'' // {
|
||||||
|
jq = pkgs.writeText "init.env.parsetarget.jq" ''
|
||||||
|
def when(c; f): if c then f else . end;
|
||||||
|
def capturesDef(i; v): .captures[i].string | when(. == null; v);
|
||||||
|
$target | match("^(?:([^@]+)@)?([^:/]+)?(?::([0-9]+))?(/.*)?$") | {
|
||||||
|
user: capturesDef(0; "root"),
|
||||||
|
host: capturesDef(1; env.system),
|
||||||
|
port: capturesDef(2; "22"),
|
||||||
|
path: capturesDef(3; "/var/src"),
|
||||||
|
} | . + {
|
||||||
|
local: (.user == env.LOGNAME and .host == env.HOSTNAME),
|
||||||
|
}
|
||||||
|
'';
|
||||||
|
};
|
||||||
|
populate = pkgs.writeScript "init.env.populate" /* sh */ ''
|
||||||
|
#! ${pkgs.dash}/bin/dash
|
||||||
|
set -efu
|
||||||
|
if \test "''${DISABLE_POPULATE-}" = 1; then
|
||||||
|
exit
|
||||||
|
fi
|
||||||
|
set -x
|
||||||
|
${pkgs.nix}/bin/nix-instantiate \
|
||||||
|
--eval \
|
||||||
|
--json \
|
||||||
|
--readonly-mode \
|
||||||
|
--show-trace \
|
||||||
|
--strict \
|
||||||
|
-I nixos-config="$config" \
|
||||||
|
-E 'with import <stockholm>; config.krebs.build.source' \
|
||||||
|
|
|
||||||
|
${spkgs.populate}/bin/populate "$qtarget" >&2
|
||||||
|
'';
|
||||||
|
proxy = pkgs.writeScript "init.env.proxy" /* sh */ ''
|
||||||
|
#! ${pkgs.dash}/bin/dash
|
||||||
|
set -efu
|
||||||
|
q() {
|
||||||
|
${pkgs.jq}/bin/jq -nr --arg x "$*" '$x | @sh "\(.)"'
|
||||||
|
}
|
||||||
|
exec ${pkgs.openssh}/bin/ssh \
|
||||||
|
"$target_user@$target_host" -p "$target_port" \
|
||||||
|
cd "$target_path/stockholm" \; \
|
||||||
|
NIX_PATH=$(q "$target_path") \
|
||||||
|
STOCKHOLM_VERSION=$STOCKHOLM_VERSION \
|
||||||
|
nix-shell \
|
||||||
|
--command $(q \
|
||||||
|
config=$config \
|
||||||
|
system=$system \
|
||||||
|
target=$target \
|
||||||
|
DISABLE_POPULATE=1 \
|
||||||
|
DISABLE_PROXY=1 \
|
||||||
|
"$*"
|
||||||
|
)
|
||||||
|
'';
|
||||||
|
};
|
||||||
|
|
||||||
|
utils.build = pkgs.writeScript "utils.build" /* sh */ ''
|
||||||
|
#! ${pkgs.dash}/bin/dash
|
||||||
|
set -efu
|
||||||
|
expr=$1
|
||||||
|
shift
|
||||||
|
${pkgs.nix}/bin/nix-build \
|
||||||
|
-Q \
|
||||||
|
--no-out-link \
|
||||||
|
--show-trace \
|
||||||
|
-E "with import <stockholm>; $expr" \
|
||||||
|
-I "$target_path" \
|
||||||
|
"$@" \
|
||||||
|
2>&1 |
|
||||||
|
${pkgs.coreutils}/bin/stdbuf -oL ${spkgs.whatsupnix}/bin/whatsupnix
|
||||||
|
'';
|
||||||
|
|
||||||
|
utils.deploy = pkgs.writeScript "utils.deploy" /* sh */ ''
|
||||||
|
#! ${pkgs.dash}/bin/dash
|
||||||
|
set -efu
|
||||||
|
PATH=/run/current-system/sw/bin nixos-rebuild \
|
||||||
|
switch \
|
||||||
|
-Q \
|
||||||
|
--show-trace \
|
||||||
|
-I "$target_path" \
|
||||||
|
"$@" \
|
||||||
|
2>&1 |
|
||||||
|
${pkgs.coreutils}/bin/stdbuf -oL ${spkgs.whatsupnix}/bin/whatsupnix
|
||||||
|
'';
|
||||||
|
|
||||||
|
hook.get-version = pkgs.writeScript "hook.get-version" /* sh */ ''
|
||||||
|
#! ${pkgs.dash}/bin/dash
|
||||||
|
set -efu
|
||||||
|
version=git.$(${pkgs.git}/bin/git describe --always --dirty)
|
||||||
|
case $version in (*-dirty)
|
||||||
|
version=$version@$HOSTNAME
|
||||||
|
esac
|
||||||
|
date=$(${pkgs.coreutils}/bin/date +%y.%m)
|
||||||
|
echo "$date.$version"
|
||||||
|
'';
|
||||||
|
|
||||||
|
hook.pkg = pkgs.runCommand "hook.pkg" {} /* sh */ ''
|
||||||
|
mkdir -p $out/bin
|
||||||
|
${lib.concatStrings (lib.mapAttrsToList (name: path: /* sh */ ''
|
||||||
|
ln -s ${path} $out/bin/${name}
|
||||||
|
'') cmds)}
|
||||||
|
'';
|
||||||
|
|
||||||
|
in pkgs.stdenv.mkDerivation {
|
||||||
|
name = "stockholm";
|
||||||
|
shellHook = ''
|
||||||
|
export NIX_PATH="stockholm=$PWD''${NIX_PATH+:$NIX_PATH}"
|
||||||
|
export PATH=${lib.makeBinPath [
|
||||||
|
hook.pkg
|
||||||
|
]}
|
||||||
|
|
||||||
|
eval "$(declare -F | ${pkgs.gnused}/bin/sed s/declare/unset/)"
|
||||||
|
shopt -u no_empty_cmd_completion
|
||||||
|
unalias -a
|
||||||
|
|
||||||
|
enable -n \
|
||||||
|
. [ alias bg bind break builtin caller cd command compgen complete \
|
||||||
|
compopt continue dirs disown eval exec false fc fg getopts hash \
|
||||||
|
help history jobs kill let local logout mapfile popd printf pushd \
|
||||||
|
pwd read readarray readonly shift source suspend test times trap \
|
||||||
|
true typeset ulimit umask unalias wait
|
||||||
|
|
||||||
|
exitHandler() {
|
||||||
|
:
|
||||||
|
}
|
||||||
|
|
||||||
|
export HOSTNAME="$(${pkgs.nettools}/bin/hostname)"
|
||||||
|
export STOCKHOLM_VERSION="''${STOCKHOLM_VERSION-$(${hook.get-version})}"
|
||||||
|
|
||||||
|
PS1='\[\e[38;5;162m\]\w\[\e[0m\] '
|
||||||
|
'';
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user