TSLiveWallpaper/TSLiveWallpaper/Data/TSDBManager/TSDBManager.swift

//
//  TSDBManager.swift
//  TSLiveWallpaper
//
//  Created by 100Years on 2025/6/10.
//


import RealmSwift
import ObjectMapper

// 1. 定义历史记录类型枚举
enum TSDBHistoryType: String,CaseIterable {
    case aiList = "kDBAIList"              //旧照片修复
}


//MARK: TSDBAIChatList - 用于存储会话及消息列表
class TSDBHistory: Object {
    @Persisted(primaryKey: true) var primaryKey: String = ""
    @Persisted var listModels = List<TSDBActionInfoModel>()
    
    var type: TSDBHistoryType {
        get { TSDBHistoryType(rawValue: primaryKey) ?? .aiList }
        set { primaryKey = newValue.rawValue }
    }
    
    convenience init(type: TSDBHistoryType) {
        self.init()
        self.primaryKey = type.rawValue
    }
    

    func getModelList() -> [TSActionInfoModel] {
        var msgModel:[TSActionInfoModel] = []
        for msgDBModel in listModels {
            msgModel.append(msgDBModel.getModel())
        }
        return msgModel
    }
    
    
    func getModelList(completion:@escaping ([TSActionInfoModel])->Void){
        let frozenList = self.listModels.freeze()
        DispatchQueue.global(qos: .userInitiated).async {
            var msgModel:[TSActionInfoModel] = []
            for msgDBModel in frozenList {
                msgModel.append(msgDBModel.getModel())
            }
            DispatchQueue.main.async {
                completion(msgModel)
            }
        }
    }
    
    func getModelList(count:Int) -> [TSActionInfoModel] {
        let listModels = Array(listModels.prefix(count))
        var msgModel:[TSActionInfoModel] = []
        for msgDBModel in listModels {
            msgModel.append(msgDBModel.getModel())
        }
        return msgModel
    }
    
    func deleteAll() {
        TSRMShared.writeThread {
            listModels.removeAll()
        }
    }
    
    func deleteListModel(id:Int) {
        TSRMShared.writeThread {
            if let index = listModels.firstIndex(where: { $0.id == id }) {
                listModels.remove(at: index)
                debugPrint("listModels.remove(at: \(index))")
            }
        }
    }
    
    func deleteListModel(index:Int) {
        TSRMShared.writeThread {
            listModels.remove(at: index)
            debugPrint("listModels.remove(at: \(index))")
        }
    }

    
    func updateData(_ actionInfoModel:TSActionInfoModel,id:Int? = nil){
        let dbModel = TSDBActionInfoModel.createDBModel(actionInfoModel: actionInfoModel)
            var replaceID = dbModel.id
            if let id = id {
                replaceID = id
            }
            
            let frozenList = self.listModels.freeze()
            if let index = frozenList.firstIndex(where: { $0.id == replaceID }) {
                TSRMShared.writeThread {
                    listModels[index] = dbModel// 如果找到，替换该元素
                }
            } else {
                print("Thread.current insert1=\(Thread.current)")
                TSRMShared.writeThread {
                    listModels.insert(dbModel, at: 0)// 如果没有找到，添加到末尾
                    print("Thread.current insert2=\(Thread.current)")
                }
            }
    }
    
    func updateDatas(_ actionInfoModels:[TSActionInfoModel]){
        for actionInfoModel in actionInfoModels {
            updateData(actionInfoModel)
        }
    }

    
    func addDatas(_ actionInfoModels:[TSActionInfoModel]){
        for actionInfoModel in actionInfoModels {
            let dbModel = TSDBActionInfoModel.createDBModel(actionInfoModel: actionInfoModel)
            TSRMShared.writeThread {
                listModels.append(dbModel)
            }
        }
    }
}

extension TSRealmManager {
    func getDBHistory(type:TSDBHistoryType) -> TSDBHistory {
        let predicate = NSPredicate(format: "primaryKey == %@",type.rawValue)
        let historys = TSRMShared.realm.objects(TSDBHistory.self).filter(predicate)
        if let history = historys.first {
            return history
        }else {
            let dbHistory = TSDBHistory(type: type)
            TSRMShared.update(dbHistory)
            return dbHistory
        }
    }
    
    func createExampleModel(id:Int,oldImageName:String,newImageName:String)->TSActionInfoModel{
        let model = TSActionInfoModel()
        model.id = id
        model.modelType = .example
        model.request.prompt = "Example"
        model.request.inputText = "Example"
        model.request.width = 330
        model.request.height = 440
        model.request.imageUrl = oldImageName
        model.response.resultUrl = newImageName
        model.status = "success"
        return model
    }
    
    //老照片修复
    var aiListDB:TSDBHistory {
        let history = getDBHistory(type: TSDBHistoryType.aiList)
        if history.listModels.count == 0,UserDefaults.standard.string(forKey: "insertAIListExampleData") == nil {
            let id = Date.timestampInt
            history.updateDatas([
                createExampleModel(id:id+1, oldImageName: "ailist_example_image_old_1", newImageName: "ailist_example_image_new_1"),
                createExampleModel(id:id+2, oldImageName: "ailist_example_image_old_0", newImageName: "ailist_example_image_new_0")
            ])
            UserDefaults.standard.set("1", forKey: "insertAIListExampleData")
            UserDefaults.standard.synchronize()
        }
        
        return history
    }
    
}