@ytchuan/flowlink 1.0.0

package/services/scanner.js

@@ -0,0 +1,254 @@
+/**
+ * 客户端安装检测（WorkBuddy / CodeBuddy）
+ * 扫描本地已安装的 AI 客户端软件
+ * Windows 检测策略：精确路径 → 注册表卸载信息（支持 CodeBuddy CN 等变体）→ 配置目录兜底
+ */
+const fs = require('fs')
+const path = require('path')
+const os = require('os')
+const { execSync } = require('child_process')
+
+/** 检查路径是否存在 */
+function checkPathExists(filePath) {
+  try {
+    return fs.existsSync(filePath)
+  } catch {
+    return false
+  }
+}
+
+/**
+ * 在指定目录下查找名称匹配的子项（模糊匹配，不区分大小写）
+ * @param {string} dir - 搜索目录
+ * @param {string} keyword - 匹配关键词
+ * @returns {string|null} 匹配到的完整路径
+ */
+function findMatchingEntry(dir, keyword) {
+  try {
+    if (!fs.existsSync(dir)) return null
+    const entries = fs.readdirSync(dir)
+    const lowerKeyword = keyword.toLowerCase()
+    for (const entry of entries) {
+      if (entry.toLowerCase().includes(lowerKeyword)) {
+        return path.join(dir, entry)
+      }
+    }
+  } catch {
+    // 目录读取失败
+  }
+  return null
+}
+
+/**
+ * 在 VS Code / Cursor 扩展目录中查找 CodeBuddy
+ * 扩展目录名格式：publisher.extension-name-version
+ */
+function findCodeBuddyExtension() {
+  const home = os.homedir()
+  const extDirs = [
+    path.join(home, '.vscode', 'extensions'),
+    path.join(home, '.cursor', 'extensions'),
+    path.join(home, '.vscode-insiders', 'extensions'),
+    path.join(home, '.windsurf', 'extensions'),
+  ]
+
+  // CodeBuddy 扩展可能的关键词
+  const keywords = ['codebuddy', 'code-buddy', 'tencent-cloud-ai']
+
+  for (const extDir of extDirs) {
+    for (const kw of keywords) {
+      const found = findMatchingEntry(extDir, kw)
+      if (found) return found
+    }
+  }
+  return null
+}
+
+/**
+ * 在 JetBrains 插件目录中查找 CodeBuddy
+ * 各平台 JetBrains 配置目录：
+ * - macOS: ~/Library/Application Support/JetBrains
+ * - Windows: %APPDATA%/JetBrains
+ * - Linux: ~/.local/share/JetBrains
+ */
+function findCodeBuddyJetBrains() {
+  const home = os.homedir()
+  const jbBaseByPlatform = {
+    darwin: path.join(home, 'Library', 'Application Support', 'JetBrains'),
+    win32: path.join(home, 'AppData', 'Roaming', 'JetBrains'),
+    linux: path.join(home, '.local', 'share', 'JetBrains'),
+  }
+  const jbBase = jbBaseByPlatform[process.platform]
+  if (!jbBase) return null
+
+  try {
+    if (!fs.existsSync(jbBase)) return null
+    const ideDirs = fs.readdirSync(jbBase)
+    for (const ideDir of ideDirs) {
+      const pluginsDir = path.join(jbBase, ideDir, 'plugins')
+      const found = findMatchingEntry(pluginsDir, 'codebuddy')
+      if (found) return found
+    }
+  } catch {
+    // 目录读取失败
+  }
+  return null
+}
+
+/**
+ * 在 Windows 注册表卸载信息中查找已安装程序
+ * 注意：必须用 `reg query ... /s` 查询全部值；若带 `/v DisplayName` 则只返回 DisplayName，拿不到 InstallLocation
+ * @param {RegExp} namePattern - DisplayName 匹配正则
+ * @returns {string|null} 匹配到的 InstallLocation（缺失时返回 DisplayName），未找到返回 null
+ */
+function findInWindowsUninstall(namePattern) {
+  if (process.platform !== 'win32') return null
+
+  const regPaths = [
+    'HKLM\\SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall',
+    'HKLM\\SOFTWARE\\WOW6432Node\\Microsoft\\Windows\\CurrentVersion\\Uninstall',
+    'HKCU\\SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall',
+  ]
+
+  for (const regPath of regPaths) {
+    let output
+    try {
+      output = execSync(`reg query "${regPath}" /s 2>nul`, {
+        encoding: 'utf-8',
+        timeout: 8000,
+      })
+    } catch {
+      // 该注册表分支不存在或查询失败
+      continue
+    }
+
+    // 每个子键块以 HKEY_ 开头
+    const blocks = output.split(/(?=HKEY_)/)
+    for (const block of blocks) {
+      const nameMatch = block.match(/DisplayName\s+REG_SZ\s+(.+)/)
+      if (!nameMatch) continue
+      const displayName = nameMatch[1].trim()
+      if (!namePattern.test(displayName)) continue
+
+      const locMatch = block.match(/InstallLocation\s+REG_SZ\s+(.+)/)
+      // 去除尾部反斜杠，避免拼接出双反斜杠路径
+      const installLocation = locMatch
+        ? locMatch[1].trim().replace(/[\\/]+$/, '')
+        : ''
+      return installLocation || displayName
+    }
+  }
+  return null
+}
+
+/** 客户端定义 */
+const CLIENT_TARGETS = [
+  {
+    key: 'workbuddy',
+    name: 'WorkBuddy',
+    description: '腾讯代码助手桌面版',
+    macOSPaths: [
+      '/Applications/WorkBuddy.app',
+      '/Applications/腾讯代码助手.app',
+      '/Applications/腾讯云AI代码助手.app',
+      path.join(os.homedir(), 'Applications/WorkBuddy.app'),
+    ],
+    win32Paths: [
+      path.join(process.env.PROGRAMFILES || 'C:\\Program Files', 'WorkBuddy'),
+      path.join(process.env['PROGRAMFILES(X86)'] || 'C:\\Program Files (x86)', 'WorkBuddy'),
+      path.join(process.env.LOCALAPPDATA || '', 'WorkBuddy'),
+      // Electron 应用默认用户级安装位置
+      path.join(process.env.LOCALAPPDATA || '', 'Programs', 'WorkBuddy'),
+    ],
+    // Windows 注册表卸载信息中的 DisplayName 匹配规则
+    win32NamePattern: /^WorkBuddy/i,
+    // 配置目录兜底：客户端运行过至少一次即会创建
+    configDir: path.join(os.homedir(), '.workbuddy'),
+    configPath: path.join(os.homedir(), '.workbuddy', 'models.json'),
+  },
+  {
+    key: 'codebuddy',
+    name: 'CodeBuddy',
+    description: '腾讯云 AI IDE 插件',
+    macOSPaths: [
+      '/Applications/CodeBuddy.app',
+      path.join(os.homedir(), 'Applications/CodeBuddy.app'),
+    ],
+    win32Paths: [
+      path.join(process.env.PROGRAMFILES || 'C:\\Program Files', 'CodeBuddy'),
+      path.join(process.env['PROGRAMFILES(X86)'] || 'C:\\Program Files (x86)', 'CodeBuddy'),
+      path.join(process.env.LOCALAPPDATA || '', 'CodeBuddy'),
+      path.join(process.env.LOCALAPPDATA || '', 'Programs', 'CodeBuddy'),
+      // 中国区独立安装包，目录名带 CN 后缀（注册表 DisplayName 为 "CodeBuddy CN (User)"）
+      path.join(process.env.LOCALAPPDATA || '', 'Programs', 'CodeBuddy CN'),
+      path.join(process.env.LOCALAPPDATA || '', 'CodeBuddy CN'),
+      path.join(process.env['PROGRAMFILES(X86)'] || 'C:\\Program Files (x86)', 'CodeBuddy CN'),
+    ],
+    win32NamePattern: /^CodeBuddy/i,
+    configDir: path.join(os.homedir(), '.codebuddy'),
+    configPath: path.join(os.homedir(), '.codebuddy', 'models.json'),
+  },
+]
+
+/**
+ * 扫描已安装的客户端
+ * @returns {Array<{ key: string, name: string, description: string, installed: boolean, configPath?: string }>}
+ */
+function scan() {
+  const platform = process.platform
+  const pathKey = platform === 'darwin' ? 'macOSPaths' : 'win32Paths'
+
+  return CLIENT_TARGETS.map((target) => {
+    // 1. 精确路径匹配
+    const paths = target[pathKey] || []
+    let installed = paths.some(checkPathExists)
+    let foundPath = paths.find(checkPathExists)
+
+    // 2. Windows 注册表卸载信息匹配（覆盖 CodeBuddy CN 等非默认目录安装）
+    if (!installed && platform === 'win32' && target.win32NamePattern) {
+      foundPath = findInWindowsUninstall(target.win32NamePattern)
+      if (foundPath) {
+        installed = true
+      }
+    }
+
+    // 3. CodeBuddy 特殊处理：在 IDE 扩展/插件目录中模糊匹配
+    if (!installed && target.key === 'codebuddy') {
+      // VS Code / Cursor 扩展
+      foundPath = findCodeBuddyExtension()
+      if (foundPath) {
+        installed = true
+      }
+
+      // JetBrains 插件
+      if (!installed) {
+        foundPath = findCodeBuddyJetBrains()
+        if (foundPath) {
+          installed = true
+        }
+      }
+    }
+
+    // 4. 配置目录兜底：客户端已运行过至少一次会创建配置目录（如 ~/.codebuddy）
+    if (!installed && target.configDir && checkPathExists(target.configDir)) {
+      foundPath = target.configDir
+      installed = true
+    }
+
+    return {
+      key: target.key,
+      name: target.name,
+      description: target.description,
+      installed,
+      configPath: installed ? foundPath : undefined,
+    }
+  })
+}
+
+/** 获取客户端配置文件路径 */
+function getConfigPath(key) {
+  const target = CLIENT_TARGETS.find((t) => t.key === key)
+  return target ? target.configPath : null
+}
+
+module.exports = { scan, getConfigPath, CLIENT_TARGETS }

