Swiftgram/submodules/TelegramCore/Sources/TelegramEngine/ItemCache/TelegramEngineItemCache.swift

import Foundation
import SwiftSignalKit
import Postbox

public typealias EngineDataBuffer = ValueBoxKey

public extension TelegramEngine {
    final class ItemCache {
        private let account: Account

        init(account: Account) {
            self.account = account
        }

        public func put<T: Codable>(collectionId: Int8, id: EngineDataBuffer, item: T) -> Signal<Never, NoError> {
            return self.account.postbox.transaction { transaction -> Void in
                if let entry = CodableEntry(item) {
                    transaction.putItemCacheEntry(id: ItemCacheEntryId(collectionId: collectionId, key: id), entry: entry)
                }
            }
            |> ignoreValues
        }
        
        public func remove(collectionId: Int8, id: EngineDataBuffer) -> Signal<Never, NoError> {
            return self.account.postbox.transaction { transaction -> Void in
                transaction.removeItemCacheEntry(id: ItemCacheEntryId(collectionId: collectionId, key: id))
            }
            |> ignoreValues
        }
        
        public func clear(collectionIds: [Int8]) -> Signal<Never, NoError> {
            return self.account.postbox.transaction { transaction -> Void in
                for id in collectionIds {
                    transaction.clearItemCacheCollection(collectionId: id)
                }
            }
            |> ignoreValues
        }
    }
}

public extension TelegramEngineUnauthorized {
    final class ItemCache {
        private let account: UnauthorizedAccount

        init(account: UnauthorizedAccount) {
            self.account = account
        }

        public func put<T: Codable>(collectionId: Int8, id: EngineDataBuffer, item: T) -> Signal<Never, NoError> {
            return self.account.postbox.transaction { transaction -> Void in
                if let entry = CodableEntry(item) {
                    transaction.putItemCacheEntry(id: ItemCacheEntryId(collectionId: collectionId, key: id), entry: entry)
                }
            }
            |> ignoreValues
        }
        
        public func get(collectionId: Int8, id: EngineDataBuffer) -> Signal<CodableEntry?, NoError> {
            return self.account.postbox.transaction { transaction -> CodableEntry? in
                return transaction.retrieveItemCacheEntry(id: ItemCacheEntryId(collectionId: collectionId, key: id))
            }
        }
        
        public func remove(collectionId: Int8, id: EngineDataBuffer) -> Signal<Never, NoError> {
            return self.account.postbox.transaction { transaction -> Void in
                transaction.removeItemCacheEntry(id: ItemCacheEntryId(collectionId: collectionId, key: id))
            }
            |> ignoreValues
        }
        
        public func clear(collectionIds: [Int8]) -> Signal<Never, NoError> {
            return self.account.postbox.transaction { transaction -> Void in
                for id in collectionIds {
                    transaction.clearItemCacheCollection(collectionId: id)
                }
            }
            |> ignoreValues
        }
    }
}