feat: added action to reload extension in case of error

wladimiiir · wladimiiir · commit 8c138c0bb381 · 2026-07-23T22:44:47.000+02:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -12,6 +12,7 @@
 - avoid showing password dialog on start up in some cases
 - added getOpenProjects to Extension Context API
 - correctly showing Updated Files in repos without any commits
+- added action to reload extension in case of error
 
 ## [0.75.0]
 
diff --git a/packages/common/src/api.ts b/packages/common/src/api.ts
@@ -161,6 +161,7 @@ export interface ApplicationAPI {
   installExtension: (extensionId: string, repositoryUrl: string, projectDir?: string) => Promise<ExtensionOperationResult>;
   uninstallExtension: (extensionId: string, projectDir?: string) => Promise<boolean>;
   updateExtension: (extensionId: string, repositoryUrl: string, projectDir?: string) => Promise<ExtensionOperationResult>;
+  reloadExtension: (filePath: string, projectDir?: string) => Promise<boolean>;
   getExtensionUIComponents: (placement?: string, projectDir?: string, taskId?: string) => Promise<ExtensionUIComponent[]>;
   getUIExtensionData: (extensionId: string, componentId: string, projectDir?: string, taskId?: string) => Promise<unknown>;
   executeUIExtensionAction: (
diff --git a/packages/common/src/locales/en.json b/packages/common/src/locales/en.json
@@ -497,10 +497,13 @@
       "updating": "Updating...",
       "uninstall": "Uninstall",
       "uninstalling": "Uninstalling...",
+      "reload": "Reload",
+      "reloading": "Reloading...",
       "success": {
         "install": "Successfully installed {{name}}",
         "update": "Successfully updated {{name}}",
         "uninstall": "Successfully uninstalled {{name}}",
+        "reload": "Successfully reloaded {{name}}",
         "saveSettings": "Successfully saved settings for {{name}}"
       },
       "errors": {
@@ -510,6 +513,7 @@
         "install": "Failed to install extension. Check logs for details.",
         "update": "Failed to update extension. Check logs for details.",
         "uninstall": "Failed to uninstall extension",
+        "reload": "Failed to reload extension",
         "saveSettings": "Failed to save extension settings",
         "settingsNotAvailable": "Extension settings are not available",
         "repositoryExists": "This repository is already added",
diff --git a/packages/common/src/locales/ru.json b/packages/common/src/locales/ru.json
@@ -491,10 +491,13 @@
       "updating": "Обновление...",
       "uninstall": "Удалить",
       "uninstalling": "Удаление...",
+      "reload": "Перезагрузить",
+      "reloading": "Перезагрузка...",
       "success": {
         "install": "«{{name}}» успешно установлен",
         "update": "«{{name}}» успешно обновлён",
         "uninstall": "«{{name}}» успешно удалён",
+        "reload": "«{{name}}» успешно перезагружен",
         "saveSettings": "Настройки для «{{name}}» успешно сохранены"
       },
       "errors": {
@@ -504,6 +507,7 @@
         "install": "Не удалось установить расширение. Подробности см. в логах.",
         "update": "Не удалось обновить расширение. Подробности см. в логах.",
         "uninstall": "Не удалось удалить расширение",
+        "reload": "Не удалось перезагрузить расширение",
         "saveSettings": "Не удалось сохранить настройки расширения",
         "settingsNotAvailable": "Настройки расширения недоступны",
         "repositoryExists": "Этот репозиторий уже добавлен",
diff --git a/packages/common/src/locales/zh.json b/packages/common/src/locales/zh.json
@@ -496,10 +496,13 @@
       "updating": "更新中...",
       "uninstall": "卸载",
       "uninstalling": "卸载中...",
+      "reload": "重新加载",
+      "reloading": "重新加载中...",
       "success": {
         "install": "成功安装 {{name}}",
         "update": "成功更新 {{name}}",
         "uninstall": "成功卸载 {{name}}",
+        "reload": "成功重新加载 {{name}}",
         "saveSettings": "成功保存 {{name}} 的设置"
       },
       "errors": {
@@ -509,6 +512,7 @@
         "install": "安装扩展失败。请查看日志了解详情。",
         "update": "更新扩展失败。请查看日志了解详情。",
         "uninstall": "卸载扩展失败",
+        "reload": "重新加载扩展失败",
         "saveSettings": "保存扩展设置失败",
         "settingsNotAvailable": "扩展设置不可用",
         "repositoryExists": "此仓库已添加",
diff --git a/src/main/events-handler.ts b/src/main/events-handler.ts
@@ -1250,6 +1250,18 @@ export class EventsHandler {
     return await this.extensionManager.uninstallExtension(extensionId, projectDir);
   }
 
+  async reloadExtension(filePath: string, projectDir?: string): Promise<boolean> {
+    let project: Project | undefined = undefined;
+    if (projectDir) {
+      project = this.projectManager.getProject(projectDir);
+      if (!project) {
+        throw new Error('Project not found');
+      }
+    }
+
+    return await this.extensionManager.reloadExtension(filePath, project);
+  }
+
   async updateExtension(extensionId: string, repositoryUrl: string, projectDir?: string): Promise<ExtensionOperationResult> {
     let project: Project | undefined = undefined;
     if (projectDir) {
diff --git a/src/main/extensions/extension-manager.ts b/src/main/extensions/extension-manager.ts
@@ -623,6 +623,36 @@ export class ExtensionManager {
     logger.debug(`[Extensions] Unloaded extension: ${filePath}`);
   }
 
+  async reloadExtension(filePath: string, project?: Project): Promise<boolean> {
+    const extension = this.findExtensionByPath(filePath);
+    if (!extension) {
+      logger.warn(`[Extensions] Cannot reload: extension not found at ${filePath}`);
+      return false;
+    }
+
+    logger.info(`[Extensions] Reloading extension: ${extension.metadata.name} (${filePath})`);
+
+    const hadUIComponents = extension.instance.getUIComponents !== undefined;
+
+    await this.unloadExtension(filePath);
+    const { success, hasUIComponents } = await this.loadAndInitializeExtension(filePath, project);
+
+    if (success && (hasUIComponents || hadUIComponents)) {
+      this.eventManager.sendExtensionUIRefresh({
+        projectDir: project?.baseDir,
+        reloadComponents: true,
+      });
+    }
+
+    const providers = this.getProviders(project);
+    if (providers.length > 0) {
+      this.modelManager.registerExtensionProviders(providers);
+    }
+
+    this.debouncedNotifyListeners();
+    return success;
+  }
+
   private findExtensionByPath(filePath: string): LoadedExtension | undefined {
     const extensions = this.registry.getExtensions();
     return extensions.find((ext) => ext.filePath === filePath);
diff --git a/src/main/ipc-handlers.ts b/src/main/ipc-handlers.ts
@@ -328,6 +328,10 @@ export const setupIpcHandlers = (eventsHandler: EventsHandler, serverController:
     return await eventsHandler.updateExtension(extensionId, repositoryUrl, projectDir);
   });
 
+  ipcMain.handle('reload-extension', async (_, filePath: string, projectDir?: string) => {
+    return await eventsHandler.reloadExtension(filePath, projectDir);
+  });
+
   ipcMain.handle('get-extension-ui-components', (_, placement?: string, projectDir?: string, taskId?: string) => {
     return eventsHandler.getUIComponents(placement, projectDir, taskId);
   });
diff --git a/src/main/server/rest-api/extensions-api.ts b/src/main/server/rest-api/extensions-api.ts
@@ -36,6 +36,11 @@ const UpdateExtensionSchema = z.object({
   projectDir: z.string().optional(),
 });
 
+const ReloadExtensionSchema = z.object({
+  filePath: z.string().min(1, 'File path is required'),
+  projectDir: z.string().optional(),
+});
+
 export class ExtensionsApi extends BaseApi {
   constructor(private readonly eventsHandler: EventsHandler) {
     super();
@@ -147,6 +152,26 @@ export class ExtensionsApi extends BaseApi {
       }),
     );
 
+    // Reload extension
+    router.post(
+      '/extensions/reload',
+      this.handleRequest(async (req, res) => {
+        const parsed = this.validateRequest(ReloadExtensionSchema, req.body, res);
+        if (!parsed) {
+          return;
+        }
+
+        const { filePath, projectDir } = parsed;
+        const success = await this.eventsHandler.reloadExtension(filePath, projectDir);
+
+        if (success) {
+          res.status(200).json({ message: 'Extension reloaded successfully', success: true });
+        } else {
+          res.status(500).json({ message: 'Failed to reload extension', success: false });
+        }
+      }),
+    );
+
     // Get extension UI components
     router.get(
       '/extensions/ui-components',
diff --git a/src/preload/index.ts b/src/preload/index.ts
@@ -134,6 +134,7 @@ const api: ApplicationAPI = {
   uninstallExtension: (extensionId: string, projectDir?: string) => ipcRenderer.invoke('uninstall-extension', extensionId, projectDir),
   updateExtension: (extensionId: string, repositoryUrl: string, projectDir?: string) =>
     ipcRenderer.invoke('update-extension', extensionId, repositoryUrl, projectDir),
+  reloadExtension: (filePath: string, projectDir?: string) => ipcRenderer.invoke('reload-extension', filePath, projectDir),
   getExtensionUIComponents: (placement?: string, projectDir?: string, taskId?: string) =>
     ipcRenderer.invoke('get-extension-ui-components', placement, projectDir, taskId),
   getUIExtensionData: (extensionId: string, componentId: string, projectDir?: string, taskId?: string) =>
diff --git a/src/renderer/src/__tests__/mocks/api.ts b/src/renderer/src/__tests__/mocks/api.ts
@@ -268,6 +268,7 @@ export const createMockApi = (overrides: Partial<ApplicationAPI> = {}): MockedOb
     installExtension: vi.fn((): Promise<ExtensionOperationResult> => Promise.resolve({ success: true })),
     uninstallExtension: vi.fn((): Promise<boolean> => Promise.resolve(true)),
     updateExtension: vi.fn((): Promise<ExtensionOperationResult> => Promise.resolve({ success: true })),
+    reloadExtension: vi.fn((): Promise<boolean> => Promise.resolve(true)),
     getExtensionUIComponents: vi.fn((): Promise<ExtensionUIComponent[]> => Promise.resolve([])),
     getUIExtensionData: vi.fn((): Promise<unknown> => Promise.resolve({})),
     executeUIExtensionAction: vi.fn((): Promise<unknown> => Promise.resolve({})),
diff --git a/src/renderer/src/api/browser-api.ts b/src/renderer/src/api/browser-api.ts
@@ -1302,6 +1302,13 @@ export class BrowserApi implements ApplicationAPI {
     });
   }
 
+  reloadExtension(filePath: string, projectDir?: string): Promise<boolean> {
+    return this.post('/extensions/reload', {
+      filePath,
+      projectDir,
+    });
+  }
+
   getExtensionUIComponents(placement?: string, projectDir?: string, taskId?: string): Promise<ExtensionUIComponent[]> {
     return this.get('/extensions/ui-components', {
       projectDir,
diff --git a/src/renderer/src/components/settings/ExtensionCard.tsx b/src/renderer/src/components/settings/ExtensionCard.tsx
@@ -37,12 +37,14 @@ type Props = {
   isUninstalling?: boolean;
   isInstalling?: boolean;
   isUpdating?: boolean;
+  isReloading?: boolean;
   hasUpdate?: boolean;
   installedFilePath?: string;
   onToggle?: (extensionFilePath: string, isDisabled: boolean) => void;
   onUninstall?: (exensionFilePath: string) => void;
   onInstall?: (extension: AvailableExtension) => void;
   onUpdate?: () => void;
+  onReload?: (extensionFilePath: string) => void;
 };
 
 export const ExtensionCard = ({
@@ -52,12 +54,14 @@ export const ExtensionCard = ({
   isUninstalling = false,
   isInstalling = false,
   isUpdating = false,
+  isReloading = false,
   hasUpdate = false,
   installedFilePath,
   onToggle,
   onUninstall,
   onInstall,
   onUpdate,
+  onReload,
 }: Props) => {
   const { t } = useTranslation();
   const [isExpanded, setIsExpanded] = useState(false);
@@ -97,6 +101,12 @@ export const ExtensionCard = ({
     onUpdate?.();
   };
 
+  const handleReload = () => {
+    if (onReload && filePath) {
+      onReload(filePath);
+    }
+  };
+
   const handleOpenSettings = () => {
     setIsSettingsOpen(true);
   };
@@ -175,6 +185,14 @@ export const ExtensionCard = ({
                   />
                 )}
 
+                <IconButton
+                  icon={<FaSync className={clsx('w-3.5 h-3.5', isReloading && 'animate-spin')} />}
+                  onClick={handleReload}
+                  disabled={isReloading}
+                  tooltip={isReloading ? t('settings.extensions.reloading') : t('settings.extensions.reload')}
+                  className="p-1"
+                />
+
                 <Toggle checked={!isDisabled} onChange={handleToggleDisabled} aria-label={t('settings.extensions.enabled')} />
 
                 {hasUpdate && (
diff --git a/src/renderer/src/components/settings/ExtensionsSettings.tsx b/src/renderer/src/components/settings/ExtensionsSettings.tsx
@@ -69,6 +69,7 @@ export const ExtensionsSettings = ({ settings, setSettings, openProjects = [], s
   const [installingExtensions, setInstallingExtensions] = useState<Set<string>>(new Set());
   const [uninstallingExtensions, setUninstallingExtensions] = useState<Set<string>>(new Set());
   const [updatingExtensions, setUpdatingExtensions] = useState<Set<string>>(new Set());
+  const [reloadingExtensions, setReloadingExtensions] = useState<Set<string>>(new Set());
   const [newRepositoryUrl, setNewRepositoryUrl] = useState('');
   const [expandedRepositories, setExpandedRepositories] = useState<Set<string>>(new Set());
   const [searchQuery, setSearchQuery] = useState('');
@@ -250,6 +251,27 @@ export const ExtensionsSettings = ({ settings, setSettings, openProjects = [], s
     }
   };
 
+  const handleReload = async (extensionFilePath: string) => {
+    setReloadingExtensions((prev) => new Set(prev).add(extensionFilePath));
+    try {
+      const success = await api.reloadExtension(extensionFilePath, projectDir);
+      if (success) {
+        showSuccessNotification(t('settings.extensions.success.reload', { name: extensionFilePath }));
+        await loadInstalledExtensions();
+      } else {
+        showErrorNotification(t('settings.extensions.errors.reload'));
+      }
+    } catch {
+      showErrorNotification(t('settings.extensions.errors.reload'));
+    } finally {
+      setReloadingExtensions((prev) => {
+        const next = new Set(prev);
+        next.delete(extensionFilePath);
+        return next;
+      });
+    }
+  };
+
   const handleAddRepository = async () => {
     const trimmedUrl = newRepositoryUrl.trim();
     if (!trimmedUrl) {
@@ -405,10 +427,12 @@ export const ExtensionsSettings = ({ settings, setSettings, openProjects = [], s
               isDisabled={disabledExtensions.includes(extension.filePath)}
               isUninstalling={uninstallingExtensions.has(extension.filePath)}
               isUpdating={updatingExtensions.has(extension.filePath)}
+              isReloading={reloadingExtensions.has(extension.filePath)}
               hasUpdate={extensionHasUpdate}
               onUpdate={availableExt ? () => handleUpdate(availableExt) : undefined}
               onToggle={handleToggleDisabled}
               onUninstall={handleUninstall}
+              onReload={handleReload}
             />
           );
         })}
@@ -476,11 +500,13 @@ export const ExtensionsSettings = ({ settings, setSettings, openProjects = [], s
             isDisabled={disabledExtensions.includes(installedExtension.filePath)}
             isUninstalling={uninstallingExtensions.has(installedExtension.filePath)}
             isUpdating={updatingExtensions.has(installedExtension.filePath)}
+            isReloading={reloadingExtensions.has(installedExtension.filePath)}
             hasUpdate={extensionHasUpdate}
             installedFilePath={installedExtension.filePath}
             onToggle={handleToggleDisabled}
             onUninstall={handleUninstall}
             onUpdate={() => handleUpdate(extension)}
+            onReload={handleReload}
           />
         );
       }
