shell: init

1 files changed, 210 insertions, 0 deletions

diff --git a/shell.nix b/shell.nix
new file mode 100644
index 000000000..d6e09e1a8
--- /dev/null
+++ b/shell.nix
@@ -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\] '
+  '';
+}