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