zooid 0.13.0 → 0.14.1

package/src/pi-extension-install.test.ts

@@ -0,0 +1,65 @@
+import { afterEach, describe, expect, it } from 'vitest'
+import { existsSync, mkdtempSync, mkdirSync, readFileSync, rmSync, writeFileSync } from 'node:fs'
+import { tmpdir } from 'node:os'
+import { join } from 'node:path'
+import { installPiExtension, resolvePiAgentDir } from './pi-extension-install.js'
+
+const dirs: string[] = []
+const scratch = () => { const dir = mkdtempSync(join(tmpdir(), 'zooid-pi-')); dirs.push(dir); return dir }
+afterEach(() => dirs.splice(0).forEach((dir) => rmSync(dir, { recursive: true, force: true })))
+
+describe('resolvePiAgentDir', () => {
+  it('resolves a relative PI_CODING_AGENT_DIR against each agent workdir', () => {
+    expect(
+      resolvePiAgentDir({
+        agentWorkdir: '/proj/agents/alice',
+        daemonHome: '/home/op',
+        env: { PI_CODING_AGENT_DIR: '.pi-agent' },
+      }),
+    ).toEqual({ dir: '/proj/agents/alice/.pi-agent', scope: 'project' })
+  })
+
+  it('honours an absolute override and falls back to the operator home', () => {
+    expect(
+      resolvePiAgentDir({
+        agentWorkdir: '/proj/agents/alice',
+        daemonHome: '/home/op',
+        env: { PI_CODING_AGENT_DIR: '/shared/pi' },
+      }),
+    ).toEqual({ dir: '/shared/pi', scope: 'project' })
+    expect(
+      resolvePiAgentDir({ agentWorkdir: '/proj/agents/alice', daemonHome: '/home/op', env: {} }),
+    ).toEqual({ dir: '/home/op/.pi/agent', scope: 'home' })
+  })
+})
+
+describe('installPiExtension', () => {
+  it('only installs into an existing Pi home and preserves other extensions', () => {
+    const home = scratch(); const source = join(scratch(), 'bundle.js')
+    writeFileSync(source, '// bundle')
+    const agentDir = join(home, '.pi', 'agent')
+    expect(installPiExtension({ agentDir, bundlePath: source })).toMatchObject({ status: 'skipped' })
+    expect(existsSync(join(home, '.pi'))).toBe(false)
+    const extensions = join(agentDir, 'extensions')
+    mkdirSync(extensions, { recursive: true }); writeFileSync(join(extensions, 'user.js'), '// user')
+    expect(installPiExtension({ agentDir, bundlePath: source })).toMatchObject({ status: 'installed' })
+    expect(readFileSync(join(extensions, 'zooid-tasks.js'), 'utf8')).toBe('// bundle')
+    expect(readFileSync(join(extensions, 'user.js'), 'utf8')).toBe('// user')
+    expect(installPiExtension({ agentDir, bundlePath: source })).toMatchObject({ status: 'unchanged' })
+  })
+
+  it('creates a project agent dir that does not exist yet', () => {
+    const workdir = scratch(); const source = join(scratch(), 'bundle.js')
+    writeFileSync(source, '// bundle')
+    const { dir } = resolvePiAgentDir({
+      agentWorkdir: workdir,
+      daemonHome: '/home/op',
+      env: { PI_CODING_AGENT_DIR: '.pi-agent' },
+    })
+    expect(installPiExtension({ agentDir: dir, bundlePath: source, createMissing: true })).toMatchObject({
+      status: 'installed',
+      target: join(workdir, '.pi-agent', 'extensions', 'zooid-tasks.js'),
+    })
+    expect(readFileSync(join(workdir, '.pi-agent', 'extensions', 'zooid-tasks.js'), 'utf8')).toBe('// bundle')
+  })
+})

package/src/pi-extension-install.ts

@@ -0,0 +1,56 @@
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'node:fs'
+import { dirname, isAbsolute, join, resolve } from 'node:path'
+
+export interface InstallPiExtensionResult {
+  status: 'installed' | 'unchanged' | 'skipped'
+  reason?: string
+  target?: string
+}
+
+export interface PiAgentDir {
+  dir: string
+  /** `project` dirs are ours to create; `home` is the operator's ~/.pi. */
+  scope: 'project' | 'home'
+}
+
+/**
+ * Resolve pi's agent dir the way pi does: PI_CODING_AGENT_DIR when set,
+ * otherwise ~/.pi/agent.  `zooid init` writes that variable as a *relative*
+ * path on purpose (ZOD075) so one value is correct under both runtimes — it
+ * resolves against the agent's cwd, and `agents/<name>` locally and
+ * `/workspace` in a container are the same directory.  We always resolve
+ * against the host workdir, which is the host side of that same mount.
+ */
+export function resolvePiAgentDir(opts: {
+  agentWorkdir: string
+  daemonHome: string
+  env?: { PI_CODING_AGENT_DIR?: string | undefined }
+}): PiAgentDir {
+  const override = opts.env?.PI_CODING_AGENT_DIR
+  if (override) {
+    return {
+      dir: isAbsolute(override) ? override : resolve(opts.agentWorkdir, override),
+      scope: 'project',
+    }
+  }
+  return { dir: join(opts.daemonHome, '.pi', 'agent'), scope: 'home' }
+}
+
+/** Install our bundle without replacing an operator's existing Pi extensions. */
+export function installPiExtension(opts: {
+  agentDir: string
+  bundlePath: string
+  /** Project dirs belong to the workspace, so we may create them. ~/.pi is the
+   *  operator's: install into it only if they already use pi. */
+  createMissing?: boolean
+}): InstallPiExtensionResult {
+  if (!opts.createMissing && !existsSync(opts.agentDir)) {
+    return { status: 'skipped', reason: 'no Pi home' }
+  }
+  const target = join(opts.agentDir, 'extensions', 'zooid-tasks.js')
+  mkdirSync(dirname(target), { recursive: true })
+  const source = readFileSync(opts.bundlePath)
+  if (existsSync(target) && readFileSync(target).equals(source)) return { status: 'unchanged', target }
+  writeFileSync(target, source)
+  return { status: 'installed', target }
+}