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,155 @@
+import Foundation
+
+enum OpenCodeGoZenBalanceParser {
+    private static let billingScale = 100_000_000.0
+
+    static func parse(text: String) -> Double? {
+        if let value = self.parseJSON(text: text) {
+            return value
+        }
+        let localizedPattern = [
+            #"(?i)(?:current\s+balance|zen\s+balance|現在の残高)"#,
+            #"[^$]{0,80}\$\s*([0-9][0-9,]*(?:\.[0-9]+)?)"#,
+        ].joined()
+        if let value = self.extractDollarValue(pattern: localizedPattern, text: text) {
+            return value
+        }
+        let nearbyPattern = #"(?i)(?:balance|残高)[\s\S]{0,120}?\$\s*([0-9][0-9,]*(?:\.[0-9]+)?)"#
+        return self.extractDollarValue(pattern: nearbyPattern, text: text)
+    }
+
+    static func parseBillingServerResponse(text: String) -> Double? {
+        if let data = text.data(using: .utf8),
+           let object = try? JSONSerialization.jsonObject(with: data, options: []),
+           let rawBalance = self.findRawBillingBalance(in: object)
+        {
+            return rawBalance / self.billingScale
+        }
+
+        let customerPattern =
+            #"(?:\"customerID\"|customerID)\s*:\s*(?:\$R\[\d+\]\s*=\s*)?\"[^\"]+\""#
+        guard self.containsMatch(pattern: customerPattern, text: text) else {
+            return nil
+        }
+        let pattern = #"(?:\"balance\"|balance)\s*:\s*(?:\$R\[\d+\]\s*=\s*)?(-?[0-9]+(?:\.[0-9]+)?)"#
+        guard let rawBalance = self.extractNumber(pattern: pattern, text: text) else {
+            return nil
+        }
+        return rawBalance / self.billingScale
+    }
+
+    private static func parseJSON(text: String) -> Double? {
+        guard let data = text.data(using: .utf8),
+              let object = try? JSONSerialization.jsonObject(with: data, options: [])
+        else {
+            return nil
+        }
+        return self.findBalanceValue(in: object, path: [])
+    }
+
+    private static func findBalanceValue(in object: Any, path: [String]) -> Double? {
+        if let dict = object as? [String: Any] {
+            for (key, value) in dict {
+                let nextPath = path + [key]
+                if self.isExplicitBalanceAmountKey(key),
+                   let number = self.doubleValue(from: value)
+                {
+                    return number
+                }
+                if let found = self.findBalanceValue(in: value, path: nextPath) {
+                    return found
+                }
+            }
+            return nil
+        }
+        if let array = object as? [Any] {
+            for (index, value) in array.enumerated() {
+                if let found = self.findBalanceValue(in: value, path: path + ["[\(index)]"]) {
+                    return found
+                }
+            }
+        }
+        return nil
+    }
+
+    private static func findRawBillingBalance(in object: Any) -> Double? {
+        if let dict = object as? [String: Any] {
+            if dict["balance"] != nil {
+                guard let customerID = dict["customerID"] as? String,
+                      !customerID.isEmpty
+                else {
+                    return nil
+                }
+                guard let rawBalance = self.doubleValue(from: dict["balance"]) else {
+                    return nil
+                }
+                return rawBalance
+            }
+            for value in dict.values {
+                if let found = self.findRawBillingBalance(in: value) {
+                    return found
+                }
+            }
+        } else if let array = object as? [Any] {
+            for value in array {
+                if let found = self.findRawBillingBalance(in: value) {
+                    return found
+                }
+            }
+        }
+        return nil
+    }
+
+    private static func containsMatch(pattern: String, text: String) -> Bool {
+        guard let regex = try? NSRegularExpression(pattern: pattern, options: []) else { return false }
+        let nsrange = NSRange(text.startIndex..<text.endIndex, in: text)
+        return regex.firstMatch(in: text, options: [], range: nsrange) != nil
+    }
+
+    private static func isExplicitBalanceAmountKey(_ key: String) -> Bool {
+        let normalized = key
+            .lowercased()
+            .filter { $0.isLetter || $0.isNumber }
+        return [
+            "zenbalance",
+            "zencurrentbalance",
+            "currentbalance",
+            "currentbalanceusd",
+            "balanceusd",
+            "usdbalance",
+        ].contains(normalized)
+    }
+
+    private static func extractDollarValue(pattern: String, text: String) -> Double? {
+        self.extractNumber(pattern: pattern, text: text)
+    }
+
+    private static func extractNumber(pattern: String, text: String) -> Double? {
+        guard let regex = try? NSRegularExpression(pattern: pattern, options: []) else { return nil }
+        let nsrange = NSRange(text.startIndex..<text.endIndex, in: text)
+        guard let match = regex.firstMatch(in: text, options: [], range: nsrange),
+              let range = Range(match.range(at: 1), in: text)
+        else {
+            return nil
+        }
+        return Double(text[range].replacingOccurrences(of: ",", with: ""))
+    }
+
+    private static func doubleValue(from value: Any?) -> Double? {
+        switch value {
+        case is Bool:
+            nil
+        case let number as Double:
+            number
+        case let number as NSNumber:
+            number.doubleValue
+        case let string as String:
+            Double(
+                string
+                    .trimmingCharacters(in: .whitespacesAndNewlines)
+                    .replacingOccurrences(of: ",", with: ""))
+        default:
+            nil
+        }
+    }
+}