yattee/Model/Cache/ChannelPlaylistsCacheModel.swift

import Cache
import Foundation
import Logging
import SwiftyJSON

struct ChannelPlaylistsCacheModel: CacheModel {
    static let shared = ChannelPlaylistsCacheModel()
    let logger = Logger(label: "stream.yattee.cache.channel-playlists")

    static let diskConfig = DiskConfig(name: "channel-playlists")
    static let memoryConfig = MemoryConfig()

    var storage = try? Storage<String, JSON>(
        diskConfig: Self.diskConfig,
        memoryConfig: Self.memoryConfig,
        transformer: BaseCacheModel.jsonTransformer
    )

    func storePlaylist(playlist: ChannelPlaylist) {
        let date = iso8601DateFormatter.string(from: Date())
        logger.info("STORE \(playlistCacheKey(playlist.id)) -- \(date)")
        let feedTimeObject: JSON = ["date": date]
        let playlistObject: JSON = ["playlist": playlist.json.object]
        try? storage?.setObject(feedTimeObject, forKey: playlistTimeCacheKey(playlist.id))
        try? storage?.setObject(playlistObject, forKey: playlistCacheKey(playlist.id))
    }

    func retrievePlaylist(_ id: ChannelPlaylist.ID) -> ChannelPlaylist? {
        logger.info("RETRIEVE \(playlistCacheKey(id))")

        if let json = try? storage?.object(forKey: playlistCacheKey(id)).dictionaryValue["playlist"] {
            return ChannelPlaylist.from(json)
        }

        return nil
    }

    func getPlaylistsTime(_ id: ChannelPlaylist.ID) -> Date? {
        if let json = try? storage?.object(forKey: playlistTimeCacheKey(id)),
           let string = json.dictionaryValue["date"]?.string,
           let date = iso8601DateFormatter.date(from: string)
        {
            return date
        }

        return nil
    }

    func getFormattedPlaylistTime(_ id: ChannelPlaylist.ID) -> String {
        getFormattedDate(getPlaylistsTime(id))
    }

    private func playlistCacheKey(_ playlist: ChannelPlaylist.ID) -> String {
        "channelplaylists-\(playlist)"
    }

    private func playlistTimeCacheKey(_ playlist: ChannelPlaylist.ID) -> String {
        "\(playlistCacheKey(playlist))-time"
    }
}