@becrafter/prompt-manager 0.0.16 → 0.0.19

package/app/desktop/main.js

@@ -1,496 +1,214 @@
-const electron = require('electron');
-const { app, BrowserWindow, Tray, Menu, clipboard, dialog, nativeImage, shell } = electron;
-const path = require('path');
-const fs = require('fs');
-const os = require('os');
-const { pathToFileURL } = require('url');
-const tar = require('tar');
-const { spawn } = require('child_process');
-
-let tray = null;
-let adminWindow = null;
-let serviceState = 'stopped';
-let currentServerState = null;
-let serverModule = null;
-let serverModuleVersion = 0;
-let serverModuleLoading = null;
-let isQuitting = false;
-let runtimeServerRoot = null;
-
-const desktopPackageJson = JSON.parse(
-  fs.readFileSync(path.join(__dirname, 'package.json'), 'utf8')
-);
-
-function resolveBaseServerRoot() {
-  return path.resolve(__dirname, '..', '..');
-}
-
-async function ensureRuntimeServerRoot() {
-  if (runtimeServerRoot) {
-    return runtimeServerRoot;
-  }
-
-  if (!app.isPackaged) {
-    runtimeServerRoot = resolveBaseServerRoot();
-    return runtimeServerRoot;
-  }
-
-  const packagedRoot = path.join(process.resourcesPath, 'prompt-manager');
-  const runtimeRoot = path.join(app.getPath('userData'), 'prompt-manager');
-
-  try {
-    await fs.promises.access(runtimeRoot, fs.constants.F_OK);
-  } catch (error) {
-    await fs.promises.mkdir(runtimeRoot, { recursive: true });
-    await fs.promises.cp(packagedRoot, runtimeRoot, { recursive: true });
-  }
-
-  runtimeServerRoot = runtimeRoot;
-  return runtimeServerRoot;
-}
-
-async function loadServerModule(options = {}) {
-  if (options.forceReload) {
-    serverModule = null;
-    serverModuleVersion += 1;
-  }
+/**
+ * 应用主入口
+ * 重构后的主程序，遵循SRP、KISS、DRY、YAGNI原则
+ */
 
-  if (serverModule) {
-    return serverModule;
-  }
+const { app, BrowserWindow, ipcMain } = require('electron');
+const path = require('path');
 
-  if (serverModuleLoading) {
-    return serverModuleLoading;
+// 导入管理器模块
+const AppState = require('./src/core/state-manager');
+const Logger = require('./src/core/logger');
+const ErrorHandler = require('./src/core/error-handler');
+const RuntimeManager = require('./src/services/runtime-manager');
+const ModuleLoader = require('./src/services/module-loader');
+const ServiceManager = require('./src/services/service-manager');
+const TrayManager = require('./src/ui/tray-manager');
+const UpdateManager = require('./src/services/update-manager');
+const AboutDialogManager = require('./src/ui/about-dialog-manager');
+const IconManager = require('./src/utils/icon-manager');
+
+class PromptManagerApp {
+  constructor() {
+    this.stateManager = new AppState();
+    this.logger = new Logger({ debugEnabled: false });
+    this.errorHandler = new ErrorHandler(this.logger);
+    this.runtimeManager = new RuntimeManager(this.logger, this.errorHandler);
+    this.moduleLoader = new ModuleLoader(this.logger, this.errorHandler);
+    this.iconManager = new IconManager();
+    this.serviceManager = new ServiceManager(this.logger, this.errorHandler, this.moduleLoader);
+    this.updateManager = new UpdateManager(this.logger, this.errorHandler, this.runtimeManager);
+    this.aboutDialogManager = new AboutDialogManager(this.logger, this.runtimeManager, this.iconManager);
+    this.trayManager = new TrayManager(this.logger, this.errorHandler, this.iconManager);
+    
+    this.isInitialized = false;
+  }
+
+  async initialize() {
+    if (this.isInitialized) return;
+
+    try {
+      this.logger.info('Initializing Prompt Manager Desktop Application');
+      
+      // 初始化日志系统
+      await this.logger.initialize();
+      
+      // 设置应用菜单
+      this.setupApplicationMenu();
+      
+      // 初始化系统托盘
+      await this.trayManager.initialize(this.stateManager);
+      
+      // 设置事件监听
+      this.setupEventListeners();
+      
+      // 确保运行时环境
+      const serverRoot = await this.runtimeManager.ensureRuntimeEnvironment();
+      this.stateManager.set('runtimeRoot', serverRoot);
+      
+      // 启动服务
+      await this.startService();
+      
+      this.isInitialized = true;
+      this.logger.info('Application initialized successfully');
+      
+    } catch (error) {
+      this.logger.error('Application initialization failed', error);
+      this.errorHandler.handleError('APP_INIT_FAILED', error, {
+        logFilePath: this.logger.getLogFilePath()
+      });
+      
+      throw error;
+    }
   }
 
-  const serverRoot = await ensureRuntimeServerRoot();
-  const serverEntry = path.join(serverRoot, 'packages', 'server', 'server.js');
-  const entryUrl = pathToFileURL(serverEntry);
-  entryUrl.searchParams.set('v', String(serverModuleVersion));
-  serverModuleLoading = import(entryUrl.href)
-    .then((mod) => {
-      serverModule = mod;
-      return mod;
-    })
-    .finally(() => {
-      serverModuleLoading = null;
+  setupApplicationMenu() {
+    if (process.platform === 'darwin') {
+      app.dock.hide();
+    }
+    
+    // 隐藏默认应用菜单
+    const { Menu } = require('electron');
+    Menu.setApplicationMenu(null);
+  }
+
+  setupEventListeners() {
+    // 托盘事件
+    this.trayManager.on('service-start-requested', () => this.startService());
+    this.trayManager.on('service-stop-requested', () => this.stopService());
+    this.trayManager.on('update-check-requested', () => this.checkForUpdates());
+    this.trayManager.on('about-dialog-requested', () => this.showAboutDialog());
+    this.trayManager.on('quit-requested', () => this.quitApplication());
+    
+    // 服务管理器事件
+    this.serviceManager.on('status-update', () => this.trayManager.updateMenu());
+    this.serviceManager.on('restart-requested', () => this.restartApplication());
+    
+    // IPC事件
+    ipcMain.on('about-window-click', (event, data) => {
+      this.aboutDialogManager.handleAboutWindowClick(data);
+    });
+    
+    // 应用事件
+    app.on('window-all-closed', (event) => {
+      event.preventDefault(); // 防止应用退出
+    });
+    
+    app.on('before-quit', async (event) => {
+      if (this.stateManager.get('isQuitting')) return;
+      
+      event.preventDefault();
+      await this.quitApplication();
+    });
+    
+    app.on('activate', () => {
+      if (!this.trayManager.tray) {
+        this.trayManager.initialize(this.stateManager);
+      }
     });
-  return serverModuleLoading;
-}
-
-function getServerStatusLabel() {
-  switch (serviceState) {
-    case 'running':
-      return '运行中';
-    case 'starting':
-      return '启动中';
-    case 'stopping':
-      return '停止中';
-    case 'error':
-      return '启动失败';
-    default:
-      return '已停止';
-  }
-}
-
-async function startService() {
-  if (serviceState === 'running' || serviceState === 'starting') {
-    return;
-  }
-
-  serviceState = 'starting';
-  refreshTrayMenu();
-
-  try {
-    const module = await loadServerModule();
-    await module.startServer();
-    currentServerState = module.getServerState();
-    serviceState = 'running';
-  } catch (error) {
-    serviceState = 'error';
-    currentServerState = null;
-    dialog.showErrorBox('服务启动失败', error?.message || String(error));
-  } finally {
-    refreshTrayMenu();
-  }
-}
-
-async function stopService() {
-  if (serviceState !== 'running') {
-    return;
-  }
-
-  serviceState = 'stopping';
-  refreshTrayMenu();
-
-  try {
-    const module = await loadServerModule();
-    await module.stopServer();
-    currentServerState = null;
-    serviceState = 'stopped';
-  } catch (error) {
-    dialog.showErrorBox('停止服务失败', error?.message || String(error));
-    serviceState = 'error';
-  } finally {
-    refreshTrayMenu();
-  }
-}
-
-function ensureTray() {
-  if (tray) {
-    return tray;
   }
 
-  // 尝试多种可能的图标路径
-  const possiblePaths = [
-    path.join(__dirname, 'assets', 'icon.png'),
-    path.join(process.resourcesPath, 'assets', 'icon.png'),
-    path.join(__dirname, '..', 'assets', 'icon.png')
-  ];
-
-  let iconPath = null;
-  let icon = null;
-  
-  console.log('App is packaged:', app.isPackaged);
-  console.log('__dirname:', __dirname);
-  console.log('process.resourcesPath:', process.resourcesPath);
-  
-  for (const possiblePath of possiblePaths) {
-    console.log('Checking icon path:', possiblePath);
-    console.log('File exists:', fs.existsSync(possiblePath));
-    if (fs.existsSync(possiblePath)) {
-      iconPath = possiblePath;
-      break;
+  async startService() {
+    const serverRoot = this.stateManager.get('runtimeRoot');
+    if (!serverRoot) {
+      this.logger.error('Cannot start service: runtime root not set');
+      return false;
     }
-  }
-  
-  if (!iconPath) {
-    console.error('Icon file not found in any of the expected locations');
-    // 创建一个简单的文本托盘作为后备
-    tray = new Tray(nativeImage.createEmpty());
-    tray.setToolTip('Prompt Server - Icon Missing');
-    refreshTrayMenu();
-    return tray;
-  }
-  
-  console.log('Attempting to load icon from:', iconPath);
-  
-  try {
-    icon = nativeImage.createFromPath(iconPath);
-    console.log('Icon loaded successfully');
-    console.log('Icon is empty:', icon.isEmpty());
-    console.log('Icon size:', icon.getSize());
     
-    if (process.platform === 'darwin') {
-      icon = icon.resize({ width: 18, height: 18 });
-      icon.setTemplateImage(true);
-      console.log('Icon resized and set as template image');
-    }
-
-    tray = new Tray(icon);
-    tray.setToolTip('Prompt Server');
-    console.log('Tray created successfully');
-    refreshTrayMenu();
-    return tray;
-  } catch (error) {
-    console.error('Failed to create tray:', error);
-    // 创建一个简单的文本托盘作为后备
-    tray = new Tray(nativeImage.createEmpty());
-    tray.setToolTip('Prompt Server - Error');
-    refreshTrayMenu();
-    return tray;
+    return await this.serviceManager.startService(serverRoot, this.stateManager);
   }
-}
 
-function refreshTrayMenu() {
-  if (!tray) {
-    return;
+  async stopService() {
+    return await this.serviceManager.stopService(this.stateManager);
   }
 
-  const address = currentServerState?.address ?? 'http://127.0.0.1:5621';
-  const adminUrl = currentServerState?.adminPath ? `${address}${currentServerState.adminPath}` : `${address}/admin`;
-
-  const template = [
-    { label: `状态：${getServerStatusLabel()}`, enabled: false },
-    { type: 'separator' },
-    {
-      label: serviceState === 'running' ? '停止服务' : '启动服务',
-      click: () => (serviceState === 'running' ? stopService() : startService())
-    },
-    {
-      label: '复制服务地址',
-      enabled: serviceState === 'running',
-      click: () => clipboard.writeText(`${address}/mcp`)
-    },
-    {
-      label: '打开管理后台',
-      enabled: serviceState === 'running',
-      click: () => openAdminWindow(adminUrl)
-    },
-    { type: 'separator' },
-    {
-      label: '检查更新',
-      click: () => checkForUpdates()
-    },
-    {
-      label: '关于服务',
-      click: () => showAboutDialog()
-    },
-    { type: 'separator' },
-    {
-      label: '退出服务',
-      click: async () => {
-        isQuitting = true;
-        await stopService();
-        app.quit();
-      }
+  async checkForUpdates() {
+    try {
+      await this.updateManager.checkForUpdates();
+    } catch (error) {
+      this.logger.error('Update check failed', error);
     }
-  ];
-
-  const menu = Menu.buildFromTemplate(template);
-  tray.setContextMenu(menu);
-}
-
-function openAdminWindow(url) {
-  if (adminWindow) {
-    adminWindow.focus();
-    return;
   }
 
-  adminWindow = new BrowserWindow({
-    width: 1200,
-    height: 800,
-    title: 'Prompt Server 管理后台',
-    webPreferences: {
-      contextIsolation: true,
-      preload: undefined
-    }
-  });
-
-  adminWindow.loadURL(url);
-  adminWindow.on('closed', () => {
-    adminWindow = null;
-  });
-}
-
-async function getCurrentServiceVersion() {
-  try {
-    const serverRoot = await ensureRuntimeServerRoot();
-    const pkgRaw = await fs.promises.readFile(path.join(serverRoot, 'package.json'), 'utf8');
-    const pkg = JSON.parse(pkgRaw);
-    return pkg.version;
-  } catch (error) {
-    return 'unknown';
+  showAboutDialog() {
+    this.aboutDialogManager.showAboutDialog();
   }
-}
-
-function compareVersions(a, b) {
-  const toNumbers = (value = '') => value.split('.').map((part) => parseInt(part, 10) || 0);
-  const [a1, a2, a3] = toNumbers(a);
-  const [b1, b2, b3] = toNumbers(b);
-  if (a1 !== b1) return a1 - b1;
-  if (a2 !== b2) return a2 - b2;
-  return a3 - b3;
-}
 
-async function checkForUpdates() {
-  try {
-    const currentVersion = await getCurrentServiceVersion();
-
-    const response = await fetch('https://registry.npmjs.org/@becrafter/prompt-manager');
-    if (!response.ok) {
-      throw new Error('无法获取最新版本信息');
-    }
-
-    const metadata = await response.json();
-    const latestVersion = metadata?.['dist-tags']?.latest;
-
-    if (!latestVersion) {
-      throw new Error('未能解析最新版本号');
-    }
-
-    if (compareVersions(latestVersion, currentVersion) <= 0) {
-      await dialog.showMessageBox({
-        type: 'info',
-        message: '已是最新版本',
-        detail: `服务当前版本：${currentVersion}`
-      });
-      return;
-    }
-
-    const { response: action } = await dialog.showMessageBox({
-      type: 'question',
-      buttons: ['立即升级', '打开发布页', '取消'],
-      defaultId: 0,
-      cancelId: 2,
-      message: `发现新版本 ${latestVersion}`,
-      detail: '升级期间服务会短暂停止，是否继续？'
-    });
-
-    if (action === 1) {
-      shell.openExternal('https://github.com/BeCrafter/prompt-manager/releases/latest');
-      return;
-    }
-
-    if (action !== 0) {
-      return;
-    }
-
-    const wasRunning = serviceState === 'running';
-    if (wasRunning) {
-      await stopService();
-    }
-
-    await performInPlaceUpgrade(latestVersion);
-
-    await dialog.showMessageBox({
-      type: 'info',
-      message: '升级成功',
-      detail: `服务已升级到 ${latestVersion}`
-    });
-
-    if (wasRunning) {
-      await startService();
-    }
-  } catch (error) {
-    dialog.showErrorBox('检查更新失败', error?.message || String(error));
+  handleAboutWindowClick(data) {
+    // 委托给AboutDialogManager处理
+    this.aboutDialogManager.handleAboutWindowClick(data);
   }
-}
-
-async function performInPlaceUpgrade(version) {
-  const tarballUrl = `https://registry.npmjs.org/@becrafter/prompt-manager/-/@becrafter/prompt-manager-${version}.tgz`;
-  const tmpDir = await fs.promises.mkdtemp(path.join(os.tmpdir(), 'prompt-manager-upgrade-'));
-  const tarballPath = path.join(tmpDir, `${version}.tgz`);
-
-  try {
-    const response = await fetch(tarballUrl);
-    if (!response.ok) {
-      throw new Error('无法下载升级包');
-    }
-
-    const buffer = Buffer.from(await response.arrayBuffer());
-    await fs.promises.writeFile(tarballPath, buffer);
-
-    await tar.x({ file: tarballPath, cwd: tmpDir });
-    const extractedPath = path.join(tmpDir, 'package');
-
-    const serverRoot = await ensureRuntimeServerRoot();
-    const examplesDir = path.join(serverRoot, 'examples', 'prompts');
-    const examplesBackup = path.join(tmpDir, 'examples-prompts');
 
-    if (await pathExists(examplesDir)) {
-      await fs.promises.cp(examplesDir, examplesBackup, { recursive: true });
-    }
-
-    await fs.promises.rm(serverRoot, { recursive: true, force: true });
-    await fs.promises.mkdir(serverRoot, { recursive: true });
-    await fs.promises.cp(extractedPath, serverRoot, { recursive: true });
-
-    if (await pathExists(examplesBackup)) {
-      const targetExamples = path.join(serverRoot, 'examples', 'prompts');
-      await fs.promises.mkdir(path.dirname(targetExamples), { recursive: true });
-      await fs.promises.cp(examplesBackup, targetExamples, { recursive: true });
+  async restartApplication() {
+    this.logger.info('Restarting application');
+    
+    try {
+      await this.stopService();
+      app.relaunch();
+      app.exit(0);
+    } catch (error) {
+      this.logger.error('Failed to restart application', error);
     }
-
-    await installServerDependencies(serverRoot);
-    await loadServerModule({ forceReload: true });
-  } finally {
-    await fs.promises.rm(tmpDir, { recursive: true, force: true });
   }
-}
 
-async function pathExists(targetPath) {
-  try {
-    await fs.promises.access(targetPath, fs.constants.F_OK);
-    return true;
-  } catch (error) {
-    return false;
+  async quitApplication() {
+    this.logger.info('Quitting application');
+    
+    try {
+      this.stateManager.set('isQuitting', true);
+      
+      // 停止服务
+      await this.stopService();
+      
+      // 清理资源
+      await this.aboutDialogManager.closeAboutDialog();
+      
+      this.trayManager.destroy();
+      await this.logger.close();
+      await this.runtimeManager.cleanup();
+      
+      app.quit();
+    } catch (error) {
+      this.logger.error('Error during application shutdown', error);
+      app.quit(); // 强制退出
+    }
   }
 }
 
-async function installServerDependencies(targetDir) {
-  const npmCommand = process.platform === 'win32' ? 'npm.cmd' : 'npm';
-  const args = ['install', '--omit=dev'];
-
-  await new Promise((resolve, reject) => {
-    const child = spawn(npmCommand, args, {
-      cwd: targetDir,
-      stdio: ['ignore', 'pipe', 'pipe']
-    });
-
-    let stderr = '';
-    child.stdout.on('data', (data) => {
-      console.log(`[npm] ${data.toString().trim()}`);
-    });
-    child.stderr.on('data', (data) => {
-      stderr += data.toString();
-      console.error(`[npm] ${data.toString().trim()}`);
-    });
-
-    child.on('error', (error) => {
-      reject(error);
-    });
-
-    child.on('close', (code) => {
-      if (code === 0) {
-        resolve();
-      } else {
-        reject(new Error(`npm install failed with exit code ${code}: ${stderr}`));
-      }
-    });
-  });
-}
-
-async function showAboutDialog() {
-  const serviceVersion = await getCurrentServiceVersion();
-  const lines = [
-    // `桌面应用版本：${desktopPackageJson.version}`,
-    `服务版本：${serviceVersion}`,
-    `Electron：${process.versions.electron}`,
-    // `Chromium：${process.versions.chrome}`,
-    `Node.js：${process.versions.node}`
-  ];
-
-  await dialog.showMessageBox({
-    type: 'info',
-    title: '关于 Prompt Server',
-    message: '组件版本信息',
-    detail: lines.join('\n')
-  });
-}
+// 应用启动
+const promptManagerApp = new PromptManagerApp();
 
 app.whenReady().then(async () => {
-  console.log('App is packaged:', app.isPackaged);
-  console.log('App data path:', app.getPath('userData'));
-  console.log('Resources path:', process.resourcesPath);
-  console.log('__dirname:', __dirname);
-  
-  Menu.setApplicationMenu(null);
-  if (process.platform === 'darwin') {
-    app.dock.hide();
+  try {
+    await promptManagerApp.initialize();
+  } catch (error) {
+    console.error('Failed to start application:', error);
+    app.quit();
   }
-
-  await ensureRuntimeServerRoot();
-  ensureTray();
-  await startService();
-});
-
-app.on('window-all-closed', (event) => {
-  event.preventDefault();
 });
 
-app.on('before-quit', async (event) => {
-  if (isQuitting) {
-    return;
+// 异常处理
+process.on('uncaughtException', (error) => {
+  console.error('Uncaught Exception:', error);
+  if (promptManagerApp.logger) {
+    promptManagerApp.logger.error('Uncaught Exception', error);
   }
-  event.preventDefault();
-  isQuitting = true;
-  await stopService();
   app.quit();
 });
 
-app.on('activate', () => {
-  if (!tray) {
-    ensureTray();
+process.on('unhandledRejection', (reason, promise) => {
+  console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+  if (promptManagerApp.logger) {
+    promptManagerApp.logger.error('Unhandled Rejection', reason);
   }
-});
+});