Tools View & Crash Report & OOM Report

2026-04-23 08:12:52 +08:00
parent 73180f0230
commit b48f4ad97f
46 changed files with 4279 additions and 533 deletions
@@ -0,0 +1,164 @@
+import Foundation
+import os
+import SwiftUI
+
+private let logger = Logger(category: "OOMReportManager")
+
+public struct OOMReport: Identifiable, Hashable, Sendable {
+    public let id: String
+    public let date: Date
+    public let fileURL: URL
+    public var isRead: Bool
+    public let origin: String?
+}
+
+public struct OOMReportFile: Identifiable, Hashable, Sendable {
+    public enum Kind: String, Sendable {
+        case metadata
+        case configContent
+        case profile
+    }
+
+    public let id: String
+    public let kind: Kind
+    public let displayName: String
+    public let fileURL: URL
+}
+
+@MainActor
+public class OOMReportManager: ObservableObject {
+    @Published public private(set) var reports: [OOMReport] = []
+    @Published public private(set) var unreadCount: Int = 0
+
+    public init() {}
+
+    public nonisolated func refresh() async {
+        let reports = await BlockingIO.run {
+            #if os(macOS)
+                if Variant.useSystemExtension {
+                    Self.collectAndArchiveOOMReportsViaHelper()
+                }
+            #endif
+            return Self.scanReports()
+        }
+        await MainActor.run {
+            self.reports = reports
+            self.unreadCount = reports.filter { !$0.isRead }.count
+        }
+    }
+
+    private nonisolated static func scanReports() -> [OOMReport] {
+        let dir = OOMReportArchive.reportsDirectory
+        guard let files = try? FileManager.default.contentsOfDirectory(
+            at: dir, includingPropertiesForKeys: [.contentModificationDateKey, .isDirectoryKey],
+            options: .skipsHiddenFiles
+        ) else {
+            return []
+        }
+
+        return files
+            .filter {
+                (try? $0.resourceValues(forKeys: [.isDirectoryKey]).isDirectory) ?? false
+            }
+            .compactMap { url -> OOMReport? in
+                let date = OOMReportArchive.reportDate(for: url)
+                    ?? (try? url.resourceValues(forKeys: [.contentModificationDateKey]).contentModificationDate)
+                    ?? Date.distantPast
+                let origin = OOMReportArchive.readMetadata(for: url)?.deviceOrigin
+                return OOMReport(
+                    id: url.lastPathComponent,
+                    date: date,
+                    fileURL: url,
+                    isRead: FileManager.default.fileExists(atPath: url.appendingPathComponent(ReportArchive.readMarkerFileName).path),
+                    origin: origin
+                )
+            }
+            .sorted { $0.date > $1.date }
+    }
+
+    public nonisolated func availableFiles(for report: OOMReport) async -> [OOMReportFile] {
+        await BlockingIO.run {
+            let fm = FileManager.default
+            var files: [OOMReportFile] = []
+
+            let metadataURL = OOMReportArchive.metadataURL(for: report.fileURL)
+            if fm.fileExists(atPath: metadataURL.path) {
+                files.append(OOMReportFile(id: "metadata", kind: .metadata, displayName: "Metadata", fileURL: metadataURL))
+            }
+
+            let configURL = OOMReportArchive.configURL(for: report.fileURL)
+            if fm.fileExists(atPath: configURL.path) {
+                files.append(OOMReportFile(id: "config", kind: .configContent, displayName: "Configuration", fileURL: configURL))
+            }
+
+            for profileURL in OOMReportArchive.profileFiles(for: report.fileURL) {
+                let name = profileURL.lastPathComponent
+                files.append(OOMReportFile(id: name, kind: .profile, displayName: name, fileURL: profileURL))
+            }
+
+            return files
+        }
+    }
+
+    public func markAsRead(_ report: OOMReport) {
+        FileManager.default.createFile(atPath: report.fileURL.appendingPathComponent(ReportArchive.readMarkerFileName).path, contents: nil)
+        if let idx = reports.firstIndex(where: { $0.id == report.id }), !reports[idx].isRead {
+            reports[idx].isRead = true
+            unreadCount = max(0, unreadCount - 1)
+        }
+    }
+
+    public nonisolated func delete(_ report: OOMReport) async {
+        await BlockingIO.run {
+            OOMReportArchive.removeArtifact(at: report.fileURL)
+        }
+        await MainActor.run {
+            let wasUnread = reports.first { $0.id == report.id }.map { !$0.isRead } ?? false
+            reports.removeAll { $0.id == report.id }
+            if wasUnread {
+                unreadCount = max(0, unreadCount - 1)
+            }
+        }
+    }
+
+    public nonisolated func deleteAll() async {
+        let dir = OOMReportArchive.reportsDirectory
+        await BlockingIO.run {
+            try? FileManager.default.removeItem(at: dir)
+        }
+        await MainActor.run {
+            reports.removeAll()
+            unreadCount = 0
+        }
+    }
+
+    #if os(macOS)
+        private nonisolated static func collectAndArchiveOOMReportsViaHelper() {
+            guard HelperServiceManager.rootHelperStatus == .enabled else {
+                return
+            }
+
+            let artifacts: OOMReportArtifactsResult
+            do {
+                artifacts = try RootHelperClient.shared.collectOOMReportArtifacts()
+            } catch {
+                logger.warning("collectOOMReportArtifacts: \(error.localizedDescription)")
+                return
+            }
+
+            let reportsDir = OOMReportArchive.reportsDirectory
+            for report in artifacts.reports {
+                let destURL = reportsDir.appendingPathComponent(report.directoryName, isDirectory: true)
+                do {
+                    try FileManager.default.createDirectory(at: destURL, withIntermediateDirectories: true)
+                    for file in report.files {
+                        let fileURL = destURL.appendingPathComponent(file.name)
+                        try file.data.write(to: fileURL, options: .atomic)
+                    }
+                } catch {
+                    logger.warning("write OOM report \(report.directoryName): \(error.localizedDescription)")
+                }
+            }
+        }
+    #endif
+}