@agent-webui/ai-desk 1.0.59 → 1.0.61-beta1

package/bin/aidesk.js

@@ -6,7 +6,8 @@ const { spawnSync } = require('child_process');
 
 const PACKAGE_NAME = '@agent-webui/ai-desk';
 const SKIP_UPDATE_ENV = 'AI_DESK_SKIP_SELF_UPDATE';
-const UPDATE_COMMANDS = new Set(['start', 'restart']);
+const UPDATE_COMMANDS = new Set(['start', 'restart', 'upgrade']);
+const OFFICIAL_NPM_REGISTRY = 'https://registry.npmjs.org/';
 
 function readJSON(filePath) {
   return JSON.parse(fs.readFileSync(filePath, 'utf8'));
@@ -68,14 +69,40 @@ function runNpm(args, options = {}) {
   });
 }
 
+function npmOutput(result) {
+  return `${result.stderr || ''}${result.stdout || ''}`.trim();
+}
+
+function withOfficialRegistry(args) {
+  return [...args, `--registry=${OFFICIAL_NPM_REGISTRY}`];
+}
+
+function runNpmWithRegistryFallback(args, options = {}) {
+  const primaryResult = runNpm(withOfficialRegistry(args), options);
+  if (primaryResult.status === 0) {
+    return primaryResult;
+  }
+
+  const fallbackResult = runNpm(args, options);
+  if (fallbackResult.status === 0) {
+    return fallbackResult;
+  }
+
+  fallbackResult.primaryError = npmOutput(primaryResult);
+  return fallbackResult;
+}
+
 function latestPublishedVersion() {
-  const result = runNpm(['view', PACKAGE_NAME, 'version'], {
+  const result = runNpmWithRegistryFallback(['view', PACKAGE_NAME, 'version'], {
     stdio: ['ignore', 'pipe', 'pipe'],
   });
 
   if (result.status !== 0) {
-    const output = `${result.stderr || ''}${result.stdout || ''}`.trim();
-    throw new Error(output || 'Unable to check npm registry');
+    const output = npmOutput(result);
+    const primaryError = result.primaryError ? `Official registry error: ${result.primaryError}\n` : '';
+    const fallbackError = output ? `Fallback registry error: ${output}` : '';
+    const errorMessage = `${primaryError}${fallbackError}`.trim();
+    throw new Error(errorMessage || 'Unable to check npm registry');
   }
 
   return String(result.stdout || '').trim();
@@ -133,13 +160,13 @@ function installLatestVersion(latestVersion) {
   const installTarget = `${PACKAGE_NAME}@${latestVersion}`;
 
   if (projectRoot) {
-    return runNpm(['install', installTarget, '--include=optional'], {
+    return runNpmWithRegistryFallback(['install', installTarget, '--include=optional'], {
       cwd: projectRoot,
       stdio: 'inherit',
     });
   }
 
-  return runNpm(['install', '-g', installTarget, '--include=optional'], {
+  return runNpmWithRegistryFallback(['install', '-g', installTarget, '--include=optional'], {
     stdio: 'inherit',
   });
 }
@@ -207,11 +234,18 @@ function updateBeforeCommand(args) {
   }
 
   const installedVersion = currentVersion();
-  if (compareVersions(latestVersion, installedVersion) <= 0) {
+  const forceUpgrade = command === 'upgrade' && args.includes('--force');
+  const alreadyLatest = compareVersions(latestVersion, installedVersion) <= 0;
+  if (alreadyLatest && !forceUpgrade) {
+    if (command === 'upgrade') {
+      console.log(`AI Desk is already up to date (${installedVersion}).`);
+      process.exit(0);
+    }
     return;
   }
 
-  console.log(`Updating AI Desk from ${installedVersion} to ${latestVersion}...`);
+  const verb = alreadyLatest ? 'Reinstalling' : 'Updating';
+  console.log(`${verb} AI Desk from ${installedVersion} to ${latestVersion}...`);
   const result = installLatestVersion(latestVersion);
   if (result.error) {
     throw result.error;
@@ -222,6 +256,9 @@ function updateBeforeCommand(args) {
 
   const updatedVersion = currentVersion();
   console.log(`AI Desk updated to ${updatedVersion}.`);
+  if (command === 'upgrade') {
+    process.exit(0);
+  }
   console.log(`Restarting aidesk ${command} with the updated CLI...`);
   restartWithUpdatedCli(args);
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agent-webui/ai-desk",
-  "version": "1.0.59",
+  "version": "1.0.61-beta1",
   "description": "Single-install AI Desk package with daemon and default harnesses",
   "bin": {
     "aidesk": "bin/aidesk.js"
@@ -14,8 +14,8 @@
     "README.md"
   ],
   "dependencies": {
-    "@agent-webui/ai-desk-daemon": "1.0.59",
-    "@agent-webui/ai-desk-harness-gimp": "1.0.59"
+    "@agent-webui/ai-desk-daemon": "1.0.61-beta1",
+    "@agent-webui/ai-desk-harness-gimp": "1.0.61-beta1"
   },
   "license": "MIT"
 }