feat: add custom addon path functionality and enhance GPU acceleration settings

Author: buxuku

main/helpers/addonManager.ts

@@ -172,7 +172,7 @@ export function registerInstalledAddon(
 }
 
 /**
- * 选择加速包版本
+ * 选择加速包版本（与自定义路径互斥）
  */
 export function selectAddonVersion(version: CudaVersion | null): void {
   const config = getAddonConfig();
@@ -183,10 +183,56 @@ export function selectAddonVersion(version: CudaVersion | null): void {
   }
 
   config.selectedVersion = version;
+  // 选择版本时清除自定义路径（互斥）
+  if (version) {
+    config.customAddonPath = null;
+  }
   saveAddonConfig(config);
   logMessage(`Selected addon version: ${version}`, 'info');
 }
 
+/**
+ * 设置自定义 addon.node 文件路径（与版本选择互斥）
+ */
+export function setCustomAddonPath(filePath: string | null): void {
+  const config = getAddonConfig();
+
+  if (filePath) {
+    // 验证文件是否存在
+    if (!fs.existsSync(filePath)) {
+      throw new Error(`File not found: ${filePath}`);
+    }
+    // 验证文件扩展名
+    if (!filePath.endsWith('.node')) {
+      throw new Error('File must have .node extension');
+    }
+  }
+
+  config.customAddonPath = filePath;
+  // 设置自定义路径时清除版本选择（互斥）
+  if (filePath) {
+    config.selectedVersion = null;
+  }
+  saveAddonConfig(config);
+  logMessage(`Custom addon path set: ${filePath}`, 'info');
+}
+
+/**
+ * 获取自定义 addon.node 文件路径
+ */
+export function getCustomAddonPath(): string | null {
+  const config = getAddonConfig();
+  const customPath = config.customAddonPath;
+
+  // 验证路径是否仍然有效
+  if (customPath && !fs.existsSync(customPath)) {
+    logMessage(`Custom addon path no longer exists: ${customPath}`, 'warning');
+    return customPath; // 仍然返回路径，让 UI 层决定如何处理
+  }
+
+  return customPath || null;
+}
+
 /**
  * 获取当前选中的加速包版本
  */
@@ -355,14 +401,17 @@ export function getAddonSummary(): {
   selectedVersion: CudaVersion | null;
   installedCount: number;
   installedVersions: CudaVersion[];
+  customAddonPath: string | null;
 } {
   const installed = getInstalledAddons();
   const selected = getSelectedAddonVersion();
+  const customPath = getCustomAddonPath();
 
   return {
     hasInstalled: installed.length > 0,
     selectedVersion: selected,
     installedCount: installed.length,
     installedVersions: installed.map((i) => i.version),
+    customAddonPath: customPath,
   };
 }

main/helpers/addonVersions.ts

@@ -123,6 +123,15 @@ function shouldForceUpdate(): boolean {
   return process.env.DEV_FORCE_ADDON_UPDATE === 'true';
 }
 
+/**
+ * 标准化版本号格式
+ * 统一将分隔符转换为点号，以便正确比较
+ * 例如 "2026-02-06" -> "2026.02.06"
+ */
+function normalizeVersion(version: string): string {
+  return version.replace(/-/g, '.');
+}
+
 /**
  * 检查指定版本是否有更新
  */
@@ -154,7 +163,10 @@ export async function checkVersionUpdate(
   }
 
   const remoteInfo = remoteVersions[version];
