AIEmoji/AIEmoji/Business/TSTextGeneralPictureVC/TSTextPicGennerateVC/TSTextPicGennerateVC.swift

//
//  TSTextPicGennerateVC.swift
//  AIEmoji
//
//  Created by 100Years on 2025/2/20.
//

class TSTextPicGennerateVC: TSAIPhotoGeneratorBaseVC {
    
    var imageModel:TSGenmojiModel?
    var complete:((TSGenmojiModel)->Void)
    var aiText:String
    var style:String
    var progressState = TSProgressState.none
    init(aiText: String,style: String,complete:@escaping ((TSGenmojiModel)->Void)) {
        self.aiText = aiText
        self.style = style
        self.complete = complete
        super.init()
    }
    
    var viewModel:TSTextPicGennerateVM = TSTextPicGennerateVM()
    @MainActor required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }

    lazy var generateInView : TSGeneratorloadingView = {
        let generateInView = TSGeneratorloadingView()
        return generateInView
    }()
    

    override func createView() {

        contentView.addSubview(generateInView)
        generateInView.snp.makeConstraints { make in
            make.edges.equalToSuperview()
        }
        
        super.createView()

    }
    
    override func closePage() {
        if progressState.isResult {
            viewModel.cancelAllRequest()
            self.dismiss(animated: true, completion: nil)
        }else{
            TSCustomAlertController.show(in: self, config: TSCustomAlertController.AlertConfig(
                message: "As you leave, your generation 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")
                    viewModel.cancelAllRequest()
                    self.dismiss(animated: true, completion: nil)
                },
                confirmAction: {
                    print("用户点击了Stay")
                }
            ))
        }
    }
    
    
    //重试
    @objc override func clickTryAgainBtn(){
        clickRegenerateBtn()
    }
    
    //重新生成
    @objc override func clickRegenerateBtn(){
        //判断 vip
        if kJudgeVip(externalBool: kPurchaseDefault.freeNumAvailable(type: .textGeneratePic) == false, vc: self) {[weak self] in
            guard let self = self else { return }
        }{ return }
        
        viewModel.creatImageEmoji(text:aiText,style: style)
    }
    

    //保存功能
    @objc override func clickSaveBtn(){
        if let image = getSuccessImage() {
            PhotoManagerShared.saveImageToAlbum(image) { success, error in
                if success {
                    kSavePhotoSuccesswShared.show(atView:self.view)
                }else{
                    debugPrint(error)
                }
            }
        }
    }
    
    override func dealThings() {
        viewModel.creatImageEmoji(text: self.aiText,style: style)
        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)
    }
    
}
extension TSTextPicGennerateVC {
    
    
    func getSuccessImage()->UIImage?{
        if let image = netWorkImageView.image {
            return image.pngImage
        }
        return nil
    }
    
}
extension TSTextPicGennerateVC {
    func upDateView(state:TSProgressState,model:TSGenmojiModel?){
        progressState = state
        switch state {
            case .failed(let errorStr):
                showError(text: errorStr)
            case .success:
                if let model = model {
                    showSuccess(model: model)
                }else{
                    showError(text: nil)
                }
            case .progressString(let string):
                showProgress(text: string)
            default:
                showLoading()
        }
    }
    func showProgress(text:String) {
        generateInView.isHidden = false
        generateInView.showLoading(text: text)
        generateInView.isRotating = true

        isClickTheBlankClosePage = false
        bottomView.isHidden = true
        netWorkImageView.isHidden = true
    }
    
    func showLoading(){
        generateInView.isHidden = false
        generateInView.showLoading(text: "Generating".localized + " ...".localized)
        generateInView.isRotating = true
        
        isClickTheBlankClosePage = false
        bottomView.isHidden = true
        netWorkImageView.isHidden = true
    }
    
    func showError(text:String?){
        let msg = "Failed to generate, please try later".localized
        generateInView.isHidden = false
        generateInView.showError(text: msg)
        generateInView.isRotating = false
        
        isClickTheBlankClosePage = true
        
        tryAgainBtn.isHidden = false
        bigSaveBtn.isHidden = true
        bottomView.isHidden = false
        netWorkImageView.isHidden = true
    }
    
    func showSuccess(model:TSGenmojiModel){
        generateInView.isHidden = true
        generateInView.isRotating = false

        imageModel = model
        isClickTheBlankClosePage = true
        
        tryAgainBtn.isHidden = false
        bigSaveBtn.isHidden = false
        bottomView.isHidden = false
        netWorkImageView.isHidden = false
        
        
        self.netWorkImageView.setAsyncImage(urlString: model.response.resultUrl,placeholder:kPlaceholderImage,backgroundColor:netWorkImageView.backgroundColor!){ [weak self] image in
            guard let self = self else { return }
            
            netWorkImageView.adaptiveScale()
        }
        
        kPurchaseDefault.useOnceForFree(type: .textGeneratePic)
        
        
        if let model = imageModel {
            complete(model)
        }
    }
}