KMNWatermarkPropertyController.swift 37 KB

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