@dvai-bridge/capacitor-llama 4.0.0

package/android/src/main/java/co/deepvoiceai/bridge/llama/Plugin.kt

@@ -0,0 +1,177 @@
+package co.deepvoiceai.bridge.llama
+
+import com.getcapacitor.JSArray
+import com.getcapacitor.JSObject
+import com.getcapacitor.Plugin
+import com.getcapacitor.PluginCall
+import com.getcapacitor.PluginMethod
+import com.getcapacitor.annotation.CapacitorPlugin
+import co.deepvoiceai.bridge.llama.core.ModelDownloader
+import co.deepvoiceai.bridge.llama.core.PluginState
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.SupervisorJob
+import kotlinx.coroutines.cancel
+import kotlinx.coroutines.launch
+
+// ---------------------------------------------------------------------------
+// JSObject ↔ Map translation helpers — needed because PluginState's API is
+// Capacitor-neutral (Map<String, Any?>) while the JS-bridge layer works with
+// Capacitor's JSObject (which extends JSONObject and has no .toMap()).
+// ---------------------------------------------------------------------------
+
+private fun JSObject.toAnyMap(): Map<String, Any?> {
+    val out = mutableMapOf<String, Any?>()
+    val it = keys()
+    while (it.hasNext()) {
+        val k = it.next()
+        out[k] = this.opt(k)  // returns Any (or JSONObject.NULL); fine for our usage
+    }
+    return out
+}
+
+private fun Map<String, Any?>.toJSObject(): JSObject {
+    val obj = JSObject()
+    for ((k, v) in this) {
+        if (v != null) obj.put(k, v)
+    }
+    return obj
+}
+
+@CapacitorPlugin(name = "DVAIBridgeLlama")
+class DVAIBridgeLlamaPlugin : Plugin() {
+    private val state = PluginState()
+    private val scope = CoroutineScope(Dispatchers.IO + SupervisorJob())
+    private val downloader: ModelDownloader by lazy { ModelDownloader(context) }
+
+    override fun handleOnDestroy() {
+        super.handleOnDestroy()
+        scope.cancel()
+    }
+
+    @PluginMethod
+    fun start(call: PluginCall) {
+        scope.launch {
+            notifyListeners("progress", JSObject().apply { put("phase", "load") })
+            try {
+                val resultMap = state.start((call.data ?: JSObject()).toAnyMap())
+                notifyListeners("progress", JSObject().apply { put("phase", "ready") })
+                call.resolve(resultMap.toJSObject())
+            } catch (e: Exception) {
+                notifyListeners("progress", JSObject().apply {
+                    put("phase", "error")
+                    put("message", e.message ?: "Start failed")
+                })
+                call.reject(e.message ?: "Start failed", e)
+            }
+        }
+    }
+
+    @PluginMethod
+    fun stop(call: PluginCall) {
+        scope.launch {
+            try {
+                state.stop()
+                call.resolve()
+            } catch (e: Exception) {
+                call.reject(e.message ?: "Stop failed", e)
+            }
+        }
+    }
+
+    @PluginMethod
+    fun status(call: PluginCall) {
+        scope.launch {
+            call.resolve(state.statusInfo().toJSObject())
+        }
+    }
+
+    @PluginMethod
+    fun downloadModel(call: PluginCall) {
+        val url = call.getString("url") ?: return call.reject("url is required")
+        val sha = call.getString("sha256")?.takeIf { it.isNotEmpty() }?.lowercase()
+            ?: return call.reject("sha256 is required")
+        val destFilename = call.getString("destFilename") ?: url.substringAfterLast('/')
+        val headers: Map<String, String> = call.getObject("headers")?.let { obj ->
+            val map = mutableMapOf<String, String>()
+            obj.keys().forEachRemaining { k -> obj.getString(k)?.let { map[k] = it } }
+            map
+        } ?: emptyMap()
+
+        scope.launch {
+            try {
+                val (path, cached) = downloader.downloadModel(
+                    url = url,
+                    expectedSha256 = sha,
+                    destFilename = destFilename,
+                    headers = headers,
+                ) { bytesDone, bytesTotal ->
+                    val payload = JSObject().apply {
+                        put("phase", "download")
+                        put("bytesReceived", bytesDone)
+                        if (bytesTotal != null) {
+                            put("bytesTotal", bytesTotal)
+                            if (bytesTotal > 0) {
+                                put("percent", bytesDone.toDouble() / bytesTotal.toDouble() * 100.0)
+                            }
+                        }
+                    }
+                    notifyListeners("progress", payload)
+                }
+                call.resolve(JSObject().apply {
+                    put("path", path)
+                    put("cached", cached)
+                })
+            } catch (e: Exception) {
+                call.reject(e.message ?: "Download failed", e)
+            }
+        }
+    }
+
+    @PluginMethod
+    fun listCachedModels(call: PluginCall) {
+        scope.launch {
+            try {
+                val infos = downloader.listCached()
+                val arr = JSArray()
+                infos.forEach { info ->
+                    arr.put(JSObject().apply {
+                        put("filename", info.filename)
+                        put("path", info.path)
+                        put("bytes", info.bytes)
+                        put("sha256", info.sha256)
+                    })
+                }
+                call.resolve(JSObject().apply { put("models", arr) })
+            } catch (e: Exception) {
+                call.reject(e.message ?: "List failed", e)
+            }
+        }
+    }
+
+    @PluginMethod
+    fun deleteCachedModel(call: PluginCall) {
+        val filename = call.getString("filename")?.takeIf { it.isNotEmpty() }
+            ?: return call.reject("filename is required")
+        scope.launch {
+            try {
+                downloader.deleteCached(filename)
+                call.resolve()
+            } catch (e: Exception) {
+                call.reject(e.message ?: "Delete failed", e)
+            }
+        }
+    }
+
+    @PluginMethod
+    fun cacheDir(call: PluginCall) {
+        scope.launch {
+            try {
+                val path = downloader.cacheDir().absolutePath
+                call.resolve(JSObject().apply { put("path", path) })
+            } catch (e: Exception) {
+                call.reject(e.message ?: "cacheDir failed", e)
+            }
+        }
+    }
+}