package/services/scheduler.js

@@ -0,0 +1,197 @@
+/**
+ * 本地调度服务（端口 51173 代理，占用时依次回退 51172、51171）
+ * 启动/停止本地 HTTP 代理服务，将 AI 请求原样转发到后端网关
+ */
+const http = require('http')
+const https = require('https')
+const { URL } = require('url')
+const { EventEmitter } = require('events')
+
+/** 调度服务默认端口列表 */
+const SCHEDULER_PORTS = [51173, 51172, 51171]
+
+/** 默认后端网关地址（支持通过环境变量注入，便于打包时适配不同环境） */
+const DEFAULT_GATEWAY_URL = 'http://localhost:9000'
+
+/**
+ * 尝试在指定端口上启动 HTTP 服务
+ * @param {http.Server} server
+ * @param {number} port
+ * @returns {Promise<void>}
+ */
+function tryListen(server, port) {
+  return new Promise((resolve, reject) => {
+    const onError = (err) => {
+      server.off('error', onError)
+      reject(err)
+    }
+    server.on('error', onError)
+    server.listen(port, '127.0.0.1', () => {
+      server.off('error', onError)
+      resolve()
+    })
+  })
+}
+
+class SchedulerService extends EventEmitter {
+  constructor() {
+    super()
+    this.server = null
+    this.running = false
+    this.port = SCHEDULER_PORTS[0]
+    this.gatewayUrl = DEFAULT_GATEWAY_URL
+    this.config = {}
+  }
+
+  /**
+   * 启动调度服务
+   * @param {{ apiKey?: string, deviceId?: string, gatewayUrl?: string }} config
+   * @returns {{ success: boolean, message: string }}
+   */
+  async start(config = {}) {
+    this.config = config
+    if (config.gatewayUrl) {
+      this.gatewayUrl = config.gatewayUrl // 远程获取的配置覆盖本地
+    }
+
+    if (this.running) {
+      return { success: true, message: '调度服务已在运行中' }
+    }
+
+    this.server = http.createServer((req, res) => {
+      this.handleRequest(req, res)
+    })
+
+    for (const port of SCHEDULER_PORTS) {
+      try {
+        await tryListen(this.server, port)
+        this.port = port
+        this.running = true
+        this.emit('status-changed', { running: true, port: this.port })
+        console.log(`[Scheduler] 已启动 http://127.0.0.1:${this.port} → ${this.gatewayUrl}`)
+        return { success: true, message: '调度服务已启动' }
+      } catch (err) {
+        if (err.code !== 'EADDRINUSE') {
+          return { success: false, message: `启动失败：${err.message}` }
+        }
+        console.warn(`[Scheduler] 端口 ${port} 已被占用，尝试下一个端口`)
+      }
+    }
+
+    this.server = null
+    return { success: false, message: '端口被占用' }
+  }
+
+  /**
+   * 停止调度服务
+   * @returns {{ success: boolean, message: string }}
+   */
+  async stop() {
+    if (!this.running || !this.server) {
+      this.running = false
+      return { success: true, message: '调度服务已停止' }
+    }
+
+    try {
+      await new Promise((resolve) => {
+        this.server.close(() => resolve())
+      })
+      this.server = null
+      this.running = false
+      this.emit('status-changed', { running: false, port: this.port })
+      return { success: true, message: '调度服务已停止' }
+    } catch (err) {
+      return { success: false, message: `停止失败：${err.message}` }
+    }
+  }
+
+  /**
+   * 获取服务状态
+   * @returns {{ running: boolean, port: number }}
+   */
+  getStatus() {
+    return { running: this.running, port: this.port }
+  }
+
+  /**
+   * 处理请求：健康检查本地响应，其余请求原样转发到后端网关
+   * 转发时透传请求头/请求体/响应流（支持 SSE 流式响应）
+   */
+  handleRequest(req, res) {
+    // CORS 头
+    res.setHeader('Access-Control-Allow-Origin', '*')
+    res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
+    res.setHeader('Access-Control-Allow-Headers', 'Content-Type, Authorization')
+
+    if (req.method === 'OPTIONS') {
+      res.writeHead(204)
+      res.end()
+      return
+    }
+
+    // 健康检查端点
+    if (req.url === '/health' && req.method === 'GET') {
+      res.writeHead(200, { 'Content-Type': 'application/json' })
+      res.end(JSON.stringify({ status: 'ok', running: true }))
+      return
+    }
+
+    // 代理转发到后端网关
+    let target
+    try {
+      target = new URL(this.gatewayUrl)
+    } catch {
+      res.writeHead(500, { 'Content-Type': 'application/json' })
+      res.end(JSON.stringify({ error: 'INVALID_GATEWAY', message: '后端网关地址无效' }))
+      return
+    }
+
+    const isHttps = target.protocol === 'https:'
+    const client = isHttps ? https : http
+
+    // 透传请求头，重写 host
+    const headers = { ...req.headers }
+    headers.host = target.host
+
+    const proxyReq = client.request(
+      {
+        hostname: target.hostname,
+        port: target.port || (isHttps ? 443 : 80),
+        path: req.url,
+        method: req.method,
+        headers,
+      },
+      (proxyRes) => {
+        // 透传响应头与状态码（流式响应直接管道输出）
+        res.writeHead(proxyRes.statusCode || 502, proxyRes.headers)
+        proxyRes.pipe(res)
+      }
+    )
+
+    proxyReq.on('error', (err) => {
+      console.error('[Scheduler] 转发失败:', err.message)
+      if (!res.headersSent) {
+        res.writeHead(502, { 'Content-Type': 'application/json' })
+        res.end(JSON.stringify({
+          error: 'GATEWAY_UNREACHABLE',
+          message: `无法连接后端网关 ${this.gatewayUrl}：${err.message}`,
+        }))
+      } else {
+        res.end()
+      }
+    })
+
+    // 客户端中断时终止上游请求
+    req.on('aborted', () => {
+      proxyReq.destroy()
+    })
+
+    // 透传请求体
+    req.pipe(proxyReq)
+  }
+}
+
+// 单例
+const schedulerService = new SchedulerService()
+
+module.exports = schedulerService

