chore: import upstream snapshot with attribution

2026-07-13 12:22:33 +08:00
commit fc4fcbab58
1848 changed files with 472303 additions and 0 deletions
@@ -0,0 +1,109 @@
+import Foundation
+
+#if os(macOS)
+import SweetCookieKit
+
+public enum QoderCookieImporter {
+    private static let log = CodexBarLog.logger(LogCategories.qoderCookie)
+    private static let cookieClient = BrowserCookieClient()
+    private static let cookieImportOrder: BrowserCookieImportOrder =
+        ProviderDefaults.metadata[.qoder]?.browserCookieOrder ?? Browser.defaultImportOrder
+
+    public struct SessionInfo: Sendable {
+        public let cookies: [HTTPCookie]
+        public let sourceLabel: String
+        public let site: QoderWebSite
+
+        public init(cookies: [HTTPCookie], sourceLabel: String, site: QoderWebSite) {
+            self.cookies = cookies
+            self.sourceLabel = sourceLabel
+            self.site = site
+        }
+
+        public var cookieHeader: String {
+            self.cookies.map { "\($0.name)=\($0.value)" }.joined(separator: "; ")
+        }
+    }
+
+    public static func importSession(
+        browserDetection: BrowserDetection = BrowserDetection(),
+        preferredBrowsers: [Browser] = [],
+        logger: ((String) -> Void)? = nil) throws -> SessionInfo
+    {
+        guard let session = try self.importSessions(
+            browserDetection: browserDetection,
+            preferredBrowsers: preferredBrowsers,
+            logger: logger).first
+        else {
+            throw QoderUsageError.missingCredentials
+        }
+        return session
+    }
+
+    public static func importSessions(
+        browserDetection: BrowserDetection = BrowserDetection(),
+        preferredBrowsers: [Browser] = [],
+        logger: ((String) -> Void)? = nil) throws -> [SessionInfo]
+    {
+        let installedBrowsers = preferredBrowsers.isEmpty
+            ? self.cookieImportOrder.cookieImportCandidates(using: browserDetection)
+            : preferredBrowsers.cookieImportCandidates(using: browserDetection)
+        var sessions: [SessionInfo] = []
+
+        for browserSource in installedBrowsers {
+            for site in QoderWebSite.allCases {
+                do {
+                    let query = Self.cookieQuery(for: site)
+                    let sources = try Self.cookieClient.codexBarRecords(
+                        matching: query,
+                        in: browserSource,
+                        logger: { msg in self.emit(msg, logger: logger) })
+                    for source in sources where !source.records.isEmpty {
+                        let records = self.records(source.records, for: site)
+                        let cookies = BrowserCookieClient.makeHTTPCookies(records, origin: query.origin)
+                        guard !cookies.isEmpty else { continue }
+                        self.emit("Found \(cookies.count) cookies in \(source.label)", logger: logger)
+                        sessions.append(SessionInfo(cookies: cookies, sourceLabel: source.label, site: site))
+                    }
+                } catch {
+                    BrowserCookieAccessGate.recordIfNeeded(error)
+                    self.emit(
+                        "\(browserSource.displayName) cookie import failed: \(error.localizedDescription)",
+                        logger: logger)
+                }
+            }
+        }
+
+        guard !sessions.isEmpty else {
+            throw QoderUsageError.missingCredentials
+        }
+        return sessions
+    }
+
+    static func cookieQuery(for site: QoderWebSite) -> BrowserCookieQuery {
+        BrowserCookieQuery(domains: site.cookieDomains, domainMatch: .exact)
+    }
+
+    static func records(_ records: [BrowserCookieRecord], for site: QoderWebSite) -> [BrowserCookieRecord] {
+        records.filter { record in
+            self.cookieDomainMatchesSite(record.domain, site: site)
+        }
+    }
+
+    private static func cookieDomainMatchesSite(_ rawDomain: String, site: QoderWebSite) -> Bool {
+        let domain = self.normalizedCookieDomain(rawDomain)
+        return site.cookieDomains.contains { self.normalizedCookieDomain($0) == domain }
+    }
+
+    private static func normalizedCookieDomain(_ rawDomain: String) -> String {
+        let trimmed = rawDomain.trimmingCharacters(in: .whitespacesAndNewlines).lowercased()
+        guard trimmed.hasPrefix(".") else { return trimmed }
+        return String(trimmed.dropFirst())
+    }
+
+    private static func emit(_ message: String, logger: ((String) -> Void)?) {
+        logger?("[qoder-cookie] \(message)")
+        self.log.debug(message)
+    }
+}
+#endif