openAgent

1import type { CliRenderer } from "@opentui/core"
2import { existsSync, readdirSync, readFileSync, statSync } from "node:fs"
3import { readFile, rm, writeFile } from "node:fs/promises"
4import os from "node:os"
5import path from "node:path"
6import { spawn } from "node:child_process"
7import type { Stream } from "node:stream"
8import { resolveZedDbPath, resolveZedSelection } from "./editor-zed"
9 
10type EditorStdio = "inherit" | "pipe" | "ignore" | number | Stream
11 
12export function normalizePromptContent(content: string) {
13  if (content.endsWith("\r\n")) {
14    const body = content.slice(0, -2)
15    return !body.includes("\n") && !body.includes("\r") ? body : content
16  }
17 
18  if (content.endsWith("\n")) {
19    const body = content.slice(0, -1)
20    return !body.includes("\n") && !body.includes("\r") ? body : content
21  }
22 
23  return content
24}
25 
26export async function openEditor(input: { value: string; renderer: CliRenderer; cwd?: string; stdin?: EditorStdio }) {
27  const editor = process.env.VISUAL || process.env.EDITOR
28  if (!editor) return
29  const file = path.join(os.tmpdir(), `${Date.now()}.md`)
30  await writeFile(file, input.value)
31  input.renderer.suspend()
32  input.renderer.currentRenderBuffer.clear()
33  try {
34    await new Promise<void>((resolve, reject) => {
35      const parts = editor.split(" ")
36      const child = spawn(parts[0]!, [...parts.slice(1), file], {
37        cwd: input.cwd && existsSync(input.cwd) ? input.cwd : process.cwd(),
38        stdio: [input.stdin ?? "inherit", "inherit", "inherit"],
39        shell: process.platform === "win32",
40      })
41      child.on("error", reject)
42      child.on("exit", (code, signal) => {
43        if (code === 0) return resolve()
44        reject(new Error(`Editor exited with ${signal ? `signal ${signal}` : `code ${code}`}`))
45      })
46    })
47    return (await readFile(file, "utf8")) || undefined
48  } finally {
49    await rm(file, { force: true }).catch(() => {})
50    input.renderer.currentRenderBuffer.clear()
51    input.renderer.resume()
52    input.renderer.requestRender()
53  }
54}
55 
56export function discoverEditorConnection(directory: string) {
57  const root = path.join(os.homedir(), ".claude", "ide")
58  const contains = (parent: string) => {
59    const resolved = path.resolve(parent)
60    const relative = path.relative(resolved, path.resolve(directory))
61    return relative === "" || (!relative.startsWith("..") && !path.isAbsolute(relative)) ? resolved.length : 0
62  }
63  try {
64    return readdirSync(root)
65      .filter((entry) => entry.endsWith(".lock"))
66      .flatMap((entry) => {
67        const file = path.join(root, entry)
68        const port = Number.parseInt(path.basename(file, ".lock"), 10)
69        if (!Number.isInteger(port) || port <= 0 || port > 65535) return []
70        try {
71          const value = JSON.parse(readFileSync(file, "utf8")) as Record<string, unknown>
72          if (value.transport !== undefined && value.transport !== "ws") return []
73          const folders = Array.isArray(value.workspaceFolders)
74            ? value.workspaceFolders.filter((item): item is string => typeof item === "string")
75            : []
76          const score = Math.max(0, ...folders.map(contains))
77          if (!score) return []
78          return [
79            {
80              url: `ws://127.0.0.1:${port}`,
81              authToken: typeof value.authToken === "string" ? value.authToken : undefined,
82              source: `lock:${port}`,
83              score,
84              mtime: statSync(file).mtimeMs,
85            },
86          ]
87        } catch {
88          return []
89        }
90      })
91      .sort((left, right) => right.score - left.score || right.mtime - left.mtime)
92      .map(({ url, authToken, source }) => ({ url, authToken, source }))[0]
93  } catch {
94    return undefined
95  }
96}
97 
98export const editorIntegration = {
99  connection: discoverEditorConnection,
100  selection: (directory: string) => resolveZedSelection(resolveZedDbPath() ?? "", directory),
101}
102