@ytchuan/flowlink 1.0.0

package/services/config-writer.js

@@ -0,0 +1,415 @@
+/**
+ * 客户端配置写入/备份/删除
+ * 向 WorkBuddy / CodeBuddy 写入 API 配置文件
+ */
+const fs = require('fs')
+const path = require('path')
+const os = require('os')
+const { app, shell } = require('electron')
+const scanner = require('./scanner')
+
+/** 备份目录 */
+function getBackupDir() {
+  return path.join(app.getPath('userData'), 'config-backups')
+}
+
+/** 客户端配置文件路径映射 */
+function getClientConfigPath(targetKey) {
+  const configPath = scanner.getConfigPath(targetKey)
+  if (configPath) return configPath
+
+  // 降级：使用默认路径
+  const home = os.homedir()
+  const map = {
+    workbuddy: path.join(home, '.workbuddy', 'models.json'),
+    codebuddy: path.join(home, '.codebuddy', 'models.json'),
+  }
+  return map[targetKey] || null
+}
+
+/**
+ * 生成 WorkBuddy 自定义模型条目
+ * @param {{ apiEndpoint: string, apiKey: string }} config
+ * @param {Array<{ code: string }>} models - app.ts modelPage 返回的模型列表
+ * @returns {Array<object>}
+ */
+function buildWorkBuddyModels(config, models) {
+  return models.map((model) => ({
+    id: model.code,
+    name: model.code,
+    vendor: 'Custom',
+    url: config.apiEndpoint,
+    apiKey: config.apiKey,
+    supportsToolCall: true,
+    supportsImages: true,
+    supportsReasoning: true,
+    useCustomProtocol: false,
+    reasoning: {
+      supportedEfforts: ['low', 'medium', 'high', 'xhigh', 'max'],
+    },
+  }))
+}
+
+/**
+ * 生成 CodeBuddy 自定义模型条目
+ * CodeBuddy 格式：vendor 为 'user'，无 reasoning/useCustomProtocol 字段
+ * @param {{ apiEndpoint: string, apiKey: string }} config
+ * @param {Array<{ code: string }>} models
+ * @returns {Array<object>}
+ */
+function buildCodeBuddyModels(config, models) {
+  return models.map((model) => ({
+    id: model.code,
+    name: model.code,
+    vendor: 'user',
+    url: config.apiEndpoint,
+    apiKey: config.apiKey,
+    supportsToolCall: true,
+    supportsImages: true,
+    supportsReasoning: true,
+  }))
+}
+
+/** 读取 JSON 数组文件（不存在或格式异常时返回空数组） */
+function readJsonArray(filePath) {
+  try {
+    if (fs.existsSync(filePath)) {
+      const data = JSON.parse(fs.readFileSync(filePath, 'utf-8'))
+      if (Array.isArray(data)) return data
+    }
+  } catch {
+    // 文件损坏视为无内容
+  }
+  return []
+}
+
+/**
+ * 读取 { "models": [...] } 格式 JSON 文件（不存在或格式异常时返回空数组）
+ * @param {string} filePath
+ * @returns {Array<object>}
+ */
+function readJsonModelsWrapper(filePath) {
+  try {
+    if (fs.existsSync(filePath)) {
+      const data = JSON.parse(fs.readFileSync(filePath, 'utf-8'))
+      if (data && Array.isArray(data.models)) return data.models
+    }
+  } catch {
+    // 文件损坏视为无内容
+  }
+  return []
+}
+
+/** 备份文件到备份目录 */
+function backupFile(filePath, targetKey, suffix = '') {
+  if (!fs.existsSync(filePath)) return
+  const backupDir = getBackupDir()
+  const backupName = `${targetKey}${suffix}-${Date.now()}.json`
+  fs.mkdirSync(backupDir, { recursive: true })
+  fs.copyFileSync(filePath, path.join(backupDir, backupName))
+}
+
+/**
+ * 写入 WorkBuddy 模型配置（合并写入：保留用户其他模型，替换本软件管理的模型）
+ * @param {string} configPath
+ * @param {boolean} backup
+ * @param {{ apiEndpoint: string, apiKey: string }} config
+ * @param {Array<{ code: string }>} models
+ */
+function applyWorkBuddyConfig(configPath, backup, config, models) {
+  if (backup) {
+    backupFile(configPath, 'workbuddy')
+  }
+
+  const managedIds = models.map((m) => m.code)
+  const existing = readJsonArray(configPath)
+  // 移除同 ID 旧条目后追加最新配置
+  const others = existing.filter((m) => !managedIds.includes(m && m.id))
+  const merged = [...others, ...buildWorkBuddyModels(config, models)]
+
+  fs.mkdirSync(path.dirname(configPath), { recursive: true })
+  fs.writeFileSync(configPath, JSON.stringify(merged, null, 2), 'utf-8')
+}
+
+/**
+ * 删除 WorkBuddy 模型配置（仅移除本软件管理的模型条目）
+ * @param {string} configPath
+ * @param {Array<{ code: string }>} models
+ * @returns {boolean} 是否有文件被处理
+ */
+function deleteWorkBuddyConfig(configPath, models) {
+  if (!fs.existsSync(configPath)) return false
+
+  const managedIds = models.map((m) => m.code)
+  // 无托管模型信息时跳过，避免把用户配置原样重写却误报「已删除」
+  if (managedIds.length === 0) return false
+
+  backupFile(configPath, 'workbuddy', '-delete')
+
+  const existing = readJsonArray(configPath)
+  const remaining = existing.filter((m) => !managedIds.includes(m && m.id))
+
+  if (remaining.length === 0) {
+    fs.unlinkSync(configPath)
+  } else {
+    fs.writeFileSync(configPath, JSON.stringify(remaining, null, 2), 'utf-8')
+  }
+  return true
+}
+
+/**
+ * 写入 CodeBuddy 模型配置
+ * CodeBuddy 格式：{ "models": [...] }，合并写入：保留用户其他模型，替换本软件管理的模型
+ * @param {string} configPath
+ * @param {boolean} backup
+ * @param {{ apiEndpoint: string, apiKey: string }} config
+ * @param {Array<{ code: string }>} models
+ */
+function applyCodeBuddyConfig(configPath, backup, config, models) {
+  if (backup) {
+    backupFile(configPath, 'codebuddy')
+  }
+
+  const managedIds = models.map((m) => m.code)
+  const existing = readJsonModelsWrapper(configPath)
+  // 移除同 ID 旧条目后追加最新配置
+  const others = existing.filter((m) => !managedIds.includes(m && m.id))
+  const merged = [...others, ...buildCodeBuddyModels(config, models)]
+
+  fs.mkdirSync(path.dirname(configPath), { recursive: true })
+  fs.writeFileSync(configPath, JSON.stringify({ models: merged }, null, 2), 'utf-8')
+}
+
+/**
+ * 删除 CodeBuddy 模型配置（仅移除本软件管理的模型条目）
+ * @param {string} configPath
+ * @param {Array<{ code: string }>} models
+ * @returns {boolean} 是否有文件被处理
+ */
+function deleteCodeBuddyConfig(configPath, models) {
+  if (!fs.existsSync(configPath)) return false
+
+  const managedIds = models.map((m) => m.code)
+  // 无托管模型信息时跳过，避免把用户配置原样重写却误报「已删除」
+  if (managedIds.length === 0) return false
+
+  backupFile(configPath, 'codebuddy', '-delete')
+
+  const existing = readJsonModelsWrapper(configPath)
+  const remaining = existing.filter((m) => !managedIds.includes(m && m.id))
+
+  if (remaining.length === 0) {
+    fs.unlinkSync(configPath)
+  } else {
+    fs.writeFileSync(configPath, JSON.stringify({ models: remaining }, null, 2), 'utf-8')
+  }
+  return true
+}
+
+/** 生成通用客户端配置内容（CodeBuddy 等） */
+function generateConfigContent(apiEndpoint, apiKey) {
+  return JSON.stringify(
+    {
+      apiEndpoint,
+      apiKey,
+      updatedAt: new Date().toISOString(),
+    },
+    null,
+    2
+  )
+}
+
+/**
+ * 写入客户端配置
+ * @param {string[]} targets - 目标客户端 key 数组
+ * @param {boolean} backup - 是否备份原文件
+ * @param {{ apiEndpoint: string, apiKey: string }} config - 配置内容
+ * @param {Array<{ code: string }>} models - app.ts modelPage 返回的模型列表
+ * @returns {Array<{ target: string, success: boolean, message: string }>}
+ */
+function applyConfig(targets, backup, config, models = []) {
+  const results = []
+
+  for (const targetKey of targets) {
+    try {
+      const configPath = getClientConfigPath(targetKey)
+      if (!configPath) {
+        results.push({ target: targetKey, success: false, message: '未知的客户端' })
+        continue
+      }
+
+      // 检查客户端是否已安装
+      const clients = scanner.scan()
+      const client = clients.find((c) => c.key === targetKey)
+      if (client && !client.installed) {
+        results.push({
+          target: targetKey,
+          success: false,
+          message: `${client.name} 未安装，无法配置`,
+        })
+        continue
+      }
+
+      if (targetKey === 'workbuddy') {
+        // WorkBuddy：写入 ~/.workbuddy/models.json 自定义模型列表（顶层数组格式）
+        applyWorkBuddyConfig(configPath, backup, config, models)
+      } else if (targetKey === 'codebuddy') {
+        // CodeBuddy：写入 ~/.codebuddy/models.json（{ "models": [...] } 包装格式）
+        applyCodeBuddyConfig(configPath, backup, config, models)
+      } else {
+        // 其他客户端：写入通用配置格式
+        if (backup) {
+          backupFile(configPath, targetKey)
+        }
+        fs.mkdirSync(path.dirname(configPath), { recursive: true })
+        fs.writeFileSync(configPath, generateConfigContent(config.apiEndpoint, config.apiKey), 'utf-8')
+      }
+
+      results.push({
+        target: targetKey,
+        success: true,
+        message: `${client?.name || targetKey} 配置成功`,
+      })
+    } catch (err) {
+      results.push({
+        target: targetKey,
+        success: false,
+        message: `配置失败：${err.message}`,
+      })
+    }
+  }
+
+  return results
+}
+
+/**
+ * 删除客户端配置
+ * @param {string[]} targets - 目标客户端 key 数组
+ * @param {Array<{ code: string }>} models - app.ts modelPage 返回的模型列表
+ * @returns {Array<{ target: string, success: boolean, message: string }>}
+ */
+function deleteConfig(targets, models = []) {
+  const results = []
+
+  for (const targetKey of targets) {
+    try {
+      const configPath = getClientConfigPath(targetKey)
+      if (!configPath) {
+        results.push({ target: targetKey, success: false, message: '未知的客户端' })
+        continue
+      }
+
+      if (targetKey === 'workbuddy') {
+        // WorkBuddy：仅清除本软件写入的模型条目
+        const handled = deleteWorkBuddyConfig(configPath, models)
+        results.push({
+          target: targetKey,
+          success: true,
+          message: handled ? '配置已删除' : '无托管模型配置，跳过',
+        })
+        continue
+      }
+
+      if (targetKey === 'codebuddy') {
+        // CodeBuddy：仅清除本软件写入的模型条目
+        const handled = deleteCodeBuddyConfig(configPath, models)
+        results.push({
+          target: targetKey,
+          success: true,
+          message: handled ? '配置已删除' : '无托管模型配置，跳过',
+        })
+        continue
+      }
+
+      if (fs.existsSync(configPath)) {
+        // 备份后删除
+        backupFile(configPath, targetKey, '-delete')
+        fs.unlinkSync(configPath)
+        results.push({ target: targetKey, success: true, message: '配置已删除' })
+      } else {
+        results.push({ target: targetKey, success: true, message: '无配置文件，跳过' })
+      }
+    } catch (err) {
+      results.push({
+        target: targetKey,
+        success: false,
+        message: `删除失败：${err.message}`,
+      })
+    }
+  }
+
+  return results
+}
+
+/**
+ * 获取目标客户端的最新一次备份文件路径
+ * 仅匹配「配置前备份」（如 workbuddy-1787847532450.json）；
+ * 排除删除备份（workbuddy-delete-xxx.json）——它是删除前的快照（含托管配置），
+ * 还原它会把已删除的托管配置写回，导致「删除后配置复活」
+ * @param {string} targetKey
+ * @returns {string|null}
+ */
+function getLatestBackup(targetKey) {
+  const backupDir = getBackupDir()
+  if (!fs.existsSync(backupDir)) return null
+
+  // 严格匹配 `${targetKey}-时间戳.json`，排除 -delete 后缀备份
+  const backupPattern = new RegExp(`^${targetKey}-(\\d+)\\.json$`)
+  const files = fs
+    .readdirSync(backupDir)
+    .filter((f) => backupPattern.test(f))
+    .sort((a, b) => {
+      const tsA = parseInt(a.match(backupPattern)?.[1] || '0', 10)
+      const tsB = parseInt(b.match(backupPattern)?.[1] || '0', 10)
+      return tsB - tsA
+    })
+
+  return files.length > 0 ? path.join(backupDir, files[0]) : null
+}
+
+/**
+ * 还原客户端配置（将最后一次自动备份的原文件复制回配置路径）
+ * @param {string[]} targets - 目标客户端 key 数组
+ * @returns {Array<{ target: string, success: boolean, message: string }>}
+ */
+function restoreConfig(targets) {
+  const results = []
+
+  for (const targetKey of targets) {
+    try {
+      const configPath = getClientConfigPath(targetKey)
+      if (!configPath) {
+        results.push({ target: targetKey, success: false, message: '未知的客户端' })
+        continue
+      }
+
+      const backupPath = getLatestBackup(targetKey)
+      if (!backupPath) {
+        // 无备份时降级为删除当前配置
+        if (fs.existsSync(configPath)) {
+          fs.unlinkSync(configPath)
+          results.push({ target: targetKey, success: true, message: '无备份，已删除当前配置' })
+        } else {
+          results.push({ target: targetKey, success: true, message: '无配置文件，跳过' })
+        }
+        continue
+      }
+
+      fs.copyFileSync(backupPath, configPath)
+      results.push({ target: targetKey, success: true, message: '配置已还原' })
+    } catch (err) {
+      results.push({ target: targetKey, success: false, message: `还原失败：${err.message}` })
+    }
+  }
+
+  return results
+}
+
+/** 打开备份目录 */
+async function openBackupDir() {
+  const backupDir = getBackupDir()
+  fs.mkdirSync(backupDir, { recursive: true })
+  await shell.openPath(backupDir)
+}
+
+module.exports = { applyConfig, deleteConfig, restoreConfig, openBackupDir, getBackupDir }