package/android/src/main/res/xml/dvai_network_security_config.xml

@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="utf-8"?>
+<network-security-config>
+    <domain-config cleartextTrafficPermitted="true">
+        <domain includeSubdomains="false">localhost</domain>
+        <domain includeSubdomains="false">127.0.0.1</domain>
+    </domain-config>
+</network-security-config>

package/android/src/test/java/co/deepvoiceai/bridge/llama/SmokeTest.kt

@@ -0,0 +1,11 @@
+package co.deepvoiceai.bridge.llama
+
+import org.junit.Assert.assertNotNull
+import org.junit.Test
+
+class SmokeTest {
+    @Test
+    fun pluginClassExists() {
+        assertNotNull(DVAIBridgeLlamaPlugin::class.java)
+    }
+}

package/dist/index.cjs

@@ -0,0 +1,34 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var index_exports = {};
+__export(index_exports, {
+  DVAIBridgeLlama: () => DVAIBridgeLlama,
+  default: () => index_default
+});
+module.exports = __toCommonJS(index_exports);
+var import_core = require("@capacitor/core");
+var DVAIBridgeLlama = (0, import_core.registerPlugin)("DVAIBridgeLlama");
+var index_default = DVAIBridgeLlama;
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  DVAIBridgeLlama
+});
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import { registerPlugin } from \"@capacitor/core\";\r\nimport type { NativePluginInterface } from \"@dvai-bridge/capacitor\";\r\n\r\nconst DVAIBridgeLlama = registerPlugin<NativePluginInterface>(\"DVAIBridgeLlama\");\r\n\r\nexport default DVAIBridgeLlama;\r\nexport { DVAIBridgeLlama };\r\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,kBAA+B;AAG/B,IAAM,sBAAkB,4BAAsC,iBAAiB;AAE/E,IAAO,gBAAQ;","names":[]}

package/dist/index.d.cts

@@ -0,0 +1,5 @@
+import { NativePluginInterface } from '@dvai-bridge/capacitor';
+
+declare const DVAIBridgeLlama: NativePluginInterface;
+
+export { DVAIBridgeLlama, DVAIBridgeLlama as default };

package/dist/index.d.ts

@@ -0,0 +1,5 @@
+import { NativePluginInterface } from '@dvai-bridge/capacitor';
+
+declare const DVAIBridgeLlama: NativePluginInterface;
+
+export { DVAIBridgeLlama, DVAIBridgeLlama as default };

package/dist/index.js

@@ -0,0 +1,9 @@
+// src/index.ts
+import { registerPlugin } from "@capacitor/core";
+var DVAIBridgeLlama = registerPlugin("DVAIBridgeLlama");
+var index_default = DVAIBridgeLlama;
+export {
+  DVAIBridgeLlama,
+  index_default as default
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import { registerPlugin } from \"@capacitor/core\";\r\nimport type { NativePluginInterface } from \"@dvai-bridge/capacitor\";\r\n\r\nconst DVAIBridgeLlama = registerPlugin<NativePluginInterface>(\"DVAIBridgeLlama\");\r\n\r\nexport default DVAIBridgeLlama;\r\nexport { DVAIBridgeLlama };\r\n"],"mappings":";AAAA,SAAS,sBAAsB;AAG/B,IAAM,kBAAkB,eAAsC,iBAAiB;AAE/E,IAAO,gBAAQ;","names":[]}

