Feed cache

Model/Cache/CacheModel.swift

@@ -1,16 +1,40 @@
+import Cache
 import Foundation
 import Logging
+import SwiftyJSON
 
 struct CacheModel {
     static var shared = CacheModel()
 
+    static let jsonToDataTransformer: (JSON) -> Data = { try! $0.rawData() }
+    static let jsonFromDataTransformer: (Data) -> JSON = { try! JSON(data: $0) }
+    static let jsonTransformer = Transformer(toData: jsonToDataTransformer, fromData: jsonFromDataTransformer)
+
     let logger = Logger(label: "stream.yattee.cache")
 
     static let bookmarksGroup = "group.stream.yattee.app.bookmarks"
     let bookmarksDefaults = UserDefaults(suiteName: Self.bookmarksGroup)
 
-    func removeAll() {
+    func clearBookmarks() {
         guard let bookmarksDefaults else { return }
         bookmarksDefaults.dictionaryRepresentation().keys.forEach(bookmarksDefaults.removeObject(forKey:))
     }
+
+    func clear() {
+        FeedCacheModel.shared.clear()
+        VideosCacheModel.shared.clear()
+    }
+
+    var totalSize: Int {
+        (FeedCacheModel.shared.storage.totalDiskStorageSize ?? 0) +
+            (VideosCacheModel.shared.storage.totalDiskStorageSize ?? 0)
+    }
+
+    var totalSizeFormatted: String {
+        totalSizeFormatter.string(fromByteCount: Int64(totalSize))
+    }
+
+    private var totalSizeFormatter: ByteCountFormatter {
+        .init()
+    }
 }

Model/Cache/FeedCacheModel.swift

@@ -0,0 +1,62 @@
+import Cache
+import Foundation
+import Logging
+import SwiftyJSON
+
+struct FeedCacheModel {
+    static let shared = FeedCacheModel()
+    let logger = Logger(label: "stream.yattee.cache.feed")
+
+    static let diskConfig = DiskConfig(name: "feed")
+    static let memoryConfig = MemoryConfig()
+
+    let storage = try! Storage<String, JSON>(
+        diskConfig: Self.diskConfig,
+        memoryConfig: Self.memoryConfig,
+        transformer: CacheModel.jsonTransformer
+    )
+
+    func storeFeed(account: Account, videos: [Video]) {
+        let date = dateFormatter.string(from: Date())
+        logger.info("caching feed \(account.feedCacheKey) -- \(date)")
+        let feedTimeObject: JSON = ["date": date]
+        let videosObject: JSON = ["videos": videos.map(\.json).map(\.object)]
+        try? storage.setObject(feedTimeObject, forKey: feedTimeCacheKey(account.feedCacheKey))
+        try? storage.setObject(videosObject, forKey: account.feedCacheKey)
+    }
+
+    func retrieveFeed(account: Account) -> [Video] {
+        logger.info("retrieving cache for \(account.feedCacheKey)")
+
+        if let json = try? storage.object(forKey: account.feedCacheKey),
+           let videos = json.dictionaryValue["videos"]
+        {
+            return videos.arrayValue.map { Video.from($0) }
+        }
+
+        return []
+    }
+
+    func getFeedTime(account: Account) -> Date? {
+        if let json = try? storage.object(forKey: feedTimeCacheKey(account.feedCacheKey)),
+           let string = json.dictionaryValue["date"]?.string,
+           let date = dateFormatter.date(from: string)
+        {
+            return date
+        }
+
+        return nil
+    }
+
+    func clear() {
+        try? storage.removeAll()
+    }
+
+    private var dateFormatter: ISO8601DateFormatter {
+        .init()
+    }
+
+    private func feedTimeCacheKey(_ feedCacheKey: String) -> String {
+        "\(feedCacheKey)-feedTime"
+    }
+}

Model/Cache/VideosCacheModel.swift

@@ -7,31 +7,31 @@ struct VideosCacheModel {
     static let shared = VideosCacheModel()
     let logger = Logger(label: "stream.yattee.cache.videos")
 
-    static let jsonToDataTransformer: (JSON) -> Data = { try! $0.rawData() }
-    static let jsonFromDataTransformer: (Data) -> JSON = { try! JSON(data: $0) }
-    static let jsonTransformer = Transformer(toData: jsonToDataTransformer, fromData: jsonFromDataTransformer)
+    static let diskConfig = DiskConfig(name: "videos")
+    static let memoryConfig = MemoryConfig()
 
-    static let videosStorageDiskConfig = DiskConfig(name: "videos")
-    static let vidoesStorageMemoryConfig = MemoryConfig()
-
-    let videosStorage = try! Storage<String, JSON>(
-        diskConfig: Self.videosStorageDiskConfig,
-        memoryConfig: Self.vidoesStorageMemoryConfig,
-        transformer: Self.jsonTransformer
+    let storage = try! Storage<String, JSON>(
+        diskConfig: Self.diskConfig,
+        memoryConfig: Self.memoryConfig,
+        transformer: CacheModel.jsonTransformer
     )
 
     func storeVideo(_ video: Video) {
         logger.info("caching \(video.cacheKey)")
-        try? videosStorage.setObject(video.json, forKey: video.cacheKey)
+        try? storage.setObject(video.json, forKey: video.cacheKey)
     }
 
     func retrieveVideo(_ cacheKey: String) -> Video? {
         logger.info("retrieving cache for \(cacheKey)")
 
-        if let json = try? videosStorage.object(forKey: cacheKey) {
+        if let json = try? storage.object(forKey: cacheKey) {
             return Video.from(json)
         }
 
         return nil
     }
+
+    func clear() {
+        try? storage.removeAll()
+    }
 }