0101-agents 0.1.3 → 0.1.5

package/bin/cli.js

@@ -6,7 +6,7 @@
 //   0101-agents logout
 //   0101-agents whoami
 //   0101-agents install <agent> [--force]
-//   0101-agents update <agent>
+//   0101-agents update <agent> [-f]
 //   0101-agents list
 //   0101-agents help
 
@@ -131,31 +131,64 @@ function ensureUnzipAvailable() {
   }
 }
 
-function unzipInto(zipPath, destDir, { skipPrefix } = {}) {
+function unzipInto(zipPath, destDir, { exclude = [] } = {}) {
   // Use the system unzip binary. Available everywhere except bare Windows.
+  // -o overwrites; a single -x takes the whole pattern list (after -d), used on
+  // update to keep the manifest's `preserve` paths (e.g. data/, .env) intact.
   mkdirSync(destDir, { recursive: true })
-  if (skipPrefix) {
-    // unzip's -x excludes paths matching glob patterns. `<prefix>/*` catches
-    // every file under the directory; the bare directory entry (e.g. `data/`)
-    // is harmless to re-extract because it just re-creates the dir shell.
-    execSync(
-      `unzip -q -o "${zipPath}" -d "${destDir}" -x "${skipPrefix}/*"`,
-      { stdio: 'inherit' },
-    )
-  } else {
-    execSync(`unzip -q -o "${zipPath}" -d "${destDir}"`, { stdio: 'inherit' })
+  const xs = exclude.length ? `-x ${exclude.map((p) => `"${p}"`).join(' ')}` : ''
+  execSync(`unzip -q -o "${zipPath}" -d "${destDir}" ${xs}`.trim(), { stdio: 'inherit' })
+}
+
+// Read the installed artifact's own manifest.json (NOT the release manifest
+// from the API). Drives per-artifact behavior: which paths to keep on update
+// (`preserve`), an optional post-install command (`postInstall`), and how
+// `start` launches. Fails soft to {} so agents without these fields behave
+// exactly as before.
+function readManifest(agent) {
+  try {
+    return JSON.parse(readFileSync(join(agentInstallDir(agent), 'manifest.json'), 'utf8'))
+  } catch {
+    return {}
   }
 }
 
-function backupDataDir(agent) {
-  const dataDir = join(agentInstallDir(agent), 'data')
-  if (!existsSync(dataDir)) return null
+// Paths an update must NOT overwrite. Agents keep their data/ by default; the
+// runner declares [".env", ".runner-state"]. Always returns a non-empty array.
+function preserveList(manifest) {
+  const p = manifest?.preserve
+  return Array.isArray(p) && p.length ? p : ['data']
+}
+
+// unzip -x patterns: drop both the entry and everything under it, so a
+// preserved file (.env) and a preserved dir (data/) are each left untouched.
+function excludePatterns(preserve) {
+  return preserve.flatMap((p) => [p, `${p}/*`])
+}
+
+// Run the manifest's postInstall command (if any) in the install dir — e.g. the
+// runner's `npm ci && node scripts/postinstall.mjs`. Trusted: the zip is the
+// license-gated artifact we shipped. No-op for agents (no postInstall field).
+function runPostInstall(agent, manifest) {
+  const cmd = manifest?.postInstall
+  if (!cmd || typeof cmd !== 'string') return
+  info('Running post-install…')
+  execSync(cmd, { cwd: agentInstallDir(agent), stdio: 'inherit' })
+}
+
+// Back up the paths an update would replace, so a bad release is recoverable.
+// Tars only the preserve paths that actually exist (data/ for agents; .env +
+// .runner-state/ for the runner).
+function backupPreserved(agent, preserve) {
+  const dir = agentInstallDir(agent)
+  const existing = preserve.filter((p) => existsSync(join(dir, p)))
+  if (!existing.length) return null
   mkdirSync(BACKUPS_DIR, { recursive: true })
   const ts = new Date().toISOString().replace(/[:.]/g, '-')
   const oldVersion = installedVersion(agent) ?? 'unknown'
   const backupPath = join(BACKUPS_DIR, `${agent}-${oldVersion}-${ts}.tar.gz`)
   execSync(
-    `tar -czf "${backupPath}" -C "${agentInstallDir(agent)}" data`,
+    `tar -czf "${backupPath}" -C "${dir}" ${existing.map((p) => `"${p}"`).join(' ')}`,
     { stdio: 'inherit' },
   )
   return backupPath
@@ -220,8 +253,9 @@ async function cmdInstall(args) {
   }
   mkdirSync(installDir, { recursive: true })
   unzipInto(zipPath, installDir)
-  writeInstalledVersion(agent, manifest.latest)
   rmSync(zipPath, { force: true })
+  runPostInstall(agent, readManifest(agent))
+  writeInstalledVersion(agent, manifest.latest)
 
   console.log('')
   ok(`Installed ${bold(agent)} ${manifest.latest} → ${installDir}`)
@@ -230,8 +264,9 @@ async function cmdInstall(args) {
 }
 
 async function cmdUpdate(args) {
-  const agent = args[0]
-  if (!agent) die('Usage: 0101-agents update <agent>')
+  const agent = args.find((a) => !a.startsWith('-'))
+  const assumeYes = args.some((a) => ['-f', '--force', '-y', '--yes'].includes(a))
+  if (!agent) die('Usage: 0101-agents update <agent> [-f]')
 
   ensureUnzipAvailable()
   const installDir = agentInstallDir(agent)
@@ -249,25 +284,29 @@ async function cmdUpdate(args) {
     return
   }
 
-  console.log(dim('Your data/ folder is preserved (memory, projects, custom files).'))
-  const ans = await prompt(`Update ${agent} ${current ?? '?'} → ${manifest.latest}? [Y/n] `)
-  if (ans && !/^y(es)?$/i.test(ans)) {
-    info('Cancelled.')
-    return
+  console.log(dim('Your local data is preserved across the update.'))
+  if (!assumeYes) {
+    const ans = await prompt(`Update ${agent} ${current ?? '?'} → ${manifest.latest}? [Y/n] `)
+    if (ans && !/^y(es)?$/i.test(ans)) {
+      info('Cancelled.')
+      return
+    }
   }
 
-  const backup = backupDataDir(agent)
-  if (backup) info(`Backed up data → ${backup}`)
+  const preserve = preserveList(readManifest(agent))
+  const backup = backupPreserved(agent, preserve)
+  if (backup) info(`Backed up ${preserve.join(', ')} → ${backup}`)
 
   const zipPath = join(tmpdir(), `0101-${agent}-${manifest.latest}.zip`)
   info(`Downloading…`)
   await downloadZip(manifest.download_url, zipPath)
 
-  // Extract into the install dir, skipping the data/ subtree so user state
-  // survives. unzip's -x flag drops matching entries.
-  unzipInto(zipPath, installDir, { skipPrefix: 'data' })
-  writeInstalledVersion(agent, manifest.latest)
+  // Extract over the install, skipping the manifest's `preserve` paths so user
+  // state survives (agents: data/; runner: .env + .runner-state/).
+  unzipInto(zipPath, installDir, { exclude: excludePatterns(preserve) })
   rmSync(zipPath, { force: true })
+  runPostInstall(agent, readManifest(agent))
+  writeInstalledVersion(agent, manifest.latest)
 
   console.log('')
   ok(`Updated ${bold(agent)} → ${manifest.latest}`)
@@ -308,6 +347,17 @@ async function cmdStart(args) {
     // ignore
   }
 
+  // Artifacts can declare how `start` launches via manifest.start. The runner
+  // sets "npm start"; agents leave it unset and fall through to launching
+  // Claude Code below.
+  const startCmd = readManifest(agent).start
+  if (startCmd) {
+    const child = spawn(startCmd, { cwd: dir, stdio: 'inherit', shell: true })
+    child.on('error', (e) => die(`Failed to start: ${e.message}`))
+    child.on('exit', (code) => process.exit(code ?? 0))
+    return
+  }
+
   // Everything after the agent name passes through to `claude`. Supports both:
   //   0101-agents start marketer --resume
   //   0101-agents start marketer -- --resume  (explicit separator)
@@ -349,7 +399,7 @@ Commands:
   ${bold('logout')}                     Forget your license key
   ${bold('whoami')}                     Show current license
   ${bold('install')} <agent> [--force]    Install an agent
-  ${bold('update')}  <agent>             Update an installed agent (preserves data/)
+  ${bold('update')}  <agent> [-f]        Update an installed agent (preserves data/). -f skips the prompt
   ${bold('start')}   <agent> [--tg] [-- args]  Open the agent (runs claude in its dir)
                                        --tg adds the right flags to run as a
                                        Telegram channel bot.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "0101-agents",
-  "version": "0.1.3",
+  "version": "0.1.5",
   "description": "",
   "bin": {
     "0101-agents": "./bin/cli.js"