131 lines
4.3 KiB
Swift
131 lines
4.3 KiB
Swift
import Foundation
|
|
import os
|
|
import PeekabooAutomationKit
|
|
import PeekabooBridge
|
|
import PeekabooFoundation
|
|
import Security
|
|
|
|
@MainActor
|
|
final class PeekabooBridgeHostCoordinator {
|
|
static let shared = PeekabooBridgeHostCoordinator()
|
|
|
|
private let logger = Logger(subsystem: "com.clawdbot", category: "PeekabooBridge")
|
|
|
|
private var host: PeekabooBridgeHost?
|
|
private var services: ClawdbotPeekabooBridgeServices?
|
|
|
|
func setEnabled(_ enabled: Bool) async {
|
|
if enabled {
|
|
await self.startIfNeeded()
|
|
} else {
|
|
await self.stop()
|
|
}
|
|
}
|
|
|
|
func stop() async {
|
|
guard let host else { return }
|
|
await host.stop()
|
|
self.host = nil
|
|
self.services = nil
|
|
self.logger.info("PeekabooBridge host stopped")
|
|
}
|
|
|
|
private func startIfNeeded() async {
|
|
guard self.host == nil else { return }
|
|
|
|
var allowlistedTeamIDs: Set<String> = ["Y5PE65HELJ"]
|
|
if let teamID = Self.currentTeamID() {
|
|
allowlistedTeamIDs.insert(teamID)
|
|
}
|
|
let allowlistedBundles: Set<String> = []
|
|
|
|
let services = ClawdbotPeekabooBridgeServices()
|
|
let server = PeekabooBridgeServer(
|
|
services: services,
|
|
hostKind: .gui,
|
|
allowlistedTeams: allowlistedTeamIDs,
|
|
allowlistedBundles: allowlistedBundles)
|
|
|
|
let host = PeekabooBridgeHost(
|
|
socketPath: PeekabooBridgeConstants.clawdbotSocketPath,
|
|
server: server,
|
|
allowedTeamIDs: allowlistedTeamIDs,
|
|
requestTimeoutSec: 10)
|
|
|
|
self.services = services
|
|
self.host = host
|
|
|
|
await host.start()
|
|
self.logger
|
|
.info("PeekabooBridge host started at \(PeekabooBridgeConstants.clawdbotSocketPath, privacy: .public)")
|
|
}
|
|
|
|
private static func currentTeamID() -> String? {
|
|
var code: SecCode?
|
|
guard SecCodeCopySelf(SecCSFlags(), &code) == errSecSuccess,
|
|
let code
|
|
else {
|
|
return nil
|
|
}
|
|
|
|
var staticCode: SecStaticCode?
|
|
guard SecCodeCopyStaticCode(code, SecCSFlags(), &staticCode) == errSecSuccess,
|
|
let staticCode
|
|
else {
|
|
return nil
|
|
}
|
|
|
|
var infoCF: CFDictionary?
|
|
guard SecCodeCopySigningInformation(
|
|
staticCode,
|
|
SecCSFlags(rawValue: kSecCSSigningInformation),
|
|
&infoCF) == errSecSuccess,
|
|
let info = infoCF as? [String: Any]
|
|
else {
|
|
return nil
|
|
}
|
|
|
|
return info[kSecCodeInfoTeamIdentifier as String] as? String
|
|
}
|
|
}
|
|
|
|
@MainActor
|
|
private final class ClawdbotPeekabooBridgeServices: PeekabooBridgeServiceProviding {
|
|
let permissions: PermissionsService
|
|
let screenCapture: any ScreenCaptureServiceProtocol
|
|
let automation: any UIAutomationServiceProtocol
|
|
let windows: any WindowManagementServiceProtocol
|
|
let applications: any ApplicationServiceProtocol
|
|
let menu: any MenuServiceProtocol
|
|
let dock: any DockServiceProtocol
|
|
let dialogs: any DialogServiceProtocol
|
|
let snapshots: any SnapshotManagerProtocol
|
|
|
|
init() {
|
|
let logging = LoggingService(subsystem: "com.clawdbot.peekaboo")
|
|
let feedbackClient: any AutomationFeedbackClient = NoopAutomationFeedbackClient()
|
|
|
|
let snapshots = InMemorySnapshotManager(options: .init(
|
|
snapshotValidityWindow: 600,
|
|
maxSnapshots: 50,
|
|
deleteArtifactsOnCleanup: false))
|
|
let applications = ApplicationService(feedbackClient: feedbackClient)
|
|
|
|
let screenCapture = ScreenCaptureService(loggingService: logging)
|
|
|
|
self.permissions = PermissionsService()
|
|
self.snapshots = snapshots
|
|
self.applications = applications
|
|
self.screenCapture = screenCapture
|
|
self.automation = UIAutomationService(
|
|
snapshotManager: snapshots,
|
|
loggingService: logging,
|
|
searchPolicy: .balanced,
|
|
feedbackClient: feedbackClient)
|
|
self.windows = WindowManagementService(applicationService: applications, feedbackClient: feedbackClient)
|
|
self.menu = MenuService(applicationService: applications, feedbackClient: feedbackClient)
|
|
self.dock = DockService(feedbackClient: feedbackClient)
|
|
self.dialogs = DialogService(feedbackClient: feedbackClient)
|
|
}
|
|
}
|