TSPTPInputVC.swift 20 KB

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