writedock-mcp 0.4.4 → 0.4.7

package/source/src/setup.ts

@@ -0,0 +1,88 @@
+import { spawn } from 'node:child_process'
+import { copyFile, mkdir, readFile, writeFile } from 'node:fs/promises'
+import { homedir } from 'node:os'
+import { dirname, join } from 'node:path'
+import { bundledExtension, defaultExtensionDirectory, findInstalledExtension, installExtensionFiles } from './extension-update'
+
+const COMMAND = 'npx'
+const ARGS = ['-y', '--prefer-online', 'writedock-mcp@latest']
+
+async function readIfPresent(path: string) {
+  try { return await readFile(path, 'utf8') }
+  catch (error) { if ((error as NodeJS.ErrnoException).code === 'ENOENT') return undefined; throw error }
+}
+
+async function save(path: string, content: string, previous: string | undefined) {
+  await mkdir(dirname(path), { recursive: true })
+  if (previous !== undefined) await copyFile(path, `${path}.writedock-backup`)
+  await writeFile(path, content)
+}
+
+/** Keeps the user's other settings; an existing WriteDock entry stays, because it may hold a custom port or data directory. */
+export async function configureCodex(path: string): Promise<'added' | 'present'> {
+  const previous = await readIfPresent(path)
+  if (previous !== undefined && /^\s*\[mcp_servers\.writedock\]/m.test(previous)) return 'present'
+  const entry = `[mcp_servers.writedock]\ncommand = "${COMMAND}"\nargs = ${JSON.stringify(ARGS)}\nstartup_timeout_sec = 120\n`
+  await save(path, previous ? `${previous.replace(/\s*$/, '')}\n\n${entry}` : entry, previous)
+  return 'added'
+}
+
+export async function configureJsonClient(path: string): Promise<'added' | 'present'> {
+  const previous = await readIfPresent(path)
+  const config = previous?.trim() ? JSON.parse(previous) as { mcpServers?: Record<string, unknown> } : {}
+  if (config.mcpServers?.writedock) return 'present'
+  config.mcpServers = { ...config.mcpServers, writedock: { command: COMMAND, args: ARGS } }
+  await save(path, `${JSON.stringify(config, null, 2)}\n`, previous)
+  return 'added'
+}
+
+const exists = async (path: string) => (await readIfPresent(path)) !== undefined
+
+/** Best effort: shows the folder so the user can pick it, or drag it onto the extensions page. */
+function revealFolder(path: string) {
+  const command = process.platform === 'darwin' ? 'open' : process.platform === 'win32' ? 'explorer' : 'xdg-open'
+  try { const child = spawn(command, [path], { detached: true, stdio: 'ignore' }); child.on('error', () => {}); child.unref() } catch { /* The path is printed as well. */ }
+}
+
+/** Only clients already on this computer are configured, so setup never creates settings for an app the user does not have. */
+export async function runSetup(home = homedir(), log: (line: string) => void = console.log) {
+  const clients: { name: string; path: string; detect: string; configure: (path: string) => Promise<'added' | 'present'> }[] = [
+    { name: 'Codex', path: join(home, '.codex/config.toml'), detect: join(home, '.codex/config.toml'), configure: configureCodex },
+    { name: 'Claude Code', path: join(home, '.claude.json'), detect: join(home, '.claude.json'), configure: configureJsonClient },
+    { name: 'Cursor', path: join(home, '.cursor/mcp.json'), detect: join(home, '.cursor/mcp.json'), configure: configureJsonClient },
+    { name: 'Claude Desktop', path: join(home, 'Library/Application Support/Claude/claude_desktop_config.json'), detect: join(home, 'Library/Application Support/Claude/claude_desktop_config.json'), configure: configureJsonClient },
+  ]
+  let found = 0
+  for (const client of clients) {
+    if (!await exists(client.detect)) continue
+    found++
+    try {
+      const state = await client.configure(client.path)
+      log(state === 'added' ? `✓ ${client.name}：已添加 WriteDock（原配置备份为 .writedock-backup）` : `✓ ${client.name}：已有 WriteDock 配置，未改动`)
+    } catch (error) { log(`✗ ${client.name}：配置失败（${error instanceof Error ? error.message : String(error)}），原文件未改动`) }
+  }
+  if (!found) {
+    log('没有找到已知的 agent 客户端。请在你的客户端里添加这个 MCP 服务：')
+    log(JSON.stringify({ mcpServers: { writedock: { command: COMMAND, args: ARGS } } }, null, 2))
+  }
+  // Keep whichever folder the browser already loads, so a repeat setup does not ask for another manual load.
+  const bundle = bundledExtension()
+  const loaded = bundle ? await findInstalledExtension().catch(() => undefined) : undefined
+  let extensionFolder: string | undefined
+  if (bundle) {
+    try { extensionFolder = (await installExtensionFiles(bundle.directory, loaded || defaultExtensionDirectory(home))).directory }
+    catch (error) { log(`✗ 扩展：安装失败（${error instanceof Error ? error.message : String(error)}）`) }
+  }
+  log('')
+  log('接下来：')
+  log('1. 重启你的 agent 客户端，让它加载 WriteDock。')
+  if (extensionFolder && !loaded) {
+    log('2. 在 Chrome 或 Edge 打开 chrome://extensions，打开右上角的“开发者模式”，点“加载已解压的扩展程序”，选择这个文件夹：')
+    log(`   ${extensionFolder}`)
+    revealFolder(extensionFolder)
+    log('3. 打开 WriteDock 扩展，点“连接”。')
+  } else if (extensionFolder) {
+    log(`2. 扩展已更新。在 chrome://extensions 点一次 WriteDock 的刷新图标，然后点“连接”。`)
+  } else log('2. 打开浏览器的 WriteDock 扩展，点“连接”。')
+  return found
+}