package/services/selfcheck.js

@@ -0,0 +1,154 @@
+/**
+ * 环境自检
+ * 诊断服务连通性问题，检查配置完整性
+ */
+const http = require('http')
+const net = require('net')
+const scheduler = require('./scheduler')
+const scanner = require('./scanner')
+const settings = require('./settings')
+
+/**
+ * 检查端口连通性
+ * @param {number} port
+ * @param {string} host
+ * @returns {Promise<boolean>}
+ */
+function checkPort(port, host = '127.0.0.1') {
+    return new Promise((resolve) => {
+        const socket = new net.Socket()
+        socket.setTimeout(2000)
+        socket.on('connect', () => {
+            socket.destroy()
+            resolve(true)
+        })
+        socket.on('timeout', () => {
+            socket.destroy()
+            resolve(false)
+        })
+        socket.on('error', () => {
+            resolve(false)
+        })
+        socket.connect(port, host)
+    })
+}
+
+/**
+ * 检查远程网关连通性
+ * @returns {Promise<boolean>}
+ */
+async function checkRemoteGateway(port) {
+    return new Promise((resolve) => {
+        const req = http.get(
+            {hostname: '127.0.0.1', port, path: '/health', timeout: 3000},
+            (res) => {
+                resolve(res.statusCode === 200)
+                res.resume()
+            }
+        )
+        req.on('error', () => resolve(false))
+        req.on('timeout', () => {
+            req.destroy()
+            resolve(false)
+        })
+    })
+}
+
+/**
+ * 运行环境自检
+ * @param {{ deviceId: string, activated?: boolean, packageCount?: number, balance?: number, apiKey?: string }} context
+ * @returns {Promise<Array<{ item: string, status: 'pass'|'fail'|'warning', message: string }>>}
+ */
+async function runSelfCheck(context = {}) {
+    const results = []
+
+    // 1. 本地端口连通性
+    const schedulerStatus = scheduler.getStatus()
+    const port = schedulerStatus.port
+    const portAvailable = await checkPort(port)
+    results.push({
+        item: `本地端口连通性 (127.0.0.1:${port})`,
+        status: schedulerStatus.running && portAvailable ? 'pass' : 'warning',
+        message: schedulerStatus.running && portAvailable
+            ? `端口 ${port} 可访问，调度服务运行中`
+            : `调度服务未运行，端口 ${port} 不可访问`,
+    })
+
+    // 2. 远程网关连通性（通过本地代理检查）
+    if (schedulerStatus.running) {
+        const gatewayOk = await checkRemoteGateway(port)
+        results.push({
+            item: '远程网关连通性',
+            status: gatewayOk ? 'pass' : 'warning',
+            message: gatewayOk ? '网关响应正常' : '网关响应异常，请检查网络',
+        })
+    } else {
+        results.push({
+            item: '远程网关连通性',
+            status: 'warning',
+            message: '调度服务未运行，跳过网关检查',
+        })
+    }
+
+    // 3. 设备标识有效性
+    const hasDeviceId = !!context.deviceId && context.deviceId.startsWith('ST-')
+    results.push({
+        item: '设备标识有效性',
+        status: hasDeviceId ? 'pass' : 'fail',
+        message: hasDeviceId
+            ? `设备标识正常：${context.deviceId.slice(0, 16)}...`
+            : '设备标识缺失或格式错误',
+    })
+
+    // 4. 积分余额检查（以真实账户数据为准）
+    const packageCount = Number(context.packageCount) || 0
+    const balance = Number(context.balance) || 0
+    const hasPackage = context.activated || packageCount > 0 || balance > 0
+    results.push({
+        item: '积分余额检查',
+        status: hasPackage ? 'pass' : 'fail',
+        message: hasPackage
+            ? `已激活积分包，剩余 ${balance} 积分（共 ${packageCount} 个积分包）`
+            : '未激活积分包，请先激活',
+    })
+
+    // 5. 客户端配置完整性
+    const clients = scanner.scan()
+    const installedClients = clients.filter((c) => c.installed)
+    if (installedClients.length === 0) {
+        results.push({
+            item: '客户端配置完整性',
+            status: 'warning',
+            message: '未检测到已安装的客户端',
+        })
+    } else {
+        const fs = require('fs')
+        const configuredCount = installedClients.filter((c) => {
+            const configPath = scanner.getConfigPath(c.key)
+            return configPath && fs.existsSync(configPath)
+        }).length
+
+        results.push({
+            item: '客户端配置完整性',
+            status: configuredCount > 0 ? 'pass' : 'warning',
+            message:
+                configuredCount > 0
+                    ? `已配置 ${configuredCount}/${installedClients.length} 个客户端`
+                    : `检测到 ${installedClients.length} 个客户端，但尚未配置`,
+        })
+    }
+
+    // 6. API Key 有效性（格式：SK-UUID）
+    const hasApiKey = !!context.apiKey && context.apiKey.startsWith('SK-')
+    results.push({
+        item: 'API Key 有效性',
+        status: hasApiKey ? 'pass' : 'warning',
+        message: hasApiKey
+            ? 'API Key 格式正确'
+            : 'API Key 缺失或格式错误，请重新生成',
+    })
+
+    return results
+}
+
+module.exports = {runSelfCheck}

