AIEmoji/AIEmoji/Business/TSAILIstVC/TSAIUploadPhotoBaseVC/TSAIUploadPhotoBaseVC.swift

//
//  TSAIUploadPhotoBaseVC.swift
//  AIEmoji
//
//  Created by 100Years on 2025/4/9.
//

import PhotosUI
enum TSGeneratorImageStyle {
    case ageOld      //变老
    case ageChild   //变年轻
    case oldPhoto   //旧照片修复
    case eyeOpen    //开眼
    case pretty     //变美
    case photoLive     //把照片变活
}

class TSAIUploadPhotoBaseVC: TSBaseVC {
    var generatorStyle:TSGeneratorImageStyle
    var titleString:String
    var imageMaxBitSize:Int
    init(titleString:String,upLoadImage:UIImage,imageMaxBitSize:Int,generatorStyle:TSGeneratorImageStyle) {
        self.titleString = titleString
        self.upLoadImage = upLoadImage
        self.imageMaxBitSize = imageMaxBitSize
        self.generatorStyle = generatorStyle
        super.init()
    }
    
    @MainActor required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    //#####################################数据区#####################################
    let lineSpacing = 6.0
    lazy var photoPickerManager: TSPhotoPickerManager = {
        let photoPickerManager = TSPhotoPickerManager(viewController: self)
        return photoPickerManager
    }()
    //#####################################底部基础#####################################
    lazy var cusStackView: TSCustomStackView = {
        let cusStackView = TSCustomStackView(axis: .vertical,spacing: 0)
        return cusStackView
    }()

    
    lazy var submitBtn: UIButton = {
        let submitBtn = kCreateNormalSubmitBtn(title: generatingText) { [weak self]  in
            guard let self = self else { return }
            generateImage()
        }
        submitBtn.cornerRadius = 24.0
        return submitBtn
    }()

    //#####################################上传 view#####################################
    var upLoadImage:UIImage? = nil {
        didSet{
            if let image = upLoadImage {
                upLoadView.isHidden = true
                uploadImageView.isHidden = false
                deleteBtn.isHidden = false
                uploadImageView.image = image
                submitBtn.isEnabled = true
            }else {
                upLoadView.isHidden = false
                uploadImageView.isHidden = true
                deleteBtn.isHidden = true
                uploadImageView.image = nil
                submitBtn.isEnabled = false
            }
        }
    }
    
    lazy var upLoadView: UIView = {
        let bgView = UIView()
        
        let addImageView = UIImageView.createImageView(imageName: "add")
        bgView.addSubview(addImageView)
        addImageView.snp.makeConstraints { make in
            make.top.equalTo(0)
            make.centerX.equalToSuperview()
            make.width.height.equalTo(24)
        }
        
        let textLabel = UILabel.createLabel(text: "Upload Photo".localized,font: .font(size: 16),textColor: .white)
        bgView.addSubview(textLabel)
        textLabel.snp.makeConstraints { make in
            make.top.equalTo(addImageView.snp.bottom).offset(16)
            make.centerX.equalToSuperview()
            make.bottom.equalToSuperview()
        }
        
        return bgView
    }()

