vibelet 0.1.16 → 0.1.17

package/bin/vibelet.mjs

@@ -25,20 +25,102 @@ const stdoutLogPath = join(logDir, 'daemon.stdout.log');
 const stderrLogPath = join(logDir, 'daemon.stderr.log');
 const pidFilePath = join(vibeletDir, 'daemon.pid');
 const relayConfigPath = join(vibeletDir, 'relay.json');
+const updateCheckPath = join(vibeletDir, 'update-check.json');
 const OFFICIAL_SITE_URL = 'https://vibelet.icu';
+const UPDATE_CHECK_INTERVAL_MS = 4 * 60 * 60 * 1000; // 4 hours
 
 let officialSitePrinted = false;
+let updateMessage = '';
 
 function printOfficialSite() {
   if (officialSitePrinted) return;
   officialSitePrinted = true;
   try {
+    if (updateMessage) writeSync(2, `\n${updateMessage}\n`);
     writeSync(1, `\nOfficial site: ${OFFICIAL_SITE_URL}\n`);
   } catch {
     // Best-effort branding footer; ignore broken pipes and closed stdio.
   }
 }
 
+// ─── Update check ──────────────────────────────────────────────────────────────
+
+function readUpdateCheck() {
+  try {
+    return JSON.parse(readFileSync(updateCheckPath, 'utf8'));
+  } catch {
+    return null;
+  }
+}
+
+function writeUpdateCheck(data) {
+  try {
+    mkdirSync(vibeletDir, { recursive: true });
+    writeFileSync(updateCheckPath, JSON.stringify(data, null, 2) + '\n', 'utf8');
+  } catch {
+    // Best-effort; failing to persist is fine.
+  }
+}
+
+function compareVersions(a, b) {
+  const pa = a.split('.').map(Number);
+  const pb = b.split('.').map(Number);
+  for (let i = 0; i < 3; i++) {
+    if ((pa[i] || 0) < (pb[i] || 0)) return -1;
+    if ((pa[i] || 0) > (pb[i] || 0)) return 1;
+  }
+  return 0;
+}
+
+function checkForUpdateFromCache() {
+  const cached = readUpdateCheck();
+  if (!cached?.latestVersion) return;
+  if (compareVersions(packageJson.version, cached.latestVersion) < 0) {
+    updateMessage =
+      `\x1b[33m╭───────────────────────────────────────────╮\x1b[0m\n` +
+      `\x1b[33m│\x1b[0m  Update available: \x1b[90m${packageJson.version}\x1b[0m → \x1b[32m${cached.latestVersion}\x1b[0m${' '.repeat(Math.max(0, 14 - packageJson.version.length - cached.latestVersion.length))}\x1b[33m│\x1b[0m\n` +
+      `\x1b[33m│\x1b[0m  Run \x1b[36mnpx @vibelet/cli@latest\x1b[0m to upgrade   \x1b[33m│\x1b[0m\n` +
+      `\x1b[33m╰───────────────────────────────────────────╯\x1b[0m`;
+  }
+}
+
+function fetchLatestVersionInBackground() {
+  const cached = readUpdateCheck();
+  const now = Date.now();
+  if (cached?.checkedAt && now - cached.checkedAt < UPDATE_CHECK_INTERVAL_MS) {
+    return; // Checked recently; skip.
+  }
+
+  // Fire-and-forget: spawn a detached process to query the registry so we
+  // never block the CLI. Results are read on the *next* invocation.
+  const script = `
+    const https = await import('node:https');
+    const fs = await import('node:fs');
+    const url = 'https://registry.npmjs.org/@vibelet/cli/latest';
+    https.get(url, { headers: { 'Accept': 'application/json' }, timeout: 8000 }, (res) => {
+      let data = '';
+      res.on('data', (c) => data += c);
+      res.on('end', () => {
+        try {
+          const version = JSON.parse(data).version;
+          if (version) {
+            fs.writeFileSync(${JSON.stringify(updateCheckPath)}, JSON.stringify({
+              latestVersion: version,
+              checkedAt: Date.now(),
+            }, null, 2) + '\\n');
+          }
+        } catch {}
+      });
+    }).on('error', () => {});
+  `;
+
+  const child = spawn(process.execPath, ['--input-type=module', '-e', script], {
+    detached: true,
+    stdio: 'ignore',
+  });
+  child.unref();
+}
+
 // ─── Helpers ────────────────────────────────────────────────────────────────────
 
 function fail(message, details) {
@@ -563,6 +645,10 @@ function clearRelayConfig() {
 }
 
 async function main() {
+  // Update check: read cached result (sync, instant) and spawn background fetch.
+  checkForUpdateFromCache();
+  fetchLatestVersionInBackground();
+
   const relayArg = parseRelayArg();
   const hostArg = parseNamedArg('host', '100.x.x.x');
   const fallbackHostsArg = parseNamedArg('fallback-hosts', '100.x.x.x,192.168.1.x');