333 lines
12 KiB
Swift
333 lines
12 KiB
Swift
import ClawdisKit
|
|
import Observation
|
|
import SwiftUI
|
|
import WebKit
|
|
|
|
@MainActor
|
|
@Observable
|
|
final class ScreenController {
|
|
enum Mode: Sendable {
|
|
case canvas
|
|
case web
|
|
}
|
|
|
|
let webView: WKWebView
|
|
private let navigationDelegate: ScreenNavigationDelegate
|
|
private let a2uiActionHandler: CanvasA2UIActionMessageHandler
|
|
|
|
var urlString: String = ""
|
|
var errorText: String?
|
|
|
|
var mode: Mode {
|
|
self.urlString.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty ? .canvas : .web
|
|
}
|
|
|
|
/// Callback invoked when a clawdis:// deep link is tapped in the canvas
|
|
var onDeepLink: ((URL) -> Void)?
|
|
|
|
/// Callback invoked when the user clicks an A2UI action (e.g. button) inside the canvas web UI.
|
|
var onA2UIAction: (([String: Any]) -> Void)?
|
|
|
|
init() {
|
|
let config = WKWebViewConfiguration()
|
|
config.websiteDataStore = .nonPersistent()
|
|
let a2uiActionHandler = CanvasA2UIActionMessageHandler()
|
|
let userContentController = WKUserContentController()
|
|
userContentController.add(a2uiActionHandler, name: CanvasA2UIActionMessageHandler.messageName)
|
|
config.userContentController = userContentController
|
|
self.navigationDelegate = ScreenNavigationDelegate()
|
|
self.a2uiActionHandler = a2uiActionHandler
|
|
self.webView = WKWebView(frame: .zero, configuration: config)
|
|
// Canvas scaffold is a fully self-contained HTML page; avoid relying on transparency underlays.
|
|
self.webView.isOpaque = true
|
|
self.webView.backgroundColor = .black
|
|
self.webView.scrollView.backgroundColor = .black
|
|
self.webView.scrollView.contentInsetAdjustmentBehavior = .never
|
|
self.webView.scrollView.contentInset = .zero
|
|
self.webView.scrollView.scrollIndicatorInsets = .zero
|
|
self.webView.scrollView.automaticallyAdjustsScrollIndicatorInsets = false
|
|
// Disable scroll to allow touch events to pass through to canvas
|
|
self.webView.scrollView.isScrollEnabled = false
|
|
self.webView.scrollView.bounces = false
|
|
self.webView.navigationDelegate = self.navigationDelegate
|
|
self.navigationDelegate.controller = self
|
|
a2uiActionHandler.controller = self
|
|
self.reload()
|
|
}
|
|
|
|
func navigate(to urlString: String) {
|
|
let trimmed = urlString.trimmingCharacters(in: .whitespacesAndNewlines)
|
|
self.urlString = (trimmed == "/" ? "" : trimmed)
|
|
self.reload()
|
|
}
|
|
|
|
func reload() {
|
|
let trimmed = self.urlString.trimmingCharacters(in: .whitespacesAndNewlines)
|
|
if trimmed.isEmpty {
|
|
guard let url = Self.canvasScaffoldURL else { return }
|
|
self.webView.loadFileURL(url, allowingReadAccessTo: url.deletingLastPathComponent())
|
|
return
|
|
} else {
|
|
guard let url = URL(string: trimmed) else { return }
|
|
if url.isFileURL {
|
|
self.webView.loadFileURL(url, allowingReadAccessTo: url.deletingLastPathComponent())
|
|
} else {
|
|
self.webView.load(URLRequest(url: url))
|
|
}
|
|
}
|
|
}
|
|
|
|
func showDefaultCanvas() {
|
|
self.urlString = ""
|
|
self.reload()
|
|
}
|
|
|
|
func waitForA2UIReady(timeoutMs: Int) async -> Bool {
|
|
let clock = ContinuousClock()
|
|
let deadline = clock.now.advanced(by: .milliseconds(timeoutMs))
|
|
while clock.now < deadline {
|
|
do {
|
|
let res = try await self.eval(javaScript: """
|
|
(() => {
|
|
try {
|
|
return !!globalThis.clawdisA2UI && typeof globalThis.clawdisA2UI.applyMessages === 'function';
|
|
} catch (_) { return false; }
|
|
})()
|
|
""")
|
|
if res == "true" { return true }
|
|
} catch {
|
|
// ignore; page likely still loading
|
|
}
|
|
try? await Task.sleep(nanoseconds: 120_000_000)
|
|
}
|
|
return false
|
|
}
|
|
|
|
func eval(javaScript: String) async throws -> String {
|
|
try await withCheckedThrowingContinuation { cont in
|
|
self.webView.evaluateJavaScript(javaScript) { result, error in
|
|
if let error {
|
|
cont.resume(throwing: error)
|
|
return
|
|
}
|
|
if let result {
|
|
cont.resume(returning: String(describing: result))
|
|
} else {
|
|
cont.resume(returning: "")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func snapshotPNGBase64(maxWidth: CGFloat? = nil) async throws -> String {
|
|
let config = WKSnapshotConfiguration()
|
|
if let maxWidth {
|
|
config.snapshotWidth = NSNumber(value: Double(maxWidth))
|
|
}
|
|
let image: UIImage = try await withCheckedThrowingContinuation { cont in
|
|
self.webView.takeSnapshot(with: config) { image, error in
|
|
if let error {
|
|
cont.resume(throwing: error)
|
|
return
|
|
}
|
|
guard let image else {
|
|
cont.resume(throwing: NSError(domain: "Screen", code: 2, userInfo: [
|
|
NSLocalizedDescriptionKey: "snapshot failed",
|
|
]))
|
|
return
|
|
}
|
|
cont.resume(returning: image)
|
|
}
|
|
}
|
|
guard let data = image.pngData() else {
|
|
throw NSError(domain: "Screen", code: 1, userInfo: [
|
|
NSLocalizedDescriptionKey: "snapshot encode failed",
|
|
])
|
|
}
|
|
return data.base64EncodedString()
|
|
}
|
|
|
|
func snapshotBase64(
|
|
maxWidth: CGFloat? = nil,
|
|
format: ClawdisCanvasSnapshotFormat,
|
|
quality: Double? = nil) async throws -> String
|
|
{
|
|
let config = WKSnapshotConfiguration()
|
|
if let maxWidth {
|
|
config.snapshotWidth = NSNumber(value: Double(maxWidth))
|
|
}
|
|
let image: UIImage = try await withCheckedThrowingContinuation { cont in
|
|
self.webView.takeSnapshot(with: config) { image, error in
|
|
if let error {
|
|
cont.resume(throwing: error)
|
|
return
|
|
}
|
|
guard let image else {
|
|
cont.resume(throwing: NSError(domain: "Screen", code: 2, userInfo: [
|
|
NSLocalizedDescriptionKey: "snapshot failed",
|
|
]))
|
|
return
|
|
}
|
|
cont.resume(returning: image)
|
|
}
|
|
}
|
|
|
|
let data: Data?
|
|
switch format {
|
|
case .png:
|
|
data = image.pngData()
|
|
case .jpeg:
|
|
let q = (quality ?? 0.82).clamped(to: 0.1...1.0)
|
|
data = image.jpegData(compressionQuality: q)
|
|
}
|
|
guard let data else {
|
|
throw NSError(domain: "Screen", code: 1, userInfo: [
|
|
NSLocalizedDescriptionKey: "snapshot encode failed",
|
|
])
|
|
}
|
|
return data.base64EncodedString()
|
|
}
|
|
|
|
// SwiftPM flattens resource directories; ensure resource filenames are unique.
|
|
private static let canvasScaffoldURL: URL? = ClawdisKitResources.bundle.url(
|
|
forResource: "scaffold",
|
|
withExtension: "html")
|
|
private static let a2uiIndexURL: URL? = ClawdisKitResources.bundle.url(
|
|
forResource: "index",
|
|
withExtension: "html")
|
|
|
|
func isTrustedCanvasUIURL(_ url: URL) -> Bool {
|
|
guard url.isFileURL else { return false }
|
|
let std = url.standardizedFileURL
|
|
if let expected = Self.canvasScaffoldURL,
|
|
std == expected.standardizedFileURL
|
|
{
|
|
return true
|
|
}
|
|
if let expected = Self.a2uiIndexURL,
|
|
std == expected.standardizedFileURL
|
|
{
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func isLocalNetworkCanvasURL(_ url: URL) -> Bool {
|
|
guard let scheme = url.scheme?.lowercased(), scheme == "http" || scheme == "https" else {
|
|
return false
|
|
}
|
|
guard let host = url.host?.trimmingCharacters(in: .whitespacesAndNewlines), !host.isEmpty else {
|
|
return false
|
|
}
|
|
if host == "localhost" { return true }
|
|
if host.hasSuffix(".local") { return true }
|
|
if let ipv4 = Self.parseIPv4(host) {
|
|
return Self.isLocalNetworkIPv4(ipv4)
|
|
}
|
|
return false
|
|
}
|
|
|
|
private static func parseIPv4(_ host: String) -> (UInt8, UInt8, UInt8, UInt8)? {
|
|
let parts = host.split(separator: ".", omittingEmptySubsequences: false)
|
|
guard parts.count == 4 else { return nil }
|
|
let bytes: [UInt8] = parts.compactMap { UInt8($0) }
|
|
guard bytes.count == 4 else { return nil }
|
|
return (bytes[0], bytes[1], bytes[2], bytes[3])
|
|
}
|
|
|
|
private static func isLocalNetworkIPv4(_ ip: (UInt8, UInt8, UInt8, UInt8)) -> Bool {
|
|
let (a, b, _, _) = ip
|
|
// 10.0.0.0/8
|
|
if a == 10 { return true }
|
|
// 172.16.0.0/12
|
|
if a == 172, (16...31).contains(Int(b)) { return true }
|
|
// 192.168.0.0/16
|
|
if a == 192, b == 168 { return true }
|
|
// 127.0.0.0/8
|
|
if a == 127 { return true }
|
|
// 169.254.0.0/16 (link-local)
|
|
if a == 169, b == 254 { return true }
|
|
// Tailscale: 100.64.0.0/10
|
|
if a == 100, (64...127).contains(Int(b)) { return true }
|
|
return false
|
|
}
|
|
|
|
static func parseA2UIActionBody(_ body: Any) -> [String: Any]? {
|
|
if let dict = body as? [String: Any] { return dict.isEmpty ? nil : dict }
|
|
if let str = body as? String,
|
|
let data = str.data(using: .utf8),
|
|
let json = try? JSONSerialization.jsonObject(with: data) as? [String: Any]
|
|
{
|
|
return json.isEmpty ? nil : json
|
|
}
|
|
if let dict = body as? [AnyHashable: Any] {
|
|
let mapped = dict.reduce(into: [String: Any]()) { acc, pair in
|
|
guard let key = pair.key as? String else { return }
|
|
acc[key] = pair.value
|
|
}
|
|
return mapped.isEmpty ? nil : mapped
|
|
}
|
|
return nil
|
|
}
|
|
}
|
|
|
|
extension Double {
|
|
fileprivate func clamped(to range: ClosedRange<Double>) -> Double {
|
|
if self < range.lowerBound { return range.lowerBound }
|
|
if self > range.upperBound { return range.upperBound }
|
|
return self
|
|
}
|
|
}
|
|
|
|
// MARK: - Navigation Delegate
|
|
|
|
/// Handles navigation policy to intercept clawdis:// deep links from canvas
|
|
private final class ScreenNavigationDelegate: NSObject, WKNavigationDelegate {
|
|
weak var controller: ScreenController?
|
|
|
|
func webView(
|
|
_ webView: WKWebView,
|
|
decidePolicyFor navigationAction: WKNavigationAction,
|
|
decisionHandler: @escaping (WKNavigationActionPolicy) -> Void)
|
|
{
|
|
guard let url = navigationAction.request.url else {
|
|
decisionHandler(.allow)
|
|
return
|
|
}
|
|
|
|
// Intercept clawdis:// deep links
|
|
if url.scheme == "clawdis" {
|
|
decisionHandler(.cancel)
|
|
Task { @MainActor in
|
|
self.controller?.onDeepLink?(url)
|
|
}
|
|
return
|
|
}
|
|
|
|
decisionHandler(.allow)
|
|
}
|
|
}
|
|
|
|
private final class CanvasA2UIActionMessageHandler: NSObject, WKScriptMessageHandler {
|
|
static let messageName = "clawdisCanvasA2UIAction"
|
|
|
|
weak var controller: ScreenController?
|
|
|
|
func userContentController(_: WKUserContentController, didReceive message: WKScriptMessage) {
|
|
guard message.name == Self.messageName else { return }
|
|
guard let controller else { return }
|
|
|
|
guard let url = message.webView?.url else { return }
|
|
if url.isFileURL {
|
|
guard controller.isTrustedCanvasUIURL(url) else { return }
|
|
} else {
|
|
// For security, only accept actions from local-network pages (e.g. the canvas host).
|
|
guard controller.isLocalNetworkCanvasURL(url) else { return }
|
|
}
|
|
|
|
guard let body = ScreenController.parseA2UIActionBody(message.body) else { return }
|
|
|
|
controller.onA2UIAction?(body)
|
|
}
|
|
}
|