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

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

import Kingfisher
struct 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
    }
}

class TSAIListPhotoGeneratorVC: TSAIPhotoDetailsVC {

    init(generatorModel:TSAIListPhotoGeneratorModel,complete:@escaping ((TSActionInfoModel)->Void)) {
        self.complete = complete
        self.viewModel = TSAIListPhotoGeneratorBaseVM(generatorModel: generatorModel)
        super.init()
    }
    @MainActor required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    var complete:((TSActionInfoModel)->Void)
    var progressState = TSProgressState.none
    
    var isSavePhotoMark:Bool = false
    var isNeedSavePhoto:Bool {
        if isSavePhotoMark == false,let _ = imageModel {
            return true
        }
        return false
    }
    
    var viewModel: TSAIListPhotoGeneratorBaseVM!

    lazy var generateInView : TSGeneratorView = {
        let generateInView = TSGeneratorView()
        generateInView.isUploadImage = false
        generateInView.animationView.setText(time: String(format: "～ %d min".localized, 2), info: "Lots of people are creating 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.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()
        }
    }

    
    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")
                        viewModel.cancelAllRequest()
                        self.dismiss(animated: true, completion: nil)
                    },
                    confirmAction: { [weak self]  in
                        guard let self = self else { return }
                        print("用户点击了Stay")
                        clickSaveBtn()
                    }
                ))
            }else{
                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 func clickRegenerateBtn(){
        //判断 vip
        if kJudgeVip(externalBool: true , vc: self){ return }
        
        viewModel.uploadAndCreatImage()
    }

    override func dealThings() {
        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)
    }
    
}

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: "Generating".localized + " ...".localized)

    }
    
    func showError(text:String,code:Int = 0){
        generateInView.updateShowError(text: text,code: code)
    }
    
    func showSuccess(model:TSActionInfoModel){
        generateInView.updateShowSuccess()

        imageModel = model
//        if viewModel.generatorModel.generatorStyle == .futureBaby {
//            rotatingPictureBtn.isHidden = false
//        }else{
//            switchOriginalPictureBtn.isHidden = false
//        }
        if let model = imageModel {
            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 = imageModel {
//                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: imageModel?.response.resultUrl,placeholder:kPlaceholderImage,backgroundColor:netWorkImageView.backgroundColor!)
//            expandAreaView.showImageView.image = self.viewModel.generatorModel.upLoadImage
//        }
    }
}