Merge pull request #1031 from DimensionDev/ios/20250616

Refactoring code logic - 8

Author: 110

iosApp/iosApp/UI/Page/Compose/SensitiveContentAnalyzer/SensitiveContentAnalyzer.swift

@@ -3,77 +3,151 @@ import Kingfisher
 import SensitiveContentAnalysis
 import UIKit
 
+// The SensitiveContentAnalysis entitlement is not available for Enterprise development or for people with free accounts.
+// https://developer.apple.com/documentation/sensitivecontentanalysis/detecting-nudity-in-media-and-providing-intervention-options
+// fuck
 class SensitiveContentAnalyzer {
     static let shared = SensitiveContentAnalyzer()
 
     private let analyzer = SCSensitivityAnalyzer()
     private let cache = SensitiveContentCache.shared
+    private let urlSession = URLSession.shared
 
     private init() {}
 
     func analyzeImage(url: String) async -> Bool {
-        // 1. cache
-        if let cachedResult = cache.get(for: url) {
-            return cachedResult
+        guard let imageURL = URL(string: url) else {
+            return false
+        }
+        do {
+            return try await analyzeImageDownLocal(url: imageURL)
+        } catch {
+            print("analyzeImageDownLocal error: \(error)")
+            return false
         }
 
-        // 2.   Kingfisher cache
-//        if let cachedImage = await getCachedImage(url: url) {
-//            do {
-//                let response = try await analyzer.analyzeImage(cachedImage as! CGImage)
-//                let isSensitive = response.isSensitive
-//
-//
-//                cache.set(url: url, isSensitive: isSensitive)
-//                return isSensitive
-//            } catch {
-//                print("Kingfisher cache fail: \(error)")
-//             }
-//        }
+        // 1. cache
+        // if let cachedResult = cache.get(for: url) {
+        //     return cachedResult
+        // }
+
+        // // 2.   Kingfisher cache
+        // if let cachedImage = await getCachedImage(url: url) {
+        //     do {
+        //         let response = try await analyzer.analyzeImage(cachedImage as! CGImage)
+        //         let isSensitive = response.isSensitive
+
+        //         cache.set(url: url, isSensitive: isSensitive)
+        //         return isSensitive
+        //     } catch {
+        //         print("Kingfisher cache fail: \(error)")
+        //         return false
+        //     }
+        // }
+        // return false
+
+//        var xsds = analyzer.analysisPolicy
 
         // 3. analyze NetworkImage
-        do {
-            guard let imageURL = URL(string: url) else {
-                return false
-            }
+        // do {
+        //     guard let imageURL = URL(string: url) else {
+        //         return false
+        //     }
 
-            let response = try await analyzer.analyzeImage(at: imageURL)
-            let isSensitive = response.isSensitive
+        //     let response = try await analyzer.analyzeImage(at: imageURL)
+        //     let isSensitive = response.isSensitive
 
-            cache.set(url: url, isSensitive: isSensitive)
+        //     cache.set(url: url, isSensitive: isSensitive)
 
-            return isSensitive
-        } catch {
-            print("analyze NetworkImage check faile: \(error)")
-            return false
+        //     return isSensitive
+        // } catch {
+        //     print("analyze NetworkImage check faile: \(error)")
+        //     return false
+        // }
+    }
+
+    private func checkPolicy() -> Bool {
+        if analyzer.analysisPolicy == .disabled {
+            false
+        } else {
+            true
         }
     }
 
     private func getCachedImage(url: String) async -> UIImage? {
         await withCheckedContinuation { continuation in
-            guard let imageURL = URL(string: url) else {
+            print("[DEBUG] 开始检查缓存: \(url)")
+
+            // check custom cache
+            if let cachedResult = cache.get(for: url) {
+                print("[DEBUG] 自定义缓存命中")
                 continuation.resume(returning: nil)
                 return
             }
 
-            // check  Kingfisher mem cache
-            let cache = ImageCache.default
-            let key = imageURL.cacheKey
+            let kingfisherCache = ImageCache.default
+            // 使用与KFImage相同的缓存键
+            let key = url // 改为使用absoluteString
 
-            if let image = cache.retrieveImageInMemoryCache(forKey: key) {
+            if let image = kingfisherCache.retrieveImageInMemoryCache(forKey: key) {
+                print("[DEBUG] 内存缓存命中: \(image)")
                 continuation.resume(returning: image)
                 return
             }
 
-            // check Kingfisher disk cache
-            cache.retrieveImage(forKey: key, options: nil) { result in
+            print("[DEBUG] 检查磁盘缓存...")
+            kingfisherCache.retrieveImage(forKey: key, options: nil) { result in
                 switch result {
                 case let .success(value):
-                    continuation.resume(returning: value.image)
-                case .failure:
+                    if let image = value.image {
+                        print("[DEBUG] 磁盘缓存成功: \(image)")
+                        continuation.resume(returning: image)
+                    } else {
+                        print("[DEBUG] 磁盘缓存损坏，清理缓存")
+                        kingfisherCache.removeImage(forKey: key)
+                        continuation.resume(returning: nil)
+                    }
+                case let .failure(error):
+                    print("[DEBUG] 磁盘缓存失败: \(error)")
                     continuation.resume(returning: nil)
                 }
             }
         }
     }
+
+    private var cacheDirectory: URL {
+        FileManager.default.temporaryDirectory
+            .appendingPathComponent("SensitiveContentCache", isDirectory: true)
+    }
+
+    func analyzeImageDownLocal(url: URL) async throws -> Bool {
+        let cachedFile = cacheDirectory.appendingPathComponent(url.lastPathComponent)
+
+        if !FileManager.default.fileExists(atPath: cachedFile.path) {
+            try await downloadImage(from: url, to: cachedFile)
+        }
+
+        let response = try await analyzer.analyzeImage(at: cachedFile)
+        return response.isSensitive
+    }
+
+    private func downloadImage(from url: URL, to localURL: URL) async throws {
+        try FileManager.default.createDirectory(
+            at: cacheDirectory,
+            withIntermediateDirectories: true
+        )
+
+        let (tempURL, _) = try await urlSession.download(from: url)
+
+        if FileManager.default.fileExists(atPath: localURL.path) {
+            try FileManager.default.removeItem(at: localURL)
+        }
+        try FileManager.default.moveItem(at: tempURL, to: localURL)
+    }
+
+    func clearCache() throws {
+        if FileManager.default.fileExists(atPath: cacheDirectory.path) {
+            try FileManager.default.removeItem(at: cacheDirectory)
+        }
+    }
 }

iosApp/iosApp/UI/Page/Compose/Timeline/ShareButton/ShareButton.swift

@@ -290,7 +290,7 @@ struct ShareButton: View {
                     .imageScale(.medium)
                     .foregroundColor(theme.labelColor)
                     .font(.system(size: 13))
-                    // .foregroundColor(theme.tintColor)
+                // .foregroundColor(theme.tintColor)
                 Spacer()
             }
             .frame(maxWidth: .infinity)

iosApp/iosApp/UI/Page/Compose/Timeline/ShareButton/StatusRowSelectableTextView.swift

@@ -69,22 +69,21 @@ struct StatusRowSelectableTextView: View {
 
         // 1. 提取特殊元素
         let (specialTags, cleanedText) = extractSpecialElements(from: markdownText)
- 
-        // 2. 文本清理 
+
+        // 2. 文本清理
         let cleanedTextSegments = cleanTextForTokenization(cleanedText)
 
         // 3. 语言检测
         let languageRecognizer = NLLanguageRecognizer()
         languageRecognizer.processString(cleanedText)
         detectedLanguage = languageRecognizer.dominantLanguage
 
-
-        // 4. 分词处理  
+        // 4. 分词处理
         var tempTokenizedTags: [Tag] = []
         for segment in cleanedTextSegments {
             tempTokenizedTags.append(contentsOf: tokenizeText(segment, unit: selectedGranularity, language: detectedLanguage))
         }
-        
+
         // 5. 合并所有标签
         allTags = specialTags + tempTokenizedTags
     }
@@ -164,7 +163,7 @@ struct StatusRowSelectableTextView: View {
 
     private func cleanTextForTokenization(_ text: String) -> [String] {
         let segments = text.components(separatedBy: "<br />")
-        
+
         return segments.compactMap { segment -> String? in
             let cleanedSegment = segment
                 // HTML标签
@@ -175,7 +174,7 @@ struct StatusRowSelectableTextView: View {
                 .replacingOccurrences(of: "\\s+", with: " ", options: .regularExpression)
                 .trimmingCharacters(in: .whitespacesAndNewlines)
                 .replacingOccurrences(of: "{ }", with: "")
-            
+
             return cleanedSegment.isEmpty ? nil : cleanedSegment
         }
     }