VoiceInput/Sources/Core/AgentBridge.swift

import Foundation

// MARK: - Agent 事件类型

enum AgentEvent {
    case status(mode: String)
    case toolStart(tool: String, display: String)
    case toolEnd(tool: String, resultLines: [String])
    case assistantChunk(content: String)
    case done(content: String, elapsedMs: Int, totalTokens: Int)
    case error(message: String)
}

// MARK: - AgentBridge

final class AgentBridge {

    private var process: Process?
    private var stdin: Pipe?
    private let agentDir: String
    private let logPath: String
    private var eventSeq = 0

    var onEvent: ((AgentEvent) -> Void)?
    var onExit: ((Int32) -> Void)?

    init(agentDir: String) {
        self.agentDir = agentDir
        // 日志写到用户目录（app bundle 内无写权限）
        let appSupport = FileManager.default.urls(for: .applicationSupportDirectory, in: .userDomainMask).first!
        let logDir = appSupport.appendingPathComponent("VoiceInput").path
        if !FileManager.default.fileExists(atPath: logDir) {
            try? FileManager.default.createDirectory(atPath: logDir, withIntermediateDirectories: true)
        }
        self.logPath = logDir + "/agent_debug.log"
        // 清空旧日志
        try? "".write(toFile: logPath, atomically: true, encoding: .utf8)
        log("======== AgentBridge 日志开始 ========")
        log("agentDir: \(agentDir)")
    }

    private func log(_ msg: String) {
        let ts = ISO8601DateFormatter().string(from: Date())
        let line = "[\(ts)] \(msg)\n"
        if let handle = FileHandle(forWritingAtPath: logPath) {
            handle.seekToEndOfFile()
            handle.write(line.data(using: .utf8)!)
            handle.closeFile()
        } else {
            try? line.write(toFile: logPath, atomically: true, encoding: .utf8)
        }
    }

    /// 启动 Node.js 桥接进程
    func start() -> Bool {
        guard let nodePath = findNode() else {
            print("[AgentBridge] ❌ Node.js not found")
            return false
        }

        let bridgePath = "\(agentDir)/src/bridge.js"
        guard FileManager.default.fileExists(atPath: bridgePath) else {
            print("[AgentBridge] ❌ bridge.js not found at \(bridgePath)")
            return false
        }

        let process = Process()
        process.executableURL = URL(fileURLWithPath: nodePath)
        process.arguments = [bridgePath, "--config", Config.shared.userAgentModelsPath]
        process.currentDirectoryURL = URL(fileURLWithPath: agentDir)

        // stdin → 向 bridge 发送输入
        let stdinPipe = Pipe()
        process.standardInput = stdinPipe
        self.stdin = stdinPipe

        // stdout → 读取 JSON 事件
        let stdoutPipe = Pipe()
        process.standardOutput = stdoutPipe

        // stderr → 只打印日志
        let stderrPipe = Pipe()
        process.standardError = stderrPipe

        // 读取 stdout
        stdoutPipe.fileHandleForReading.readabilityHandler = { [weak self] handle in
            let data = handle.availableData
            guard !data.isEmpty, let self = self else { return }
            if let str = String(data: data, encoding: .utf8) {
                let lines = str.components(separatedBy: "\n").filter { !$0.isEmpty }
                for line in lines {
                    self.parseLine(line)
                }
            }
        }

        // 读取 stderr
        stderrPipe.fileHandleForReading.readabilityHandler = { [weak self] handle in
            let data = handle.availableData
            if !data.isEmpty, let str = String(data: data, encoding: .utf8) {
                if !str.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty {
                    let trimmed = str.trimmingCharacters(in: .newlines)
                    print("[Agent] \(trimmed)")
                    self?.log("stderr: \(trimmed)")
                }
            }
        }

        process.terminationHandler = { [weak self] proc in
            let code = proc.terminationStatus
            let reason = proc.terminationReason
            self?.log("进程退出: status=\(code), reason=\(reason.rawValue)")
            self?.onExit?(code)
        }

        do {
            try process.run()
            self.process = process
            print("[AgentBridge] ✅ Started (pid \(process.processIdentifier))")
            return true
        } catch {
            print("[AgentBridge] ❌ Failed to start: \(error)")
            return false
        }
    }

    /// 发送对话请求（第一轮）
    func send(message: String, workspace: String, allowMode: String = "full_access", modelKey: String? = nil, newConversation: Bool = false) {
        sendInternal(message: message, workspace: workspace, allowMode: allowMode, modelKey: modelKey, newConversation: newConversation)
    }

