niveum/packages/pi.nix

{
  runCommand,
  nodejs,
  writeShellApplication,
  lib,
  jq,
  cacert,
  pi-llm,
}:
let
  # Pre-install pi plugins into a fake npm global prefix
  pluginPrefixRaw =
    runCommand "pi-plugins-raw"
      {
        nativeBuildInputs = [
          nodejs
          cacert
        ];
        outputHashMode = "recursive";
        outputHashAlgo = "sha256";
        outputHash = "sha256-ruCukrbH89/uZHhxYSmrcs5NUi2ZG8nuwTEuJKHP2rw=";
        impureEnvVars = [
          "http_proxy"
          "https_proxy"
        ];
        SSL_CERT_FILE = "${cacert}/etc/ssl/certs/ca-bundle.crt";
      }
      ''
        export HOME=$TMPDIR
        export npm_config_prefix=$out
        npm install -g pi-hooks shitty-extensions
      '';

  # Remove the resistance extension (annoying terminator quote widget)
  pluginPrefix = runCommand "pi-plugins" { } ''
    cp -a ${pluginPrefixRaw} $out
    chmod -R u+w $out
    pkg=$out/lib/node_modules/shitty-extensions/package.json
    ${lib.getExe jq} '.pi.extensions |= map(select(contains("resistance") | not))' "$pkg" > "$pkg.tmp"
    mv "$pkg.tmp" "$pkg"
  '';
in
writeShellApplication {
  name = "pi";
  runtimeInputs = [ nodejs ];
  text = ''
    set -efu
    export npm_config_prefix="${pluginPrefix}"

    # Ensure settings.json has our plugins listed
    SETTINGS_DIR="''${PI_CODING_AGENT_DIR:-$HOME/.pi/agent}"
    SETTINGS_FILE="$SETTINGS_DIR/settings.json"
    mkdir -p "$SETTINGS_DIR"

    # Add packages to settings if not already present
    if [ ! -f "$SETTINGS_FILE" ]; then
      echo '{"packages":["npm:pi-hooks","npm:shitty-extensions"]}' > "$SETTINGS_FILE"
    else
      for pkg in "npm:pi-hooks" "npm:shitty-extensions"; do
        if ! grep -q "$pkg" "$SETTINGS_FILE"; then
          ${lib.getExe jq} --arg p "$pkg" '.packages = ((.packages // []) + [$p] | unique)' "$SETTINGS_FILE" > "$SETTINGS_FILE.tmp"
          mv "$SETTINGS_FILE.tmp" "$SETTINGS_FILE"
        fi
      done
    fi

    exec ${lib.getExe pi-llm} "$@"
  '';
}