Trio/Trio Watch App Extension/WatchLogger.swift

import Foundation
import WatchConnectivity

actor WatchLogger {
    static let shared = WatchLogger()

    private var logs: [String] = []
    private let maxEntries = 3000
    private let flushInterval: TimeInterval = 3 * 60
    private let flushSizeThreshold = 1000
    private var lastFlush = Date()

    private let session = WCSession.default
    private var timerTask: Task<Void, Never>?

    private init() {
        Task {
            await startFlushTimer()
        }
    }

    private var dateFormatter: DateFormatter {
        let formatter = DateFormatter()
        formatter.dateFormat = "yyyy-MM-dd'T'HH:mm:ssZ"
        return formatter
    }
    private func startFlushTimer() async {
        timerTask = Task {
            while true {
                try? await Task.sleep(nanoseconds: UInt64(flushInterval * 1_000_000_000))
                await flushIfNeeded(force: false)
            }
        }
    }

    func log(
        _ message: String,
        force: Bool = false,
        function: String = #function,
        file: String = #fileID,
        line: Int = #line
    ) async {
        let shortFile = (file as NSString).lastPathComponent
        let timestamp = dateFormatter.string(from: Date())
        let entry = "[\(timestamp)] [\(shortFile):\(line)] \(function) → \(message)"

        logs.append(entry)
        if logs.count > maxEntries {
            logs.removeFirst(logs.count - maxEntries)
        }

        print(entry)
        await flushIfNeeded(force: force)
    }

    func flushIfNeeded(force: Bool = false) async {
        let now = Date()
        let shouldFlush = force || now.timeIntervalSince(lastFlush) >= flushInterval || logs.count >= flushSizeThreshold

        if shouldFlush {
            await flushToPhone()
        } else {
            await log("⌚️ Skipping flush — force: \(force), logs: \(logs.count), interval: \(now.timeIntervalSince(lastFlush))")
        }
    }

    private func flushToPhone() async {
        guard !logs.isEmpty else {
            await log("⌚️ No logs to flush")
            return
        }

        let payload: [String: Any] = ["watchLogs": logs.joined(separator: "\n")]

        if session.activationState != .activated {
            session.activate()
        }

        if session.isReachable {
            await log("⌚️ Sending logs to phone: \(logs.count) entries")
            session.sendMessage(payload, replyHandler: nil) { error in
                Task {
                    await self.log("⌚️ Failed to flush logs to phone: \(error.localizedDescription)")
                    await self.persistLogsLocally()
                }
            }
        } else {
            await log("⌚️ Phone not reachable, persisting logs locally")
            await persistLogsLocally()
        }

        lastFlush = Date()
        logs.removeAll()
    }

    func persistLogsLocally() async {
        let logDir = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first!
            .appendingPathComponent("logs", isDirectory: true)

        try? FileManager.default.createDirectory(at: logDir, withIntermediateDirectories: true)

        let logFile = logDir.appendingPathComponent("watch_log.txt")
        let previousLogFile = logDir.appendingPathComponent("watch_log_prev.txt")
        let startOfDay = Calendar.current.startOfDay(for: Date())

        if let attributes = try? FileManager.default.attributesOfItem(atPath: logFile.path),
           let creationDate = attributes[.creationDate] as? Date,
           creationDate < startOfDay
        {
            try? FileManager.default.removeItem(at: previousLogFile)
            try? FileManager.default.moveItem(at: logFile, to: previousLogFile)
            FileManager.default.createFile(atPath: logFile.path, contents: nil, attributes: [.creationDate: startOfDay])
        }

        let fullLog = logs.joined(separator: "\n") + "\n"
        if let data = fullLog.data(using: .utf8) {
            if let handle = try? FileHandle(forWritingTo: logFile) {
                _ = try? handle.seekToEnd()
                handle.write(data)
                try? handle.close()
            } else {
                try? data.write(to: logFile)
            }
        }

        await log("⌚️ Persisted \(logs.count) logs locally to \(logFile.lastPathComponent)")
    }

    func flushPersistedLogs() async {
        let logDir = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first!
            .appendingPathComponent("logs", isDirectory: true)
        let logFile = logDir.appendingPathComponent("watch_log.txt")

        guard let data = try? Data(contentsOf: logFile),
              let logString = String(data: data, encoding: .utf8),
              !logString.isEmpty
        else { return }

        let payload: [String: Any] = ["watchLogs": logString]

        if session.activationState != .activated {
            session.activate()
        }

        if session.isReachable {
            await log("⌚️ Sending persisted logs to phone")
            session.sendMessage(payload, replyHandler: nil) { error in
                Task {
                    await self.log("⌚️ Failed to resend persisted logs: \(error.localizedDescription)")
                }
            }
            try? FileManager.default.removeItem(at: logFile)
        } else {
            await log("⌚️ Transferring persisted logs via userInfo")
            _ = session.transferUserInfo(payload)
            try? FileManager.default.removeItem(at: logFile)
        }
    }
}