create-walle 0.3.3 → 0.4.1

package/bin/create-walle.js

@@ -1,19 +1,24 @@
 #!/usr/bin/env node
 'use strict';
 
-const { execFileSync, execFile, spawn } = require('child_process');
+const { execFileSync, spawn } = require('child_process');
 const fs = require('fs');
 const path = require('path');
 
 const BOLD = '\x1b[1m';
 const DIM = '\x1b[2m';
 const GREEN = '\x1b[32m';
+const YELLOW = '\x1b[33m';
 const RED = '\x1b[31m';
 const CYAN = '\x1b[36m';
 const RESET = '\x1b[0m';
 
 const TEMPLATE_DIR = path.join(__dirname, '..', 'template');
 const LABEL = 'com.walle.server';
+const INSTALL_PATH_FILE = path.join(process.env.HOME, '.walle', 'install-path');
+
+// Files to preserve during update (user config, not code)
+const PRESERVE_ON_UPDATE = ['.env', 'wall-e/wall-e-config.json'];
 
 // ── CLI Router ──
 
@@ -21,6 +26,8 @@ const command = process.argv[2] || '';
 
 if (command === 'install') {
   install(process.argv[3]);
+} else if (command === 'update' || command === 'upgrade') {
+  update();
 } else if (command === 'start') {
   start();
 } else if (command === 'stop') {
@@ -33,8 +40,10 @@ if (command === 'install') {
   uninstall();
 } else if (command === '--help' || command === '-h' || command === 'help') {
   usage();
+} else if (command === '--version' || command === '-v') {
+  const pkg = require('../package.json');
+  console.log(pkg.version);
 } else if (command && !command.startsWith('-')) {
-  // Backwards compat: bare directory name = install
   install(command);
 } else {
   usage();
@@ -48,28 +57,46 @@ ${BOLD}${CYAN}  Wall-E${RESET} — Your personal digital twin
 
   ${BOLD}Usage:${RESET}
 
-    ${BOLD}npx create-walle install <directory>${RESET}  Install Wall-E and start the service
-    ${BOLD}npx create-walle start${RESET}               Start Wall-E (auto-restarts on reboot)
-    ${BOLD}npx create-walle stop${RESET}                Stop Wall-E
-    ${BOLD}npx create-walle status${RESET}              Check if Wall-E is running
-    ${BOLD}npx create-walle logs${RESET}                Tail the service logs
-    ${BOLD}npx create-walle uninstall${RESET}           Remove the auto-start service
+    ${BOLD}npx create-walle install <dir>${RESET}   Install (or detect existing install and update)
+    ${BOLD}npx create-walle update${RESET}          Update code to latest, keep your config
+    ${BOLD}npx create-walle start${RESET}           Start Wall-E (auto-restarts on reboot)
+    ${BOLD}npx create-walle stop${RESET}            Stop Wall-E
+    ${BOLD}npx create-walle status${RESET}          Check if Wall-E is running
+    ${BOLD}npx create-walle logs${RESET}            Tail the service logs
+    ${BOLD}npx create-walle uninstall${RESET}       Remove the auto-start service
 
-  ${DIM}Run 'start' and 'stop' from inside your Wall-E directory,
-  or set WALLE_DIR to point to it.${RESET}
+  ${DIM}Commands like start/stop/update auto-detect your install directory.
+  Or set WALLE_DIR to point to it.${RESET}
 `);
 }
 
 function install(targetDir) {
+  // If no target dir given, check for existing install
   if (!targetDir) {
+    const existing = findWalleDirSilent();
+    if (existing) {
+      console.log(`\n  Found existing install at ${DIM}${existing}${RESET}`);
+      console.log(`  Running update instead...\n`);
+      update();
+      return;
+    }
     console.log(`\n  ${RED}Missing directory name.${RESET} Usage: npx create-walle install ./my-agent\n`);
     process.exit(1);
   }
 
+  // If target dir exists and has Wall-E, treat as update
+  if (fs.existsSync(targetDir) && fs.existsSync(path.join(targetDir, 'claude-task-manager', 'server.js'))) {
+    console.log(`\n  ${CYAN}Wall-E already installed at ${targetDir}.${RESET} Updating...\n`);
+    saveWalleDir(path.resolve(targetDir));
+    update();
+    return;
+  }
+
+  // Fresh install
   console.log(`\n${BOLD}${CYAN}  Wall-E${RESET} — Installing...\n`);
 
   if (fs.existsSync(targetDir)) {
-    console.log(`  ${RED}Directory "${targetDir}" already exists.${RESET}\n`);
+    console.log(`  ${RED}Directory "${targetDir}" exists but is not a Wall-E install.${RESET}\n`);
     process.exit(1);
   }
   if (!fs.existsSync(TEMPLATE_DIR)) {
@@ -77,7 +104,6 @@ function install(targetDir) {
     process.exit(1);
   }
 
-  // Auto-detect
   const ownerName = detectName().replace(/[\r\n=]/g, '').trim().slice(0, 200);
   const timezone = detectTimezone();
   const nameParts = ownerName.split(/\s+/);
@@ -88,19 +114,11 @@ function install(targetDir) {
   console.log(`  ${DIM}Timezone: ${timezone}${RESET}`);
   console.log(`  ${DIM}Port:     ${port}${RESET}`);
 
-  // Copy template
   console.log(`\n  Copying files...`);
   copyRecursive(TEMPLATE_DIR, targetDir);
 
-  // Install deps
   console.log(`  Installing dependencies...\n`);
-  try {
-    execFileSync('npm', ['install', '--loglevel=warn'], { cwd: path.join(targetDir, 'claude-task-manager'), stdio: 'inherit' });
-    execFileSync('npm', ['install', '--loglevel=warn'], { cwd: path.join(targetDir, 'wall-e'), stdio: 'inherit' });
-  } catch {
-    console.error(`\n  ${RED}npm install failed.${RESET} Try manually:\n    cd ${targetDir}/claude-task-manager && npm install\n    cd ${targetDir}/wall-e && npm install\n`);
-    process.exit(1);
-  }
+  npmInstall(targetDir);
 
   // .env
   const envLines = [
@@ -108,7 +126,7 @@ function install(targetDir) {
     `# Finish setup at http://localhost:${port}/setup.html`,
     '',
     `WALLE_OWNER_NAME=${ownerName}`,
-    '# ANTHROPIC_API_KEY=sk-ant-...',
+    '# ANTHROPIC_API_KEY=',
     '',
     `CTM_PORT=${port}`,
     `WALL_E_PORT=${wallePort}`,
@@ -130,57 +148,95 @@ function install(targetDir) {
     { mode: 0o600 }
   );
 
-  // Data dir
   fs.mkdirSync(path.join(process.env.HOME, '.walle', 'data'), { recursive: true });
-
-  // Git init
   try { execFileSync('git', ['init', '-q'], { cwd: targetDir, stdio: 'ignore' }); } catch {}
 
-  // Save install path for start/stop
   saveWalleDir(path.resolve(targetDir));
 
+  // Start the service
+  console.log(`  Starting Wall-E...`);
+  startForegroundOrService(path.resolve(targetDir), port);
+
   console.log(`
-  ${GREEN}${BOLD}Done!${RESET}
+  ${GREEN}${BOLD}Wall-E is running!${RESET}
+
+  ${BOLD}Open:${RESET}  http://localhost:${port}
+
+  ${BOLD}Commands:${RESET}
+    npx create-walle update      ${DIM}Update to latest version${RESET}
+    npx create-walle stop        ${DIM}Stop the service${RESET}
+    npx create-walle start       ${DIM}Restart${RESET}
+    npx create-walle status      ${DIM}Check status${RESET}
+    npx create-walle logs        ${DIM}View logs${RESET}
+`);
+}
+
+function update() {
+  const dir = findWalleDir();
+  const port = readPort(dir);
+  const pkg = require('../package.json');
 
-  ${BOLD}Start Wall-E:${RESET}
-    cd ${targetDir} && node claude-task-manager/server.js
+  console.log(`${BOLD}${CYAN}  Wall-E${RESET} — Updating to v${pkg.version}...\n`);
+  console.log(`  ${DIM}Directory: ${dir}${RESET}`);
 
-  ${BOLD}Then open:${RESET}  http://localhost:${port}
+  if (!fs.existsSync(TEMPLATE_DIR)) {
+    console.error('  Template not found. Try: npm cache clean --force && npx create-walle@latest update');
+    process.exit(1);
+  }
+
+  // 1. Stop if running
+  console.log(`  Stopping Wall-E...`);
+  stopQuiet(dir, port);
+
+  // 2. Backup preserved files
+  const backups = {};
+  for (const relPath of PRESERVE_ON_UPDATE) {
+    const fullPath = path.join(dir, relPath);
+    if (fs.existsSync(fullPath)) {
+      backups[relPath] = fs.readFileSync(fullPath);
+    }
+  }
+
+  // 3. Copy new code (overwrite everything except node_modules, .git, data)
+  console.log(`  Updating code...`);
+  copyRecursive(TEMPLATE_DIR, dir);
+
+  // 4. Restore preserved files
+  for (const [relPath, content] of Object.entries(backups)) {
+    const fullPath = path.join(dir, relPath);
+    fs.mkdirSync(path.dirname(fullPath), { recursive: true });
+    fs.writeFileSync(fullPath, content);
+  }
+
+  // 5. Reinstall deps (in case package.json changed)
+  console.log(`  Installing dependencies...\n`);
+  npmInstall(dir);
 
-  ${BOLD}Auto-start on login (optional):${RESET}
-    npx create-walle start
+  // 6. Start again
+  console.log(`\n  Starting Wall-E...`);
+  startForegroundOrService(dir, port);
+
+  console.log(`
+  ${GREEN}${BOLD}Updated to v${pkg.version}!${RESET}  http://localhost:${port}
 
-  ${BOLD}Other commands:${RESET}
-    npx create-walle stop       ${DIM}Stop the service${RESET}
-    npx create-walle status     ${DIM}Check status${RESET}
-    npx create-walle logs       ${DIM}View logs${RESET}
+  ${DIM}Your .env and config were preserved.${RESET}
 `);
 }
 
 function start() {
   const dir = findWalleDir();
   const port = readPort(dir);
-
   console.log(`\n  Starting Wall-E from ${DIM}${dir}${RESET} on port ${port}...`);
   installService(dir, port);
   console.log(`  ${GREEN}Running!${RESET}  http://localhost:${port}\n`);
 }
 
 function stop() {
-  const plist = path.join(process.env.HOME, 'Library', 'LaunchAgents', `${LABEL}.plist`);
-  if (process.platform === 'darwin' && fs.existsSync(plist)) {
-    try { execFileSync('launchctl', ['unload', plist], { stdio: 'ignore' }); } catch {}
-    console.log(`\n  ${GREEN}Wall-E stopped.${RESET}`);
-    console.log(`  Run ${BOLD}npx create-walle start${RESET} to restart.\n`);
-  } else {
-    // Fallback: try killing the process
-    try {
-      const dir = findWalleDir();
-      const port = readPort(dir);
-      execFileSync('curl', ['-sX', 'POST', `http://localhost:${port}/api/stop/walle`], { timeout: 3000 });
-    } catch {}
-    console.log(`\n  Stopped (or was not running).\n`);
-  }
+  const dir = findWalleDirSilent();
+  const port = dir ? readPort(dir) : '3456';
+  stopQuiet(dir, port);
+  console.log(`\n  ${GREEN}Wall-E stopped.${RESET}`);
+  console.log(`  Run ${BOLD}npx create-walle start${RESET} to restart.\n`);
 }
 
 function status() {
@@ -213,12 +269,52 @@ function uninstall() {
   const plist = path.join(process.env.HOME, 'Library', 'LaunchAgents', `${LABEL}.plist`);
   try { execFileSync('launchctl', ['unload', plist], { stdio: 'ignore' }); } catch {}
   try { fs.unlinkSync(plist); } catch {}
-  try { fs.unlinkSync(path.join(process.env.HOME, '.walle', 'install-path')); } catch {}
+  try { fs.unlinkSync(INSTALL_PATH_FILE); } catch {}
   console.log(`\n  ${GREEN}Service uninstalled.${RESET} Your data in ~/.walle/data is preserved.\n`);
 }
 
 // ── Service Management ──
 
+function stopQuiet(dir, port) {
+  // Try launchctl first
+  const plist = path.join(process.env.HOME, 'Library', 'LaunchAgents', `${LABEL}.plist`);
+  if (process.platform === 'darwin' && fs.existsSync(plist)) {
+    try { execFileSync('launchctl', ['unload', plist], { stdio: 'ignore' }); } catch {}
+  }
+  // Also try API
+  try { execFileSync('curl', ['-sX', 'POST', `http://localhost:${port}/api/restart/ctm`], { timeout: 2000 }); } catch {}
+  // Wait for port to free up
+  for (let i = 0; i < 10; i++) {
+    try {
+      execFileSync('curl', ['-s', '--max-time', '1', `http://localhost:${port}/api/services/status`], { timeout: 2000 });
+      // Still running, wait
+      execFileSync('sleep', ['0.5']);
+    } catch {
+      break; // Port is free
+    }
+  }
+}
+
+function startForegroundOrService(dir, port) {
+  // If launchd plist exists, reload it. Otherwise start in background.
+  const plist = path.join(process.env.HOME, 'Library', 'LaunchAgents', `${LABEL}.plist`);
+  if (process.platform === 'darwin' && fs.existsSync(plist)) {
+    // Update plist with latest paths, then reload
+    installService(dir, port);
+  } else {
+    // Start in background without launchd
+    const logDir = path.join(process.env.HOME, '.walle', 'logs');
+    fs.mkdirSync(logDir, { recursive: true });
+    const child = spawn('node', ['claude-task-manager/server.js'], {
+      cwd: dir,
+      detached: true,
+      stdio: ['ignore', fs.openSync(path.join(logDir, 'walle.log'), 'a'), fs.openSync(path.join(logDir, 'walle.err'), 'a')],
+      env: { ...process.env, CTM_PORT: port, WALL_E_PORT: String(parseInt(port) + 1) },
+    });
+    child.unref();
+  }
+}
+
 function installService(walleDir, port) {
   const logDir = path.join(process.env.HOME, '.walle', 'logs');
   fs.mkdirSync(logDir, { recursive: true });
@@ -229,7 +325,6 @@ function installService(walleDir, port) {
     fs.mkdirSync(plistDir, { recursive: true });
     const plistPath = path.join(plistDir, `${LABEL}.plist`);
 
-    // Build env vars from .env file
     let envDict = `    <key>PATH</key>\n    <string>/usr/local/bin:/opt/homebrew/bin:/usr/bin:/bin</string>\n`;
     envDict += `    <key>HOME</key>\n    <string>${process.env.HOME}</string>\n`;
     envDict += `    <key>CTM_PORT</key>\n    <string>${port}</string>\n`;
@@ -274,7 +369,6 @@ ${envDict}  </dict>
     try { execFileSync('launchctl', ['unload', plistPath], { stdio: 'ignore' }); } catch {}
     execFileSync('launchctl', ['load', plistPath]);
   } else {
-    // Linux/other: just start directly in background
     const child = spawn('node', ['claude-task-manager/server.js'], {
       cwd: walleDir,
       detached: true,
@@ -288,30 +382,42 @@ ${envDict}  </dict>
 
 // ── Helpers ──
 
+function npmInstall(dir) {
+  try {
+    execFileSync('npm', ['install', '--loglevel=warn'], { cwd: path.join(dir, 'claude-task-manager'), stdio: 'inherit' });
+    execFileSync('npm', ['install', '--loglevel=warn'], { cwd: path.join(dir, 'wall-e'), stdio: 'inherit' });
+  } catch {
+    console.error(`\n  ${RED}npm install failed.${RESET} Try manually:\n    cd ${dir}/claude-task-manager && npm install\n    cd ${dir}/wall-e && npm install\n`);
+    process.exit(1);
+  }
+}
+
 function saveWalleDir(dir) {
   const metaDir = path.join(process.env.HOME, '.walle');
   fs.mkdirSync(metaDir, { recursive: true });
-  fs.writeFileSync(path.join(metaDir, 'install-path'), dir + '\n');
+  fs.writeFileSync(INSTALL_PATH_FILE, dir + '\n');
 }
 
 function findWalleDir() {
-  // 1. WALLE_DIR env
+  const dir = findWalleDirSilent();
+  if (dir) return dir;
+  console.error(`\n  ${RED}Can't find Wall-E installation.${RESET}`);
+  console.error(`  Run ${BOLD}npx create-walle install ./my-agent${RESET} first.\n`);
+  process.exit(1);
+}
+
+function findWalleDirSilent() {
   if (process.env.WALLE_DIR && fs.existsSync(path.join(process.env.WALLE_DIR, 'claude-task-manager'))) {
     return process.env.WALLE_DIR;
   }
-  // 2. Current directory
   if (fs.existsSync(path.join(process.cwd(), 'claude-task-manager', 'server.js'))) {
     return process.cwd();
   }
-  // 3. Saved install path
-  const saved = path.join(process.env.HOME, '.walle', 'install-path');
-  if (fs.existsSync(saved)) {
-    const dir = fs.readFileSync(saved, 'utf8').trim();
+  if (fs.existsSync(INSTALL_PATH_FILE)) {
+    const dir = fs.readFileSync(INSTALL_PATH_FILE, 'utf8').trim();
     if (fs.existsSync(path.join(dir, 'claude-task-manager', 'server.js'))) return dir;
   }
-  console.error(`\n  ${RED}Can't find Wall-E installation.${RESET}`);
-  console.error(`  Run from inside the Wall-E directory, or set WALLE_DIR.\n`);
-  process.exit(1);
+  return null;
 }
 
 function readPort(dir) {
@@ -320,7 +426,7 @@ function readPort(dir) {
     const m = env.match(/^CTM_PORT=(\d+)/m);
     if (m) return m[1];
   } catch {}
-  return '4567';
+  return '3456';
 }
 
 function detectName() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "create-walle",
-  "version": "0.3.3",
+  "version": "0.4.1",
   "description": "Set up Wall-E — your personal digital twin",
   "bin": {
     "create-walle": "bin/create-walle.js"