package/services/device.js

@@ -0,0 +1,104 @@
+/**
+ * 设备标识生成（硬件指纹）
+ * 格式：ST-xxxxxxxx-xxxxxxxx-xxxxxxxx-xxxxxxxx
+ * 持久化到 userData/device-id.json
+ */
+const { app } = require('electron')
+const fs = require('fs')
+const path = require('path')
+const crypto = require('crypto')
+const os = require('os')
+const { execSync } = require('child_process')
+
+/** 获取硬件 UUID */
+function getHardwareUUID() {
+  try {
+    if (process.platform === 'darwin') {
+      return execSync(
+        'ioreg -d 2 -c IOPlatformExpertDevice | awk -F\\" \'/IOPlatformUUID/{print $4}\'',
+        { encoding: 'utf-8' }
+      ).trim()
+    } else if (process.platform === 'win32') {
+      // 优先 wmic（老版本 Windows）
+      try {
+        const wmicResult = execSync('wmic csproduct get UUID', { encoding: 'utf-8', timeout: 5000 })
+          .split('\n')
+          .filter((l) => l.trim() && !l.includes('UUID'))[0]
+          ?.trim()
+        if (wmicResult) return wmicResult
+      } catch {
+        // wmic 不可用（Windows 11 25H2+ 已移除），降级到 PowerShell
+      }
+      return execSync(
+        'powershell -NoProfile -Command "(Get-CimInstance -ClassName Win32_ComputerSystemProduct).UUID"',
+        { encoding: 'utf-8', timeout: 10000 }
+      ).trim()
+    } else if (process.platform === 'linux') {
+      return (
+        fs.readFileSync('/etc/machine-id', 'utf-8').trim() ||
+        fs.readFileSync('/var/lib/dbus/machine-id', 'utf-8').trim()
+      )
+    }
+  } catch {
+    // 降级：使用 hostname + networkInterfaces
+  }
+  return ''
+}
+
+/** 生成设备标识 */
+function generateDeviceId() {
+  const uuid = getHardwareUUID()
+  const hostname = os.hostname()
+  const platform = process.platform
+  const nics = Object.values(os.networkInterfaces())
+    .flat()
+    .filter((ni) => ni && !ni.internal && ni.mac !== '00:00:00:00:00:00')
+    .map((ni) => ni.mac)
+    .join(',')
+
+  const raw = `${uuid}|${hostname}|${platform}|${nics}`
+  const hash = crypto.createHash('sha256').update(raw).digest('hex')
+
+  // 格式化为 ST-xxxxxxxx-xxxxxxxx-xxxxxxxx-xxxxxxxx
+  const parts = [
+    hash.slice(0, 8),
+    hash.slice(8, 16),
+    hash.slice(16, 24),
+    hash.slice(24, 32),
+  ]
+  return `ST-${parts.join('-')}`
+}
+
+/** 持久化文件路径 */
+function getDeviceIdFilePath() {
+  return path.join(app.getPath('userData'), 'device-id.json')
+}
+
+/** 获取设备 ID（不存在则生成并持久化） */
+function getDeviceId() {
+  const filePath = getDeviceIdFilePath()
+
+  // 尝试读取已保存的设备 ID
+  try {
+    if (fs.existsSync(filePath)) {
+      const data = JSON.parse(fs.readFileSync(filePath, 'utf-8'))
+      if (data.deviceId && data.deviceId.startsWith('ST-')) {
+        return data.deviceId
+      }
+    }
+  } catch {
+    // 文件损坏，重新生成
+  }
+
+  // 生成新设备 ID 并保存
+  const deviceId = generateDeviceId()
+  try {
+    fs.mkdirSync(path.dirname(filePath), { recursive: true })
+    fs.writeFileSync(filePath, JSON.stringify({ deviceId, createdAt: new Date().toISOString() }, null, 2))
+  } catch {
+    // 保存失败也返回设备 ID（内存中可用）
+  }
+  return deviceId
+}
+
+module.exports = { getDeviceId, generateDeviceId }