native macOS codings agent orchestrator prowl.onev.cat
Something went wrong. Try again.
8.1 kB · 218 lines
Swift
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219import Foundation
/// Known agent install/launcher binary names (lowercased). Aliases map to the/// same agent; extend this table when onboarding a new CLI.private let knownAgentBinaries: [String: DetectedAgent] = [ "pi": .pi, "omp": .omp, "oh-my-pi": .omp, "claude": .claude, "claude-code": .claude, "codex": .codex, "omx": .codex, "oh-my-codex": .codex, "gemini": .gemini, "cursor": .cursor, "cursor-agent": .cursor, "cline": .cline, "opencode": .opencode, "open-code": .opencode, "copilot": .copilot, "github-copilot": .copilot, "ghcs": .copilot, "kimi": .kimi, "kimi code": .kimi, "devin": .devin, "droid": .droid, "amp": .amp, "amp-local": .amp, "qodercli": .qoder, "qwen": .qwen, "grok": .grok, // Bare `antigravity` stays unmapped: the Antigravity IDE ships an // `antigravity` launcher (and an `agy` shim) that opens the desktop app. "agy": .antigravity, "antigravity-cli": .antigravity, "antigravity_cli": .antigravity,]
func identifyAgent(processName: String) -> DetectedAgent? { let lower = processName.lowercased() if let agent = knownAgentBinaries[lower] { return agent } // Versioned install binary only, e.g. `grok-0.2.101-macos-aarch64`. // Must not match model-id tokens like `grok-4` / `grok-4.5` that show up // as argv fragments of other agents (score-40 wrapped-runtime candidates). if isGrokVersionedBinaryName(lower) { return .grok } return nil}
/// Install packages are named `grok-<semver>-<platform>-<arch>` (verified/// against `~/.grok/downloads/grok-0.2.101-macos-aarch64`). Model ids/// (`grok-4`, `grok-4.5`) never include a platform segment.private func isGrokVersionedBinaryName(_ lower: String) -> Bool { guard lower.hasPrefix("grok-") else { return false } return lower.contains("-macos-") || lower.contains("-linux-") || lower.contains("-windows-")}
struct IdentifiedAgentProcess: Equatable, Sendable { let agent: DetectedAgent let name: String let process: ForegroundProcess /// The process the shell launched for this agent (`ForegroundJob.launchProcessID(of:)`). /// `process` is what state and sessions are read from; this is what a launch is /// identified by, so an engine child taking over `process` is not a replacement. let launchProcessID: pid_t
init(agent: DetectedAgent, name: String, process: ForegroundProcess, launchProcessID: pid_t? = nil) { self.agent = agent self.name = name self.process = process self.launchProcessID = launchProcessID ?? process.pid }
/// Icon token for `CommandIconMap`. The shared `agent` entrypoint name maps /// to the Cursor icon there, so alias-identified agents resolve through the /// detected agent instead of the raw process name. var iconLookupToken: String { name == "agent" ? agent.iconLookupToken : name }}
func identifyAgentInJob(_ job: ForegroundJob) -> IdentifiedAgentProcess? { var best: AgentCandidate?
for process in job.processes { for candidate in agentCandidates(for: process) { guard let agent = identifyAgent(candidate: candidate, process: process) else { continue } if best == nil || candidate.score > best!.score { best = AgentCandidate(score: candidate.score, agent: agent, name: candidate.name, process: process) } } }
return best.map { IdentifiedAgentProcess( agent: $0.agent, name: $0.name, process: $0.process, launchProcessID: job.launchProcessID(of: $0.process.pid) ) }}
private func agentCandidates(for process: ForegroundProcess) -> [(name: String, score: Int)] { var candidates: [(String, Int)] = [] let arguments = process.arguments ?? process.cmdline?.split(whereSeparator: \.isWhitespace).map(String.init) ?? [] // agy spawns a transient `--bg-updater` child in the same foreground job // (verified argv on 1.3.1: `agy --bg-updater --app_data_dir=…`). It shares // argv0 but holds no presence lock, so the interactive TUI must always win // the pick. The pinned first position keeps a prompt payload that merely // mentions the token from demoting the real TUI. // agy's Go-style flags accept one dash too, so `-bg-updater` spells the // same updater mode. let isAntigravityUpdater = arguments.dropFirst().first == "--bg-updater" || arguments.dropFirst().first == "-bg-updater"
if let argv0 = process.argv0, let name = normalizedProcessName(argv0) { // The REPL delegates to `devin acp`, which owns the native session lock. // Keep the shell-launched parent as launchProcessID while resolving that child. let isDevinEngine = name == "devin" && arguments.dropFirst().first == "acp" let demoted = isAntigravityUpdater && identifyAgent(processName: name) == .antigravity candidates.append((name, isDevinEngine ? 90 : demoted ? 60 : 80)) } if let name = normalizedProcessName(process.name) { // Demote the comm-name candidate too (50 clears the score-40 wrapped-runtime // guard), or a missing TUI argv0 would tie the pick at 70. let demoted = isAntigravityUpdater && identifyAgent(processName: name) == .antigravity candidates.append((name, demoted ? 50 : 70)) }
let primaryName = normalizedProcessName(process.argv0 ?? process.name) ?? process.name.lowercased() if isWrappedRuntime(primaryName), let cmdline = process.cmdline { for token in cmdline.split(whereSeparator: \.isWhitespace) { guard let name = normalizedProcessName(String(token)) else { continue } candidates.append((name, 40)) } }
return candidates}
private func identifyAgent(candidate: (name: String, score: Int), process: ForegroundProcess) -> DetectedAgent? { if candidate.name == "agent" { // Cursor and Grok Build both ship an `agent` entrypoint. Disambiguate from // path/cmdline evidence only — bare `agent` stays unknown. if isCursorAgentAlias(process) { return .cursor } if isGrokAgentAlias(process) { return .grok } return nil } // Grok Build, Devin, and Antigravity are native executables, not wrapped-runtime // scripts. Their names can also appear as model arguments or quoted prompt // tokens in a wrapper's cmdline; require argv0/name evidence. if ["grok", "devin", "agy", "antigravity-cli", "antigravity_cli"].contains(candidate.name), candidate.score == 40 { return nil } return identifyAgent(processName: candidate.name)}
private func isCursorAgentAlias(_ process: ForegroundProcess) -> Bool { let haystack = agentAliasHaystack(process) return haystack.contains("cursor-agent") || haystack.contains("cursor.app")}
private func isGrokAgentAlias(_ process: ForegroundProcess) -> Bool { // Production `argv0` is basename-only (`ProcessDetection` strips the path); // the full executable path is the first whitespace token of `cmdline`. // Only inspect those two executable locations — never later argv tokens // (e.g. `agent --model grok-4` must stay unknown). let executablePaths = [ process.argv0, process.cmdline?.split(whereSeparator: \.isWhitespace).first.map(String.init), ] .compactMap { $0?.lowercased() }
for path in executablePaths { if path.contains("/.grok/") { return true } if let basename = ProcessDetection.basename(path), isGrokVersionedBinaryName(basename) { return true } } return false}
private func agentAliasHaystack(_ process: ForegroundProcess) -> String { [ process.argv0, process.cmdline, ] .compactMap(\.self) .joined(separator: " ") .lowercased()}
private struct AgentCandidate { let score: Int let agent: DetectedAgent let name: String let process: ForegroundProcess}
private func normalizedProcessName(_ raw: String) -> String? { guard let basename = ProcessDetection.basename(raw) else { return nil } let lower = basename.lowercased() if lower.hasSuffix(".js") { return String(lower.dropLast(3)) } return lower}
private func isWrappedRuntime(_ name: String) -> Bool { [ "node", "bun", "python", "python3", "ruby", "deno", "sh", "bash", "zsh", "fish", "tmux", "npx", "bunx", ].contains(name)}