TSPTPInputVC.swift 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547
  1. //
  2. // TSPTPInputVC.swift
  3. // AIEmoji
  4. //
  5. // Created by 100Years on 2025/4/7.
  6. //
  7. //import IQKeyboardManagerSwift
  8. import PhotosUI
  9. import MXParallaxHeader
  10. class TSPTPInputVC: TSBaseVC {
  11. lazy var viewModel: TSPTPInputVM = {
  12. let viewModel = TSPTPInputVM()
  13. viewModel.isCanGennerateBlock = { [weak self] isCan in
  14. guard let self = self else { return }
  15. submitBtn.isEnabled = isCan
  16. }
  17. return viewModel
  18. }()
  19. lazy var photoPickerManager: TSPhotoPickerManager = {
  20. let photoPickerManager = TSPhotoPickerManager(viewController: self)
  21. return photoPickerManager
  22. }()
  23. private var keyboardHeight: CGFloat = 0
  24. var hintBaseVC = TSAIListHintBaseVC(config: .defaultConfig)
  25. //###################################### 导航栏 view ######################################
  26. lazy var vipBtn: UIButton = {
  27. let vipBtn = UIButton.createButton(image: UIImage(named: "nav_vip")) { [weak self] in
  28. guard let self = self else { return }
  29. TSPurchaseVC.show(target: self) {}
  30. }
  31. return vipBtn
  32. }()
  33. lazy var navBarView: TSBaseNavContentBarView = {
  34. let navBarView = TSBaseNavContentBarView()
  35. let titleImageView = UIImageView.createImageView(imageName: "nav_title_pic",contentMode: .scaleToFill)
  36. navBarView.barView.addSubview(titleImageView)
  37. titleImageView.snp.makeConstraints { make in
  38. make.centerY.equalToSuperview()
  39. make.left.equalTo(16)
  40. }
  41. navBarView.barView.addSubview(vipBtn)
  42. vipBtn.snp.makeConstraints { make in
  43. make.centerY.equalToSuperview()
  44. make.trailing.equalTo(-16)//(-60)
  45. make.width.height.equalTo(24)
  46. }
  47. return navBarView
  48. }()
  49. //###################################### cusStackView ######################################
  50. lazy var cusStackView: TSCustomStackView = {
  51. let cusStackView = TSCustomStackView(axis: .vertical,spacing: 0)
  52. cusStackView.scrollView.isScrollEnabled = false
  53. return cusStackView
  54. }()
  55. //###################################### 上传图片 ######################################
  56. lazy var uploadView: TSPTPUploadView = {
  57. let uploadView = TSPTPUploadView()
  58. uploadView.clickHandel = { [weak self] index in
  59. guard let self = self else { return }
  60. if index == 0 {//删除
  61. viewModel.upLoadImage = nil
  62. uploadView.upLoadImage = nil
  63. }else{//添加
  64. if TSAIListHintBaseVC.isShowUploadImageHint{
  65. TSAIListHintBaseVC.isShowUploadImageHint = false
  66. presentModalHintVC()
  67. }else {
  68. pickSinglePhoto()
  69. }
  70. }
  71. }
  72. return uploadView
  73. }()
  74. func pickSinglePhoto() {
  75. // photoPickerManager.pickSinglePhoto(maxBitSize: kUploadImageMaxBit10Size) { [weak self] image, errorString in
  76. photoPickerManager.pickCustomSinglePhoto() { [weak self] image, errorString in
  77. guard let self = self else { return }
  78. if let errorString = errorString {
  79. TSToastShared.showToast(text: errorString)
  80. }else{
  81. viewModel.upLoadImage = image
  82. uploadView.upLoadImage = image
  83. }
  84. kDelayMainShort {
  85. self.photoPickerManager.dismissPageVC()
  86. }
  87. }
  88. }
  89. //###################################### 选择风格 ######################################
  90. lazy var selectStyleView: TSPTPSelectStyleView = {
  91. let selectStyleView = TSPTPSelectStyleView()
  92. selectStyleView.currentIndexPath = IndexPath(item: viewModel.selectedStyleIndex, section: 0)
  93. selectStyleView.dataArray = viewModel.ptpStyleModels
  94. selectStyleView.clickHandle = { [weak self] model in
  95. guard let self = self else { return }
  96. viewModel.selectedPTPStyleModel = model
  97. updateVipView()
  98. updateTextFiledView()
  99. }
  100. return selectStyleView
  101. }()
  102. //###################################### 输入框 ######################################
  103. lazy var customTextView: TSPlaceholderTextView = {
  104. let customTextView = TSPlaceholderTextView(
  105. placeholder: "Describe how you want to transform".localized,
  106. text: "",
  107. font: .font(size: 14),
  108. textColor: .white,
  109. backgroundColor: .clear
  110. )
  111. customTextView.delegate = self
  112. customTextView.returnKeyType = .done
  113. return customTextView
  114. }()
  115. lazy var clearBtn: TSUIExpandedTouchButton = {
  116. let clearBtn = TSUIExpandedTouchButton()
  117. clearBtn.setUpButton(
  118. image: UIImage(named: "clear_text")
  119. )
  120. { [weak self] in
  121. guard let self = self else { return }
  122. customTextView.text = ""
  123. textViewDidChange(customTextView)
  124. }
  125. clearBtn.isHidden = true
  126. return clearBtn
  127. }()
  128. var promptTextViewH:CGFloat = 96.0
  129. lazy var promptTextView:UIView = {
  130. let promptTextView = UIView()
  131. promptTextView.isHidden = true
  132. promptTextView.clipsToBounds = true
  133. let bgView = UIView()
  134. bgView.backgroundColor = "#333333".uiColor
  135. bgView.cornerRadius = 16.0
  136. promptTextView.addSubview(bgView)
  137. bgView.snp.makeConstraints { make in
  138. make.edges.equalTo(UIEdgeInsets(top: 16, left: 16, bottom: 16, right: 16))
  139. }
  140. bgView.addSubview(customTextView)
  141. bgView.addSubview(clearBtn)
  142. customTextView.snp.makeConstraints { make in
  143. make.centerY.equalToSuperview()
  144. make.leading.equalTo(12.0)
  145. make.top.equalTo(12.0)
  146. make.bottom.equalTo(-12.0)
  147. make.trailing.equalTo(-20)
  148. }
  149. clearBtn.snp.makeConstraints { make in
  150. make.width.height.equalTo(16.0)
  151. make.trailing.equalTo(-12)
  152. make.bottom.equalTo(-12.0)
  153. }
  154. return promptTextView
  155. }()
  156. //###################################### 集合视图 ######################################
  157. let collectionViewBtootm:CGFloat = 80
  158. lazy var collectionComponent: TSCollectionViewComponent = {
  159. let layout = UICollectionViewFlowLayout()
  160. let cp = TSCollectionViewComponent(frame: CGRect.zero, layout: layout, attributes: [:])
  161. cp.collectionView.contentInset = UIEdgeInsets(top: 0, left: 0, bottom: collectionViewBtootm, right: 0)
  162. // 禁用自动 contentInset 调整
  163. if #available(iOS 11.0, *) {
  164. cp.collectionView.contentInsetAdjustmentBehavior = .never
  165. } else {
  166. automaticallyAdjustsScrollViewInsets = false
  167. }
  168. cp.sectionActionHandler = { [weak self] cellCp, indexPath in
  169. guard let self = self else { return }
  170. if let cmd = cellCp as? String, cmd == "delete" {
  171. showCustomAlert(message: "Are you sure to delete".localized, deleteHandler: {
  172. self.viewModel.removeAllHistoryList()
  173. self.collectionComponent.clear()
  174. self.collectionComponent.reloadView(with: self.viewModel.colDataArray)
  175. })
  176. }
  177. }
  178. cp.itemDidSelectedHandler = { [weak self] (object, indexPath) in
  179. guard let self = self else { return }
  180. if let sections = viewModel.colDataArray.safeObj(At: indexPath.section) as? TSGenmojiCoLSectionModel{
  181. var dataModelArray:[TSActionInfoModel] = []
  182. for itemModel in sections.items {
  183. dataModelArray.append(itemModel.dataModel)
  184. }
  185. let browseVC = TSAIPhotoBrowseVC()
  186. browseVC.dataModelArray = dataModelArray
  187. browseVC.currentIndex = indexPath.item
  188. kPresentModalVC(target: self, modelVC: browseVC,transitionStyle: .crossDissolve)
  189. }
  190. }
  191. cp.collectionView.keyboardDismissMode = .interactive
  192. return cp
  193. }()
  194. //###################################### 按钮 ######################################
  195. lazy var submitBtn: UIButton = {
  196. let submitBtn = kCreateNormalSubmitBtn(title: getVipText()) { [weak self] in
  197. guard let self = self else { return }
  198. generateImage()
  199. }
  200. submitBtn.cornerRadius = 24.0
  201. submitBtn.isEnabled = false
  202. return submitBtn
  203. }()
  204. override func createView() {
  205. navBarContentView.addSubview(navBarView)
  206. navBarView.snp.makeConstraints { make in
  207. make.edges.equalToSuperview()
  208. }
  209. contentView.addSubview(collectionComponent.collectionView)
  210. collectionComponent.collectionView.snp.makeConstraints { make in
  211. make.edges.equalToSuperview()
  212. }
  213. let tapGesture = UITapGestureRecognizer(target: self, action: #selector(clickView))
  214. tapGesture.cancelsTouchesInView = false
  215. view.addGestureRecognizer(tapGesture)
  216. collectionComponent.clear()
  217. collectionComponent.reloadView(with:viewModel.colDataArray)
  218. contentView.addSubview(submitBtn)
  219. submitBtn.snp.makeConstraints { make in
  220. make.bottom.equalTo(-16)
  221. make.leading.equalTo(16)
  222. make.trailing.equalTo(-16)
  223. make.height.equalTo(48)
  224. }
  225. setUpCusStackView()
  226. setHeaderCusStackView()
  227. kDelayMainShort {
  228. self.setHeaderCusStackView()
  229. }
  230. }
  231. func setHeaderCusStackView(){
  232. self.collectionComponent.collectionView.parallaxHeader.view = self.cusStackView
  233. self.collectionComponent.collectionView.parallaxHeader.height = self.cusStackViewH
  234. self.collectionComponent.collectionView.parallaxHeader.mode = .bottom
  235. self.collectionComponent.collectionView.parallaxHeader.minimumHeight = 0
  236. }
  237. override func dealThings() {
  238. NotificationCenter.default.addObserver(self, selector: #selector(vipInfoChanged), name: .kPurchaseDidChanged, object: nil)
  239. updateVipView()
  240. TSAIListHintBaseVC.userDefaultsKey = "isFirstUploadImagePTP"
  241. // // 监听键盘事件
  242. NotificationCenter.default.addObserver(self, selector: #selector(keyboardWillShow(_:)), name: UIResponder.keyboardWillShowNotification, object: nil)
  243. NotificationCenter.default.addObserver(self, selector: #selector(keyboardWillHide(_:)), name: UIResponder.keyboardWillHideNotification, object: nil)
  244. // collectionViewObserverHandle()
  245. }
  246. // private var collectionViewObserver: CollectionViewObserver!
  247. // func collectionViewObserverHandle(){
  248. // collectionViewObserver = CollectionViewObserver(collectionView: collectionComponent.collectionView)
  249. //
  250. // collectionViewObserver.onContentSizeChange = { size in
  251. // print("collectionViewObserver 内容大小变化: \(size)")
  252. // }
  253. //
  254. // collectionViewObserver.onContentInsetChange = { inset in
  255. // print("collectionViewObserver 内边距变化: \(inset)")
  256. // }
  257. //
  258. // collectionViewObserver.onContentOffsetChange = { offset in
  259. // print("collectionViewObserver 偏移量变化: \(offset)")
  260. // }
  261. //
  262. // }
  263. //
  264. //
  265. // func scrollViewDidScroll(_ scrollView: UIScrollView) {
  266. // print("contentOffset 变化: \(scrollView.contentOffset)")
  267. // // 实时监听偏移量变化
  268. // }
  269. }
  270. extension TSPTPInputVC {
  271. var cusStackViewH:CGFloat{
  272. get {
  273. if cusStackView.viewH > 0{
  274. dePrint("collectionViewObserver cusStackView.viewH == \(cusStackView.viewH)")
  275. return cusStackView.viewH
  276. }
  277. return 454
  278. }
  279. }
  280. func presentModalHintVC(){
  281. hintBaseVC = TSAIListHintBaseVC(config: .defaultConfig) { [weak self] image in
  282. guard let self = self else { return }
  283. viewModel.upLoadImage = image
  284. uploadView.upLoadImage = image
  285. hintBaseVC.dismissPageVC()
  286. }
  287. kPresentModalVC(target: self, modelVC: hintBaseVC,transitionStyle: .crossDissolve)
  288. }
  289. func setUpCusStackView(){
  290. let uploadPhotoTitleView = TSTitleView.creatTitleView(title: "Upload Photo".localized, subTitle: "")
  291. cusStackView.addSubviewToStack(uploadPhotoTitleView)
  292. uploadPhotoTitleView.snp.makeConstraints { make in
  293. make.height.equalTo(uploadPhotoTitleView.viewH)
  294. make.width.equalTo(k_ScreenWidth)
  295. }
  296. let hintBtn = TSUIExpandedTouchButton()
  297. hintBtn.setUpButton(image: UIImage(named: "ptp_hint")){ [weak self] in
  298. guard let self = self else { return }
  299. presentModalHintVC()
  300. }
  301. uploadPhotoTitleView.contentView.addSubview(hintBtn)
  302. hintBtn.snp.makeConstraints { make in
  303. make.centerY.equalToSuperview().offset(kSectionTitleViewCenterYOffset)
  304. make.trailing.equalTo(-16)
  305. make.width.height.equalTo(16)
  306. }
  307. cusStackView.addSubviewToStack(uploadView)
  308. uploadView.snp.makeConstraints { make in
  309. make.height.equalTo(uploadView.viewH)
  310. make.width.equalTo(k_ScreenWidth)
  311. }
  312. let selectStyleTitleView = TSTitleView.creatTitleView(title: "Select Style".localized)
  313. cusStackView.addSubviewToStack(selectStyleTitleView)
  314. selectStyleTitleView.snp.makeConstraints { make in
  315. make.height.equalTo(selectStyleTitleView.viewH)
  316. make.width.equalTo(k_ScreenWidth)
  317. }
  318. cusStackView.addSubviewToStack(selectStyleView)
  319. selectStyleView.snp.makeConstraints { make in
  320. make.height.equalTo(selectStyleView.viewH)
  321. make.width.equalTo(k_ScreenWidth)
  322. }
  323. cusStackView.addSubviewToStack(promptTextView,length: promptTextViewH,animate: false)
  324. promptTextView.isHidden = !viewModel.selectedPTPStyleModel.input
  325. }
  326. override func viewDidLayoutSubviews() {
  327. super.viewDidLayoutSubviews()
  328. }
  329. }
  330. extension TSPTPInputVC: UITextViewDelegate{
  331. // 实现 UITextViewDelegate 协议方法,控制 return 键行为
  332. func textView(_ textView: UITextView, shouldChangeTextIn range: NSRange, replacementText text: String) -> Bool {
  333. if text == "\n" {
  334. // 当输入为换行符(即按下 return 键)时,执行相应操作
  335. //sendBolck?(textView.text)
  336. clickView()
  337. return false
  338. }
  339. return true
  340. }
  341. func textViewDidChange(_ textView: UITextView){
  342. clearBtn.isHidden = textView.text.count <= 0
  343. viewModel.selectedPTPStyleModel.inputText = textView.text
  344. viewModel.gennerateChange()
  345. }
  346. }
  347. extension TSPTPInputVC {
  348. @objc func vipInfoChanged() {
  349. kExecuteOnMainThread {
  350. self.updateVipView()
  351. }
  352. }
  353. func updateVipView() {
  354. kExecuteOnMainThread {
  355. self.vipBtn.isHidden = PurchaseManager.default.isVip
  356. var showVip = kPurchaseDefault.generateVipShow(type: .picToPic)
  357. if showVip == false {
  358. showVip = self.viewModel.selectedPTPStyleModel.isVip
  359. }
  360. kSetBtnVipIcon(btn: self.submitBtn, show: showVip)
  361. }
  362. }
  363. func getVipText()->String{
  364. return "Generate".localized
  365. }
  366. }
  367. extension TSPTPInputVC {
  368. @objc func clickView() {
  369. view.endEditing(true)
  370. }
  371. func generateImage() {
  372. var isVip = kPurchaseDefault.freeNumAvailable(type: .picToPic) == false
  373. if viewModel.selectedPTPStyleModel.isVip == true {
  374. isVip = true
  375. }
  376. if kJudgeVip(externalBool: isVip, vc: self) { return } //判断 vip
  377. viewModel.selectedPTPStyleModel.upLoadImage = viewModel.upLoadImage
  378. viewModel.selectedPTPStyleModel.upLoadImageUrl = nil
  379. let gennerateVC = TSPTPGeneratorVC(generateStyleModel: viewModel.selectedPTPStyleModel) { [weak self] model in
  380. guard let self = self else { return }
  381. if viewModel.saveModel(model:model) {
  382. collectionComponent.clear()
  383. collectionComponent.reloadView(with:viewModel.colDataArray)
  384. }else{
  385. collectionComponent.reloadData()
  386. }
  387. updateVipView()
  388. }
  389. gennerateVC.reloadViewBlock = { [weak self] in
  390. guard let self = self else { return }
  391. collectionComponent.reloadData()
  392. }
  393. kPresentModalVC(target: self, modelVC: gennerateVC,transitionStyle: .crossDissolve)
  394. }
  395. }
  396. extension TSPTPInputVC {
  397. // MARK: - 键盘弹出时滚动到 TextView
  398. @objc func keyboardWillShow(_ notification: Notification) {
  399. guard let keyboardFrame = notification.userInfo?[UIResponder.keyboardFrameEndUserInfoKey] as? CGRect else { return }
  400. let textView = customTextView
  401. guard let scrollView = collectionComponent.collectionView else { return }
  402. // 1. 计算键盘高度(减去安全区域)
  403. let keyboardHeight = keyboardFrame.height - view.safeAreaInsets.bottom
  404. dePrint("keyboardHeight = \(keyboardHeight )")
  405. // 2. 直接获取 TextView 在 ScrollView 中的位置
  406. let textViewFrame = scrollView.convert(textView.frame, from: textView.superview)
  407. dePrint("textViewFrame = \(textViewFrame)")
  408. // 3. 计算需要滚动的距离(TextView 底部 - (屏幕高度 - 键盘高度))
  409. var scrollDistance = textViewFrame.maxY - (scrollView.bounds.height - keyboardHeight)
  410. dePrint("scrollDistance = \(scrollDistance)")
  411. // 4. 如果需要滚动,调整 contentOffset
  412. var y = scrollDistance
  413. scrollView.setContentOffset(CGPoint(x: 0, y: y),animated: true)
  414. }
  415. // MARK: - 键盘隐藏时恢复
  416. @objc private func keyboardWillHide(_ notification: Notification) {
  417. self.collectionComponent.collectionView.contentOffset = CGPoint(x: 0, y: -cusStackViewH)
  418. }
  419. }
  420. extension TSPTPInputVC {
  421. func updateTextFiledView () {
  422. if viewModel.selectedPTPStyleModel.input {
  423. cusStackView.showWithDrawerAnimation(view: promptTextView, viewHeight: promptTextViewH) { [weak self] in
  424. self?.updateViewCusStackViewH()
  425. }
  426. }else{
  427. cusStackView.hideWithDrawerAnimation(view: promptTextView, viewHeight: promptTextViewH) { [weak self] in
  428. self?.updateViewCusStackViewH()
  429. }
  430. }
  431. }
  432. func updateViewCusStackViewH() {
  433. self.collectionComponent.collectionView.parallaxHeader.updateHeaderHeight(height: self.cusStackViewH)
  434. // self.collectionComponent.collectionView.parallaxHeader.height = self.cusStackViewH
  435. //
  436. // let newHeight = self.cusStackViewH
  437. // let headerView = self.collectionComponent.collectionView.parallaxHeader.view!
  438. // for constraint in headerView.constraints {
  439. // if constraint.firstAttribute == .height &&
  440. // constraint.firstItem as? UIView == headerView &&
  441. // constraint.secondItem == nil {
  442. // constraint.constant = newHeight
  443. // break
  444. // }
  445. // }
  446. }
  447. }
  448. extension MXParallaxHeader {
  449. func updateHeaderHeight(height:CGFloat){
  450. self.height = height
  451. let headerView = view!
  452. for constraint in headerView.constraints {
  453. if constraint.firstAttribute == .height &&
  454. constraint.firstItem as? UIView == headerView &&
  455. constraint.secondItem == nil {
  456. constraint.constant = height
  457. break
  458. }
  459. }
  460. }
  461. }