697 lines
27 KiB
Swift
697 lines
27 KiB
Swift
import ClawdisIPC
|
||
import Darwin
|
||
import Foundation
|
||
|
||
@main
|
||
struct ClawdisCLI {
|
||
static func main() async {
|
||
do {
|
||
var args = Array(CommandLine.arguments.dropFirst())
|
||
let jsonOutput = args.contains("--json")
|
||
args.removeAll(where: { $0 == "--json" })
|
||
|
||
let parsed = try parseCommandLine(args: args)
|
||
let response = try await send(request: parsed.request)
|
||
|
||
if jsonOutput {
|
||
try self.printJSON(parsed: parsed, response: response)
|
||
} else {
|
||
try self.printText(parsed: parsed, response: response)
|
||
}
|
||
|
||
exit(response.ok ? 0 : 1)
|
||
} catch CLIError.help {
|
||
self.printHelp()
|
||
exit(0)
|
||
} catch CLIError.version {
|
||
self.printVersion()
|
||
exit(0)
|
||
} catch {
|
||
fputs("clawdis-mac error: \(error)\n", stderr)
|
||
exit(2)
|
||
}
|
||
}
|
||
|
||
private struct ParsedCLIRequest {
|
||
var request: Request
|
||
var kind: Kind
|
||
|
||
enum Kind {
|
||
case uiScreens
|
||
case uiScreenshot
|
||
case generic
|
||
}
|
||
}
|
||
|
||
// swiftlint:disable cyclomatic_complexity
|
||
private static func parseCommandLine(args: [String]) throws -> ParsedCLIRequest {
|
||
var args = args
|
||
guard let command = args.first else { throw CLIError.help }
|
||
args = Array(args.dropFirst())
|
||
|
||
switch command {
|
||
case "--help", "-h", "help":
|
||
throw CLIError.help
|
||
|
||
case "--version", "-V", "version":
|
||
throw CLIError.version
|
||
|
||
case "notify":
|
||
var title: String?
|
||
var body: String?
|
||
var sound: String?
|
||
var priority: NotificationPriority?
|
||
var delivery: NotificationDelivery?
|
||
while !args.isEmpty {
|
||
let arg = args.removeFirst()
|
||
switch arg {
|
||
case "--title": title = args.popFirst()
|
||
case "--body": body = args.popFirst()
|
||
case "--sound": sound = args.popFirst()
|
||
case "--priority":
|
||
if let val = args.popFirst(), let p = NotificationPriority(rawValue: val) { priority = p }
|
||
case "--delivery":
|
||
if let val = args.popFirst(), let d = NotificationDelivery(rawValue: val) { delivery = d }
|
||
default: break
|
||
}
|
||
}
|
||
guard let t = title, let b = body else { throw CLIError.help }
|
||
return ParsedCLIRequest(
|
||
request: .notify(title: t, body: b, sound: sound, priority: priority, delivery: delivery),
|
||
kind: .generic)
|
||
|
||
case "ensure-permissions":
|
||
var caps: [Capability] = []
|
||
var interactive = false
|
||
while !args.isEmpty {
|
||
let arg = args.removeFirst()
|
||
switch arg {
|
||
case "--cap":
|
||
if let val = args.popFirst(), let cap = Capability(rawValue: val) { caps.append(cap) }
|
||
case "--interactive": interactive = true
|
||
default: break
|
||
}
|
||
}
|
||
if caps.isEmpty { caps = Capability.allCases }
|
||
return ParsedCLIRequest(request: .ensurePermissions(caps, interactive: interactive), kind: .generic)
|
||
|
||
case "ui":
|
||
guard let sub = args.first else { throw CLIError.help }
|
||
args = Array(args.dropFirst())
|
||
|
||
switch sub {
|
||
case "screens":
|
||
return ParsedCLIRequest(request: .uiListScreens, kind: .uiScreens)
|
||
case "screenshot":
|
||
var screenIndex: Int?
|
||
var windowID: UInt32?
|
||
while !args.isEmpty {
|
||
let arg = args.removeFirst()
|
||
switch arg {
|
||
case "--screen-index": screenIndex = args.popFirst().flatMap(Int.init)
|
||
case "--window-id": windowID = args.popFirst().flatMap(UInt32.init)
|
||
default: break
|
||
}
|
||
}
|
||
return ParsedCLIRequest(request: .uiScreenshot(screenIndex: screenIndex, windowID: windowID), kind: .uiScreenshot)
|
||
default:
|
||
throw CLIError.help
|
||
}
|
||
|
||
case "run":
|
||
var cwd: String?
|
||
var env: [String: String] = [:]
|
||
var timeout: Double?
|
||
var needsSR = false
|
||
var cmd: [String] = []
|
||
while !args.isEmpty {
|
||
let arg = args.removeFirst()
|
||
switch arg {
|
||
case "--cwd": cwd = args.popFirst()
|
||
|
||
case "--env":
|
||
if let pair = args.popFirst(), let eq = pair.firstIndex(of: "=") {
|
||
let k = String(pair[..<eq]); let v = String(pair[pair.index(after: eq)...]); env[k] = v
|
||
}
|
||
|
||
case "--timeout": if let val = args.popFirst(), let dbl = Double(val) { timeout = dbl }
|
||
|
||
case "--needs-screen-recording": needsSR = true
|
||
|
||
default:
|
||
cmd.append(arg)
|
||
}
|
||
}
|
||
return ParsedCLIRequest(request: .runShell(
|
||
command: cmd,
|
||
cwd: cwd,
|
||
env: env.isEmpty ? nil : env,
|
||
timeoutSec: timeout,
|
||
needsScreenRecording: needsSR), kind: .generic)
|
||
|
||
case "status":
|
||
return ParsedCLIRequest(request: .status, kind: .generic)
|
||
|
||
case "rpc-status":
|
||
return ParsedCLIRequest(request: .rpcStatus, kind: .generic)
|
||
|
||
case "agent":
|
||
var message: String?
|
||
var thinking: String?
|
||
var session: String?
|
||
var deliver = false
|
||
var to: String?
|
||
|
||
while !args.isEmpty {
|
||
let arg = args.removeFirst()
|
||
switch arg {
|
||
case "--message": message = args.popFirst()
|
||
case "--thinking": thinking = args.popFirst()
|
||
case "--session": session = args.popFirst()
|
||
case "--deliver": deliver = true
|
||
case "--to": to = args.popFirst()
|
||
default:
|
||
// Support bare message as last argument
|
||
if message == nil {
|
||
message = arg
|
||
}
|
||
}
|
||
}
|
||
|
||
guard let message else { throw CLIError.help }
|
||
return ParsedCLIRequest(
|
||
request: .agent(message: message, thinking: thinking, session: session, deliver: deliver, to: to),
|
||
kind: .generic)
|
||
|
||
case "node":
|
||
guard let sub = args.first else { throw CLIError.help }
|
||
args = Array(args.dropFirst())
|
||
|
||
switch sub {
|
||
case "list":
|
||
return ParsedCLIRequest(request: .nodeList, kind: .generic)
|
||
|
||
case "invoke":
|
||
var nodeId: String?
|
||
var command: String?
|
||
var paramsJSON: String?
|
||
while !args.isEmpty {
|
||
let arg = args.removeFirst()
|
||
switch arg {
|
||
case "--node": nodeId = args.popFirst()
|
||
case "--command": command = args.popFirst()
|
||
case "--params-json": paramsJSON = args.popFirst()
|
||
default: break
|
||
}
|
||
}
|
||
guard let nodeId, let command else { throw CLIError.help }
|
||
return ParsedCLIRequest(
|
||
request: .nodeInvoke(nodeId: nodeId, command: command, paramsJSON: paramsJSON),
|
||
kind: .generic)
|
||
|
||
default:
|
||
throw CLIError.help
|
||
}
|
||
|
||
case "canvas":
|
||
guard let sub = args.first else { throw CLIError.help }
|
||
args = Array(args.dropFirst())
|
||
|
||
switch sub {
|
||
case "show":
|
||
var session = "main"
|
||
var path: String?
|
||
var x: Double?
|
||
var y: Double?
|
||
var width: Double?
|
||
var height: Double?
|
||
while !args.isEmpty {
|
||
let arg = args.removeFirst()
|
||
switch arg {
|
||
case "--session": session = args.popFirst() ?? session
|
||
case "--path": path = args.popFirst()
|
||
case "--x": x = args.popFirst().flatMap(Double.init)
|
||
case "--y": y = args.popFirst().flatMap(Double.init)
|
||
case "--width": width = args.popFirst().flatMap(Double.init)
|
||
case "--height": height = args.popFirst().flatMap(Double.init)
|
||
default: break
|
||
}
|
||
}
|
||
let placement = (x != nil || y != nil || width != nil || height != nil)
|
||
? CanvasPlacement(x: x, y: y, width: width, height: height)
|
||
: nil
|
||
return ParsedCLIRequest(
|
||
request: .canvasShow(session: session, path: path, placement: placement),
|
||
kind: .generic)
|
||
|
||
case "hide":
|
||
var session = "main"
|
||
while !args.isEmpty {
|
||
let arg = args.removeFirst()
|
||
switch arg {
|
||
case "--session": session = args.popFirst() ?? session
|
||
default: break
|
||
}
|
||
}
|
||
return ParsedCLIRequest(request: .canvasHide(session: session), kind: .generic)
|
||
|
||
case "goto":
|
||
var session = "main"
|
||
var path: String?
|
||
var x: Double?
|
||
var y: Double?
|
||
var width: Double?
|
||
var height: Double?
|
||
while !args.isEmpty {
|
||
let arg = args.removeFirst()
|
||
switch arg {
|
||
case "--session": session = args.popFirst() ?? session
|
||
case "--path": path = args.popFirst()
|
||
case "--x": x = args.popFirst().flatMap(Double.init)
|
||
case "--y": y = args.popFirst().flatMap(Double.init)
|
||
case "--width": width = args.popFirst().flatMap(Double.init)
|
||
case "--height": height = args.popFirst().flatMap(Double.init)
|
||
default: break
|
||
}
|
||
}
|
||
guard let path else { throw CLIError.help }
|
||
let placement = (x != nil || y != nil || width != nil || height != nil)
|
||
? CanvasPlacement(x: x, y: y, width: width, height: height)
|
||
: nil
|
||
return ParsedCLIRequest(
|
||
request: .canvasGoto(session: session, path: path, placement: placement),
|
||
kind: .generic)
|
||
|
||
case "eval":
|
||
var session = "main"
|
||
var js: String?
|
||
while !args.isEmpty {
|
||
let arg = args.removeFirst()
|
||
switch arg {
|
||
case "--session": session = args.popFirst() ?? session
|
||
case "--js": js = args.popFirst()
|
||
default: break
|
||
}
|
||
}
|
||
guard let js else { throw CLIError.help }
|
||
return ParsedCLIRequest(request: .canvasEval(session: session, javaScript: js), kind: .generic)
|
||
|
||
case "snapshot":
|
||
var session = "main"
|
||
var outPath: String?
|
||
while !args.isEmpty {
|
||
let arg = args.removeFirst()
|
||
switch arg {
|
||
case "--session": session = args.popFirst() ?? session
|
||
case "--out": outPath = args.popFirst()
|
||
default: break
|
||
}
|
||
}
|
||
return ParsedCLIRequest(request: .canvasSnapshot(session: session, outPath: outPath), kind: .generic)
|
||
|
||
default:
|
||
throw CLIError.help
|
||
}
|
||
|
||
default:
|
||
throw CLIError.help
|
||
}
|
||
}
|
||
|
||
// swiftlint:enable cyclomatic_complexity
|
||
|
||
private static func printText(parsed: ParsedCLIRequest, response: Response) throws {
|
||
guard response.ok else {
|
||
let msg = response.message ?? "failed"
|
||
fputs("\(msg)\n", stderr)
|
||
return
|
||
}
|
||
|
||
switch parsed.kind {
|
||
case .uiScreens:
|
||
let screens = try self.decodePayload([UIScreenInfo].self, payload: response.payload)
|
||
if screens.isEmpty {
|
||
FileHandle.standardOutput.write(Data("No screens\n".utf8))
|
||
return
|
||
}
|
||
for s in screens {
|
||
let primary = s.isPrimary ? " (primary)" : ""
|
||
let size = "\(Int(s.frame.width))×\(Int(s.frame.height))"
|
||
let scale = String(format: "%.1f", Double(s.scaleFactor))
|
||
let line = "Display \(s.index + 1)\(primary): \(s.name) \(size) @\(scale)x (id \(s.displayID))\n"
|
||
FileHandle.standardOutput.write(Data(line.utf8))
|
||
}
|
||
|
||
case .uiScreenshot:
|
||
let result = try self.decodePayload(UIScreenshotResult.self, payload: response.payload)
|
||
FileHandle.standardOutput.write(Data((result.path + "\n").utf8))
|
||
|
||
case .generic:
|
||
if let payload = response.payload, let text = String(data: payload, encoding: .utf8), !text.isEmpty {
|
||
FileHandle.standardOutput.write(payload)
|
||
if !text.hasSuffix("\n") { FileHandle.standardOutput.write(Data([0x0A])) }
|
||
return
|
||
}
|
||
if let message = response.message, !message.isEmpty {
|
||
FileHandle.standardOutput.write(Data((message + "\n").utf8))
|
||
}
|
||
}
|
||
}
|
||
|
||
private static func printJSON(parsed: ParsedCLIRequest, response: Response) throws {
|
||
var output: [String: Any] = [
|
||
"ok": response.ok,
|
||
"message": response.message ?? "",
|
||
]
|
||
|
||
switch parsed.kind {
|
||
case .uiScreens:
|
||
if let payload = response.payload,
|
||
let obj = try? JSONSerialization.jsonObject(with: payload) {
|
||
output["result"] = obj
|
||
} else {
|
||
output["result"] = []
|
||
}
|
||
|
||
case .uiScreenshot:
|
||
if let payload = response.payload,
|
||
let obj = try? JSONSerialization.jsonObject(with: payload) {
|
||
output["result"] = obj
|
||
} else {
|
||
output["result"] = NSNull()
|
||
}
|
||
|
||
case .generic:
|
||
if let payload = response.payload, !payload.isEmpty {
|
||
if let obj = try? JSONSerialization.jsonObject(with: payload) {
|
||
output["result"] = obj
|
||
} else if let text = String(data: payload, encoding: .utf8) {
|
||
output["payload"] = text
|
||
}
|
||
}
|
||
}
|
||
|
||
let json = try JSONSerialization.data(withJSONObject: output, options: [.prettyPrinted])
|
||
FileHandle.standardOutput.write(json)
|
||
FileHandle.standardOutput.write(Data([0x0A]))
|
||
}
|
||
|
||
private static func decodePayload<T: Decodable>(_ type: T.Type, payload: Data?) throws -> T {
|
||
guard let payload else { throw POSIXError(.EINVAL) }
|
||
return try JSONDecoder().decode(T.self, from: payload)
|
||
}
|
||
|
||
private static func printHelp() {
|
||
let usage = """
|
||
clawdis-mac — talk to the running Clawdis.app XPC service
|
||
|
||
Usage:
|
||
clawdis-mac [--json] <command> ...
|
||
clawdis-mac notify --title <t> --body <b> [--sound <name>] [--priority <passive|active|timeSensitive>] [--delivery <system|overlay|auto>]
|
||
clawdis-mac ensure-permissions
|
||
[--cap <notifications|accessibility|screenRecording|microphone|speechRecognition>]
|
||
[--interactive]
|
||
clawdis-mac ui screens
|
||
clawdis-mac ui screenshot [--screen-index <n>] [--window-id <u32>]
|
||
clawdis-mac run [--cwd <path>] [--env KEY=VAL] [--timeout <sec>] [--needs-screen-recording] <command ...>
|
||
clawdis-mac status
|
||
clawdis-mac rpc-status
|
||
clawdis-mac agent --message <text> [--thinking <low|default|high>]
|
||
[--session <key>] [--deliver] [--to <E.164>]
|
||
clawdis-mac node list
|
||
clawdis-mac node invoke --node <id> --command <name> [--params-json <json>]
|
||
clawdis-mac canvas show [--session <key>] [--path </...>]
|
||
[--x <screenX> --y <screenY>] [--width <w> --height <h>]
|
||
clawdis-mac canvas hide [--session <key>]
|
||
clawdis-mac canvas goto --path </...> [--session <key>]
|
||
[--x <screenX> --y <screenY>] [--width <w> --height <h>]
|
||
clawdis-mac canvas eval --js <code> [--session <key>]
|
||
clawdis-mac canvas snapshot [--out <path>] [--session <key>]
|
||
clawdis-mac --help
|
||
|
||
Default output is text. Use --json for machine-readable output.
|
||
"""
|
||
print(usage)
|
||
}
|
||
|
||
private static func printVersion() {
|
||
let info = self.loadInfo()
|
||
let version = (info["CFBundleShortVersionString"] as? String) ?? self.loadPackageJSONVersion() ?? "unknown"
|
||
var build = info["CFBundleVersion"] as? String ?? ""
|
||
if build.isEmpty, version != "unknown" {
|
||
build = version
|
||
}
|
||
let git = info["ClawdisGitCommit"] as? String ?? "unknown"
|
||
let ts = info["ClawdisBuildTimestamp"] as? String ?? "unknown"
|
||
|
||
let buildPart = build.isEmpty ? "" : " (\(build))"
|
||
print("clawdis-mac \(version)\(buildPart) git:\(git) built:\(ts)")
|
||
}
|
||
|
||
private static func loadInfo() -> [String: Any] {
|
||
if let dict = Bundle.main.infoDictionary, !dict.isEmpty { return dict }
|
||
|
||
guard let exeURL = self.resolveExecutableURL() else { return [:] }
|
||
|
||
var dir = exeURL.deletingLastPathComponent()
|
||
for _ in 0..<10 {
|
||
let candidate = dir.appendingPathComponent("Info.plist")
|
||
if let dict = self.loadPlistDictionary(at: candidate) {
|
||
return dict
|
||
}
|
||
let parent = dir.deletingLastPathComponent()
|
||
if parent.path == dir.path { break }
|
||
dir = parent
|
||
}
|
||
|
||
return [:]
|
||
}
|
||
|
||
private static func loadPlistDictionary(at url: URL) -> [String: Any]? {
|
||
guard let data = try? Data(contentsOf: url) else { return nil }
|
||
return try? PropertyListSerialization
|
||
.propertyList(from: data, options: [], format: nil) as? [String: Any]
|
||
}
|
||
|
||
private static func resolveExecutableURL() -> URL? {
|
||
var size = UInt32(PATH_MAX)
|
||
var buffer = [CChar](repeating: 0, count: Int(size))
|
||
|
||
let result = buffer.withUnsafeMutableBufferPointer { ptr in
|
||
_NSGetExecutablePath(ptr.baseAddress, &size)
|
||
}
|
||
|
||
if result != 0 {
|
||
buffer = [CChar](repeating: 0, count: Int(size))
|
||
let result2 = buffer.withUnsafeMutableBufferPointer { ptr in
|
||
_NSGetExecutablePath(ptr.baseAddress, &size)
|
||
}
|
||
guard result2 == 0 else { return nil }
|
||
}
|
||
|
||
let nulIndex = buffer.firstIndex(of: 0) ?? buffer.count
|
||
let bytes = buffer.prefix(nulIndex).map { UInt8(bitPattern: $0) }
|
||
let path = String(decoding: bytes, as: UTF8.self)
|
||
return URL(fileURLWithPath: path).resolvingSymlinksInPath()
|
||
}
|
||
|
||
private static func loadPackageJSONVersion() -> String? {
|
||
guard let exeURL = self.resolveExecutableURL() else { return nil }
|
||
|
||
var dir = exeURL.deletingLastPathComponent()
|
||
for _ in 0..<12 {
|
||
let candidate = dir.appendingPathComponent("package.json")
|
||
if let version = self.loadPackageJSONVersion(at: candidate) {
|
||
return version
|
||
}
|
||
let parent = dir.deletingLastPathComponent()
|
||
if parent.path == dir.path { break }
|
||
dir = parent
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
private static func loadPackageJSONVersion(at url: URL) -> String? {
|
||
guard let data = try? Data(contentsOf: url) else { return nil }
|
||
guard let obj = try? JSONSerialization.jsonObject(with: data) as? [String: Any] else { return nil }
|
||
guard obj["name"] as? String == "clawdis" else { return nil }
|
||
return obj["version"] as? String
|
||
}
|
||
|
||
private static func send(request: Request) async throws -> Response {
|
||
try await self.ensureAppRunning()
|
||
|
||
let timeout = self.rpcTimeoutSeconds(for: request)
|
||
return try await self.sendViaSocket(request: request, timeoutSeconds: timeout)
|
||
}
|
||
|
||
/// Attempt a direct UNIX socket call; falls back to XPC if unavailable.
|
||
private static func sendViaSocket(request: Request, timeoutSeconds: TimeInterval) async throws -> Response {
|
||
let path = controlSocketPath
|
||
let deadline = Date().addingTimeInterval(timeoutSeconds)
|
||
let fd = socket(AF_UNIX, SOCK_STREAM, 0)
|
||
guard fd >= 0 else { throw POSIXError(.ECONNREFUSED) }
|
||
defer { close(fd) }
|
||
|
||
var noSigPipe: Int32 = 1
|
||
_ = setsockopt(fd, SOL_SOCKET, SO_NOSIGPIPE, &noSigPipe, socklen_t(MemoryLayout.size(ofValue: noSigPipe)))
|
||
|
||
let flags = fcntl(fd, F_GETFL)
|
||
if flags != -1 {
|
||
_ = fcntl(fd, F_SETFL, flags | O_NONBLOCK)
|
||
}
|
||
|
||
var addr = sockaddr_un()
|
||
addr.sun_family = sa_family_t(AF_UNIX)
|
||
let capacity = MemoryLayout.size(ofValue: addr.sun_path)
|
||
let copied = path.withCString { cstr -> Int in
|
||
strlcpy(&addr.sun_path.0, cstr, capacity)
|
||
}
|
||
guard copied < capacity else { throw POSIXError(.ENAMETOOLONG) }
|
||
addr.sun_len = UInt8(MemoryLayout.size(ofValue: addr))
|
||
let len = socklen_t(MemoryLayout.size(ofValue: addr))
|
||
let result = withUnsafePointer(to: &addr) { ptr -> Int32 in
|
||
ptr.withMemoryRebound(to: sockaddr.self, capacity: 1) { sockPtr in
|
||
connect(fd, sockPtr, len)
|
||
}
|
||
}
|
||
if result != 0 {
|
||
let err = errno
|
||
if err == EINPROGRESS {
|
||
try self.waitForSocket(
|
||
fd: fd,
|
||
events: Int16(POLLOUT),
|
||
until: deadline,
|
||
timeoutSeconds: timeoutSeconds)
|
||
var soError: Int32 = 0
|
||
var soLen = socklen_t(MemoryLayout.size(ofValue: soError))
|
||
_ = getsockopt(fd, SOL_SOCKET, SO_ERROR, &soError, &soLen)
|
||
if soError != 0 { throw POSIXError(POSIXErrorCode(rawValue: soError) ?? .ECONNREFUSED) }
|
||
} else {
|
||
throw POSIXError(POSIXErrorCode(rawValue: err) ?? .ECONNREFUSED)
|
||
}
|
||
}
|
||
|
||
let payload = try JSONEncoder().encode(request)
|
||
try payload.withUnsafeBytes { buf in
|
||
guard let base = buf.baseAddress else { return }
|
||
var written = 0
|
||
while written < payload.count {
|
||
try self.ensureDeadline(deadline, timeoutSeconds: timeoutSeconds)
|
||
let n = write(fd, base.advanced(by: written), payload.count - written)
|
||
if n > 0 {
|
||
written += n
|
||
continue
|
||
}
|
||
if n == -1, errno == EINTR { continue }
|
||
if n == -1, errno == EAGAIN {
|
||
try self.waitForSocket(
|
||
fd: fd,
|
||
events: Int16(POLLOUT),
|
||
until: deadline,
|
||
timeoutSeconds: timeoutSeconds)
|
||
continue
|
||
}
|
||
throw POSIXError(POSIXErrorCode(rawValue: errno) ?? .EIO)
|
||
}
|
||
}
|
||
shutdown(fd, SHUT_WR)
|
||
|
||
var data = Data()
|
||
let decoder = JSONDecoder()
|
||
var buffer = [UInt8](repeating: 0, count: 8192)
|
||
let bufSize = buffer.count
|
||
while true {
|
||
try self.ensureDeadline(deadline, timeoutSeconds: timeoutSeconds)
|
||
try self.waitForSocket(
|
||
fd: fd,
|
||
events: Int16(POLLIN),
|
||
until: deadline,
|
||
timeoutSeconds: timeoutSeconds)
|
||
let n = buffer.withUnsafeMutableBytes { read(fd, $0.baseAddress!, bufSize) }
|
||
if n > 0 {
|
||
data.append(buffer, count: n)
|
||
if let resp = try? decoder.decode(Response.self, from: data) {
|
||
return resp
|
||
}
|
||
continue
|
||
}
|
||
if n == 0 { break }
|
||
if n == -1, errno == EINTR { continue }
|
||
if n == -1, errno == EAGAIN { continue }
|
||
throw POSIXError(POSIXErrorCode(rawValue: errno) ?? .EIO)
|
||
}
|
||
guard !data.isEmpty else { throw POSIXError(.ECONNRESET) }
|
||
return try decoder.decode(Response.self, from: data)
|
||
}
|
||
|
||
private static func rpcTimeoutSeconds(for request: Request) -> TimeInterval {
|
||
switch request {
|
||
case let .runShell(_, _, _, timeoutSec, _):
|
||
// Allow longer for commands; still cap overall to a sane bound.
|
||
min(300, max(10, (timeoutSec ?? 10) + 2))
|
||
default:
|
||
// Fail-fast so callers (incl. SSH tool calls) don't hang forever.
|
||
10
|
||
}
|
||
}
|
||
|
||
private static func ensureDeadline(_ deadline: Date, timeoutSeconds: TimeInterval) throws {
|
||
if Date() >= deadline {
|
||
throw CLITimeoutError(seconds: timeoutSeconds)
|
||
}
|
||
}
|
||
|
||
private static func waitForSocket(
|
||
fd: Int32,
|
||
events: Int16,
|
||
until deadline: Date,
|
||
timeoutSeconds: TimeInterval) throws
|
||
{
|
||
while true {
|
||
let remaining = deadline.timeIntervalSinceNow
|
||
if remaining <= 0 { throw CLITimeoutError(seconds: timeoutSeconds) }
|
||
var pfd = pollfd(fd: fd, events: events, revents: 0)
|
||
let ms = Int32(max(1, min(remaining, 0.5) * 1000)) // small slices so we enforce total timeout
|
||
let n = poll(&pfd, 1, ms)
|
||
if n > 0 { return }
|
||
if n == 0 { continue }
|
||
if errno == EINTR { continue }
|
||
throw POSIXError(POSIXErrorCode(rawValue: errno) ?? .EIO)
|
||
}
|
||
}
|
||
|
||
private static func ensureAppRunning() async throws {
|
||
let appURL = URL(fileURLWithPath: CommandLine.arguments.first ?? "")
|
||
.resolvingSymlinksInPath()
|
||
.deletingLastPathComponent() // MacOS
|
||
.deletingLastPathComponent() // Contents
|
||
let proc = Process()
|
||
proc.launchPath = "/usr/bin/open"
|
||
proc.arguments = ["-n", appURL.path]
|
||
proc.standardOutput = Pipe()
|
||
proc.standardError = Pipe()
|
||
try proc.run()
|
||
try? await Task.sleep(nanoseconds: 100_000_000)
|
||
}
|
||
}
|
||
|
||
enum CLIError: Error { case help, version }
|
||
|
||
struct CLITimeoutError: Error, CustomStringConvertible {
|
||
let seconds: TimeInterval
|
||
var description: String {
|
||
let rounded = Int(max(1, seconds.rounded(.toNearestOrEven)))
|
||
return "timed out after \(rounded)s"
|
||
}
|
||
}
|
||
|
||
extension [String] {
|
||
mutating func popFirst() -> String? {
|
||
guard let first else { return nil }
|
||
self = Array(self.dropFirst())
|
||
return first
|
||
}
|
||
}
|