KMBatchWatermarkView.swift 9.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271
  1. //
  2. // KMBatchWatermarkView.swift
  3. // PDF Master
  4. //
  5. // Created by lizhe on 2023/1/16.
  6. //
  7. import Cocoa
  8. import KMComponentLibrary
  9. class KMBatchWatermarkModel: KMBatchSettingItemViewModel {
  10. var watermarkModel: KMWatermarkModel?
  11. }
  12. class KMBatchWatermarkView: KMBatchSettingItemView {
  13. @IBOutlet weak var waterContentView: NSView!
  14. @IBOutlet var saveTemplateButton: ComponentButton!
  15. @IBOutlet var applyButton: ComponentButton!
  16. @IBOutlet var applyBtnLeftConst: NSLayoutConstraint!
  17. var propertyController: KMNWatermarkPropertyController = KMNWatermarkPropertyController.init()
  18. var templatesController: KMNWatermarkTemplateController = KMNWatermarkTemplateController()
  19. var saveTemplateMessage: ComponentMessage = ComponentMessage()
  20. var pdfDocument: CPDFDocument?
  21. var currentWatermarkData: KMWatermarkModel?
  22. weak open var delegate: KMWatermarkControllerDelegate?
  23. var editSubType: KMPDFEditSubModeType = .none {
  24. didSet {
  25. resetUI()
  26. }
  27. }
  28. var data: KMBatchWatermarkModel = KMBatchWatermarkModel()
  29. //MARK: - func
  30. override func setup() {
  31. super.setup()
  32. propertyController.view.frame = waterContentView.bounds
  33. propertyController.view.autoresizingMask = [.width, .height]
  34. propertyController.delegate = self
  35. propertyController.view.isHidden = true
  36. propertyController.isInBatchMode = true
  37. waterContentView.addSubview(propertyController.view)
  38. templatesController.view.frame = waterContentView.bounds
  39. templatesController.view.autoresizingMask = [.width, .height]
  40. templatesController.delegate = self
  41. templatesController.view.isHidden = true
  42. waterContentView.addSubview(templatesController.view)
  43. setupProperty()
  44. reloadInfoView()
  45. }
  46. override func reloadData() {
  47. super.reloadData()
  48. var applyEnable = true
  49. if let model = self.currentWatermarkData {
  50. if model.watermarkType == .text {
  51. if model.text == nil || model.text?.count == 0 {
  52. applyEnable = false
  53. }
  54. } else if model.watermarkType == .image {
  55. if model.imagePath == nil {
  56. applyEnable = false
  57. }
  58. }
  59. } else {
  60. applyEnable = false
  61. }
  62. saveTemplateButton.properties.isDisabled = !applyEnable
  63. saveTemplateButton.reloadData()
  64. if self.filesData.count == 0 {
  65. applyEnable = false
  66. }
  67. applyButton.properties.isDisabled = !applyEnable
  68. applyButton.reloadData()
  69. }
  70. func reloadInfoView() {
  71. if editSubType == .template {
  72. templatesController.reloadData()
  73. } else if editSubType == .add {
  74. propertyController.editSubType = editSubType
  75. propertyController.watermarkData = KMWatermarkManager.defaultManager.defaultWatermarkData
  76. propertyController.watermarkData.cPDFWatermark = CPDFWatermark(document: pdfDocument, type: .text)
  77. propertyController.reloadData()
  78. currentWatermarkData = propertyController.watermarkData
  79. } else if editSubType == .edit {
  80. propertyController.editSubType = editSubType
  81. propertyController.reloadData()
  82. }
  83. }
  84. func setupProperty() {
  85. saveTemplateMessage.properties = ComponentMessageProperty(messageType: .success, title: KMLocalizedString("Saved Template"))
  86. saveTemplateButton.properties = ComponentButtonProperty(type: .default_tertiary,
  87. size: .m,
  88. state: .normal,
  89. onlyIcon: true,
  90. icon: NSImage(named: "icon_wm_template_save"),
  91. keepPressState: false)
  92. saveTemplateButton.setTarget(self, action: #selector(saveTemplateButtonAction(_:)))
  93. saveTemplateButton.reloadData()
  94. applyButton.properties = ComponentButtonProperty(type: .primary,
  95. size: .m,
  96. state: .normal,
  97. buttonText: KMLocalizedString("Apply"),
  98. keepPressState: false)
  99. applyButton.setTarget(self, action: #selector(applyButtonAction(_:)))
  100. applyButton.reloadData()
  101. }
  102. func resetUI() {
  103. propertyController.view.isHidden = true
  104. templatesController.view.isHidden = true
  105. if editSubType == .template {
  106. templatesController.view.isHidden = false
  107. applyBtnLeftConst.constant = 16
  108. saveTemplateButton.isHidden = true
  109. } else if editSubType == .add {
  110. propertyController.view.isHidden = false
  111. propertyController.editSubType = editSubType
  112. applyBtnLeftConst.constant = 68
  113. saveTemplateButton.isHidden = false
  114. } else if editSubType == .edit {
  115. propertyController.view.isHidden = false
  116. propertyController.editSubType = editSubType
  117. applyBtnLeftConst.constant = 16
  118. saveTemplateButton.isHidden = true
  119. }
  120. }
  121. //MARK: - Action
  122. @objc func saveTemplateButtonAction(_ sender: NSButton) {
  123. propertyController.saveButtonClicked(propertyController.saveButton)
  124. }
  125. @objc func applyButtonAction(_ sender: NSButton) {
  126. if KMMemberInfo.shared.isLogin == false {
  127. KMLoginWindowsController.shared.showWindow(nil)
  128. return
  129. }
  130. self.data.watermarkModel = self.currentWatermarkData
  131. if self.batchExport != nil {
  132. self.batchExport!(self, self.data)
  133. }
  134. }
  135. }
  136. //MARK: - KMNWatermarkPropertyControllerDelegate
  137. extension KMBatchWatermarkView: KMNWatermarkPropertyControllerDelegate {
  138. func watermarkPropertyControllerDidRemoveWatermark(_ controller: KMNWatermarkPropertyController, _ type: CPDFWatermarkType) {
  139. if type == .text {
  140. controller.watermarkData.cPDFWatermark = nil
  141. controller.watermarkData.cPDFWatermark = CPDFWatermark(document: pdfDocument, type: .text)
  142. } else if type == .image {
  143. controller.watermarkData.cPDFWatermark = nil
  144. controller.watermarkData.cPDFWatermark = CPDFWatermark(document: pdfDocument, type: .image)
  145. }
  146. }
  147. func watermarkPropertyControllerDidUpdate(_ controller: KMNWatermarkPropertyController) {
  148. currentWatermarkData = controller.watermarkData
  149. reloadData()
  150. }
  151. func watermarkPropertyControllerDidChangetoTemplate(_ controller: KMNWatermarkPropertyController) {
  152. editSubType = .template
  153. currentWatermarkData = nil
  154. }
  155. func watermarkPropertyControllerSaveTemplateSuccess(_ controller: KMNWatermarkPropertyController, _ data: KMWatermarkModel) {
  156. saveTemplateMessage.frame = CGRectMake((CGRectGetWidth(self.frame) - saveTemplateMessage.properties.propertyInfo.viewWidth)/2,
  157. CGRectGetHeight(self.frame) - saveTemplateMessage.properties.propertyInfo.viewHeight - 8,
  158. saveTemplateMessage.properties.propertyInfo.viewWidth,
  159. saveTemplateMessage.properties.propertyInfo.viewHeight)
  160. saveTemplateMessage.reloadData()
  161. saveTemplateMessage.show(inView: self, autoHideSeconde: 2)
  162. editSubType = .template
  163. templatesController.selectedWatermark = data
  164. templatesController.reloadData()
  165. currentWatermarkData = templatesController.selectedWatermark
  166. reloadData()
  167. }
  168. func watermarkPropertyControllerCancelTemplateEdit(_ controller: KMNWatermarkPropertyController) {
  169. currentWatermarkData = nil
  170. reloadData()
  171. editSubType = .template
  172. templatesController.reloadData()
  173. }
  174. }
  175. //MARK: - KMNWatermarkTemplateControllerDelegate
  176. extension KMBatchWatermarkView: KMNWatermarkTemplateControllerDelegate {
  177. func templateControllerDidAddWatermarkData(_ controller: KMNWatermarkTemplateController) {
  178. editSubType = .add
  179. propertyController.watermarkData = KMWatermarkManager.defaultManager.defaultWatermarkData
  180. if let value = propertyController.watermarkData.imagePath {
  181. propertyController.watermarkData.cPDFWatermark = CPDFWatermark(document: pdfDocument, type: .image)
  182. } else {
  183. propertyController.watermarkData.cPDFWatermark = CPDFWatermark(document: pdfDocument, type: .text)
  184. }
  185. propertyController.reloadData()
  186. currentWatermarkData = propertyController.watermarkData
  187. reloadData()
  188. }
  189. func templateControllerDidEditWatermarkData(_ controller: KMNWatermarkTemplateController, _ watermark: KMWatermarkModel) {
  190. editSubType = .edit
  191. propertyController.watermarkData = watermark
  192. propertyController.originalDataDict = KMWatermarkManager.defaultManager.parseWaterMark(model: watermark)
  193. propertyController.reloadData()
  194. currentWatermarkData = propertyController.watermarkData
  195. reloadData()
  196. templatesController.selectedWatermark = nil
  197. }
  198. func templateControllerDidSelectedChanged(_ controller: KMNWatermarkTemplateController) {
  199. currentWatermarkData = controller.selectedWatermark
  200. reloadData()
  201. }
  202. }