package/ios/Package.swift

@@ -0,0 +1,57 @@
+// swift-tools-version: 5.9
+import PackageDescription
+
+let package = Package(
+    name: "DVAICapacitorLlama",
+    platforms: [.iOS(.v14), .macOS(.v12)],
+    products: [
+        .library(name: "DVAICapacitorLlama", targets: ["DVAICapacitorLlama"]),
+    ],
+    dependencies: [
+        // Capacitor SPM artifact (existing)
+        .package(url: "https://github.com/ionic-team/capacitor-swift-pm", branch: "main"),
+        // Core package — relative path during dev, replaced with version-pin
+        // or git URL at publish time. The path is relative to *this* Package.swift's
+        // location (`packages/dvai-bridge-capacitor-llama/ios/`), so two `..` get
+        // us to the `packages/` parent and `dvai-bridge-ios-llama-core/ios` is the
+        // sibling package's SPM root.
+        //
+        // The core's Package.swift lives at the PACKAGE ROOT (not under `ios/`)
+        // so that SPM derives identity "dvai-bridge-ios-llama-core" rather than
+        // "ios" for this dependency. With both Package.swifts at `ios/`, SPM's
+        // path-dep identity-from-last-dir-name rule aliased them and triggered
+        // a false `cyclic dependency between packages DVAICapacitorLlama ->
+        // DVAICapacitorLlama` resolution error. The bare product reference
+        // ("DVAILlamaCore") in the target dependency list below works because
+        // SPM auto-resolves unambiguous product names across the dep graph.
+        .package(path: "../../dvai-bridge-ios-llama-core"),
+    ],
+    targets: [
+        .target(
+            name: "DVAICapacitorLlama",
+            dependencies: [
+                .product(name: "Capacitor", package: "capacitor-swift-pm"),
+                .product(name: "Cordova", package: "capacitor-swift-pm"),
+                // The package's identity is derived from the last dir of the
+                // path dep (`dvai-bridge-ios-llama-core`), not from the
+                // manifest's `name:` field. Bare-name product references don't
+                // cross package boundaries reliably; explicit form is required.
+                .product(name: "DVAILlamaCore", package: "dvai-bridge-ios-llama-core"),
+            ],
+            path: "Sources/DVAICapacitorLlama",
+            exclude: ["PluginProxy.m"]
+        ),
+        .testTarget(
+            name: "DVAICapacitorLlamaTests",
+            dependencies: [
+                "DVAICapacitorLlama",
+                // RealModelSmokeTest reaches the core directly (LlamaCppBridge,
+                // ModelDownloader, MTMD_MEDIA_MARKER) — declare both core products
+                // explicitly so Xcode resolves them at link time.
+                .product(name: "DVAILlamaCore", package: "dvai-bridge-ios-llama-core"),
+                .product(name: "DVAILlamaCoreObjC", package: "dvai-bridge-ios-llama-core"),
+            ],
+            path: "Tests/DVAICapacitorLlamaTests"
+        ),
+    ]
+)

package/ios/Sources/DVAICapacitorLlama/Plugin.swift

