Extensions update command (#8100)

packages/cli/src/ui/commands/extensionsCommand.ts

@@ -4,43 +4,131 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
+import {
+  updateExtensionByName,
+  updateAllUpdatableExtensions,
+  type ExtensionUpdateInfo,
+} from '../../config/extension.js';
+import { getErrorMessage } from '../../utils/errors.js';
+import { MessageType } from '../types.js';
 import {
   type CommandContext,
   type SlashCommand,
   CommandKind,
 } from './types.js';
-import { MessageType } from '../types.js';
 
-export const extensionsCommand: SlashCommand = {
-  name: 'extensions',
-  description: 'list active extensions',
-  kind: CommandKind.BUILT_IN,
-  action: async (context: CommandContext): Promise<void> => {
-    const activeExtensions = context.services.config
-      ?.getExtensions()
-      .filter((ext) => ext.isActive);
-    if (!activeExtensions || activeExtensions.length === 0) {
+async function listAction(context: CommandContext) {
+  const activeExtensions = context.services.config
+    ?.getExtensions()
+    .filter((ext) => ext.isActive);
+  if (!activeExtensions || activeExtensions.length === 0) {
+    context.ui.addItem(
+      {
+        type: MessageType.INFO,
+        text: 'No active extensions.',
+      },
+      Date.now(),
+    );
+    return;
+  }
+
+  const extensionLines = activeExtensions.map(
+    (ext) => `  - \u001b[36m${ext.name} (v${ext.version})\u001b[0m`,
+  );
+  const message = `Active extensions:\n\n${extensionLines.join('\n')}\n`;
+
+  context.ui.addItem(
+    {
+      type: MessageType.INFO,
+      text: message,
+    },
+    Date.now(),
+  );
+}
+
+const updateOutput = (info: ExtensionUpdateInfo) =>
+  `Extension "${info.name}" successfully updated: ${info.originalVersion} → ${info.updatedVersion}.`;
+
+async function updateAction(context: CommandContext, args: string) {
+  const updateArgs = args.split(' ').filter((value) => value.length > 0);
+  const all = updateArgs.length === 1 && updateArgs[0] === '--all';
+  const names = all ? undefined : updateArgs;
+  let updateInfos: ExtensionUpdateInfo[] = [];
+  try {
+    if (all) {
+      updateInfos = await updateAllUpdatableExtensions();
+    } else if (names?.length) {
+      for (const name of names) {
+        updateInfos.push(await updateExtensionByName(name));
+      }
+    } else {
       context.ui.addItem(
         {
-          type: MessageType.INFO,
-          text: 'No active extensions.',
+          type: MessageType.ERROR,
+          text: 'Usage: /extensions update <extension-names>|--all',
         },
         Date.now(),
       );
       return;
     }
 
-    const extensionLines = activeExtensions.map(
-      (ext) => `  - \u001b[36m${ext.name} (v${ext.version})\u001b[0m`,
+    // Filter to the actually updated ones.
+    updateInfos = updateInfos.filter(
+      (info) => info.originalVersion !== info.updatedVersion,
     );
-    const message = `Active extensions:\n\n${extensionLines.join('\n')}\n`;
+
+    if (updateInfos.length === 0) {
+      context.ui.addItem(
+        {
+          type: MessageType.INFO,
+          text: 'No extensions to update.',
+        },
+        Date.now(),
+      );
+      return;
+    }
 
     context.ui.addItem(
       {
         type: MessageType.INFO,
-        text: message,
+        text: [
+          ...updateInfos.map((info) => updateOutput(info)),
+          'Restart gemini-cli to see the changes.',
+        ].join('\n'),
       },
       Date.now(),
     );
-  },
+  } catch (error) {
+    context.ui.addItem(
+      {
+        type: MessageType.ERROR,
+        text: getErrorMessage(error),
+      },
+      Date.now(),
+    );
+  }
+}
+
+const listExtensionsCommand: SlashCommand = {
+  name: 'list',
+  description: 'List active extensions',
+  kind: CommandKind.BUILT_IN,
+  action: listAction,
+};
+
+const updateExtensionsCommand: SlashCommand = {
+  name: 'update',
+  description: 'Update extensions. Usage: update <extension-names>|--all',
+  kind: CommandKind.BUILT_IN,
+  action: updateAction,
+};
+
+export const extensionsCommand: SlashCommand = {
+  name: 'extensions',
+  description: 'Manage extensions',
+  kind: CommandKind.BUILT_IN,
+  subCommands: [listExtensionsCommand, updateExtensionsCommand],
+  action: (context, args) =>
+    // Default to list if no subcommand is provided
+    listExtensionsCommand.action!(context, args),
 };