KMNWatermarkPropertyController.swift 41 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908
  1. //
  2. // KMNWatermarkPropertyController.swift
  3. // PDF Reader Pro
  4. //
  5. // Created by Niehaoyu on 2024/11/1.
  6. //
  7. import Cocoa
  8. import KMComponentLibrary
  9. @objc protocol KMNWatermarkPropertyControllerDelegate: AnyObject {
  10. @objc optional func watermarkPropertyControllerDidRemoveWatermark(_ controller: KMNWatermarkPropertyController, _ type: CPDFWatermarkType)
  11. //水印数据有更新
  12. @objc optional func watermarkPropertyControllerDidUpdate(_ controller: KMNWatermarkPropertyController)
  13. //切换到模板界面
  14. @objc optional func watermarkPropertyControllerDidChangetoTemplate(_ controller: KMNWatermarkPropertyController)
  15. //成功保存到模板
  16. @objc optional func watermarkPropertyControllerSaveTemplateSuccess(_ controller: KMNWatermarkPropertyController, _ data: KMWatermarkModel)
  17. //取消修改模板信息
  18. @objc optional func watermarkPropertyControllerCancelTemplateEdit(_ controller: KMNWatermarkPropertyController)
  19. }
  20. class KMNWatermarkPropertyController: NSViewController {
  21. @IBOutlet var contendView: NSView!
  22. @IBOutlet var leftTopButton: ComponentButton!
  23. @IBOutlet var titleLabel: NSTextField!
  24. @IBOutlet var templateButton: ComponentButton!
  25. @IBOutlet var typeTabsBGView: NSView!
  26. @IBOutlet var typeTabs: ComponentTabs!
  27. @IBOutlet var infoContendView: NSView!
  28. @IBOutlet var infoScrollView: KMScrollView!
  29. @IBOutlet var infoScrollTopConst: NSLayoutConstraint!
  30. @IBOutlet var infoContendBottomConst: NSLayoutConstraint!
  31. //Text
  32. @IBOutlet var textBGView: NSView!
  33. @IBOutlet var text_textareaView: ComponentTextarea!
  34. @IBOutlet var fontNameSelect: ComponentSelect!
  35. @IBOutlet var fontStyleSelect: ComponentSelect!
  36. @IBOutlet var fontSizeSelect: ComponentSelect!
  37. @IBOutlet var fontColorGroup: ComponentCColorGroup!
  38. //File
  39. @IBOutlet var fileBGView: NSView!
  40. @IBOutlet var fileInputView: ComponentInput!
  41. @IBOutlet var fileInputAddonView: ComponentInputAddon!
  42. //Appearance
  43. @IBOutlet var appearanceBGView: NSView!
  44. @IBOutlet var appearanceLabel: NSTextField!
  45. @IBOutlet var appearance_RotateSelect: ComponentSelect!
  46. @IBOutlet var appearance_OpacitySelect: ComponentSelect!
  47. @IBOutlet var appearanceScaleCheckbox: ComponentCheckBox!
  48. @IBOutlet var appearanceScaleSelect: ComponentSelect!
  49. @IBOutlet var appearanceTopPageRadio: ComponentRadio!
  50. @IBOutlet var appearanceBottomPageRadio: ComponentRadio!
  51. @IBOutlet var appearanceBGTopConst: NSLayoutConstraint!
  52. //Position
  53. @IBOutlet var positionBGView: NSView!
  54. @IBOutlet var positionLabel: NSTextField!
  55. @IBOutlet var positionItemView: ComponentCPosition!
  56. @IBOutlet var positionXInput: ComponentInputNumber!
  57. @IBOutlet var positionYInput: ComponentInputNumber!
  58. @IBOutlet var positionTileCheckbox: ComponentCheckBox!
  59. @IBOutlet var positionTileHorImage: NSImageView!
  60. @IBOutlet var positionTileVertImage: NSImageView!
  61. @IBOutlet var positionTileHoriInput: ComponentInputNumber!
  62. @IBOutlet var positionTileVertInput: ComponentInputNumber!
  63. //Save
  64. @IBOutlet var saveTemplateBGView: NSView!
  65. @IBOutlet var saveButton: ComponentButton!
  66. private var textTabProperty = ComponentTabsProperty(tabsType: .underline_Fill, state: .normal, showIcon: false, title: KMLocalizedString("Text"))
  67. private var fileTabProperty = ComponentTabsProperty(tabsType: .underline_Fill, state: .normal, showIcon: false, title: KMLocalizedString("File"))
  68. private var familyNames = CPDFFont.familyNames
  69. var chooseFileURL: URL?
  70. var watermarkData: KMWatermarkModel = KMWatermarkManager.defaultManager.defaultWatermarkData
  71. public var editSubType: KMPDFEditSubModeType = .add
  72. public var isInBatchMode: Bool = false
  73. var originalDataDict: Dictionary<String, Any>?
  74. weak open var delegate: KMNWatermarkPropertyControllerDelegate?
  75. /**
  76. 1、新增模板:新增完成后,重置缓存数据
  77. 2、修改模板:
  78. */
  79. //MARK: - func
  80. override func viewDidLoad() {
  81. super.viewDidLoad()
  82. // Do view setup here.
  83. infoScrollView.documentView = infoContendView
  84. infoScrollView.scrollerStyle = .overlay
  85. infoScrollView.layout()
  86. setupProperty()
  87. reloadData()
  88. refreshScrollView()
  89. }
  90. func setupProperty() {
  91. contendView.wantsLayer = true
  92. contendView.layer?.backgroundColor = ComponentLibrary.shared.getComponentColorFromKey("colorBg/layout-middle").cgColor
  93. titleLabel.textColor = ComponentLibrary.shared.getComponentColorFromKey("colorText/2")
  94. titleLabel.font = ComponentLibrary.shared.getFontFromKey("mac/body-m-bold")
  95. leftTopButton.properties = ComponentButtonProperty(type: .text_gray, size: .xxs, onlyIcon: true, icon: NSImage(named: "watermark_arrowLeft"), keepPressState: false)
  96. leftTopButton.setTarget(self, action: #selector(leftTopButtonClicked(_:)))
  97. templateButton.properties = ComponentButtonProperty(type: .text_gray, size: .xxs, onlyIcon: true, icon: NSImage(named: "icon_wm_template"), keepPressState: false)
  98. templateButton.setTarget(self, action: #selector(templateButtonClicked(_:)))
  99. typeTabs.updateItemProperty([textTabProperty, fileTabProperty])
  100. typeTabs.delegate = self
  101. //Text
  102. text_textareaView.properties = ComponentTextareaProperty(size: .s, placeholderString: KMLocalizedString("Please enter..."), totalCount: -1, text: "")
  103. text_textareaView.delegate = self
  104. fontNameSelect.properties = ComponentSelectProperties(size: .s,
  105. state: .normal,
  106. text: "")
  107. var menuItemArr: [ComponentMenuitemProperty] = []
  108. for string in familyNames {
  109. let item = ComponentMenuitemProperty(type: .normal, text: string, identifier: string)
  110. menuItemArr.append(item)
  111. }
  112. fontNameSelect.updateMenuItemsArr(menuItemArr)
  113. fontNameSelect.selectItemAtIndex(0)
  114. fontNameSelect.delegate = self
  115. fontStyleSelect.properties = ComponentSelectProperties(size: .s,
  116. state: .normal,
  117. text: "")
  118. fontStyleSelect.delegate = self
  119. fontSizeSelect.properties = ComponentSelectProperties(size: .s,
  120. state: .normal,
  121. creatable: true,
  122. text: "12 pt",
  123. textUnit: " pt",
  124. regexString: "0123456789 pt")
  125. var sizeItemArr: [ComponentMenuitemProperty] = []
  126. for string in ["6","8","10","12","14",
  127. "16","18","20","22","24",
  128. "26","28","30","32","34",
  129. "36","40","48","64","80",
  130. "96","112"] {
  131. let item = ComponentMenuitemProperty(type: .normal, text: string, identifier: string)
  132. sizeItemArr.append(item)
  133. }
  134. fontSizeSelect.updateMenuItemsArr(sizeItemArr)
  135. fontSizeSelect.selectItemAtIndex(0)
  136. fontSizeSelect.delegate = self
  137. let colorAProperty = ComponentCColorProperty(colorType: .color, state: .normal, isCustom: false, color: KMAnnotationPropertiesColorManager.manager.watermark_Colors[0])
  138. let colorBProperty = ComponentCColorProperty(colorType: .color, state: .normal, isCustom: false, color: KMAnnotationPropertiesColorManager.manager.watermark_Colors[1])
  139. let colorCProperty = ComponentCColorProperty(colorType: .color, state: .normal, isCustom: false, color: KMAnnotationPropertiesColorManager.manager.watermark_Colors[2])
  140. let colorDProperty = ComponentCColorProperty(colorType: .color, state: .normal, isCustom: false, color: KMAnnotationPropertiesColorManager.manager.watermark_Colors[3])
  141. let colorEProperty = ComponentCColorProperty(colorType: .color, state: .normal, isCustom: true, color: KMAnnotationPropertiesColorManager.manager.watermark_Colors[4])
  142. fontColorGroup.setUpWithColorPropertys([colorAProperty, colorBProperty, colorCProperty, colorDProperty], customItemProperty: colorEProperty)
  143. fontColorGroup.delegate = self
  144. //File
  145. fileInputView.properties = ComponentInputProperty(size: .s,
  146. state:.pressed ,
  147. placeholder: "Select Source File...",
  148. text: "",
  149. creatable: false)
  150. fileInputView.properties.propertyInfo.cornerRadius_topLeft = 0
  151. fileInputView.properties.propertyInfo.cornerRadius_topRight = 0
  152. fileInputView.properties.propertyInfo.cornerRadius_bottomLeft = 0
  153. fileInputView.properties.propertyInfo.cornerRadius_bottomRight = 0
  154. fileInputView.reloadData()
  155. fileInputView.delegate = self
  156. fileInputAddonView.properties = ComponentInputAddonProperty(size: .s,
  157. state: .normal,
  158. addOnBefore: false,
  159. onlyRead: false,
  160. addonType: .imageWithColor,
  161. iconImage: NSImage(named: "icon_folder"))
  162. fileInputAddonView.setTarget(self, action: #selector(chooseURLAction(_ :)))
  163. //Appearance
  164. appearanceLabel.stringValue = KMLocalizedString("Appearance")
  165. appearanceLabel.textColor = ComponentLibrary.shared.getComponentColorFromKey("colorText/2")
  166. appearanceLabel.font = ComponentLibrary.shared.getFontFromKey("mac/body-s-medium")
  167. appearance_RotateSelect.properties = ComponentSelectProperties(size: .s,
  168. state: .normal,
  169. creatable: true,
  170. text: "0°",
  171. textUnit: "°",
  172. regexString: "0123456789°-")
  173. var rotateItems: [ComponentMenuitemProperty] = []
  174. for string in ["0°", "45°", "-45°", "90°", "-90°"] {
  175. let item = ComponentMenuitemProperty(type: .normal, text: string)
  176. rotateItems.append(item)
  177. }
  178. appearance_RotateSelect.updateMenuItemsArr(rotateItems)
  179. appearance_RotateSelect.delegate = self
  180. appearance_OpacitySelect.properties = ComponentSelectProperties(size: .s,
  181. state: .normal,
  182. creatable: true,
  183. text: "100%",
  184. textUnit: "%",
  185. regexString: "0123456789%")
  186. var opacityItems: [ComponentMenuitemProperty] = []
  187. for string in ["25%", "50%", "75%", "100%"] {
  188. let item = ComponentMenuitemProperty(type: .normal, text: string)
  189. opacityItems.append(item)
  190. }
  191. appearance_OpacitySelect.updateMenuItemsArr(opacityItems)
  192. appearance_OpacitySelect.delegate = self
  193. appearanceScaleCheckbox.properties = ComponentCheckBoxProperty(size: .s, text: KMLocalizedString("Scale relative to target page"), checkboxType: .normal)
  194. appearanceScaleCheckbox.setTarget(self, action: #selector(checkBoxClicked(_:)))
  195. appearanceScaleSelect.properties = ComponentSelectProperties(size: .s,
  196. state: .normal,
  197. creatable: true,
  198. text: "100%",
  199. textUnit: "%",
  200. regexString: "0123456789%")
  201. var appearanceScaleItems: [ComponentMenuitemProperty] = []
  202. for string in ["25%", "50%", "75%", "100%", "125%", "150%","200%"] {
  203. let item = ComponentMenuitemProperty(type: .normal, text: string)
  204. appearanceScaleItems.append(item)
  205. }
  206. appearanceScaleSelect.updateMenuItemsArr(appearanceScaleItems)
  207. appearanceScaleSelect.delegate = self
  208. appearanceTopPageRadio.properties = ComponentCheckBoxProperty(size: .s, text: KMLocalizedString("At the top of the page"), checkboxType: .normal)
  209. appearanceTopPageRadio.setTarget(self, action: #selector(radioClicked(_:)))
  210. appearanceBottomPageRadio.properties = ComponentCheckBoxProperty(size: .s, text: KMLocalizedString("At the bottom of the page"), checkboxType: .normal)
  211. appearanceBottomPageRadio.setTarget(self, action: #selector(radioClicked(_:)))
  212. //Position
  213. positionLabel.stringValue = KMLocalizedString("Position (mm)")
  214. positionLabel.textColor = ComponentLibrary.shared.getComponentColorFromKey("colorText/2")
  215. positionLabel.font = ComponentLibrary.shared.getFontFromKey("mac/body-s-medium")
  216. positionItemView.properties = ComponentCPositionProperty(rowCount: 3, columnCount: 3, dash: false, selRow: 1, selColumn: 1)
  217. positionItemView.delegate = self
  218. positionXInput.properties = ComponentInputNumberProperty(alignment: .left,
  219. size: .s,
  220. minSize: -1000,
  221. maxSize: 1000,
  222. text: "0")
  223. positionXInput.delegate = self
  224. positionYInput.properties = ComponentInputNumberProperty(alignment: .left,
  225. size: .s,
  226. minSize: -1000,
  227. maxSize: 1000,
  228. text: "0")
  229. positionYInput.delegate = self
  230. positionTileCheckbox.properties = ComponentCheckBoxProperty(size: .s, text: KMLocalizedString("Tile"), checkboxType: .normal)
  231. positionTileCheckbox.setTarget(self, action: #selector(checkBoxClicked(_:)))
  232. positionTileHoriInput.properties = ComponentInputNumberProperty(alignment: .left,
  233. size: .s,
  234. minSize: 0,
  235. maxSize: 1000,
  236. text: "0")
  237. positionTileHoriInput.delegate = self
  238. positionTileVertInput.properties = ComponentInputNumberProperty(alignment: .left,
  239. size: .s,
  240. minSize: 0,
  241. maxSize: 1000,
  242. text: "0")
  243. positionTileVertInput.delegate = self
  244. //Save
  245. saveTemplateBGView.wantsLayer = true
  246. saveTemplateBGView.layer?.backgroundColor = ComponentLibrary.shared.getComponentColorFromKey("colorBg/layout-middle").cgColor
  247. saveButton.properties = ComponentButtonProperty(type: .default_tertiary, size: .m, showLeftIcon: true, buttonText: KMLocalizedString("Save as Template"), icon: NSImage(named: "icon_wm_template_save"), keepPressState: false)
  248. saveButton.setTarget(self, action: #selector(saveButtonClicked(_:)))
  249. }
  250. func refreshScrollView() {
  251. infoScrollView.contentViewSize = CGSizeMake(264, 672)
  252. if watermarkData.watermarkType == .text {
  253. infoScrollView.contentViewSize = CGSizeMake(264, 672)
  254. } else if watermarkData.watermarkType == .image {
  255. infoScrollView.contentViewSize = CGSizeMake(264, 512)
  256. }
  257. infoScrollView.scrollToTop()
  258. }
  259. func reloadData() {
  260. titleLabel.stringValue = KMLocalizedString("Add Watermark")
  261. leftTopButton.isHidden = true
  262. saveTemplateBGView.isHidden = false
  263. templateButton.properties.isDisabled = false
  264. templateButton.properties.icon = NSImage(named: "icon_wm_template")
  265. typeTabsBGView.isHidden = false
  266. infoScrollTopConst.constant = 56
  267. if editSubType == .edit {
  268. typeTabsBGView.isHidden = true
  269. infoScrollTopConst.constant = 8
  270. titleLabel.stringValue = KMLocalizedString("Edit Template")
  271. leftTopButton.isHidden = false
  272. saveTemplateBGView.isHidden = true
  273. templateButton.properties.icon = NSImage(named: "edit_save")
  274. var isChanged = false
  275. if let dict = self.originalDataDict {
  276. isChanged = KMWatermarkManager.compareIsChangedModel(watermarkData, withDict: dict as NSDictionary)
  277. }
  278. if isChanged {
  279. templateButton.properties.isDisabled = false
  280. } else {
  281. templateButton.properties.isDisabled = true
  282. }
  283. }
  284. templateButton.reloadData()
  285. if self.isInBatchMode || editSubType == .edit {
  286. infoContendBottomConst.constant = 0
  287. saveTemplateBGView.isHidden = true
  288. } else {
  289. infoContendBottomConst.constant = 72
  290. saveTemplateBGView.isHidden = false
  291. }
  292. fileBGView.isHidden = true
  293. textBGView.isHidden = true
  294. textTabProperty.state = .normal
  295. fileTabProperty.state = .normal
  296. if watermarkData.watermarkType == .text {
  297. textBGView.isHidden = false
  298. appearanceBGTopConst.constant = 200
  299. textTabProperty.state = .pressed
  300. } else if watermarkData.watermarkType == .image {
  301. fileBGView.isHidden = false
  302. appearanceBGTopConst.constant = 48
  303. fileTabProperty.state = .pressed
  304. }
  305. typeTabs.refreshItems()
  306. //Text
  307. text_textareaView.properties.text = watermarkData.text ?? ""
  308. text_textareaView.reloadData()
  309. if let index = familyNames.firstIndex(of: watermarkData.fontName) {
  310. fontNameSelect.selectItemAtIndex(index)
  311. }
  312. fontNameSelect.properties.text = watermarkData.fontName
  313. fontNameSelect.reloadData()
  314. let styleNames = CPDFFont.fontNames(forFamilyName: watermarkData.fontName)
  315. var menuItemArr: [ComponentMenuitemProperty] = []
  316. for string in styleNames {
  317. let item = ComponentMenuitemProperty(type: .normal, text: string, identifier: string)
  318. menuItemArr.append(item)
  319. }
  320. fontStyleSelect.updateMenuItemsArr(menuItemArr)
  321. if watermarkData.fontStyle == nil {
  322. fontStyleSelect.selectItemAtIndex(0)
  323. watermarkData.fontStyle = styleNames.first
  324. } else {
  325. fontStyleSelect.properties.text = watermarkData.fontStyle
  326. fontStyleSelect.reloadData()
  327. }
  328. positionTileCheckbox.properties.isDisabled = false
  329. let fontSizeStr = String(format: "%.0f", watermarkData.fontSize)
  330. if let index = KMWatermarkManager.getFontSize().firstIndex(of: fontSizeStr) {
  331. fontSizeSelect.selectItemAtIndex(index)
  332. }
  333. fontSizeSelect.properties.text = String(format: "%.0f", watermarkData.fontSize) + " pt"
  334. fontSizeSelect.properties.isDisabled = false
  335. if watermarkData.isScale {
  336. fontSizeSelect.properties.isDisabled = true
  337. positionTileCheckbox.properties.isDisabled = true
  338. watermarkData.isTilePage = false
  339. }
  340. fontSizeSelect.reloadData()
  341. fontColorGroup.currentColor = watermarkData.textColor
  342. fontColorGroup.refreshUI()
  343. //File
  344. fileInputView.properties.text = watermarkData.imagePath ?? ""
  345. fileInputView.reloadData()
  346. //Appearance
  347. appearance_RotateSelect.properties.text = String(format: "%.0f", watermarkData.rotation) + "°"
  348. appearance_RotateSelect.reloadData()
  349. if watermarkData.opacity > 1 {
  350. watermarkData.opacity = 1
  351. }
  352. appearance_OpacitySelect.properties.text = String(format: "%.0f", watermarkData.opacity*100) + "%"
  353. appearance_OpacitySelect.reloadData()
  354. appearanceScaleCheckbox.properties.checkboxType = watermarkData.isScale ? .selected : .normal
  355. appearanceScaleCheckbox.reloadData()
  356. appearanceScaleSelect.properties.isDisabled = watermarkData.isScale ? false : true
  357. appearanceScaleSelect.properties.text = String(format: "%.0f", watermarkData.scale*100) + "%"
  358. appearanceScaleSelect.reloadData()
  359. appearanceTopPageRadio.properties.checkboxType = watermarkData.isFront ? .selected : .normal
  360. appearanceTopPageRadio.reloadData()
  361. appearanceBottomPageRadio.properties.checkboxType = watermarkData.isFront ? .normal : .selected
  362. appearanceBottomPageRadio.reloadData()
  363. //Position
  364. positionXInput.properties.text = String(format: "%.0f", watermarkData.tx)
  365. positionXInput.reloadData()
  366. positionYInput.properties.text = String(format: "%.0f", watermarkData.ty)
  367. positionYInput.reloadData()
  368. positionTileHoriInput.properties.text = String(format: "%.0f", watermarkData.horizontalSpacing)
  369. positionTileVertInput.properties.text = String(format: "%.0f", watermarkData.verticalSpacing)
  370. if watermarkData.isTilePage {
  371. positionTileHorImage.image = NSImage(named: "tile_spacing_horiz")
  372. positionTileVertImage.image = NSImage(named: "tile_spacing_vert")
  373. positionTileHoriInput.properties.isDisabled = false
  374. positionTileVertInput.properties.isDisabled = false
  375. } else {
  376. positionTileHorImage.image = NSImage(named: "tile_spacing_horiz_dis")
  377. positionTileVertImage.image = NSImage(named: "tile_spacing_vert_dis")
  378. positionTileHoriInput.properties.isDisabled = true
  379. positionTileVertInput.properties.isDisabled = true
  380. }
  381. positionTileHoriInput.reloadData()
  382. positionTileVertInput.reloadData()
  383. positionTileCheckbox.properties.checkboxType = watermarkData.isTilePage ? .selected : .normal
  384. positionTileCheckbox.reloadData()
  385. saveButton.properties.isDisabled = false
  386. if watermarkData.watermarkType == .text {
  387. if watermarkData.text?.count == 0 {
  388. saveButton.properties.isDisabled = true
  389. }
  390. } else if watermarkData.watermarkType == .image {
  391. if watermarkData.imagePath == nil {
  392. saveButton.properties.isDisabled = true
  393. }
  394. }
  395. saveButton.reloadData()
  396. }
  397. func clearData() {
  398. KMWatermarkManager.defaultManager.defaultWatermarkData = KMWatermarkModel.defaultWDData()
  399. watermarkData = KMWatermarkManager.defaultManager.defaultWatermarkData
  400. reloadData()
  401. }
  402. //MARK: - Action
  403. @objc func leftTopButtonClicked(_ sender: ComponentButton) {
  404. if sender == leftTopButton {
  405. var isChanged = false
  406. if let dict = self.originalDataDict {
  407. isChanged = KMWatermarkManager.compareIsChangedModel(watermarkData, withDict: dict as NSDictionary)
  408. }
  409. if isChanged == true {
  410. let alert = NSAlert()
  411. alert.alertStyle = .critical
  412. alert.messageText = KMLocalizedString("Save template changes?")
  413. alert.informativeText = KMLocalizedString("Cancel and they will not be saved.")
  414. alert.addButton(withTitle: KMLocalizedString("Save"))
  415. alert.addButton(withTitle: KMLocalizedString("Cancel"))
  416. alert.beginSheetModal(for: NSApp.mainWindow!) { (response) in
  417. if response == NSApplication.ModalResponse.alertFirstButtonReturn {
  418. self.templateButtonClicked(self.templateButton)
  419. } else {
  420. if let dict = self.originalDataDict {
  421. KMWatermarkManager.defaultManager.setDictToWatermarK(dict: dict as NSDictionary, self.watermarkData)
  422. let _ = KMWatermarkManager.defaultManager.updateWatermark(watermark: self.watermarkData)
  423. }
  424. self.delegate?.watermarkPropertyControllerCancelTemplateEdit?(self)
  425. }
  426. }
  427. } else {
  428. delegate?.watermarkPropertyControllerCancelTemplateEdit?(self)
  429. }
  430. }
  431. }
  432. @objc func templateButtonClicked(_ sender: ComponentButton) {
  433. if editSubType == .edit {
  434. let _ = KMWatermarkManager.defaultManager.updateWatermark(watermark: watermarkData)
  435. delegate?.watermarkPropertyControllerSaveTemplateSuccess?(self, self.watermarkData)
  436. } else {
  437. if watermarkData.watermarkType == .image, let _ = watermarkData.imagePath {
  438. watermarkData.text = nil
  439. watermarkData.watermarkType = .image
  440. } else {
  441. watermarkData.watermarkType = .text
  442. watermarkData.imagePath = nil
  443. }
  444. delegate?.watermarkPropertyControllerDidChangetoTemplate?(self)
  445. }
  446. }
  447. @objc func checkBoxClicked(_ sender: ComponentCheckBox) {
  448. if sender == appearanceScaleCheckbox {
  449. watermarkData.isScale = appearanceScaleCheckbox.properties.checkboxType == .selected ? true : false
  450. } else if sender == positionTileCheckbox {
  451. watermarkData.isTilePage = positionTileCheckbox.properties.checkboxType == .selected ? true : false
  452. }
  453. reloadData()
  454. delegate?.watermarkPropertyControllerDidUpdate?(self)
  455. }
  456. @objc func radioClicked(_ sender: ComponentRadio) {
  457. if sender == appearanceTopPageRadio {
  458. watermarkData.isFront = true
  459. } else if sender == appearanceBottomPageRadio {
  460. watermarkData.isFront = false
  461. }
  462. reloadData()
  463. delegate?.watermarkPropertyControllerDidUpdate?(self)
  464. }
  465. @objc func saveButtonClicked(_ sender: ComponentButton) {
  466. let saveWindow: KMWatermarkSaveWindow = KMWatermarkSaveWindow(windowNibName: "KMWatermarkSaveWindow")
  467. if watermarkData.watermarkType == .image {
  468. saveWindow.nameValue = watermarkData.imagePath?.lastPathComponent ?? ""
  469. } else if watermarkData.watermarkType == .text {
  470. saveWindow.nameValue = watermarkData.text ?? ""
  471. }
  472. saveWindow.saveHandler = {[weak self] string in
  473. guard let weakSelf = self else { return }
  474. DispatchQueue.main.async {
  475. weakSelf.watermarkData.watermarkName = string ?? ""
  476. if KMWatermarkManager.defaultManager.addWatermark(watermark: weakSelf.watermarkData) == true {
  477. weakSelf.delegate?.watermarkPropertyControllerSaveTemplateSuccess?(weakSelf, weakSelf.watermarkData)
  478. weakSelf.clearData()
  479. }
  480. }
  481. }
  482. saveWindow.own_beginSheetModal(for: view.window) { string in
  483. }
  484. saveWindow.titleLabel.stringValue = KMLocalizedString("Add Watermark")
  485. }
  486. @objc func chooseURLAction(_ sender: NSView) {
  487. self.chooseURLAction(sender, nil)
  488. }
  489. func chooseURLAction(_ sender: NSView, _ filePath: String? = nil) {
  490. let openPanel = NSOpenPanel()
  491. openPanel.canChooseDirectories = false
  492. openPanel.canChooseFiles = true
  493. openPanel.allowsMultipleSelection = false
  494. openPanel.allowedFileTypes = ["jpg", "jpeg", "png", "pdf"]
  495. if let path = filePath {
  496. openPanel.directoryURL = NSURL.fileURL(withPath: path)
  497. }
  498. openPanel.beginSheetModal(for: self.view.window!) { (result) in
  499. if result == NSApplication.ModalResponse.OK {
  500. guard let url = openPanel.url else { return }
  501. self.chooseFileURL = url
  502. let filePath = url.path
  503. let outFolder = KMWatermarkManager.defaultManager.watermarkImageFolder
  504. let fileName = filePath.getLastComponentDeleteExtension
  505. if filePath.extension.lowercased() == ".pdf" {
  506. let pdf = CPDFDocument(url: url)
  507. guard !pdf!.isEncrypted else { return }
  508. if let image = self.generateThumbnail(for: url) {
  509. if let outFolderPath = outFolder?.stringByAppendingPathComponent(fileName + ".png") {
  510. try?image.pngData()?.write(to: URL(fileURLWithPath: outFolderPath))
  511. if FileManager.default.fileExists(atPath: outFolderPath) {
  512. self.watermarkData.imagePath = outFolderPath
  513. }
  514. self.delegate?.watermarkPropertyControllerDidUpdate?(self)
  515. self.reloadData()
  516. }
  517. }
  518. } else {
  519. let image = NSImage(contentsOfFile: url.path)
  520. if image == nil || NSImage.isDamageImage(image, imagePath: url.path) {
  521. let alert = NSAlert()
  522. alert.alertStyle = .critical
  523. alert.messageText = "The file \"\(url.lastPathComponent)\" could not be opened."
  524. alert.informativeText = "It may be damaged or use a file format that PDF Reader Pro doesn’t recognize."
  525. alert.addButton(withTitle: "Cancel")
  526. alert.beginSheetModal(for: NSApp.mainWindow!) { (response) in
  527. if response == NSApplication.ModalResponse.alertFirstButtonReturn {
  528. // Cancel button clicked
  529. }
  530. }
  531. return
  532. }
  533. if let resultPath = outFolder?.stringByAppendingPathComponent(fileName + ".png") {
  534. try?FileManager.default.copyItem(atPath: url.path, toPath: resultPath)
  535. self.watermarkData.imagePath = resultPath
  536. }
  537. self.delegate?.watermarkPropertyControllerDidUpdate?(self)
  538. self.reloadData()
  539. }
  540. }
  541. }
  542. }
  543. func generateThumbnail(for pdfURL: URL) -> NSImage? {
  544. // 创建PDF文档
  545. guard let pdfDocument = CGPDFDocument(pdfURL as CFURL),
  546. let pdfPage = pdfDocument.page(at: 1) else {
  547. print("打开PDF文件失败")
  548. return nil
  549. }
  550. // 获取页面尺寸
  551. let pageRect = pdfPage.getBoxRect(.mediaBox)
  552. // 创建图形上下文
  553. let size = NSSize(width: 200, height: 300)
  554. let bitsPerComponent = 8
  555. let bytesPerPixel = 4
  556. let bitmapInfo = CGImageAlphaInfo.premultipliedLast.rawValue
  557. let context = CGContext(data: nil, width: Int(size.width), height: Int(size.height), bitsPerComponent: bitsPerComponent, bytesPerRow: 0, space: CGColorSpaceCreateDeviceRGB(), bitmapInfo: bitmapInfo)
  558. // 绘制PDF页面
  559. context?.saveGState()
  560. context?.translateBy(x: 0, y: size.height)
  561. context?.scaleBy(x: 1.0, y: -1.0) // 反转坐标系
  562. context?.scaleBy(x: size.width / pageRect.width, y: size.height / pageRect.height) // 缩放
  563. context?.drawPDFPage(pdfPage)
  564. context?.restoreGState()
  565. // 创建NSImage
  566. if let cgImage = context?.makeImage() {
  567. return NSImage(cgImage: cgImage, size: NSSize(width: size.width, height: size.height))
  568. } else {
  569. return nil
  570. }
  571. }
  572. //MARK: - Mouse
  573. override func mouseDown(with event: NSEvent) {
  574. super.mouseDown(with: event)
  575. view.window?.makeFirstResponder(nil)
  576. }
  577. }
  578. //MARK: - ComponentTabsDelegate
  579. extension KMNWatermarkPropertyController: ComponentTabsDelegate {
  580. func componentTabsDidSelected(_ view: ComponentTabs, _ property: ComponentTabsProperty) {
  581. if property == textTabProperty {
  582. if watermarkData.watermarkType != .text {
  583. delegate?.watermarkPropertyControllerDidRemoveWatermark?(self, .text)
  584. }
  585. watermarkData.watermarkType = .text
  586. } else if property == fileTabProperty {
  587. if watermarkData.watermarkType != .image {
  588. delegate?.watermarkPropertyControllerDidRemoveWatermark?(self, .image)
  589. }
  590. watermarkData.watermarkType = .image
  591. }
  592. reloadData()
  593. delegate?.watermarkPropertyControllerDidUpdate?(self)
  594. refreshScrollView()
  595. }
  596. }
  597. //MARK: - ComponentTextareaDelegate
  598. extension KMNWatermarkPropertyController: ComponentTextareaDelegate {
  599. func componentTextareaTextDidChange(_ view: ComponentTextarea) {
  600. watermarkData.text = view.properties.text
  601. delegate?.watermarkPropertyControllerDidUpdate?(self)
  602. }
  603. func componentTextareaTextDidEndEditing(_ view: ComponentTextarea) {
  604. watermarkData.text = view.properties.text
  605. delegate?.watermarkPropertyControllerDidUpdate?(self)
  606. }
  607. }
  608. //MARK: - ComponentCColorDelegate
  609. extension KMNWatermarkPropertyController: ComponentCColorDelegate {
  610. func componentCColorDidChooseColor(_ view: NSView, _ color: NSColor?) {
  611. if view == fontColorGroup {
  612. watermarkData.textColor = color ?? NSColor.clear
  613. }
  614. reloadData()
  615. delegate?.watermarkPropertyControllerDidUpdate?(self)
  616. }
  617. func componentCColorGroupColorsUpdates(_ view: NSView, _ colors: [NSColor]) {
  618. KMAnnotationPropertiesColorManager.manager.updateDefaultColors(colors, forKey: KM_Watermark_default_ColorsKey)
  619. }
  620. }
  621. //MARK: - ComponentSelectDelegate
  622. extension KMNWatermarkPropertyController: ComponentSelectDelegate {
  623. func componentSelectDidSelect(view: ComponentSelect?, menuItemProperty: ComponentMenuitemProperty?) {
  624. if view == fontNameSelect {
  625. watermarkData.fontName = menuItemProperty?.text ?? "Helvetica"
  626. watermarkData.fontStyle = nil
  627. } else if view == fontStyleSelect {
  628. watermarkData.fontStyle = menuItemProperty?.text
  629. } else if view == fontSizeSelect {
  630. if let text = menuItemProperty?.text {
  631. let result = text.stringByDeleteCharString(" pt")
  632. watermarkData.fontSize = result.stringToCGFloat()
  633. }
  634. } else if view == appearance_OpacitySelect {
  635. if let text = menuItemProperty?.text {
  636. let result = text.stringByDeleteCharString("%")
  637. watermarkData.opacity = result.stringToCGFloat()/100
  638. }
  639. } else if view == appearance_RotateSelect {
  640. if let text = menuItemProperty?.text {
  641. let result = text.stringByDeleteCharString("°")
  642. watermarkData.rotation = result.stringToCGFloat()
  643. }
  644. } else if view == appearanceScaleSelect {
  645. if let text = menuItemProperty?.text {
  646. let result = text.stringByDeleteCharString("%")
  647. watermarkData.scale = result.stringToCGFloat()/100
  648. }
  649. }
  650. reloadData()
  651. delegate?.watermarkPropertyControllerDidUpdate?(self)
  652. }
  653. func componentSelectTextDidChange(_ view: ComponentSelect) {
  654. if view == fontSizeSelect {
  655. if let string = view.properties.text {
  656. var size = string.stringToCGFloat()
  657. size = min(size, 288)
  658. watermarkData.fontSize = size
  659. delegate?.watermarkPropertyControllerDidUpdate?(self)
  660. }
  661. }
  662. }
  663. func componentSelectTextDidEndEditing(_ view: ComponentSelect) {
  664. if view == fontSizeSelect {
  665. if let string = view.properties.text {
  666. let result = string.stringByDeleteCharString(" pt")
  667. var size = result.stringToCGFloat()
  668. size = min(size, 288)
  669. watermarkData.fontSize = size
  670. }
  671. } else if view == appearance_OpacitySelect {
  672. if let text = appearance_OpacitySelect.properties.text {
  673. let result = text.stringByDeleteCharString("%")
  674. watermarkData.opacity = result.stringToCGFloat()/100
  675. }
  676. } else if view == appearance_RotateSelect {
  677. if let text = appearance_RotateSelect.properties.text {
  678. let result = text.stringByDeleteCharString("°")
  679. watermarkData.rotation = result.stringToCGFloat()
  680. }
  681. } else if view == appearanceScaleSelect {
  682. if let text = appearanceScaleSelect.properties.text {
  683. let result = text.stringByDeleteCharString("%")
  684. watermarkData.scale = result.stringToCGFloat()/100
  685. }
  686. }
  687. reloadData()
  688. delegate?.watermarkPropertyControllerDidUpdate?(self)
  689. }
  690. }
  691. //MARK: - ComponentCPositionDelegate
  692. extension KMNWatermarkPropertyController: ComponentCPositionDelegate {
  693. func componentCPositionDidChoose(_ view: NSView, _ row: Int, _ column: Int) {
  694. if view == positionItemView {
  695. if row == 0 {
  696. watermarkData.verticalPosition = .bottom
  697. } else if row == 1 {
  698. watermarkData.verticalPosition = .center
  699. } else if row == 2 {
  700. watermarkData.verticalPosition = .top
  701. }
  702. if column == 0 {
  703. watermarkData.horizontalPosition = .left
  704. } else if column == 1 {
  705. watermarkData.horizontalPosition = .center
  706. } else if column == 2 {
  707. watermarkData.horizontalPosition = .right
  708. }
  709. }
  710. reloadData()
  711. delegate?.watermarkPropertyControllerDidUpdate?(self)
  712. }
  713. }
  714. //MARK: - ComponentInputNumberDelegate
  715. extension KMNWatermarkPropertyController: ComponentInputNumberDelegate {
  716. func componentInputNumberDidValueChanged(inputNumber: ComponentInputNumber?) {
  717. if inputNumber == positionXInput {
  718. if let text = inputNumber?.properties.text {
  719. watermarkData.tx = text.stringToCGFloat()
  720. }
  721. } else if inputNumber == positionYInput {
  722. if let text = inputNumber?.properties.text {
  723. watermarkData.ty = text.stringToCGFloat()
  724. }
  725. } else if inputNumber == positionTileHoriInput {
  726. if let text = inputNumber?.properties.text {
  727. watermarkData.horizontalSpacing = text.stringToCGFloat()
  728. }
  729. } else if inputNumber == positionTileVertInput {
  730. if let text = inputNumber?.properties.text {
  731. watermarkData.verticalSpacing = text.stringToCGFloat()
  732. }
  733. }
  734. reloadData()
  735. delegate?.watermarkPropertyControllerDidUpdate?(self)
  736. }
  737. }
  738. //MARK: - ComponentInputDelegate
  739. extension KMNWatermarkPropertyController: ComponentInputDelegate {
  740. func componentInputDidCoverButtonClicked(inputView: ComponentInput) {
  741. if let fileURL = self.chooseFileURL, inputView.properties.creatable == false {
  742. self.chooseURLAction(inputView, fileURL.path)
  743. }
  744. }
  745. }