@@ -0,0 +1,154 @@
+import Foundation
+import DVAILlamaCore
+
+#if canImport(Capacitor)
+import Capacitor
+
+@objc(DVAIBridgeLlamaPlugin)
+public class DVAIBridgeLlamaPlugin: CAPPlugin {
+    private let state = PluginState()
+    private let downloader = ModelDownloader()
+
+    public override func load() {
+        super.load()
+    }
+
+    @objc func start(_ call: CAPPluginCall) {
+        let opts: [String: Any] = (call.options as? [String: Any]) ?? [:]
+        Task { [weak self] in
+            guard let self else { return }
+            self.notifyListeners("progress", data: ["phase": "load"])
+            do {
+                let result = try await self.state.start(opts: opts)
+                self.notifyListeners("progress", data: ["phase": "ready"])
+                call.resolve(result)
+            } catch {
+                self.notifyListeners("progress", data: [
+                    "phase": "error",
+                    "message": error.localizedDescription,
+                ])
+                call.reject(error.localizedDescription)
+            }
+        }
+    }
+
+    @objc func stop(_ call: CAPPluginCall) {
+        Task {
+            do {
+                try await state.stop()
+                call.resolve()
+            } catch {
+                call.reject(error.localizedDescription)
+            }
+        }
+    }
+
+    @objc func status(_ call: CAPPluginCall) {
+        Task {
+            let info = await state.statusInfo()
+            call.resolve(info)
+        }
+    }
+
+    @objc func downloadModel(_ call: CAPPluginCall) {
+        guard let urlStr = call.getString("url"), let url = URL(string: urlStr) else {
+            call.reject("url is required")
+            return
+        }
+        guard let sha = call.getString("sha256"), !sha.isEmpty else {
+            call.reject("sha256 is required")
+            return
+        }
+        let destFilename = call.getString("destFilename") ?? url.lastPathComponent
+        // Capacitor's JSObject is `[String: Any]`, so an `as? [String: String]`
+        // cast returns nil whenever any value isn't already typed as String,
+        // silently dropping ALL headers. Mirror Android's per-key extraction.
+        var headers: [String: String] = [:]
+        if let raw = call.getObject("headers") {
+            for (k, v) in raw {
+                if let s = v as? String { headers[k] = s }
+            }
+        }
+
+        Task { [weak self] in
+            guard let self else { return }
+            do {
+                let result = try await self.downloader.downloadModel(
+                    url: url,
+                    expectedSha256: sha.lowercased(),
+                    destFilename: destFilename,
+                    headers: headers,
+                    onProgress: { [weak self] bytesDone, bytesTotal in
+                        guard let self else { return }
+                        var payload: [String: Any] = [
+                            "phase": "download",
+                            "bytesReceived": bytesDone,
+                        ]
+                        if let total = bytesTotal {
+                            payload["bytesTotal"] = total
+                            if total > 0 {
+                                payload["percent"] = Double(bytesDone) / Double(total) * 100.0
+                            }
+                        }
+                        self.notifyListeners("progress", data: payload)
+                    }
+                )
+                call.resolve([
+                    "path": result.path,
+                    "cached": result.cached,
+                ])
+            } catch {
+                call.reject(error.localizedDescription)
+            }
+        }
+    }
+
+    @objc func listCachedModels(_ call: CAPPluginCall) {
+        Task { [weak self] in
+            guard let self else { return }
+            do {
+                let infos = try await self.downloader.listCachedModels()
+                let models: [[String: Any]] = infos.map {
+                    [
+                        "filename": $0.filename,
+                        "path": $0.path,
+                        "bytes": $0.bytes,
+                        "sha256": $0.sha256,
+                    ]
+                }
+                call.resolve(["models": models])
+            } catch {
+                call.reject(error.localizedDescription)
+            }
+        }
+    }
+
+    @objc func deleteCachedModel(_ call: CAPPluginCall) {
+        guard let filename = call.getString("filename"), !filename.isEmpty else {
+            call.reject("filename is required")
+            return
+        }
+        Task { [weak self] in
+            guard let self else { return }
+            do {
+                try await self.downloader.deleteCachedModel(filename: filename)
+                call.resolve()
+            } catch {
+                call.reject(error.localizedDescription)
+            }
+        }
+    }
+
+    @objc func cacheDir(_ call: CAPPluginCall) {
+        Task { [weak self] in
+            guard let self else { return }
+            do {
+                let path = try await self.downloader.cacheDirPath()
+                call.resolve(["path": path])
+            } catch {
+                call.reject(error.localizedDescription)
+            }
+        }
+    }
+}
+#endif

package/ios/Sources/DVAICapacitorLlama/PluginProxy.m

@@ -0,0 +1,15 @@
+#import <Foundation/Foundation.h>
+
+#if __has_include(<Capacitor/Capacitor.h>)
+#import <Capacitor/Capacitor.h>
+
+CAP_PLUGIN(DVAIBridgeLlamaPlugin, "DVAIBridgeLlama",
+    CAP_PLUGIN_METHOD(start, CAPPluginReturnPromise);
+    CAP_PLUGIN_METHOD(stop, CAPPluginReturnPromise);
+    CAP_PLUGIN_METHOD(status, CAPPluginReturnPromise);
+    CAP_PLUGIN_METHOD(downloadModel, CAPPluginReturnPromise);
+    CAP_PLUGIN_METHOD(listCachedModels, CAPPluginReturnPromise);
+    CAP_PLUGIN_METHOD(deleteCachedModel, CAPPluginReturnPromise);
+    CAP_PLUGIN_METHOD(cacheDir, CAPPluginReturnPromise);
+)
+#endif