TSLiveWallpaper/TSLiveWallpaper/Business/TSAIListVC/TSAIPhotoGeneratorVC/TSAIListPhotoGeneratorVC.swift

//
//  TSAIPhotoGeneratorVC.swift
//  TSLiveWallpaper
//
//  Created by 100Years on 2025/6/11.
//

import Kingfisher
class TSAIListPhotoGeneratorModel {
    var upLoadImage:UIImage
    var generatorStyle:TSGeneratorImageStyle
    var expandEdge:UIEdgeInsets
    var expandViewSizes:(CGSize,CGSize)?
    var additionalPrompt:String//追加的提示词
    
    //预测宝宝
    var upLoadImages:[UIImage]?
    
    init(upLoadImage: UIImage,
         generatorStyle: TSGeneratorImageStyle,
         expandEdge:UIEdgeInsets = UIEdgeInsets(top: 0, left: 0, bottom: 0, right: 0),
         expandViewSizes:(CGSize,CGSize)? = nil,
         additionalPrompt:String = "",
         upLoadImages:[UIImage]? = nil
    ) {
        self.upLoadImage = upLoadImage
        self.generatorStyle = generatorStyle
        self.expandEdge = expandEdge
        self.expandViewSizes = expandViewSizes
        self.additionalPrompt = additionalPrompt
        self.upLoadImages = upLoadImages
    }
    
    var prompt:String = ""
    var inputText:String = ""
    var upLoadImageUrl:String?
    var model:String = ""   //决定生图的模型
}

class TSAIListPhotoGeneratorVC: TSAIPhotoDetailsVC {
    var generatorModel:TSAIListPhotoGeneratorModel
    init(generatorModel:TSAIListPhotoGeneratorModel,infoModel:TSActionInfoModel? = nil,complete:@escaping ((TSActionInfoModel)->Void)) {
        self.complete = complete
        self.generatorModel = generatorModel
//        self.viewModel = TSAIListPhotoGeneratorBaseVM(generatorModel: generatorModel)
        super.init()
        self.infoModel = infoModel
    }
    @MainActor required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    
    
    var complete:((TSActionInfoModel)->Void)
    var backstageBlock:(()->Void)?
    
    
    var progressState = TSProgressState.none
    
    var uuidString:String = UUID().uuidString
    var operation:TSGenerateBaseOperation?

    var isSavePhotoMark:Bool = false
    var isNeedSavePhoto:Bool {
        if isSavePhotoMark == false,let _ = infoModel {
            return true
        }
        return false
    }
    
//    var viewModel: TSAIListPhotoGeneratorBaseVM!

    lazy var generateInView : TSGeneratorView = {
        let generateInView = TSGeneratorView()
        generateInView.isUploadImage = false
        generateInView.animationView.setText(time: String(format: "～ %d mins".localized, 2), info: "Lots of people are processing images right now, so this might take a bit.".localized)
        generateInView.clickErrorBlock = { [weak self] style in
            guard let self = self else { return }
            
            switch style {
            case .netWorkError:
                clickRegenerateBtn()
            default:
                self.dismiss(animated: false, completion: nil)
                break
            }
        }
        generateInView.clickBackstageBlock = { [weak self]  in
            guard let self = self else { return }
            clickBackstageBtn()
        }
        generateInView.clickCloseBlock = { [weak self]  in
            guard let self = self else { return }
            closePage()
        }
        return generateInView
    }()

    override func createView() {
        
        super.createView()
        navRightBtn.isHidden = true
        view.addSubview(generateInView)
        generateInView.snp.makeConstraints { make in
            make.edges.equalToSuperview()
        }
    }

    @objc func clickBackstageBtn() {
        self.operation?.backstageGeneration()
        NotificationCenter.default.post(name: .kAIPhotoDataChanged, object: nil)
        backstageBlock?()
        self.dismiss(animated: true, completion: nil)
    }
    
    func closePage() {
    
        if progressState.isResult {
            if isNeedSavePhoto{
                TSCustomAlertController.show(in: self, config: TSCustomAlertController.AlertConfig(
                    message: "You haven't saved the photo yet. Are you sure to quit?".localized,
                    messageColor: .white,
                    messageFont: .systemFont(ofSize: 16),
                    
                    cancelTitle: "Quit".localized,
                    cancelColor: .white,
                    
                    confirmTitle: "Save".localized,
                    confirmColor: .themeColor,
                    
                    cancelAction: { [weak self]  in
                        guard let self = self else { return }
                        print("用户点击了Leave")
                        
                        if let model = infoModel{
                            TSRMShared.aiListDB.deleteListModel(id: model.id)
                            NotificationCenter.default.post(name: .kAIPhotoDataChanged, object: nil)
                        }
            
                        cancelDidmiss()
                    },
                    confirmAction: { [weak self]  in
                        guard let self = self else { return }
                        print("用户点击了Stay")
                        clickSaveBtn()
                    }
                ))
            }else{
                cancelDidmiss()
            }
        }else{
            TSCustomAlertController.show(in: self, config: TSCustomAlertController.AlertConfig(
                message: "As you leave, your task will be interrupted and no result.".localized,
                messageColor: .white,
                messageFont: .systemFont(ofSize: 16),
                
                cancelTitle: "Leave".localized,
                cancelColor: .white,
                
                confirmTitle: "Wait".localized,
                confirmColor: .themeColor,
                
                cancelAction: { [weak self]  in
                    guard let self = self else { return }
                    print("用户点击了Leave")
                    cancelDidmiss()
                },
                confirmAction: {
                    print("用户点击了Stay")
                }
            ))
        }
    }