    lazy var uploadImageView: UIImageView = {
        let uploadImageView = UIImageView()
        uploadImageView.contentMode = .scaleAspectFit
        uploadImageView.cornerRadius = 12
        return uploadImageView
    }()
    
    
    lazy var uploadImageBgView: UIView = {
        
        let bgView = UIView()
        bgView.addGestureRecognizer(UITapGestureRecognizer(target: self, action:#selector(clickBgView)))
        let bgImageView = UIImageView.createImageView(imageName: "ailist_upload_big_bg")
        bgImageView.contentMode = .scaleToFill
        bgView.addSubview(bgImageView)
        bgImageView.snp.makeConstraints { make in
            make.leading.bottom.trailing.top.equalTo(0)
        }

        bgView.addSubview(upLoadView)
        upLoadView.snp.makeConstraints { make in
            make.top.equalTo(244*kDesignScale)
            make.centerX.equalToSuperview()
        }

        bgView.addSubview(uploadImageView)
        uploadImageView.snp.makeConstraints { make in
            make.top.leading.equalTo(8)
            make.bottom.trailing.equalTo(-8)
        }
        
        return bgView
    }()
    
    
    lazy var deleteBtn: UIButton = {
        let deleteBtn = UIButton.createButton(backgroundImage: UIImage(named: "delete_redRound")) { [weak self]  in
            guard let self = self else { return }
            upLoadImage = nil
        }
        deleteBtn.isHidden = true
        return deleteBtn
    }()
    
    
    override func createView() {
        
        addNormalNavBarView()
        setPageTitle(titleString)
        _ = setNavigationItem("", imageName: "aichat_history", direction: .right, action: #selector(clickNavRight))
        
        contentView.addSubview(submitBtn)
        submitBtn.snp.makeConstraints { make in
            make.centerX.equalToSuperview()
            make.width.equalTo(250*kDesignScale)
            make.height.equalTo(48)
            make.bottom.equalTo(-10-k_Height_safeAreaInsetsBottom())
        }
        
        contentView.addSubview(cusStackView)
        cusStackView.snp.makeConstraints { make in
            make.top.equalTo(0)
            make.leading.trailing.equalToSuperview()
            make.bottom.equalTo(submitBtn.snp.top).offset(-10)
        }
        
        setUpStackView()
        
        let image = upLoadImage
        upLoadImage = image
    }

    
    override func dealThings() {
        kSetBtnVipIcon(btn: self.submitBtn, show: true)
    }
    
}

extension TSAIUploadPhotoBaseVC {
    
    func setUpStackView(){
        //添加上传一大块
        let bgView = UIView()
        bgView.addSubview(uploadImageBgView)
        cusStackView.addSubviewToStack(bgView)
    
        uploadImageBgView.snp.makeConstraints { make in
            make.width.equalTo(296*kDesignScale)
            make.height.equalTo(528*kDesignScale)
            make.centerX.equalToSuperview()
            make.top.equalTo(8)
            make.bottom.equalTo(-16)
        }

        bgView.addSubview(deleteBtn)
        deleteBtn.snp.makeConstraints { make in
            make.top.equalTo(0)
            make.trailing.equalTo(-32*kDesignScale)
            make.width.height.equalTo(32)
        }
        
        
        //文字信息区域
        cusStackView.addSubviewToStack(getTextInfoCell(text: "Single photo with face fully visible".localized))
        cusStackView.addSpacing(height: lineSpacing)
        cusStackView.addSubviewToStack(getTextInfoCell(text: "No group photos, covered faces, nudes".localized))
    }
    
    func getTextInfoCell(text:String) -> UIView {
        
        let bgView = UIView()
        let pointView = UIView()
        pointView.backgroundColor = .white.withAlphaComponent(0.7)
        pointView.cornerRadius = 1
        bgView.addSubview(pointView)
        pointView.snp.makeConstraints { make in
            make.top.equalTo(7)
            make.leading.equalTo(22)
            make.width.height.equalTo(2)
        }
        let textLabel1 = UILabel.createLabel(text: text,font: .font(size: 12),textColor: .white.withAlphaComponent(0.6),numberOfLines: 0)
        textLabel1.setLineSpacing(lineSpacing)
        bgView.addSubview(textLabel1)
        textLabel1.snp.makeConstraints { make in
            make.top.bottom.equalToSuperview()
            make.leading.equalTo(16+16)
            make.trailing.equalTo(-16)
        }
        return bgView
    }
    
    var generatingText:String{
        if generatorStyle == .oldPhoto {
            return "Restore".localized
        }else
        if generatorStyle == .pretty {
            return "Change".localized
        }
        return "Generate".localized
    }
}

extension TSAIUploadPhotoBaseVC {

    @objc func clickNavRight() {
        kPushVC(target: self, modelVC: TSAIListHistoryBaseVC(generatorStyle: generatorStyle))
    }
    
    @objc func clickBgView() {
        pickSinglePhoto()
    }
    
    func pickSinglePhoto()  {
//        photoPickerManager.pickSinglePhoto(maxBitSize: imageMaxBitSize) { [weak self] image, errorString in
        photoPickerManager.pickCustomSinglePhoto() { [weak self] image, errorString in
            guard let self = self else { return }
            if let errorString = errorString {
                TSToastShared.showToast(text: errorString)
            }else{
                upLoadImage = image
            }
        }
    }
}


extension TSAIUploadPhotoBaseVC {
    
    func generateImage() {
        if kJudgeVip(externalBool: true, vc: self){ return } //判断 vip
        
        guard let upLoadImage = upLoadImage else { return }
        let gennerateVC = TSAIListPhotoGeneratorBaseVC(upLoadImage: upLoadImage, generatorStyle: generatorStyle) { [weak self] model in
            guard let self = self else { return }
            saveModel(model: model)
        }
        
        kPresentModalVC(target: self, modelVC: gennerateVC,transitionStyle: .crossDissolve)
    }
    
    func saveModel(model:TSActionInfoModel){
        switch generatorStyle {
        case .ageOld:
            TSChangeOldAgeHistory.shared.saveModel(model: model)
        case .ageChild:
            TSChangeBabyAgeHistory.shared.saveModel(model: model)
        case .oldPhoto:
            TSChangeOldPhotoHistory.shared.saveModel(model: model)
        case .eyeOpen:
            TSAIEyeOpenHistory.shared.saveModel(model: model)
        case .pretty:
            TSAIPhotoPrettyHistory.shared.saveModel(model: model)
        case .photoLive:
            TSAIPhotoLiveHistory.shared.saveModel(model: model)
        }
    }

}