-  const hasUpdate = remoteInfo.version > installedInfo.remoteVersion;
+  // 标准化版本号格式后再比较，避免 "2026.02.06" vs "2026-02-06" 因分隔符不同导致误判
+  const normalizedRemote = normalizeVersion(remoteInfo.version);
+  const normalizedLocal = normalizeVersion(installedInfo.remoteVersion);
+  const hasUpdate = normalizedRemote > normalizedLocal;
 
   return {
     cudaVersion: version,

main/helpers/ipcAddonHandlers.ts

@@ -1,4 +1,4 @@
-import { ipcMain, BrowserWindow } from 'electron';
+import { ipcMain, BrowserWindow, dialog } from 'electron';
 import { logMessage } from './storeManager';
 import { getCudaEnvironment, isPlatformCudaCapable } from './cudaUtils';
 import {
@@ -9,6 +9,8 @@ import {
   removeAddon,
   registerInstalledAddon,
   getAddonSummary,
+  setCustomAddonPath,
+  getCustomAddonPath,
 } from './addonManager';
 import {
   getAddonDownloader,
@@ -110,7 +112,8 @@ export function registerAddonIpcHandlers(): void {
             const remoteInfo = await getRemoteVersionInfo(config.cudaVersion);
             registerInstalledAddon(
               config.cudaVersion,
-              remoteInfo?.version || new Date().toISOString().split('T')[0],
+              remoteInfo?.version ||
+                new Date().toISOString().split('T')[0].replace(/-/g, '.'),
             );
             // 自动选中刚下载的版本
             selectAddonVersion(config.cudaVersion);
@@ -219,5 +222,54 @@ export function registerAddonIpcHandlers(): void {
     },
   );
 
+  // 选择自定义 addon.node 文件
+  ipcMain.handle('select-addon-file', async () => {
+    try {
+      const result = await dialog.showOpenDialog({
+        properties: ['openFile'],
+        title: 'Select addon.node file',
+        filters: [
+          {
+            name: 'Node Addon',
+            extensions: ['node'],
+          },
+        ],
+      });
+
+      if (result.canceled || !result.filePaths[0]) {
+        return { filePath: null, canceled: true };
+      }
+
+      return { filePath: result.filePaths[0], canceled: false };
+    } catch (error) {
+      logMessage(`Error selecting addon file: ${error}`, 'error');
+      return { filePath: null, canceled: true, error: String(error) };
+    }
+  });
+
+  // 设置自定义 addon.node 路径
+  ipcMain.handle(
+    'set-custom-addon-path',
+    async (event, filePath: string | null) => {
+      try {
+        setCustomAddonPath(filePath);
+        return { success: true };
+      } catch (error) {
+        logMessage(`Error setting custom addon path: ${error}`, 'error');
+        return { success: false, error: String(error) };
+      }
+    },
+  );
+
+  // 获取自定义 addon.node 路径
+  ipcMain.handle('get-custom-addon-path', async () => {
+    try {
+      return getCustomAddonPath();
+    } catch (error) {
+      logMessage(`Error getting custom addon path: ${error}`, 'error');
+      return null;
+    }
+  });
+
   logMessage('Addon IPC handlers registered', 'info');
 }

main/helpers/updater.ts

@@ -113,27 +113,27 @@ export function setupAutoUpdater(mainWindow: BrowserWindow) {
     });
 
     // 当自动更新出错时，提供手动下载选项
-    if (process.platform === 'darwin') {
-      // 针对Mac平台的特殊处理
-      dialog
-        .showMessageBox(mainWindow, {
-          type: 'info',
-          title: '更新失败',
-          message: '自动更新失败',
-          detail:
-            '由于macOS系统限制，自动更新失败。您可以手动下载并安装最新版本。',
-          buttons: ['手动下载', '取消'],
-          cancelId: 1,
-        })
-        .then(({ response }) => {
-          if (response === 0) {
-            // 打开GitHub发布页面，让用户手动下载
-            const releaseUrl =
-              'https://github.com/buxuku/SmartSub/releases/latest';
-            require('electron').shell.openExternal(releaseUrl);
-          }
-        });
-    }
+    // if (process.platform === 'darwin') {
+    //   // 针对Mac平台的特殊处理
+    //   dialog
+    //     .showMessageBox(mainWindow, {
+    //       type: 'info',
+    //       title: '更新失败',
+    //       message: '自动更新失败',
+    //       detail:
+    //         '由于macOS系统限制，自动更新失败。您可以手动下载并安装最新版本。',
+    //       buttons: ['手动下载', '取消'],
+    //       cancelId: 1,
+    //     })
+    //     .then(({ response }) => {
+    //       if (response === 0) {
+    //         // 打开GitHub发布页面，让用户手动下载
+    //         const releaseUrl =
+    //           'https://github.com/buxuku/SmartSub/releases/latest';
+    //         require('electron').shell.openExternal(releaseUrl);
+    //       }
+    //     });
+    // }
   });
 
   // 设置IPC处理程序

