feat(workstation): add pi

5 files changed, 120 insertions, 1 deletions

diff --git a/link-user b/link-user
index 752e38c..4c9d79d 100755
--- a/link-user
+++ b/link-user
@@ -3,7 +3,7 @@
 
 set -u
 
-cd $(dirname "$0") # change to the directory of the script
+cd "$(dirname "$(realpath "$0")")" # change to the resolved directory of the script
 cd user
 
 for dir in $(find . -type d); do
diff --git a/nixos/profiles/workstation/default.nix b/nixos/profiles/workstation/default.nix
index e6fc7e1..fef1757 100644
--- a/nixos/profiles/workstation/default.nix
+++ b/nixos/profiles/workstation/default.nix
@@ -19,6 +19,7 @@ in
     ./firefox.nix
     ./dev.nix
     ./create.nix
+    ./llm.nix
   ];
 
   config = {
diff --git a/nixos/profiles/workstation/llm.nix b/nixos/profiles/workstation/llm.nix
new file mode 100644
index 0000000..3301ff2
--- /dev/null
+++ b/nixos/profiles/workstation/llm.nix
@@ -0,0 +1,64 @@
+{ pkgs, pkgs-unstable, ... }:
+
+let
+  sbox = pkgs.writeShellScriptBin "sbox" ''
+    set -euo pipefail
+
+    bwrap_args=(
+      --die-with-parent
+      --tmpfs /
+      --dev /dev
+      --proc /proc
+      --tmpfs /tmp
+      --ro-bind /etc /etc
+      --bind /nix/store /nix/store
+      --ro-bind /usr/bin/env /usr/bin/env
+      --setenv NIX_REMOTE daemon
+      --bind-try /nix/var/nix/daemon-socket /nix/var/nix/daemon-socket
+      --dir /run
+      --ro-bind-try /run/current-system /run/current-system
+      # mount git and jj config
+      --ro-bind "$HOME/.config" "$HOME/.config"
+      --ro-bind "$HOME/repos/config" "$HOME/repos/config"
+      # jj must be able to write to ~/.config/jj/repos: https://github.com/jj-vcs/jj/issues/8841
+      --overlay-src "$HOME/.config/jj" --tmp-overlay "$HOME/.config/jj"
+    )
+
+    exec ${pkgs.bubblewrap}/bin/bwrap "''${bwrap_args[@]}" "$@"
+  '';
+in
+{
+  environment.systemPackages = [
+    sbox
+
+    (pkgs.writeShellScriptBin "pi" ''
+      set -euo pipefail
+
+      export PI_CODING_AGENT_DIR="$HOME/.config/pi"
+
+      exec ${sbox}/bin/sbox \
+        --bind "$HOME/.config/pi" "$HOME/.config/pi" \
+        --bind "$PWD" "$PWD" \
+        --chdir "$PWD" \
+        ${pkgs-unstable.pi-coding-agent}/bin/pi "$@"
+    '')
+
+    (pkgs.writeShellScriptBin "pit" ''
+      set -euo pipefail
+
+      tmp_cwd="$(${pkgs.coreutils}/bin/mktemp -d /tmp/pit.XXXXXX)"
+      cleanup() {
+        ${pkgs.coreutils}/bin/rmdir "$tmp_cwd" 2>/dev/null || true
+      }
+      trap cleanup EXIT
+
+      export PI_CODING_AGENT_DIR="$HOME/.config/pi"
+
+      ${sbox}/bin/sbox \
+        --bind "$HOME/.config/pi" "$HOME/.config/pi" \
+        --bind "$tmp_cwd" "$tmp_cwd" \
+        --chdir "$tmp_cwd" \
+        ${pkgs-unstable.pi-coding-agent}/bin/pi "$@"
+    '')
+  ];
+}
diff --git a/user/pi/extensions/stash-on-escape.ts b/user/pi/extensions/stash-on-escape.ts
new file mode 100644
index 0000000..b361a3e
--- /dev/null
+++ b/user/pi/extensions/stash-on-escape.ts
@@ -0,0 +1,46 @@
+// On Escape clear the editor an store the text in the history instead of doing nothing.
+import {
+  CustomEditor,
+  type ExtensionAPI,
+  type ExtensionContext,
+  type KeybindingsManager,
+} from "@earendil-works/pi-coding-agent";
+import type { EditorTheme, TUI } from "@earendil-works/pi-tui";
+
+class StashOnEscapeEditor extends CustomEditor {
+  constructor(
+    tui: TUI,
+    theme: EditorTheme,
+    private readonly keys: KeybindingsManager,
+    private readonly ctx: ExtensionContext,
+  ) {
+    super(tui, theme, keys);
+  }
+
+  handleInput(data: string): void {
+    const text = this.getExpandedText?.() ?? this.getText();
+
+    if (
+      this.ctx.isIdle() &&
+      text.trim() &&
+      this.keys.matches(data, "app.interrupt")
+    ) {
+      this.addToHistory(text);
+      this.setText("");
+      return;
+    }
+
+    super.handleInput(data);
+  }
+}
+
+export default function (pi: ExtensionAPI) {
+  pi.on("session_start", (_event, ctx) => {
+    if (ctx.mode !== "tui") return;
+
+    ctx.ui.setEditorComponent(
+      (tui, theme, keybindings) =>
+        new StashOnEscapeEditor(tui, theme, keybindings, ctx),
+    );
+  });
+}
diff --git a/user/pi/settings.json b/user/pi/settings.json
new file mode 100644
index 0000000..46218e0
--- /dev/null
+++ b/user/pi/settings.json
@@ -0,0 +1,8 @@
+{
+  "lastChangelogVersion": "0.87.1",
+  "theme": "dark",
+  "defaultProvider": "openai-codex",
+  "defaultModel": "gpt-5.5",
+  "defaultThinkingLevel": "medium",
+  "packages": []
+}
\ No newline at end of file