    /// 追加消息（同一对话）
    func sendContinue(message: String) {
        sendInternal(message: message, workspace: "", allowMode: "", modelKey: nil)
    }

    private func sendInternal(message: String, workspace: String, allowMode: String, modelKey: String?, newConversation: Bool = false) {
        var payload: [String: Any] = [
            "action": "chat",
            "message": message,
        ]
        if !workspace.isEmpty { payload["workspace"] = workspace }
        if !allowMode.isEmpty { payload["allowMode"] = allowMode }
        if let mk = modelKey, !mk.isEmpty { payload["modelKey"] = mk }
        payload["thinkingMode"] = Config.shared.agentThinkingMode
        if newConversation { payload["new_conversation"] = true }

        guard let data = try? JSONSerialization.data(withJSONObject: payload),
              let json = String(data: data, encoding: .utf8) else {
            log("send() ❌ JSON 编码失败")
            print("[AgentBridge] ❌ Failed to encode message")
            return
        }
        log("send() JSON: \(json.prefix(200))")

        if let handle = stdin?.fileHandleForWriting {
            handle.write((json + "\n").data(using: .utf8)!)
            log("send() ✅ 已写入 stdin")
        } else {
            log("send() ❌ stdin handle 为 nil")
        }
    }

    /// 停止进程
    func stop() {
        log("stop() 终止进程")
        process?.terminate()
        process = nil
    }

    // MARK: - Private

    private func findNode() -> String? {
        // 优先级 0：bundle 内打包的 Node.js
        let bundledNode = agentDir + "/bin/node"
        if FileManager.default.fileExists(atPath: bundledNode) {
            return bundledNode
        }
        // 回退到系统路径
        let paths = [
            "/opt/homebrew/bin/node",
            "/usr/local/bin/node",
            "/usr/bin/node"
        ]
        for p in paths where FileManager.default.fileExists(atPath: p) {
            return p
        }
        // 尝试 which
        let task = Process()
        task.executableURL = URL(fileURLWithPath: "/usr/bin/env")
        task.arguments = ["which", "node"]
        let pipe = Pipe()
        task.standardOutput = pipe
        do {
            try task.run()
            task.waitUntilExit()
            let data = pipe.fileHandleForReading.readDataToEndOfFile()
            if let path = String(data: data, encoding: .utf8)?.trimmingCharacters(in: .whitespacesAndNewlines),
               !path.isEmpty {
                return path
            }
        } catch {
            print("[AgentBridge] which node failed: \(error)")
        }
        return nil
    }

    private func parseLine(_ line: String) {
        eventSeq += 1
        guard let data = line.data(using: .utf8),
              let json = try? JSONSerialization.jsonObject(with: data) as? [String: Any],
              let type = json["type"] as? String else {
            log("parseLine #\(eventSeq) 解析失败: \(line.prefix(100))")
            return
        }
        log("parseLine #\(eventSeq) type=\(type)")

        switch type {
        case "status":
            let mode = json["mode"] as? String ?? "unknown"
            log("  → status mode=\(mode)")
            onEvent?(.status(mode: mode))
        case "tool_start":
            let tool = json["tool"] as? String ?? "unknown"
            log("  → tool_start tool=\(tool)")
            onEvent?(.toolStart(
                tool: tool,
                display: json["display"] as? String ?? ""
            ))
        case "tool_end":
            let tool = json["tool"] as? String ?? "unknown"
            let resultLines = json["result"] as? [String] ?? []
            log("  → tool_end tool=\(tool) resultLines=\(resultLines.count)")
            onEvent?(.toolEnd(tool: tool, resultLines: resultLines))
        case "assistant_chunk":
            let content = json["content"] as? String ?? ""
            log("  → assistant_chunk len=\(content.count)")
            onEvent?(.assistantChunk(content: content))
        case "done":
            let content = json["content"] as? String ?? ""
            let elapsed = json["elapsed_ms"] as? Int ?? 0
            let totalTokens = json["total_tokens"] as? Int ?? 0
            log("  → done contentLen=\(content.count) elapsedMs=\(elapsed) totalTokens=\(totalTokens)")
            onEvent?(.done(content: content, elapsedMs: elapsed, totalTokens: totalTokens))
        case "error":
            let message = json["message"] as? String ?? "未知错误"
            log("  → error message=\(message)")
            onEvent?(.error(message: message))
        default:
            log("  → 未知 type: \(type)")
            break
        }
    }
}