407 lines
15 KiB
Swift
407 lines
15 KiB
Swift
//
|
||
// RolePublishController.swift
|
||
// Crush
|
||
//
|
||
// Created by Leon on 2025/7/20.
|
||
//
|
||
|
||
import UIKit
|
||
import Combine
|
||
import TZImagePickerController
|
||
|
||
/// 生成角色页面
|
||
/// 选参考图
|
||
class RolePublishController: RoleCreateBaseController<RolePublishView> {
|
||
weak var viewModel: RoleCreateViewModel!
|
||
|
||
// @Published var image: UIImage?
|
||
private var cancellables = Set<AnyCancellable>()
|
||
|
||
var isAllImagesOK: Bool = false
|
||
|
||
override func viewDidLoad() {
|
||
super.viewDidLoad()
|
||
|
||
// Do any additional setup after loading the view.
|
||
setupViews()
|
||
setupDatas()
|
||
setupEvents()
|
||
}
|
||
|
||
override func viewWillAppear(_ animated: Bool) {
|
||
super.viewWillAppear(animated)
|
||
disabledFullScreenPan()
|
||
}
|
||
|
||
override func viewWillDisappear(_ animated: Bool) {
|
||
super.viewWillDisappear(animated)
|
||
enabledFullScreenPan()
|
||
}
|
||
|
||
private func setupViews() {
|
||
|
||
container.container.scrollView.delegate = self
|
||
}
|
||
|
||
private func setupDatas(){
|
||
// if let editAIInfo = viewModel.editAIInfo, let url = editAIInfo.imageUrl{
|
||
// var usingImage = AIUserImageQuery()
|
||
// usingImage.imageUrl = url
|
||
// usingImage.status = .completed
|
||
// container.appearanceResult.usingFigureImage = usingImage
|
||
// }
|
||
if let editAIInfo = viewModel.editAIInfo{
|
||
viewModel.requestParams.aiUserExt?.imageDesc = editAIInfo.aiUserExt?.imageDesc
|
||
viewModel.requestParams.aiUserExt?.imageStyleCode = editAIInfo.aiUserExt?.imageStyleCode
|
||
}
|
||
|
||
// -- 文案
|
||
var title = "Create"
|
||
var bottomButtonTitle = "Create"
|
||
if viewModel.editAIInfo != nil {
|
||
title = "Modify"
|
||
bottomButtonTitle = "Modify"
|
||
}
|
||
navigationView.alpha0Title = title
|
||
container.titleView.title = title
|
||
container.bottomButton.setTitle(bottomButtonTitle, for: .normal)
|
||
}
|
||
|
||
private func setupEvents(){
|
||
container.backButton.addTarget(self, action: #selector(tapBackButton), for: .touchUpInside)
|
||
container.bottomButton.addTarget(self, action: #selector(bottomButtonTapped), for: .touchUpInside)
|
||
container.generate1Button.addTarget(self, action: #selector(tapAIGeneratedIntroductionButton), for: .touchUpInside)
|
||
container.appearanceResult.generateButton.addTarget(self, action: #selector(tapRegenerateFigureButton), for: .touchUpInside)
|
||
container.uploadImageView.bgTapBlock = {[weak self] in
|
||
//self?.pickAppearance()
|
||
self?.goCreateAIGeneratePics()
|
||
}
|
||
|
||
viewModel.$batchNo.sink {[weak self] result in
|
||
self?.container.showQueryGeneratedImages(generating: !result.isEmpty)
|
||
self?.loadingGeneratingResults(batchNo: result)
|
||
}.store(in: &cancellables)
|
||
|
||
container.$introductionContent.sink {[weak self] string in
|
||
self?.viewModel.requestParams.introduction = string
|
||
}.store(in: &cancellables)
|
||
|
||
if viewModel.editAIInfo == nil{
|
||
doGenerate()
|
||
}else{
|
||
restoreEditInfo(aiInfo: viewModel.editAIInfo)
|
||
}
|
||
}
|
||
|
||
private func restoreEditInfo(aiInfo: AIUserModel?) {
|
||
guard let info = aiInfo else { return }
|
||
|
||
container.privacyPublic = (info.permission ?? 2) == 1
|
||
container.introductionContent = info.introduction ?? ""
|
||
|
||
container.showQueryGeneratedImages(generating: true)
|
||
|
||
// 图片
|
||
let last = AIUserImageQuery()
|
||
last.imageUrl = info.imageUrl
|
||
last.status = .completed
|
||
container.appearanceResult.usingFigureImage = last
|
||
container.appearanceResult.selectImageUrl = info.imageUrl ?? ""
|
||
|
||
//头像
|
||
let photo = UploadPhotoM()
|
||
photo.setupValidImageUrl(url: info.headImg)
|
||
container.avatarModel = photo
|
||
}
|
||
|
||
// MARK: - Functions
|
||
|
||
func loadingGeneratingResults(batchNo : String?){
|
||
if isAllImagesOK {
|
||
return
|
||
}
|
||
|
||
guard let queryNo = batchNo, queryNo.isEmpty == false else{
|
||
dlog("❌batchNo is nil")
|
||
return
|
||
}
|
||
|
||
container.appearanceResult.setupLastGenerateProcessStart()
|
||
|
||
dlog("📖查询图片生成中...")
|
||
AICowProvider.request(.imageGeneratedQuery(batchNo: queryNo), modelType: [AIUserImageQuery].self) { [weak self] result in
|
||
switch result {
|
||
case .success(let success):
|
||
guard let queryDatas = success else{
|
||
return
|
||
}
|
||
self?.container.appearanceResult.config(success)
|
||
var ok = true
|
||
if(queryDatas.count < 6){
|
||
ok = false
|
||
}else{
|
||
for per in queryDatas {
|
||
if per.status == .pending{
|
||
ok = false
|
||
}
|
||
}
|
||
}
|
||
self?.isAllImagesOK = ok
|
||
if ok == false{
|
||
DispatchQueue.main.asyncAfter(deadline: .now() + 6) {[weak self] in
|
||
self?.loadingGeneratingResults(batchNo: self?.viewModel.batchNo)
|
||
}
|
||
}else{
|
||
// ✅
|
||
self?.container.appearanceResult.setupLastGenerateProcessEnd()
|
||
}
|
||
case .failure:
|
||
break
|
||
}
|
||
}
|
||
|
||
}
|
||
|
||
|
||
/// 创建时免费生成图像
|
||
private func goCreateAIGeneratePics(){
|
||
|
||
let vc = RoleFigureGenerateController()
|
||
vc.viewModel = viewModel
|
||
navigationController?.pushViewController(vc, animated: true)
|
||
|
||
vc.generatedBatchNoAction = {[weak self] batchNo, imageStyleCode,description, referenceBase64 in
|
||
self?.viewModel.requestParams.aiUserExt?.imageStyleCode = imageStyleCode
|
||
self?.viewModel.requestParams.aiUserExt?.imageDesc = description
|
||
|
||
// 新图片
|
||
if batchNo.count > 0{
|
||
self?.isAllImagesOK = false
|
||
self?.container.appearanceResult.config([], reset: true) // 先loading着
|
||
self?.container.showQueryGeneratedImages(generating: true)
|
||
self?.loadingGeneratingResults(batchNo: batchNo)
|
||
}
|
||
}
|
||
}
|
||
|
||
/// 编辑额外生成图片
|
||
private func goEditAIRegeneratePics(){
|
||
let vc = RolePhotoGenerateController(type: .figure)
|
||
vc.aiId = viewModel.requestParams.aiId
|
||
vc.introduction = container.introductionContent
|
||
vc.sex = viewModel.requestParams.sex
|
||
vc.birthdayString = viewModel.requestParams.birthday
|
||
vc.preSelectstyleCode = viewModel.requestParams.aiUserExt?.imageStyleCode // "IS0020"//
|
||
vc.prePhotoDesc = viewModel.requestParams.aiUserExt?.imageDesc
|
||
presentNaviRootVc(vc: vc)
|
||
|
||
vc.confirmSelectAction = {[weak self] imageStyleCode, desc, imgs in
|
||
self?.viewModel.requestParams.aiUserExt?.imageStyleCode = imageStyleCode
|
||
self?.viewModel.requestParams.aiUserExt?.imageDesc = desc
|
||
|
||
guard let image = imgs?.first else {return}
|
||
self?.container.loadDataAndSelect(image)
|
||
}
|
||
}
|
||
|
||
private func judgeAlertToPublishNewAIRole(){
|
||
// Alert.showAIRoleCreateSuccessAlert {
|
||
// Hud.hideIndicator()
|
||
// } confirmAction: {[weak self] in
|
||
// self?.doPublishRole()
|
||
// }
|
||
|
||
if viewModel.editAIInfo != nil{
|
||
// 编辑AI
|
||
doPublishRole()
|
||
}else{
|
||
// 创建AI Role
|
||
if UserCore.shared.user?.canCreateAIRole() ?? false{
|
||
// 能创建
|
||
Alert.showAIRoleCreateSuccessAlert {
|
||
Hud.hideIndicator()
|
||
} confirmAction: {[weak self] in
|
||
self?.doPublishRole()
|
||
}
|
||
} else {
|
||
if UserCore.shared.user?.isMember == false{ // 去订VIP
|
||
Hud.hideIndicator()
|
||
let sheet = VIPSubscribeSheet()
|
||
sheet.show()
|
||
}else {
|
||
// #warning("是否主动Alert 到底创建限制")
|
||
Alert.showAIRoleCreateSuccessAlert {
|
||
Hud.hideIndicator()
|
||
} confirmAction: {[weak self] in
|
||
self?.doPublishRole()
|
||
}
|
||
}
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
private func doPublishRole(){
|
||
//#warning("test")
|
||
// guard let aiId = viewModel.requestParams.aiId else{
|
||
// return
|
||
// }
|
||
// AppRouter.goBackRootController(jumpIndex: .me)
|
||
// DispatchQueue.main.asyncAfter(deadline: .now() + 0.25) {
|
||
// AppRouter.goChatVC(aiId: aiId)
|
||
// AppRouter.alertCreateAlbumsToRoleHome(aiId: aiId)
|
||
// }
|
||
// view.hideToastActivity()
|
||
// return
|
||
|
||
// close(dismissFirst: true)
|
||
// AppRouter.goBackRootController(jumpIndex: .me)
|
||
// close(dismissFirst: true) {
|
||
// let vc = UIWindow.getTopViewController()
|
||
// if let current = UIWindow.getTopViewController(), current.isKind(of: RoleHomePagerController.self){
|
||
// // Stay in RoleHomePagerController
|
||
// }else{
|
||
// AppRouter.goBackRootController(jumpIndex: .me)
|
||
// }
|
||
// dlog("🔥current vc: \(String(describing: vc))")
|
||
// }
|
||
// return
|
||
|
||
guard let avatar = container.avatarModel?.remoteFullPath else {
|
||
dlog("invalid ai role avatar")
|
||
return
|
||
}
|
||
|
||
viewModel.requestParams.headImg = avatar
|
||
|
||
viewModel.requestParams.introduction = container.introductionContent
|
||
viewModel.requestParams.permission = container.privacyPublic ? 1 : 2
|
||
var params = viewModel.requestParams.toNonNilDictionary()
|
||
|
||
let editAIInfo = viewModel.editAIInfo
|
||
|
||
params.updateValue(container.appearanceResult.selectImageUrl, forKey: "imageUrl")
|
||
|
||
//dlog("🔥当前params: \(params)")
|
||
//Hud.showIndicator()
|
||
AIRoleProvider.request(.aiUserCreateEdit(params: params), modelType: AICreateResponse.self) {[weak self] result in
|
||
Hud.hideIndicator()
|
||
switch result {
|
||
case let .success(success):
|
||
NotificationCenter.post(name: .aiRoleCreatedOrDelete)
|
||
NotificationCenter.post(name: .aiRoleInfoChanged)
|
||
if editAIInfo != nil{
|
||
dlog("编辑AI 退出")
|
||
// 编辑保存成功
|
||
self?.close(dismissFirst: true) {
|
||
// _ = UIWindow.getTopViewController()
|
||
if let current = UIWindow.getTopViewController(), current.isKind(of: RoleHomePagerController.self){
|
||
// Stay in RoleHomePagerController
|
||
}else{
|
||
AppRouter.goBackRootController(jumpIndex: .me)
|
||
}
|
||
}
|
||
}else{
|
||
// 新建用户
|
||
dlog("新建AI 退出")
|
||
AppRouter.goBackRootController(jumpIndex: .me)
|
||
DispatchQueue.main.asyncAfter(deadline: .now() + 0.25) {
|
||
AppRouter.goChatVC(aiId: success?.aiId)
|
||
AppRouter.alertCreateAlbumsToRoleHome(aiId: success?.aiId)
|
||
}
|
||
}
|
||
|
||
case .failure:
|
||
break
|
||
}
|
||
}
|
||
}
|
||
|
||
@objc private func tapAIGeneratedIntroductionButton(){
|
||
if container.introductionContent.count > 0{
|
||
let alert = Alert(title: "内容覆盖", text: "AI创建的内容会覆盖你已经填写的内容，请确认是否继续？")
|
||
let action1 = AlertAction(title: "继续", actionStyle: .confirm) {[weak self] in
|
||
self?.doGenerate()
|
||
}
|
||
let action2 = AlertAction(title: "Cancel", actionStyle: .cancel)
|
||
alert.addAction(action1)
|
||
alert.addAction(action2)
|
||
alert.show()
|
||
}else{
|
||
doGenerate()
|
||
}
|
||
}
|
||
|
||
private func doGenerate(){
|
||
var params = viewModel.requestParams.toPartialDictionary(keys: ["nickname", "sex", "birthday", "characterCode", "tagCode", "introduction"])
|
||
/// 形象描述
|
||
params.updateValue("GEN_INTRODUCTION", forKey: "ptType")
|
||
|
||
if container.introductionContent.count > 0{
|
||
params.updateValue(container.introductionContent, forKey: "content")
|
||
}
|
||
|
||
if let dialogue = viewModel.requestParams.aiUserExt?.dialogueStyle{
|
||
params.updateValue(dialogue, forKey: "dialogue")
|
||
}
|
||
|
||
if let figure = viewModel.requestParams.aiUserExt?.profile{
|
||
params.updateValue(figure, forKey: "figure")
|
||
}
|
||
|
||
Hud.showIndicator()
|
||
AICowProvider.request(.aiContentGenerate(params: params), modelType: AIUserContentGenResponse.self) {[weak self] result in
|
||
Hud.hideIndicator()
|
||
switch result {
|
||
case .success(let success):
|
||
if let content = success?.content{
|
||
self?.container.contentAIGenerated = content
|
||
}
|
||
case .failure:
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
@objc private func tapRegenerateFigureButton(){
|
||
// 尝试重新生成图片
|
||
if viewModel.requestParams.aiId != nil{
|
||
goEditAIRegeneratePics()
|
||
}else{
|
||
goCreateAIGeneratePics()
|
||
}
|
||
}
|
||
|
||
@objc private func bottomButtonTapped(){
|
||
// 头像是否已上传？
|
||
if container.avatarModel?.remoteFullPath != nil{
|
||
Hud.showIndicator()
|
||
judgeAlertToPublishNewAIRole()
|
||
}else{
|
||
// 头像上传...
|
||
guard let photo = container.avatarModel else { return }
|
||
Hud.showIndicator()
|
||
CloudStorage.shared.s3BatchAddPhotos([photo], bucket: .ROLE) {[weak self] result in
|
||
if result{
|
||
//self?.doPublishRole()
|
||
self?.judgeAlertToPublishNewAIRole()
|
||
}else{
|
||
Hud.hideIndicator()
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
@objc private func tapBackButton() {
|
||
close()
|
||
}
|
||
}
|
||
|
||
|
||
extension RolePublishController: UIScrollViewDelegate{
|
||
func scrollViewDidScroll(_ scrollView: UIScrollView) {
|
||
NaviAlphaHandle.changeNaviTitleAlpha(scrollView: scrollView, titleLabel: navigationView.titleLabel)
|
||
}
|
||
|
||
}
|