    func cancelDidmiss(){
//        viewModel.cancelAllRequest()
        operation?.cancel()
        self.dismiss(animated: true, completion: nil)
    }
    //重新生成
    @objc func clickRegenerateBtn(){
        //判断 vip
        if kJudgeVip(externalBool: true , vc: self){ return }
        
//        viewModel.uploadAndCreatImage()
        generatorOperation()
    }

    override func dealThings() {
        self.uuidString = UUID().uuidString
//        viewModel.uploadAndCreatImage()
//        viewModel.$stateDatauPblished.receive(on: DispatchQueue.main).sink {[weak self]  (state,model) in
//            guard let self = self else { return }
//            self.upDateView(state: state, model: model)
//        }.store(in: &cancellable)
        generatorOperation()
    }
    
    func generatorOperation() {
        
        if let model = self.infoModel{
            if model.actionStatus == .failed{
                generatorCreatRing(oldModel: model)
            }else if model.response.resultUrl.count > 0 {
                upDateView(state: .success(model), model: model)
                return
            }
        }else{
            generatorNew()
        }
    }
}
extension TSAIListPhotoGeneratorVC{
    
    func generatorNew() {
        let operation:TSGenerateBasePhotoOperation = TSGenerateBasePhotoOperationQueue.shared.creatOperation(uuid: self.uuidString)
        self.operation = operation
        operation.$stateDatauPblished.receive(on: DispatchQueue.main).sink {[weak self]  (state,model) in
            guard let self = self else { return }
            self.upDateView(state: state, model: model)
        }.store(in: &cancellable)
        
        operation.uploadImage(generateStyleModel: self.generatorModel)  { [weak self] actionInfoModel in
            guard let self = self else { return }
            guard let oldModel = actionInfoModel else {return}
            generatorCreatRing(oldModel: oldModel)
        }
    }
    
    func generatorCreatRing(oldModel:TSActionInfoModel) {
        let operation:TSGenerateBasePhotoOperation = TSGenerateBasePhotoOperationQueue.shared.creatOperation(uuid: self.uuidString)
        self.operation = operation
        operation.$stateDatauPblished.receive(on: DispatchQueue.main).sink {[weak self]  (state,model) in
            guard let self = self else { return }
            self.upDateView(state: state, model: model)
        }.store(in: &cancellable)
        
        operation.creatImage(oldModel: oldModel){ [weak self]  success in
            guard let self = self else { return }
            if success {
                generateInView.setBackgroundGenerateBtnHidden(false)
            }
        }
    }
}
extension TSAIListPhotoGeneratorVC {
    func getSuccessImage()->UIImage?{
        if let image = panComparisonView.newImageView.image {
            return image.pngImage
        }
        return nil
    }
}
extension TSAIListPhotoGeneratorVC {
    
    func upDateView(state:TSProgressState,model:TSActionInfoModel?){
        progressState = state
        switch state {
            case .failed(let errorStr,let code):
                showError(text: errorStr,code:code)
            case .success:
                if let model = model {
                    showSuccess(model: model)
                }else{
                    showError(text: "")
                }
            case .progressString(let string):
                showProgress(text: string)
            default:
                showLoading()
        }
    }
    
    func showProgress(text:String) {
        generateInView.updateShowProgress(text: text)
//        setVideoHidden()
    }
    
    func showLoading(){
        generateInView.updateShowLoading(text: "Processing".localized + " " + kPercentlocalized(0))
    }
    
    func showError(text:String,code:Int = 0){
        generateInView.updateShowError(text: text,code: code)
    }
    
    func showSuccess(model:TSActionInfoModel){
        generateInView.updateShowSuccess()

        infoModel = model
//        if viewModel.generatorModel.generatorStyle == .futureBaby {
//            rotatingPictureBtn.isHidden = false
//        }else{
//            switchOriginalPictureBtn.isHidden = false
//        }
        if let model = infoModel {
            complete(model)
        }
        
//        setVideoURL()
//        setExpandAreaImage()
    }
}

extension TSAIListPhotoGeneratorVC {
    func setVideoHidden(){
//        if viewModel.generatorModel.generatorStyle == .photoLive {
//            videoPlayerVC.removeFromParent()
//            videoPlayerVC.view.removeFromSuperview()
//        }
    }
    
    func setVideoURL(){
//        if viewModel.generatorModel.generatorStyle == .photoLive {
//            if let model = infoModel {
//                switchOriginalPictureBtn.isHidden = true
//                self.videoPlayerVC = TSAIListVideoPlayerVC(videoURL: model.videoURL)
//                self.addChild(self.videoPlayerVC)
//                self.videoPlayerVC.view.frame = self.netWorkImageView.bounds
//                self.netWorkImageView.addSubview(self.videoPlayerVC.view)
//                self.videoPlayerVC.setControlsBottom(bottem: -20)
//            }
//        }
    }
}
extension TSAIListPhotoGeneratorVC {
    
    func setExpandAreaImage(){
//        if viewModel.generatorModel.generatorStyle == .photoExpand {
//            netWorkImageView.image = nil
//            expandAreaView.bgImageView.setAsyncImage(urlString: infoModel?.response.resultUrl,placeholder:kPlaceholderImage,backgroundColor:netWorkImageView.backgroundColor!)
//            expandAreaView.showImageView.image = self.viewModel.generatorModel.upLoadImage
//        }
    }
}