AIRingtone/AIRingtone/Common/Tool/OperationQueue/TSGenerateBaseOperation/TSGenerateBaseOperation.swift

//
//  TSGenerateBaseOperation.swift
//  AIRingtone
//
//  Created by 100Years on 2025/3/24.
//

import Combine
import Alamofire
class TSGenerateBaseOperationQueue: TSBaseOperationQueue {
    // 存储每个操作的 AnyCancellable
    var stateables: [String: AnyCancellable] = [:]
    
    var generateOperationStateChanged:((String)->Void)?
    
    
    override func cancelOperations(uuid: String) {
        super.cancelOperations(uuid: uuid)
        stateables.removeValue(forKey: uuid)
    }

    func handleStateDatauPblished(uuid:String,generateOperation: TSGenerateBaseOperation,notificationName:Notification.Name) {
        stateables[uuid] = generateOperation.$stateDatauPblished.sink { [weak self] state in
            guard let self = self else { return }
            DispatchQueue.main.async {
                self.generateOperationStateChanged?(uuid)
                
                let uuidData = self.getUUIDData(uuid: uuid)
                NotificationCenter.default.post(
                    name: notificationName,
                    object: nil,
                    userInfo: [
                        "uuid": uuid,
                        "count":self.queue.maxConcurrentOperationCount,
                        "state":uuidData.0,
                        "actionInfo":uuidData.1,
                    ])
            }
        }
    }
    
    func getUUIDData(uuid:String)->(TSProgressState,TSActionInfoModel?){
        return (.none,TSActionInfoModel())
    }
    
}

class TSGenerateBaseOperation: TSBaseOperation , @unchecked Sendable{
    
    var actionInfoDict:[String:Any]{
        return [
            "actionType":"music_create",
            "comments": "Success",
            "costTime":15,
            "createdTimestamp":1741338454,
            "id":1536,
            "percent":1,
            "request":"{\"prompt\": \"Create a Techno ringtone with a repetitive bassline, crisp hi-hats, and subtle synth textures. Use a BPM of 125-130 for a sleek, modern sound., Create a uplifting and modern music track blending Pop, Electronic, and Ambient elements. Use a BPM of 100-120, a catchy melody with synth or piano, warm harmonies, and a mix of electronic and organic sounds. Ensure a clear structure (Intro, Verse, Chorus, Outro) and a light, positive vibe suitable for background or casual listening\", \"duration\": 5}",
            "response":"{\"coverUrl\": \"https://be-aigc.s3-accelerate.amazonaws.com/f0fb7739-a5cc-4805-9b68-b4a5890eb285.png\", \"title\": \"Neon Pulse\\\"  \\n\\\"Horizon Glow\", \"musicUrl\": \"https://be-aigc.s3-accelerate.amazonaws.com/c47d40dd-d07c-4edc-a6d9-8382438149d1.wav\"}",
            "status":"success"
        ]
    }
    
    @Published var stateDatauPblished:(TSProgressState,TSActionInfoModel?) = (TSProgressState.none,nil){
        didSet{
            dePrint("TSBaseOperation stateDatauPblished didSet = \(stateDatauPblished)")
            if case .start = stateDatauPblished.0 {
                start()
            }else if stateDatauPblished.0.isResult {
                finished()
            }
        }
    }
    
    var creatRequest:Request?
    var queryRequest:Request?
    var stopNetwork = false
    var generatingProgress = 0
    var action_id:Int = 0
    
    var currentActionInfoModelChanged:((TSActionInfoModel)->Void)?
    @Published var currentActionInfoModel: TSActionInfoModel = TSActionInfoModel()
    
    func getActionInfo(action_id:Int){
        self.action_id = action_id
        queryRequest = TSNetworkShared.get(urlType: .actionInfo,parameters: ["action_id":action_id]) { [weak self] data,error in
            guard let self = self else { return }
            if let result = kNetWorkResultSuccess(data: data) {
                if let genmojiModel = TSActionInfoModel(JSON: result) {
                    switch genmojiModel.actionStatus {
                    case .success:
                        TSToastShared.hideLoading()
                        self.stateDatauPblished = (.success(nil),genmojiModel)
                        generatingProgress = 0
                    case .failed:
                        self.stateDatauPblished = (.failed(kNetWorkMessage(data: data) ?? ""),nil)
                        generatingProgress = 0
                    default:
                        stateDatauPblished = (.progressString(generating(progress: genmojiModel.percent)),nil)
                        if stopNetwork == false {
                            kDelayOnMainThread(1.0) {
                                self.getActionInfo(action_id: action_id)
                            }
                        }
                    }
                }
            }else{
                self.stateDatauPblished = (.failed(error?.localizedDescription ?? ""),nil)
            }
        }
    }
     func generating(progress:Float) -> String {

         //Generating 0%-100%
         var progressInt = Int(progress*100)

         if generatingProgress >= progressInt{
             return getGeneratingProgressText()
         }

         if progressInt > 99 {
             progressInt = 99
         }
         
         generatingProgress = progressInt
         return getGeneratingProgressText()
     }
     
     
    func getGeneratingProgressText()->String{
        return "Generating \(generatingProgress)%"
    }
     
     func cancelAllRequest(){
         creatRequest?.cancel()
         queryRequest?.cancel()
         stopNetwork = true
         
         cancel()
     }
     
}