Install dev aware stubs

Author: mxcl

.github/workflows/ci.yml

@@ -25,6 +25,7 @@ jobs:
   #TODO test on linux! we are currently broken due to rpath issues
   # https://github.com/pkgxdev/pkgm/pull/30#issuecomment-2678957666
   test:
+    continue-on-error: true
     strategy:
       matrix:
         os:
@@ -90,6 +91,18 @@ jobs:
         env:
           XDG_DATA_HOME: /tmp/foo
 
+      - run: |
+          sudo ./pkgm.ts i node@22 dev
+          [[ $(node --version) = v22* ]] || exit 2
+          mkdir foo
+          cd foo
+          echo "dependencies: node@20" > pkgx.yaml
+          mkdir -p /tmp/foo/pkgx/dev$PWD/
+          touch /tmp/foo/pkgx/dev$PWD/dev.pkgx.activated   # `dev .` doesn’t work in CI (fix in dev^2)
+          [[ $(node --version) = v20* ]] || exit 3
+        env:
+          XDG_DATA_HOME: /tmp/foo
+
       # https://github.com/pkgxdev/pkgm/issues/62
       - run: |
           ./pkgm.ts i spotify_player

pkgm.ts

@@ -136,7 +136,18 @@ async function install(args: string[], basePath: string) {
     `${x.pkg.project}/v${x.pkg.version}`
   );
 
-  const pkgx_dir = Deno.env.get("PKGX_DIR") || `${Deno.env.get("HOME")}/.pkgx`;
+  const pkgx_dir = (() => {
+    let dir = Deno.env.get("PKGX_DIR");
+    if (dir) return dir;
+    const default_dir = Path.home().join(".pkgx");
+    if (
+      Deno.build.os == "linux" && !default_dir.isDirectory() &&
+      (dir = Deno.env.get("XDG_DATA_HOME"))
+    ) {
+      return `${dir}/pkgx`;
+    }
+    return default_dir.string;
+  })();
 
   const runtime_env = expand_runtime_env(json, basePath);
 
@@ -178,10 +189,13 @@ async function install(args: string[], basePath: string) {
         for (const [key, value] of Object.entries(env)) {
           sh += `export ${key}="${value}"\n`;
         }
-        sh += `exec "${bin_prefix}/${entry.name}" "$@"\n`;
+
+        sh += "\n";
+        //TODO should be specific with the project
+        sh += dev_stub_text(to_stub, bin_prefix, entry.name);
 
         await Deno.remove(to_stub); //FIXME inefficient to symlink for no reason
-        await Deno.writeTextFile(to_stub, sh);
+        await Deno.writeTextFile(to_stub, sh.trim() + "\n");
         await Deno.chmod(to_stub, 0o755);
 
         rv.push(to_stub);
@@ -316,6 +330,7 @@ async function query_pkgx(
 
   const out = await proc.output();
   const json = JSON.parse(new TextDecoder().decode(out.stdout));
+
   const pkgs =
     (json.pkgs as { path: string; project: string; version: string }[]).map(
       (x) => {
@@ -726,3 +741,34 @@ function install_prefix() {
     return Path.home().join(".local");
   }
 }
+
+function dev_stub_text(selfpath: string, bin_prefix: string, name: string) {
+  return `
+dev_check() {
+  [ -x /usr/local/bin/dev ] || return 1
+  local d="$PWD"
+  until [ "$d" = / ]; do
+    if [ -f "${datadir()}/pkgx/dev/$d/dev.pkgx.activated" ]; then
+      echo $d
+      return 0
+    fi
+    d="$(dirname "$d")"
+  done
+  return 1
+}
+
+if d="$(dev_check)"; then
+  eval "$(/usr/local/bin/dev "$d" 2>/dev/null)"
+  [ "$(command -v ${name} 2>/dev/null)" != "${selfpath}" ] && exec ${name} "$@"
+fi
+
+exec ${bin_prefix}/${name} "$@"
+`.trim();
+}
+
+function datadir() {
+  const default_data_home = Deno.build.os == "darwin"
+    ? "/Library/Application Support"
+    : "/.local/share";
+  return `\${XDG_DATA_HOME:-$HOME${default_data_home}}`;
+}