import AppKit
import Foundation
import WebKit
final class WebChatWindowController: NSWindowController, WKScriptMessageHandler {
private let webView: WKWebView
private let sessionKey: String
init(sessionKey: String) {
self.sessionKey = sessionKey
let config = WKWebViewConfiguration()
let contentController = WKUserContentController()
config.userContentController = contentController
config.preferences.isElementFullscreenEnabled = true
// Inject callback receiver stub
let callbackScript = """
window.__clawdisCallbacks = new Map();
window.__clawdisReceive = function(resp) {
const entry = window.__clawdisCallbacks.get(resp.id);
if (!entry) return;
window.__clawdisCallbacks.delete(resp.id);
if (resp.ok) {
entry.resolve(resp.result);
} else {
entry.reject(resp.error || 'unknown error');
}
};
window.__clawdisSend = function(payload) {
const id = crypto.randomUUID();
return new Promise((resolve, reject) => {
window.__clawdisCallbacks.set(id, { resolve, reject });
window.webkit?.messageHandlers?.clawdis?.postMessage({ id, ...payload });
});
};
"""
let userScript = WKUserScript(source: callbackScript, injectionTime: .atDocumentStart, forMainFrameOnly: true)
contentController.addUserScript(userScript)
self.webView = WKWebView(frame: .zero, configuration: config)
let window = NSWindow(
contentRect: NSRect(x: 0, y: 0, width: 960, height: 720),
styleMask: [.titled, .closable, .resizable, .miniaturizable],
backing: .buffered,
defer: false)
window.title = "Clawd Web Chat"
window.contentView = self.webView
super.init(window: window)
contentController.add(self, name: "clawdis")
self.loadPage()
}
@available(*, unavailable)
required init?(coder: NSCoder) { fatalError() }
private func loadPage() {
guard let webChatURL = Bundle.main.url(forResource: "WebChat", withExtension: nil) else {
NSLog("WebChat resources missing")
return
}
let distPath = webChatURL.path(percentEncoded: false)
let cssPath = webChatURL.appendingPathComponent("app.css").path(percentEncoded: false)
let vendor = webChatURL.appendingPathComponent("vendor")
let piAi = vendor.appendingPathComponent("pi-ai/index.js").path(percentEncoded: false)
let miniLit = vendor.appendingPathComponent("mini-lit/index.js").path(percentEncoded: false)
let lit = vendor.appendingPathComponent("lit/index.js").path(percentEncoded: false)
let lucide = vendor.appendingPathComponent("lucide/lucide.js").path(percentEncoded: false)
let pdfjs = vendor.appendingPathComponent("pdfjs-dist/build/pdf.js").path(percentEncoded: false)
let pdfWorker = vendor.appendingPathComponent("pdfjs-dist/build/pdf.worker.min.mjs").path(percentEncoded: false)
let importMap = [
"imports": [
"@mariozechner/pi-web-ui": "file://\(distPath)/index.js",
"@mariozechner/pi-ai": "file://\(piAi)",
"@mariozechner/mini-lit": "file://\(miniLit)",
"lit": "file://\(lit)",
"lucide": "file://\(lucide)",
"pdfjs-dist": "file://\(pdfjs)",
"pdfjs-dist/build/pdf.worker.min.mjs": "file://\(pdfWorker)",
],
]
let importMapJSON: String = if let data = try? JSONSerialization.data(
withJSONObject: importMap,
options: [.prettyPrinted]),
let json = String(data: data, encoding: .utf8)
{
json
} else {
"{}"
}
let html = """
Clawd Web Chat
"""
self.webView.loadHTMLString(html, baseURL: webChatURL)
}
func userContentController(_ userContentController: WKUserContentController, didReceive message: WKScriptMessage) {
guard let body = message.body as? [String: Any],
let id = body["id"] as? String,
let type = body["type"] as? String,
type == "chat",
let payload = body["payload"] as? [String: Any],
let text = payload["text"] as? String
else { return }
Task { @MainActor in
let reply = await runAgent(text: text, sessionKey: sessionKey)
let json: [String: Any] = [
"id": id,
"ok": reply.error == nil,
"result": ["text": reply.text ?? ""],
"error": reply.error ?? NSNull(),
]
if let data = try? JSONSerialization.data(withJSONObject: json),
let js = String(data: data, encoding: .utf8)
{
_ = try? await self.webView.evaluateJavaScript("window.__clawdisReceive(" + js + ")")
}
}
}
private func runAgent(text: String, sessionKey: String) async -> (text: String?, error: String?) {
let data: Data
do {
data = try await Task.detached(priority: .utility) { () -> Data in
let process = Process()
process.executableURL = URL(fileURLWithPath: "/usr/bin/env")
process.arguments = ["pnpm", "clawdis", "agent", "--to", sessionKey, "--message", text, "--json"]
process.currentDirectoryURL = URL(fileURLWithPath: "/Users/steipete/Projects/clawdis")
let pipe = Pipe()
process.standardOutput = pipe
process.standardError = Pipe()
try process.run()
process.waitUntilExit()
let out = pipe.fileHandleForReading.readDataToEndOfFile()
guard process.terminationStatus == 0 else {
throw NSError(
domain: "ClawdisAgent",
code: Int(process.terminationStatus),
userInfo: [NSLocalizedDescriptionKey: String(data: out, encoding: .utf8) ?? "unknown error"])
}
return out
}.value
} catch {
return (nil, error.localizedDescription)
}
if let obj = try? JSONSerialization.jsonObject(with: data) as? [String: Any],
let payloads = obj["payloads"] as? [[String: Any]],
let first = payloads.first,
let text = first["text"] as? String
{
return (text, nil)
}
return (String(data: data, encoding: .utf8), nil)
}
}
@MainActor
final class WebChatManager {
static let shared = WebChatManager()
private var window: WebChatWindowController?
func show(sessionKey: String) {
if self.window == nil {
self.window = WebChatWindowController(sessionKey: sessionKey)
}
self.window?.showWindow(nil)
self.window?.window?.makeKeyAndOrderFront(nil)
NSApp.activate(ignoringOtherApps: true)
}
}