main/helpers/whisper.ts

@@ -12,6 +12,7 @@ import {
   getAddonVersionDir,
   hasDependentLibs,
   isAddonInstalled,
+  getCustomAddonPath,
 } from './addonManager';
 
 export const getPath = (key?: string) => {
@@ -288,34 +289,58 @@ export async function loadWhisperAddon(model: string) {
 
   // 检查是否是可能支持 CUDA 的平台
   if (isPlatformCudaCapable() && useCuda) {
-    // 获取用户选择的加速包版本
-    const selectedVersion = getSelectedAddonVersion();
-
-    if (selectedVersion && isAddonInstalled(selectedVersion)) {
-      // 从用户数据目录加载
-      const versionDir = getAddonVersionDir(selectedVersion);
-      const userAddonPath = path.join(versionDir, 'addon.node');
-
-      if (fs.existsSync(userAddonPath)) {
-        // 检查并设置依赖库路径（必须在 dlopen 之前）
-        if (hasDependentLibs(versionDir)) {
-          setupLibraryPath(versionDir);
-        }
+    // 优先检查自定义 addon.node 路径
+    const customPath = getCustomAddonPath();
+
+    if (customPath && fs.existsSync(customPath)) {
+      // 使用自定义路径
+      const customDir = path.dirname(customPath);
+      if (hasDependentLibs(customDir)) {
+        setupLibraryPath(customDir);
+      }
+      addonPath = customPath;
+      logMessage(`Loading custom addon from: ${addonPath}`, 'info');
+    } else if (customPath) {
+      // 自定义路径已设置但文件不存在
+      logMessage(
+        `Custom addon path not found: ${customPath}, falling back to default`,
+        'warning',
+      );
+      addonPath = path.join(getExtraResourcesPath(), 'addons', 'addon.node');
+    } else {
+      // 获取用户选择的加速包版本
+      const selectedVersion = getSelectedAddonVersion();
+
+      if (selectedVersion && isAddonInstalled(selectedVersion)) {
+        // 从用户数据目录加载
+        const versionDir = getAddonVersionDir(selectedVersion);
+        const userAddonPath = path.join(versionDir, 'addon.node');
+
+        if (fs.existsSync(userAddonPath)) {
+          // 检查并设置依赖库路径（必须在 dlopen 之前）
+          if (hasDependentLibs(versionDir)) {
+            setupLibraryPath(versionDir);
+          }
 
-        addonPath = userAddonPath;
-        logMessage(`Loading CUDA addon from userData: ${addonPath}`, 'info');
+          addonPath = userAddonPath;
+          logMessage(`Loading CUDA addon from userData: ${addonPath}`, 'info');
+        } else {
+          // 用户数据目录的 addon 不存在，回退到默认版本
+          logMessage(
+            `Selected addon version ${selectedVersion} not found, falling back to default`,
+            'warning',
+          );
+          addonPath = path.join(
+            getExtraResourcesPath(),
+            'addons',
+            'addon.node',
+          );
+        }
       } else {
-        // 用户数据目录的 addon 不存在，回退到默认版本
-        logMessage(
-          `Selected addon version ${selectedVersion} not found, falling back to default`,
-          'warning',
-        );
+        // 没有安装加速包，使用默认的 no-cuda 版本
         addonPath = path.join(getExtraResourcesPath(), 'addons', 'addon.node');
+        logMessage('No CUDA addon installed, using default addon', 'info');
       }
-    } else {
-      // 没有安装加速包，使用默认的 no-cuda 版本
-      addonPath = path.join(getExtraResourcesPath(), 'addons', 'addon.node');
-      logMessage('No CUDA addon installed, using default addon', 'info');
     }
   } else if (
     platform === 'darwin' &&

package.json

@@ -2,7 +2,7 @@
   "private": true,
   "name": "smartsub",
   "description": "视频转字幕，字幕翻译软件",
-  "version": "2.12.0-beta.0",
+  "version": "2.14.0-beta.0",
   "author": "buxuku <buxuku@gmail.com>",
   "main": "app/background.js",
   "scripts": {