package/services/settings.js

@@ -0,0 +1,58 @@
+/**
+ * 应用设置读写与持久化
+ * 存储到 userData/settings.json
+ */
+const { app } = require('electron')
+const fs = require('fs')
+const path = require('path')
+
+/** 默认设置 */
+const DEFAULT_SETTINGS = {
+  theme: 'light',
+  closeBehavior: 'quit',
+  autoStart: false,
+  autoSchedule: false,
+  backupBeforeConfig: true,
+  /** 后端网关地址（本地 51173 端点的转发目标） */
+  gatewayUrl: 'http://localhost:9000',
+}
+
+/** 设置文件路径 */
+function getSettingsFilePath() {
+  return path.join(app.getPath('userData'), 'settings.json')
+}
+
+/**
+ * 读取设置
+ * @returns {typeof DEFAULT_SETTINGS}
+ */
+function getSettings() {
+  const filePath = getSettingsFilePath()
+  try {
+    if (fs.existsSync(filePath)) {
+      const data = JSON.parse(fs.readFileSync(filePath, 'utf-8'))
+      return { ...DEFAULT_SETTINGS, ...data }
+    }
+  } catch {
+    // 文件损坏，返回默认值
+  }
+  return { ...DEFAULT_SETTINGS }
+}
+
+/**
+ * 保存设置（合并写入）
+ * @param {Partial<typeof DEFAULT_SETTINGS>} partial
+ */
+function saveSettings(partial) {
+  const current = getSettings()
+  const merged = { ...current, ...partial }
+  const filePath = getSettingsFilePath()
+  try {
+    fs.mkdirSync(path.dirname(filePath), { recursive: true })
+    fs.writeFileSync(filePath, JSON.stringify(merged, null, 2), 'utf-8')
+  } catch (err) {
+    console.error('保存设置失败:', err)
+  }
+}
+
+module.exports = { getSettings, saveSettings, DEFAULT_SETTINGS }