capacitor-native-agent 0.1.1 → 0.1.3

package/ios/Sources/NativeAgentPlugin/Generated/native_agent_ffiFFI.modulemap

@@ -0,0 +1,4 @@
+module native_agent_ffiFFI {
+    header "native_agent_ffiFFI.h"
+    export *
+}

package/ios/Sources/NativeAgentPlugin/NativeAgentPlugin.swift

@@ -0,0 +1,648 @@
+import Foundation
+import Capacitor
+
+@objc(NativeAgentPlugin)
+public class NativeAgentPlugin: CAPPlugin, CAPBridgedPlugin {
+    public let identifier = "NativeAgentPlugin"
+    public let jsName = "NativeAgent"
+    public let pluginMethods: [CAPPluginMethod] = [
+        // Lifecycle
+        CAPPluginMethod(name: "initialize", returnType: CAPPluginReturnPromise),
+        // Agent
+        CAPPluginMethod(name: "sendMessage", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "followUp", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "abort", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "steer", returnType: CAPPluginReturnPromise),
+        // Approval gate
+        CAPPluginMethod(name: "respondToApproval", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "respondToCronApproval", returnType: CAPPluginReturnPromise),
+        // Auth
+        CAPPluginMethod(name: "getAuthToken", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "setAuthKey", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "deleteAuth", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "refreshToken", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "getAuthStatus", returnType: CAPPluginReturnPromise),
+        // Sessions
+        CAPPluginMethod(name: "listSessions", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "loadSession", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "resumeSession", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "clearSession", returnType: CAPPluginReturnPromise),
+        // Cron / heartbeat
+        CAPPluginMethod(name: "addCronJob", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "updateCronJob", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "removeCronJob", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "listCronJobs", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "runCronJob", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "listCronRuns", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "handleWake", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "getSchedulerConfig", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "setSchedulerConfig", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "setHeartbeatConfig", returnType: CAPPluginReturnPromise),
+        // Skills
+        CAPPluginMethod(name: "addSkill", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "updateSkill", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "removeSkill", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "listSkills", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "startSkill", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "endSkill", returnType: CAPPluginReturnPromise),
+        // MCP
+        CAPPluginMethod(name: "startMcp", returnType: CAPPluginReturnPromise),
+        CAPPluginMethod(name: "restartMcp", returnType: CAPPluginReturnPromise),
+        // Models
+        CAPPluginMethod(name: "getModels", returnType: CAPPluginReturnPromise),
+        // Tools
+        CAPPluginMethod(name: "invokeTool", returnType: CAPPluginReturnPromise),
+    ]
+
+    private var handle: NativeAgentHandle?
+
+    // ── Helper ──────────────────────────────────────────────────────────────
+
+    /// Resolves a `files://` prefixed path to an absolute iOS path
+    /// under the app's Documents directory (always writable).
+    private func resolvePath(_ path: String) -> String {
+        if path.hasPrefix("files://") {
+            let relative = String(path.dropFirst("files://".count))
+            let base = FileManager.default
+                .urls(for: .documentDirectory, in: .userDomainMask)
+                .first!
+                .path
+            let resolved = (base as NSString).appendingPathComponent(relative)
+            // Ensure parent directory exists
+            let parentDir = (resolved as NSString).deletingLastPathComponent
+            try? FileManager.default.createDirectory(
+                atPath: parentDir,
+                withIntermediateDirectories: true,
+                attributes: nil
+            )
+            return resolved
+        }
+        return path
+    }
+
+    private func withHandle(_ call: CAPPluginCall, _ block: @escaping (NativeAgentHandle) -> Void) {
+        guard let h = handle else {
+            return call.reject("NativeAgent not initialized — call initialize() first")
+        }
+        Task {
+            block(h)
+        }
+    }
+
+    // ── Lifecycle ────────────────────────────────────────────────────────────
+
+    @objc func initialize(_ call: CAPPluginCall) {
+        guard let dbPath = call.getString("dbPath") else {
+            return call.reject("dbPath is required")
+        }
+        guard let workspacePath = call.getString("workspacePath") else {
+            return call.reject("workspacePath is required")
+        }
+        guard let authProfilesPath = call.getString("authProfilesPath") else {
+            return call.reject("authProfilesPath is required")
+        }
+
+        Task {
+            do {
+                let config = InitConfig(
+                    dbPath: resolvePath(dbPath),
+                    workspacePath: resolvePath(workspacePath),
+                    authProfilesPath: resolvePath(authProfilesPath)
+                )
+                let h = try NativeAgentHandle(config: config)
+                h.setEventCallback(callback: NativeAgentEventBridge(plugin: self))
+                self.handle = h
+                call.resolve()
+            } catch {
+                call.reject("Failed to initialize NativeAgent: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    // ── Agent ────────────────────────────────────────────────────────────────
+
+    @objc func sendMessage(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let prompt = call.getString("prompt") else {
+                return call.reject("prompt is required")
+            }
+            guard let sessionKey = call.getString("sessionKey") else {
+                return call.reject("sessionKey is required")
+            }
+            guard let systemPrompt = call.getString("systemPrompt") else {
+                return call.reject("systemPrompt is required")
+            }
+
+            do {
+                let params = SendMessageParams(
+                    prompt: prompt,
+                    sessionKey: sessionKey,
+                    model: call.getString("model"),
+                    provider: call.getString("provider"),
+                    systemPrompt: systemPrompt,
+                    maxTurns: call.getInt("maxTurns").map { UInt32($0) },
+                    allowedToolsJson: call.getString("allowedToolsJson")
+                )
+                let runId = try h.sendMessage(params: params)
+                call.resolve(["runId": runId])
+            } catch {
+                call.reject("sendMessage failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func followUp(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                try h.followUp(prompt: call.getString("prompt") ?? "")
+                call.resolve()
+            } catch {
+                call.reject("followUp failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func abort(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                try h.abort()
+                call.resolve()
+            } catch {
+                call.reject("abort failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func steer(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                try h.steer(text: call.getString("text") ?? "")
+                call.resolve()
+            } catch {
+                call.reject("steer failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    // ── Approval gate ────────────────────────────────────────────────────────
+
+    @objc func respondToApproval(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let toolCallId = call.getString("toolCallId") else {
+                return call.reject("toolCallId is required")
+            }
+            do {
+                try h.respondToApproval(
+                    toolCallId: toolCallId,
+                    approved: call.getBool("approved") ?? true,
+                    reason: call.getString("reason")
+                )
+                call.resolve()
+            } catch {
+                call.reject("respondToApproval failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func respondToCronApproval(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let requestId = call.getString("requestId") else {
+                return call.reject("requestId is required")
+            }
+            do {
+                try h.respondToCronApproval(
+                    requestId: requestId,
+                    approved: call.getBool("approved") ?? false
+                )
+                call.resolve()
+            } catch {
+                call.reject("respondToCronApproval failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    // ── Auth ─────────────────────────────────────────────────────────────────
+
+    @objc func getAuthToken(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                let result = try h.getAuthToken(provider: call.getString("provider") ?? "anthropic")
+                call.resolve([
+                    "apiKey": result.apiKey as Any,
+                    "isOAuth": result.isOauth,
+                ])
+            } catch {
+                call.reject("getAuthToken failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func setAuthKey(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let key = call.getString("key") else {
+                return call.reject("key is required")
+            }
+            do {
+                try h.setAuthKey(
+                    key: key,
+                    provider: call.getString("provider") ?? "anthropic",
+                    authType: call.getString("authType") ?? "api_key"
+                )
+                call.resolve()
+            } catch {
+                call.reject("setAuthKey failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func deleteAuth(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                try h.deleteAuth(provider: call.getString("provider") ?? "anthropic")
+                call.resolve()
+            } catch {
+                call.reject("deleteAuth failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func refreshToken(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                let result = try h.refreshToken(provider: call.getString("provider") ?? "anthropic")
+                call.resolve([
+                    "apiKey": result.apiKey as Any,
+                    "isOAuth": result.isOauth,
+                ])
+            } catch {
+                call.reject("refreshToken failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func getAuthStatus(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                let result = try h.getAuthStatus(provider: call.getString("provider") ?? "anthropic")
+                call.resolve([
+                    "hasKey": result.hasKey,
+                    "masked": result.masked,
+                    "provider": result.provider,
+                ])
+            } catch {
+                call.reject("getAuthStatus failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    // ── Sessions ─────────────────────────────────────────────────────────────
+
+    @objc func listSessions(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                let json = try h.listSessions(agentId: call.getString("agentId") ?? "main")
+                call.resolve(["sessionsJson": json])
+            } catch {
+                call.reject("listSessions failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func loadSession(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let sessKey = call.getString("sessionKey") else {
+                return call.reject("sessionKey is required")
+            }
+            do {
+                let json = try h.loadSession(sessionKey: sessKey)
+                call.resolve([
+                    "sessionKey": sessKey,
+                    "messagesJson": json,
+                ])
+            } catch {
+                call.reject("loadSession failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func resumeSession(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let sessKey = call.getString("sessionKey") else {
+                return call.reject("sessionKey is required")
+            }
+            do {
+                try h.resumeSession(
+                    sessionKey: sessKey,
+                    agentId: call.getString("agentId") ?? "main",
+                    messagesJson: call.getString("messagesJson"),
+                    provider: call.getString("provider"),
+                    model: call.getString("model")
+                )
+                call.resolve()
+            } catch {
+                call.reject("resumeSession failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func clearSession(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                try h.clearSession()
+                call.resolve()
+            } catch {
+                call.reject("clearSession failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    // ── Cron / heartbeat ─────────────────────────────────────────────────────
+
+    @objc func addCronJob(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let inputJson = call.getString("inputJson") else {
+                return call.reject("inputJson is required")
+            }
+            do {
+                let json = try h.addCronJob(inputJson: inputJson)
+                call.resolve(["recordJson": json])
+            } catch {
+                call.reject("addCronJob failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func updateCronJob(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let id = call.getString("id") else {
+                return call.reject("id is required")
+            }
+            do {
+                try h.updateCronJob(id: id, patchJson: call.getString("patchJson") ?? "{}")
+                call.resolve()
+            } catch {
+                call.reject("updateCronJob failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func removeCronJob(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let id = call.getString("id") else {
+                return call.reject("id is required")
+            }
+            do {
+                try h.removeCronJob(id: id)
+                call.resolve()
+            } catch {
+                call.reject("removeCronJob failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func listCronJobs(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                call.resolve(["jobsJson": try h.listCronJobs()])
+            } catch {
+                call.reject("listCronJobs failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func runCronJob(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let jobId = call.getString("jobId") else {
+                return call.reject("jobId is required")
+            }
+            do {
+                try h.runCronJob(jobId: jobId)
+                call.resolve()
+            } catch {
+                call.reject("runCronJob failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func listCronRuns(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                let json = try h.listCronRuns(
+                    jobId: call.getString("jobId"),
+                    limit: Int64(call.getInt("limit") ?? 100)
+                )
+                call.resolve(["runsJson": json])
+            } catch {
+                call.reject("listCronRuns failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func handleWake(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                try h.handleWake(source: call.getString("source") ?? "unknown")
+                call.resolve()
+            } catch {
+                call.reject("handleWake failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func getSchedulerConfig(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                let schedulerJson = try h.getSchedulerConfig()
+                let heartbeatJson = try h.getHeartbeatConfig()
+                call.resolve([
+                    "schedulerJson": schedulerJson,
+                    "heartbeatJson": heartbeatJson,
+                ])
+            } catch {
+                call.reject("getSchedulerConfig failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func setSchedulerConfig(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                try h.setSchedulerConfig(configJson: call.getString("configJson") ?? "{}")
+                call.resolve()
+            } catch {
+                call.reject("setSchedulerConfig failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func setHeartbeatConfig(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                try h.setHeartbeatConfig(configJson: call.getString("configJson") ?? "{}")
+                call.resolve()
+            } catch {
+                call.reject("setHeartbeatConfig failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    // ── Skills ───────────────────────────────────────────────────────────────
+
+    @objc func addSkill(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let inputJson = call.getString("inputJson") else {
+                return call.reject("inputJson is required")
+            }
+            do {
+                let json = try h.addSkill(inputJson: inputJson)
+                call.resolve(["recordJson": json])
+            } catch {
+                call.reject("addSkill failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func updateSkill(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let id = call.getString("id") else {
+                return call.reject("id is required")
+            }
+            do {
+                try h.updateSkill(id: id, patchJson: call.getString("patchJson") ?? "{}")
+                call.resolve()
+            } catch {
+                call.reject("updateSkill failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func removeSkill(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let id = call.getString("id") else {
+                return call.reject("id is required")
+            }
+            do {
+                try h.removeSkill(id: id)
+                call.resolve()
+            } catch {
+                call.reject("removeSkill failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func listSkills(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                call.resolve(["skillsJson": try h.listSkills()])
+            } catch {
+                call.reject("listSkills failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func startSkill(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let skillId = call.getString("skillId") else {
+                return call.reject("skillId is required")
+            }
+            do {
+                let sessKey = try h.startSkill(
+                    skillId: skillId,
+                    configJson: call.getString("configJson") ?? "{}",
+                    provider: call.getString("provider")
+                )
+                call.resolve(["sessionKey": sessKey])
+            } catch {
+                call.reject("startSkill failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func endSkill(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let skillId = call.getString("skillId") else {
+                return call.reject("skillId is required")
+            }
+            do {
+                try h.endSkill(skillId: skillId)
+                call.resolve()
+            } catch {
+                call.reject("endSkill failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    // ── MCP ──────────────────────────────────────────────────────────────────
+
+    @objc func startMcp(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                let count = try h.startMcp(toolsJson: call.getString("toolsJson") ?? "[]")
+                call.resolve(["toolCount": Int(count)])
+            } catch {
+                call.reject("startMcp failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    @objc func restartMcp(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                let count = try h.restartMcp(toolsJson: call.getString("toolsJson") ?? "[]")
+                call.resolve(["toolCount": Int(count)])
+            } catch {
+                call.reject("restartMcp failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    // ── Models ───────────────────────────────────────────────────────────────
+
+    @objc func getModels(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            do {
+                let json = try h.getModels(provider: call.getString("provider") ?? "anthropic")
+                call.resolve(["modelsJson": json])
+            } catch {
+                call.reject("getModels failed: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    // ── Tools ────────────────────────────────────────────────────────────────
+
+    @objc func invokeTool(_ call: CAPPluginCall) {
+        withHandle(call) { h in
+            guard let toolName = call.getString("toolName") else {
+                return call.reject("toolName is required")
+            }
+            do {
+                let resultJson = try h.invokeTool(
+                    toolName: toolName,
+                    argsJson: call.getString("argsJson") ?? "{}"
+                )
+                call.resolve(["resultJson": resultJson])
+            } catch {
+                call.reject("invokeTool failed: \(error.localizedDescription)")
+            }
+        }
+    }
+}
+
+// ── Event bridge ─────────────────────────────────────────────────────────────
+
+/// Bridges Rust NativeEventCallback → Capacitor notifyListeners
+class NativeAgentEventBridge: NativeEventCallback {
+    private weak var plugin: NativeAgentPlugin?
+
+    init(plugin: NativeAgentPlugin) {
+        self.plugin = plugin
+    }
+
+    func onEvent(eventType: String, payloadJson: String) {
+        plugin?.notifyListeners("nativeAgentEvent", data: [
+            "eventType": eventType,
+            "payloadJson": payloadJson,
+        ])
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "capacitor-native-agent",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "Native AI agent loop for Capacitor — runs LLM completions, tool execution, and cron jobs in native Rust, enabling background execution.",
   "main": "dist/esm/index.js",
   "types": "dist/esm/index.d.ts",
@@ -17,12 +17,19 @@
   "capacitor": {
     "android": {
       "src": "android"
+    },
+    "ios": {
+      "src": "ios"
     }
   },
   "files": [
     "dist/esm/",
     "android/src/main/",
     "android/build.gradle",
+    "ios/Sources/",
+    "ios/Frameworks/",
+    "CapacitorNativeAgent.podspec",
+    "Package.swift",
     "README.md",
     "LICENSE"
   ],