macOS: fold AgentRPC into GatewayConnection
This commit is contained in:
@@ -1,74 +0,0 @@
|
|||||||
import Foundation
|
|
||||||
import OSLog
|
|
||||||
|
|
||||||
struct ControlRequestParams: @unchecked Sendable {
|
|
||||||
/// Heterogeneous JSON-ish params (Bool/String/Int/Double/[...]/[String:...]).
|
|
||||||
/// `@unchecked Sendable` is intentional: values are treated as immutable payloads.
|
|
||||||
let raw: [String: Any]
|
|
||||||
}
|
|
||||||
|
|
||||||
actor AgentRPC {
|
|
||||||
static let shared = AgentRPC()
|
|
||||||
|
|
||||||
private let logger = Logger(subsystem: "com.steipete.clawdis", category: "agent.rpc")
|
|
||||||
|
|
||||||
func shutdown() async {
|
|
||||||
// no-op; socket managed by GatewayConnection
|
|
||||||
}
|
|
||||||
|
|
||||||
func setHeartbeatsEnabled(_ enabled: Bool) async -> Bool {
|
|
||||||
do {
|
|
||||||
_ = try await self.controlRequest(
|
|
||||||
method: "set-heartbeats",
|
|
||||||
params: ControlRequestParams(raw: ["enabled": enabled]))
|
|
||||||
return true
|
|
||||||
} catch {
|
|
||||||
self.logger.error("setHeartbeatsEnabled failed \(error.localizedDescription, privacy: .public)")
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func status() async -> (ok: Bool, error: String?) {
|
|
||||||
do {
|
|
||||||
let data = try await controlRequest(method: "status")
|
|
||||||
if let obj = try? JSONSerialization.jsonObject(with: data) as? [String: Any],
|
|
||||||
(obj["ok"] as? Bool) ?? true
|
|
||||||
{
|
|
||||||
return (true, nil)
|
|
||||||
}
|
|
||||||
return (false, "status error")
|
|
||||||
} catch {
|
|
||||||
return (false, error.localizedDescription)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func send(
|
|
||||||
text: String,
|
|
||||||
thinking: String?,
|
|
||||||
sessionKey: String,
|
|
||||||
deliver: Bool,
|
|
||||||
to: String?,
|
|
||||||
channel: String? = nil) async -> (ok: Bool, text: String?, error: String?)
|
|
||||||
{
|
|
||||||
do {
|
|
||||||
let params: [String: Any] = [
|
|
||||||
"message": text,
|
|
||||||
"sessionKey": sessionKey,
|
|
||||||
"thinking": thinking ?? "default",
|
|
||||||
"deliver": deliver,
|
|
||||||
"to": to ?? "",
|
|
||||||
"channel": channel ?? "",
|
|
||||||
"idempotencyKey": UUID().uuidString,
|
|
||||||
]
|
|
||||||
_ = try await self.controlRequest(method: "agent", params: ControlRequestParams(raw: params))
|
|
||||||
return (true, nil, nil)
|
|
||||||
} catch {
|
|
||||||
return (false, nil, error.localizedDescription)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func controlRequest(method: String, params: ControlRequestParams? = nil) async throws -> Data {
|
|
||||||
let rawParams = params?.raw.reduce(into: [String: AnyCodable]()) { $0[$1.key] = AnyCodable($1.value) }
|
|
||||||
return try await GatewayConnection.shared.request(method: method, params: rawParams)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -131,7 +131,7 @@ final class AppState {
|
|||||||
didSet {
|
didSet {
|
||||||
self.ifNotPreview {
|
self.ifNotPreview {
|
||||||
UserDefaults.standard.set(self.heartbeatsEnabled, forKey: heartbeatsEnabledKey)
|
UserDefaults.standard.set(self.heartbeatsEnabled, forKey: heartbeatsEnabledKey)
|
||||||
Task { _ = await AgentRPC.shared.setHeartbeatsEnabled(self.heartbeatsEnabled) }
|
Task { _ = await GatewayConnection.shared.setHeartbeatsEnabled(self.heartbeatsEnabled) }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -167,8 +167,8 @@ actor BridgeServer {
|
|||||||
let sessionKey = payload.sessionKey?.trimmingCharacters(in: .whitespacesAndNewlines).nonEmpty
|
let sessionKey = payload.sessionKey?.trimmingCharacters(in: .whitespacesAndNewlines).nonEmpty
|
||||||
?? "node-\(nodeId)"
|
?? "node-\(nodeId)"
|
||||||
|
|
||||||
_ = await AgentRPC.shared.send(
|
_ = await GatewayConnection.shared.sendAgent(
|
||||||
text: text,
|
message: text,
|
||||||
thinking: "low",
|
thinking: "low",
|
||||||
sessionKey: sessionKey,
|
sessionKey: sessionKey,
|
||||||
deliver: false,
|
deliver: false,
|
||||||
@@ -193,8 +193,8 @@ actor BridgeServer {
|
|||||||
let to = link.to?.trimmingCharacters(in: .whitespacesAndNewlines).nonEmpty
|
let to = link.to?.trimmingCharacters(in: .whitespacesAndNewlines).nonEmpty
|
||||||
let channel = link.channel?.trimmingCharacters(in: .whitespacesAndNewlines).nonEmpty
|
let channel = link.channel?.trimmingCharacters(in: .whitespacesAndNewlines).nonEmpty
|
||||||
|
|
||||||
_ = await AgentRPC.shared.send(
|
_ = await GatewayConnection.shared.sendAgent(
|
||||||
text: message,
|
message: message,
|
||||||
thinking: thinking,
|
thinking: thinking,
|
||||||
sessionKey: sessionKey,
|
sessionKey: sessionKey,
|
||||||
deliver: link.deliver,
|
deliver: link.deliver,
|
||||||
@@ -357,9 +357,7 @@ actor BridgeServer {
|
|||||||
]
|
]
|
||||||
if let ip { params["ip"] = ip }
|
if let ip { params["ip"] = ip }
|
||||||
if let version { params["version"] = version }
|
if let version { params["version"] = version }
|
||||||
_ = try await AgentRPC.shared.controlRequest(
|
_ = try await GatewayConnection.shared.controlRequest(method: "system-event", params: params)
|
||||||
method: "system-event",
|
|
||||||
params: ControlRequestParams(raw: params))
|
|
||||||
} catch {
|
} catch {
|
||||||
// Best-effort only.
|
// Best-effort only.
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -550,8 +550,8 @@ private final class CanvasA2UIActionMessageHandler: NSObject, WKScriptMessageHan
|
|||||||
: "A2UI action: \(name)\n\n```json\n\(json)\n```"
|
: "A2UI action: \(name)\n\n```json\n\(json)\n```"
|
||||||
|
|
||||||
Task {
|
Task {
|
||||||
let result = await AgentRPC.shared.send(
|
let result = await GatewayConnection.shared.sendAgent(
|
||||||
text: text,
|
message: text,
|
||||||
thinking: nil,
|
thinking: nil,
|
||||||
sessionKey: self.sessionKey,
|
sessionKey: self.sessionKey,
|
||||||
deliver: false,
|
deliver: false,
|
||||||
|
|||||||
@@ -140,7 +140,7 @@ enum ControlRequestHandler {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private static func handleRPCStatus() async -> Response {
|
private static func handleRPCStatus() async -> Response {
|
||||||
let result = await AgentRPC.shared.status()
|
let result = await GatewayConnection.shared.status()
|
||||||
return Response(ok: result.ok, message: result.error)
|
return Response(ok: result.ok, message: result.error)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -169,15 +169,15 @@ enum ControlRequestHandler {
|
|||||||
let trimmed = message.trimmingCharacters(in: .whitespacesAndNewlines)
|
let trimmed = message.trimmingCharacters(in: .whitespacesAndNewlines)
|
||||||
guard !trimmed.isEmpty else { return Response(ok: false, message: "message empty") }
|
guard !trimmed.isEmpty else { return Response(ok: false, message: "message empty") }
|
||||||
let sessionKey = session ?? "main"
|
let sessionKey = session ?? "main"
|
||||||
let rpcResult = await AgentRPC.shared.send(
|
let rpcResult = await GatewayConnection.shared.sendAgent(
|
||||||
text: trimmed,
|
message: trimmed,
|
||||||
thinking: thinking,
|
thinking: thinking,
|
||||||
sessionKey: sessionKey,
|
sessionKey: sessionKey,
|
||||||
deliver: deliver,
|
deliver: deliver,
|
||||||
to: to,
|
to: to,
|
||||||
channel: nil)
|
channel: nil)
|
||||||
return rpcResult.ok
|
return rpcResult.ok
|
||||||
? Response(ok: true, message: rpcResult.text ?? "sent")
|
? Response(ok: true, message: "sent")
|
||||||
: Response(ok: false, message: rpcResult.error ?? "failed to send")
|
: Response(ok: false, message: rpcResult.error ?? "failed to send")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -115,7 +115,7 @@ enum DebugActions {
|
|||||||
|
|
||||||
static func sendTestHeartbeat() async -> Result<ControlHeartbeatEvent?, Error> {
|
static func sendTestHeartbeat() async -> Result<ControlHeartbeatEvent?, Error> {
|
||||||
do {
|
do {
|
||||||
_ = await AgentRPC.shared.setHeartbeatsEnabled(true)
|
_ = await GatewayConnection.shared.setHeartbeatsEnabled(true)
|
||||||
await ControlChannel.shared.configure()
|
await ControlChannel.shared.configure()
|
||||||
let data = try await ControlChannel.shared.request(method: "last-heartbeat")
|
let data = try await ControlChannel.shared.request(method: "last-heartbeat")
|
||||||
if let evt = try? JSONDecoder().decode(ControlHeartbeatEvent.self, from: data) {
|
if let evt = try? JSONDecoder().decode(ControlHeartbeatEvent.self, from: data) {
|
||||||
|
|||||||
@@ -1,10 +1,11 @@
|
|||||||
import ClawdisProtocol
|
import ClawdisProtocol
|
||||||
import Foundation
|
import Foundation
|
||||||
|
import OSLog
|
||||||
|
|
||||||
/// Single, shared Gateway websocket connection for the whole app.
|
/// Single, shared Gateway websocket connection for the whole app.
|
||||||
///
|
///
|
||||||
/// This owns exactly one `GatewayChannelActor` and reuses it across all callers
|
/// This owns exactly one `GatewayChannelActor` and reuses it across all callers
|
||||||
/// (ControlChannel, AgentRPC, SwiftUI WebChat, etc.).
|
/// (ControlChannel, debug actions, SwiftUI WebChat, etc.).
|
||||||
actor GatewayConnection {
|
actor GatewayConnection {
|
||||||
static let shared = GatewayConnection()
|
static let shared = GatewayConnection()
|
||||||
|
|
||||||
@@ -112,3 +113,72 @@ actor GatewayConnection {
|
|||||||
try await GatewayEndpointStore.shared.requireConfig()
|
try await GatewayEndpointStore.shared.requireConfig()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private let gatewayControlLogger = Logger(subsystem: "com.steipete.clawdis", category: "gateway.control")
|
||||||
|
|
||||||
|
extension GatewayConnection {
|
||||||
|
private static func wrapParams(_ raw: [String: Any]?) -> [String: AnyCodable]? {
|
||||||
|
guard let raw else { return nil }
|
||||||
|
return raw.reduce(into: [String: AnyCodable]()) { acc, pair in
|
||||||
|
acc[pair.key] = AnyCodable(pair.value)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func controlRequest(
|
||||||
|
method: String,
|
||||||
|
params: [String: Any]? = nil,
|
||||||
|
timeoutMs: Double? = nil) async throws -> Data
|
||||||
|
{
|
||||||
|
try await self.request(method: method, params: Self.wrapParams(params), timeoutMs: timeoutMs)
|
||||||
|
}
|
||||||
|
|
||||||
|
func status() async -> (ok: Bool, error: String?) {
|
||||||
|
do {
|
||||||
|
let data = try await self.controlRequest(method: "status")
|
||||||
|
if let obj = try? JSONSerialization.jsonObject(with: data) as? [String: Any],
|
||||||
|
(obj["ok"] as? Bool) ?? true
|
||||||
|
{
|
||||||
|
return (true, nil)
|
||||||
|
}
|
||||||
|
return (false, "status error")
|
||||||
|
} catch {
|
||||||
|
return (false, error.localizedDescription)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func setHeartbeatsEnabled(_ enabled: Bool) async -> Bool {
|
||||||
|
do {
|
||||||
|
_ = try await self.controlRequest(method: "set-heartbeats", params: ["enabled": enabled])
|
||||||
|
return true
|
||||||
|
} catch {
|
||||||
|
gatewayControlLogger.error("setHeartbeatsEnabled failed \(error.localizedDescription, privacy: .public)")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func sendAgent(
|
||||||
|
message: String,
|
||||||
|
thinking: String?,
|
||||||
|
sessionKey: String,
|
||||||
|
deliver: Bool,
|
||||||
|
to: String?,
|
||||||
|
channel: String? = nil,
|
||||||
|
idempotencyKey: String = UUID().uuidString) async -> (ok: Bool, error: String?)
|
||||||
|
{
|
||||||
|
do {
|
||||||
|
let params: [String: Any] = [
|
||||||
|
"message": message,
|
||||||
|
"sessionKey": sessionKey,
|
||||||
|
"thinking": thinking ?? "default",
|
||||||
|
"deliver": deliver,
|
||||||
|
"to": to ?? "",
|
||||||
|
"channel": channel ?? "",
|
||||||
|
"idempotencyKey": idempotencyKey,
|
||||||
|
]
|
||||||
|
_ = try await self.controlRequest(method: "agent", params: params)
|
||||||
|
return (true, nil)
|
||||||
|
} catch {
|
||||||
|
return (false, error.localizedDescription)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -206,7 +206,6 @@ final class AppDelegate: NSObject, NSApplicationDelegate {
|
|||||||
WebChatManager.shared.close()
|
WebChatManager.shared.close()
|
||||||
WebChatManager.shared.resetTunnels()
|
WebChatManager.shared.resetTunnels()
|
||||||
Task { await RemoteTunnelManager.shared.stopAll() }
|
Task { await RemoteTunnelManager.shared.stopAll() }
|
||||||
Task { await AgentRPC.shared.shutdown() }
|
|
||||||
Task { await GatewayConnection.shared.shutdown() }
|
Task { await GatewayConnection.shared.shutdown() }
|
||||||
Task { await self.socketServer.stop() }
|
Task { await self.socketServer.stop() }
|
||||||
Task { await PeekabooBridgeHostCoordinator.shared.stop() }
|
Task { await PeekabooBridgeHostCoordinator.shared.stop() }
|
||||||
|
|||||||
@@ -48,8 +48,8 @@ enum VoiceWakeForwarder {
|
|||||||
let payload = Self.prefixedTranscript(transcript)
|
let payload = Self.prefixedTranscript(transcript)
|
||||||
let channel = options.channel.trimmingCharacters(in: .whitespacesAndNewlines).lowercased()
|
let channel = options.channel.trimmingCharacters(in: .whitespacesAndNewlines).lowercased()
|
||||||
let deliver = options.deliver && channel != "webchat"
|
let deliver = options.deliver && channel != "webchat"
|
||||||
let result = await AgentRPC.shared.send(
|
let result = await GatewayConnection.shared.sendAgent(
|
||||||
text: payload,
|
message: payload,
|
||||||
thinking: options.thinking,
|
thinking: options.thinking,
|
||||||
sessionKey: options.sessionKey,
|
sessionKey: options.sessionKey,
|
||||||
deliver: deliver,
|
deliver: deliver,
|
||||||
@@ -67,7 +67,7 @@ enum VoiceWakeForwarder {
|
|||||||
}
|
}
|
||||||
|
|
||||||
static func checkConnection() async -> Result<Void, VoiceWakeForwardError> {
|
static func checkConnection() async -> Result<Void, VoiceWakeForwardError> {
|
||||||
let status = await AgentRPC.shared.status()
|
let status = await GatewayConnection.shared.status()
|
||||||
if status.ok { return .success(()) }
|
if status.ok { return .success(()) }
|
||||||
return .failure(.rpcFailed(status.error ?? "agent rpc unreachable"))
|
return .failure(.rpcFailed(status.error ?? "agent rpc unreachable"))
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user