KMNWatermarkPropertyController.swift 42 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966
  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. public func scrollToTop() {
  260. self.infoScrollView.scrollToTop()
  261. }
  262. public func beginEditing() {
  263. if watermarkData.watermarkType == .text {
  264. self.text_textareaView.beginEditing()
  265. }
  266. }
  267. func reloadData() {
  268. titleLabel.stringValue = KMLocalizedString("Add Watermark")
  269. leftTopButton.isHidden = true
  270. saveTemplateBGView.isHidden = false
  271. templateButton.properties.isDisabled = false
  272. templateButton.properties.icon = NSImage(named: "icon_wm_template")
  273. typeTabsBGView.isHidden = false
  274. infoScrollTopConst.constant = 56
  275. if editSubType == .edit {
  276. typeTabsBGView.isHidden = true
  277. infoScrollTopConst.constant = 8
  278. titleLabel.stringValue = KMLocalizedString("Edit Template")
  279. leftTopButton.isHidden = false
  280. saveTemplateBGView.isHidden = true
  281. templateButton.properties.icon = NSImage(named: "edit_save")
  282. var isChanged = false
  283. if let dict = self.originalDataDict {
  284. isChanged = KMWatermarkManager.compareIsChangedModel(watermarkData, withDict: dict as NSDictionary)
  285. }
  286. if isChanged {
  287. templateButton.properties.isDisabled = false
  288. } else {
  289. templateButton.properties.isDisabled = true
  290. }
  291. }
  292. templateButton.reloadData()
  293. if self.isInBatchMode || editSubType == .edit {
  294. infoContendBottomConst.constant = 0
  295. saveTemplateBGView.isHidden = true
  296. } else {
  297. infoContendBottomConst.constant = 72
  298. saveTemplateBGView.isHidden = false
  299. }
  300. fileBGView.isHidden = true
  301. textBGView.isHidden = true
  302. textTabProperty.state = .normal
  303. fileTabProperty.state = .normal
  304. if watermarkData.watermarkType == .text {
  305. textBGView.isHidden = false
  306. appearanceBGTopConst.constant = 200
  307. textTabProperty.state = .pressed
  308. } else if watermarkData.watermarkType == .image {
  309. fileBGView.isHidden = false
  310. appearanceBGTopConst.constant = 48
  311. fileTabProperty.state = .pressed
  312. }
  313. typeTabs.refreshItems()
  314. //Text
  315. text_textareaView.properties.text = watermarkData.text ?? ""
  316. text_textareaView.reloadData()
  317. if let index = familyNames.firstIndex(of: watermarkData.fontName) {
  318. fontNameSelect.selectItemAtIndex(index)
  319. }
  320. fontNameSelect.properties.text = watermarkData.fontName
  321. fontNameSelect.reloadData()
  322. let styleNames = CPDFFont.fontNames(forFamilyName: watermarkData.fontName)
  323. var menuItemArr: [ComponentMenuitemProperty] = []
  324. for string in styleNames {
  325. let item = ComponentMenuitemProperty(type: .normal, text: string, identifier: string)
  326. menuItemArr.append(item)
  327. }
  328. fontStyleSelect.updateMenuItemsArr(menuItemArr)
  329. if watermarkData.fontStyle == nil {
  330. fontStyleSelect.selectItemAtIndex(0)
  331. watermarkData.fontStyle = styleNames.first
  332. } else {
  333. fontStyleSelect.properties.text = watermarkData.fontStyle
  334. fontStyleSelect.reloadData()
  335. }
  336. positionTileCheckbox.properties.isDisabled = false
  337. let fontSizeStr = String(format: "%.0f", watermarkData.fontSize)
  338. if let index = KMWatermarkManager.getFontSize().firstIndex(of: fontSizeStr) {
  339. fontSizeSelect.selectItemAtIndex(index)
  340. }
  341. fontSizeSelect.properties.text = String(format: "%.0f", watermarkData.fontSize) + " pt"
  342. fontSizeSelect.properties.isDisabled = false
  343. if watermarkData.isScale {
  344. fontSizeSelect.properties.isDisabled = true
  345. positionTileCheckbox.properties.isDisabled = true
  346. watermarkData.isTilePage = false
  347. }
  348. fontSizeSelect.reloadData()
  349. fontColorGroup.currentColor = watermarkData.textColor
  350. fontColorGroup.refreshUI()
  351. //File
  352. fileInputView.properties.text = watermarkData.imagePath ?? ""
  353. fileInputView.reloadData()
  354. //Appearance
  355. appearance_RotateSelect.properties.text = String(format: "%.0f", watermarkData.rotation) + "°"
  356. appearance_RotateSelect.reloadData()
  357. if watermarkData.opacity > 1 {
  358. watermarkData.opacity = 1
  359. }
  360. appearance_OpacitySelect.properties.text = String(format: "%.0f", watermarkData.opacity*100) + "%"
  361. appearance_OpacitySelect.reloadData()
  362. appearanceScaleCheckbox.properties.checkboxType = watermarkData.isScale ? .selected : .normal
  363. appearanceScaleCheckbox.reloadData()
  364. appearanceScaleSelect.properties.isDisabled = watermarkData.isScale ? false : true
  365. appearanceScaleSelect.properties.text = String(format: "%.0f", watermarkData.scale*100) + "%"
  366. appearanceScaleSelect.reloadData()
  367. appearanceTopPageRadio.properties.checkboxType = watermarkData.isFront ? .selected : .normal
  368. appearanceTopPageRadio.reloadData()
  369. appearanceBottomPageRadio.properties.checkboxType = watermarkData.isFront ? .normal : .selected
  370. appearanceBottomPageRadio.reloadData()
  371. //Position
  372. positionXInput.properties.text = String(format: "%.0f", watermarkData.tx)
  373. positionXInput.reloadData()
  374. positionYInput.properties.text = String(format: "%.0f", watermarkData.ty)
  375. positionYInput.reloadData()
  376. positionTileHoriInput.properties.text = String(format: "%.0f", watermarkData.horizontalSpacing)
  377. positionTileVertInput.properties.text = String(format: "%.0f", watermarkData.verticalSpacing)
  378. if watermarkData.isTilePage {
  379. positionTileHorImage.image = NSImage(named: "tile_spacing_horiz")
  380. positionTileVertImage.image = NSImage(named: "tile_spacing_vert")
  381. positionTileHoriInput.properties.isDisabled = false
  382. positionTileVertInput.properties.isDisabled = false
  383. } else {
  384. positionTileHorImage.image = NSImage(named: "tile_spacing_horiz_dis")
  385. positionTileVertImage.image = NSImage(named: "tile_spacing_vert_dis")
  386. positionTileHoriInput.properties.isDisabled = true
  387. positionTileVertInput.properties.isDisabled = true
  388. }
  389. positionTileHoriInput.reloadData()
  390. positionTileVertInput.reloadData()
  391. positionTileCheckbox.properties.checkboxType = watermarkData.isTilePage ? .selected : .normal
  392. positionTileCheckbox.reloadData()
  393. saveButton.properties.isDisabled = false
  394. if watermarkData.watermarkType == .text {
  395. if watermarkData.text?.count == 0 {
  396. saveButton.properties.isDisabled = true
  397. }
  398. } else if watermarkData.watermarkType == .image {
  399. if watermarkData.imagePath == nil {
  400. saveButton.properties.isDisabled = true
  401. }
  402. }
  403. saveButton.reloadData()
  404. var row = 0
  405. var column = 0
  406. if watermarkData.verticalPosition == .bottom {
  407. row = 0
  408. } else if watermarkData.verticalPosition == .center {
  409. row = 1
  410. } else if watermarkData.verticalPosition == .top {
  411. row = 2
  412. }
  413. if watermarkData.horizontalPosition == .left {
  414. column = 0
  415. } else if watermarkData.horizontalPosition == .center {
  416. column = 1
  417. } else if watermarkData.horizontalPosition == .right {
  418. column = 2
  419. }
  420. positionItemView.properties.selRow = row
  421. positionItemView.properties.selColumn = column
  422. positionItemView.reloadData()
  423. }
  424. func clearData() {
  425. KMWatermarkManager.defaultManager.defaultWatermarkData = KMWatermarkModel.defaultWDData()
  426. watermarkData = KMWatermarkManager.defaultManager.defaultWatermarkData
  427. reloadData()
  428. }
  429. func isOriginalDataDictChanged() -> Bool {
  430. var isChanged = false
  431. if let dict = self.originalDataDict {
  432. isChanged = KMWatermarkManager.compareIsChangedModel(watermarkData, withDict: dict as NSDictionary)
  433. }
  434. return isChanged
  435. }
  436. func showSaveEditChangeAlert(completion: @escaping (NSApplication.ModalResponse) -> Void) {
  437. let alert = NSAlert()
  438. alert.alertStyle = .critical
  439. alert.messageText = KMLocalizedString("Save template changes?")
  440. alert.informativeText = KMLocalizedString("Cancel and they will not be saved.")
  441. alert.addButton(withTitle: KMLocalizedString("Save"))
  442. alert.addButton(withTitle: KMLocalizedString("Cancel"))
  443. if let window = self.view.window {
  444. alert.beginSheetModal(for: window) { response in
  445. completion(response)
  446. }
  447. }
  448. }
  449. func cancelDataEdit() {
  450. if let dict = self.originalDataDict {
  451. KMWatermarkManager.defaultManager.setDictToWatermarK(dict: dict as NSDictionary, self.watermarkData)
  452. let _ = KMWatermarkManager.defaultManager.updateWatermark(watermark: self.watermarkData)
  453. }
  454. self.delegate?.watermarkPropertyControllerCancelTemplateEdit?(self)
  455. }
  456. func saveDataEdit() {
  457. self.templateButtonClicked(self.templateButton)
  458. }
  459. //MARK: - Action
  460. @objc func leftTopButtonClicked(_ sender: ComponentButton) {
  461. if sender == leftTopButton {
  462. var isChanged = false
  463. if let dict = self.originalDataDict {
  464. isChanged = KMWatermarkManager.compareIsChangedModel(watermarkData, withDict: dict as NSDictionary)
  465. }
  466. if isChanged == true {
  467. self.showSaveEditChangeAlert { response in
  468. if response == NSApplication.ModalResponse.alertFirstButtonReturn {
  469. //OK
  470. self.saveDataEdit()
  471. } else {
  472. //Cancel
  473. self.cancelDataEdit()
  474. }
  475. }
  476. } else {
  477. delegate?.watermarkPropertyControllerCancelTemplateEdit?(self)
  478. }
  479. }
  480. }
  481. @objc func templateButtonClicked(_ sender: ComponentButton) {
  482. if editSubType == .edit {
  483. let _ = KMWatermarkManager.defaultManager.updateWatermark(watermark: watermarkData)
  484. delegate?.watermarkPropertyControllerSaveTemplateSuccess?(self, self.watermarkData)
  485. } else {
  486. if watermarkData.watermarkType == .image, let _ = watermarkData.imagePath {
  487. watermarkData.text = nil
  488. watermarkData.watermarkType = .image
  489. } else {
  490. watermarkData.watermarkType = .text
  491. watermarkData.imagePath = nil
  492. }
  493. delegate?.watermarkPropertyControllerDidChangetoTemplate?(self)
  494. }
  495. }
  496. @objc func checkBoxClicked(_ sender: ComponentCheckBox) {
  497. if sender == appearanceScaleCheckbox {
  498. watermarkData.isScale = appearanceScaleCheckbox.properties.checkboxType == .selected ? true : false
  499. } else if sender == positionTileCheckbox {
  500. watermarkData.isTilePage = positionTileCheckbox.properties.checkboxType == .selected ? true : false
  501. }
  502. reloadData()
  503. delegate?.watermarkPropertyControllerDidUpdate?(self)
  504. }
  505. @objc func radioClicked(_ sender: ComponentRadio) {
  506. if sender == appearanceTopPageRadio {
  507. watermarkData.isFront = true
  508. } else if sender == appearanceBottomPageRadio {
  509. watermarkData.isFront = false
  510. }
  511. reloadData()
  512. delegate?.watermarkPropertyControllerDidUpdate?(self)
  513. }
  514. @objc func saveButtonClicked(_ sender: ComponentButton) {
  515. let saveWindow: KMWatermarkSaveWindow = KMWatermarkSaveWindow(windowNibName: "KMWatermarkSaveWindow")
  516. if watermarkData.watermarkType == .image {
  517. saveWindow.nameValue = watermarkData.imagePath?.lastPathComponent ?? ""
  518. } else if watermarkData.watermarkType == .text {
  519. saveWindow.nameValue = watermarkData.text ?? ""
  520. }
  521. saveWindow.saveHandler = {[weak self] string in
  522. guard let weakSelf = self else { return }
  523. DispatchQueue.main.async {
  524. weakSelf.watermarkData.watermarkName = string ?? ""
  525. if KMWatermarkManager.defaultManager.addWatermark(watermark: weakSelf.watermarkData) == true {
  526. weakSelf.delegate?.watermarkPropertyControllerSaveTemplateSuccess?(weakSelf, weakSelf.watermarkData)
  527. weakSelf.clearData()
  528. }
  529. }
  530. }
  531. saveWindow.own_beginSheetModal(for: view.window) { string in
  532. }
  533. saveWindow.titleLabel.stringValue = KMLocalizedString("Add Watermark")
  534. }
  535. @objc func chooseURLAction(_ sender: NSView) {
  536. self.chooseURLAction(sender, nil)
  537. }
  538. func chooseURLAction(_ sender: NSView, _ filePath: String? = nil) {
  539. let openPanel = NSOpenPanel()
  540. openPanel.canChooseDirectories = false
  541. openPanel.canChooseFiles = true
  542. openPanel.allowsMultipleSelection = false
  543. openPanel.allowedFileTypes = ["jpg", "jpeg", "png", "pdf"]
  544. if let path = filePath {
  545. openPanel.directoryURL = NSURL.fileURL(withPath: path)
  546. }
  547. openPanel.beginSheetModal(for: self.view.window!) { (result) in
  548. if result == NSApplication.ModalResponse.OK {
  549. guard let url = openPanel.url else { return }
  550. self.chooseFileURL = url
  551. let filePath = url.path
  552. let outFolder = KMWatermarkManager.defaultManager.watermarkImageFolder
  553. let fileName = filePath.getLastComponentDeleteExtension
  554. if filePath.extension.lowercased() == ".pdf" {
  555. let pdf = CPDFDocument(url: url)
  556. guard !pdf!.isEncrypted else { return }
  557. if let image = self.generateThumbnail(for: url) {
  558. if let outFolderPath = outFolder?.stringByAppendingPathComponent(fileName + ".png") {
  559. try?image.pngData()?.write(to: URL(fileURLWithPath: outFolderPath))
  560. if FileManager.default.fileExists(atPath: outFolderPath) {
  561. self.watermarkData.imagePath = outFolderPath
  562. }
  563. self.delegate?.watermarkPropertyControllerDidUpdate?(self)
  564. self.reloadData()
  565. }
  566. }
  567. } else {
  568. let image = NSImage(contentsOfFile: url.path)
  569. if image == nil || NSImage.isDamageImage(image, imagePath: url.path) {
  570. let alert = NSAlert()
  571. alert.alertStyle = .critical
  572. alert.messageText = "The file \"\(url.lastPathComponent)\" could not be opened."
  573. alert.informativeText = "It may be damaged or use a file format that PDF Reader Pro doesn’t recognize."
  574. alert.addButton(withTitle: "Cancel")
  575. alert.beginSheetModal(for: NSApp.mainWindow!) { (response) in
  576. if response == NSApplication.ModalResponse.alertFirstButtonReturn {
  577. // Cancel button clicked
  578. }
  579. }
  580. return
  581. }
  582. if let resultPath = outFolder?.stringByAppendingPathComponent(fileName + ".png") {
  583. try?FileManager.default.copyItem(atPath: url.path, toPath: resultPath)
  584. self.watermarkData.imagePath = resultPath
  585. }
  586. self.delegate?.watermarkPropertyControllerDidUpdate?(self)
  587. self.reloadData()
  588. }
  589. }
  590. }
  591. }
  592. func generateThumbnail(for pdfURL: URL) -> NSImage? {
  593. // 创建PDF文档
  594. guard let pdfDocument = CGPDFDocument(pdfURL as CFURL),
  595. let pdfPage = pdfDocument.page(at: 1) else {
  596. print("打开PDF文件失败")
  597. return nil
  598. }
  599. // 获取页面尺寸
  600. let pageRect = pdfPage.getBoxRect(.mediaBox)
  601. // 创建图形上下文
  602. let size = NSSize(width: 200, height: 300)
  603. let bitsPerComponent = 8
  604. let bytesPerPixel = 4
  605. let bitmapInfo = CGImageAlphaInfo.premultipliedLast.rawValue
  606. let context = CGContext(data: nil, width: Int(size.width), height: Int(size.height), bitsPerComponent: bitsPerComponent, bytesPerRow: 0, space: CGColorSpaceCreateDeviceRGB(), bitmapInfo: bitmapInfo)
  607. // 绘制PDF页面
  608. context?.saveGState()
  609. context?.translateBy(x: 0, y: size.height)
  610. context?.scaleBy(x: 1.0, y: -1.0) // 反转坐标系
  611. context?.scaleBy(x: size.width / pageRect.width, y: size.height / pageRect.height) // 缩放
  612. context?.drawPDFPage(pdfPage)
  613. context?.restoreGState()
  614. // 创建NSImage
  615. if let cgImage = context?.makeImage() {
  616. return NSImage(cgImage: cgImage, size: NSSize(width: size.width, height: size.height))
  617. } else {
  618. return nil
  619. }
  620. }
  621. //MARK: - Mouse
  622. override func mouseDown(with event: NSEvent) {
  623. super.mouseDown(with: event)
  624. view.window?.makeFirstResponder(nil)
  625. }
  626. }
  627. //MARK: - ComponentTabsDelegate
  628. extension KMNWatermarkPropertyController: ComponentTabsDelegate {
  629. func componentTabsDidSelected(_ view: ComponentTabs, _ property: ComponentTabsProperty) {
  630. if property == textTabProperty {
  631. if watermarkData.watermarkType != .text {
  632. delegate?.watermarkPropertyControllerDidRemoveWatermark?(self, .text)
  633. }
  634. watermarkData.watermarkType = .text
  635. } else if property == fileTabProperty {
  636. if watermarkData.watermarkType != .image {
  637. delegate?.watermarkPropertyControllerDidRemoveWatermark?(self, .image)
  638. }
  639. watermarkData.watermarkType = .image
  640. }
  641. reloadData()
  642. delegate?.watermarkPropertyControllerDidUpdate?(self)
  643. refreshScrollView()
  644. }
  645. }
  646. //MARK: - ComponentTextareaDelegate
  647. extension KMNWatermarkPropertyController: ComponentTextareaDelegate {
  648. func componentTextareaTextDidChange(_ view: ComponentTextarea) {
  649. watermarkData.text = view.properties.text
  650. delegate?.watermarkPropertyControllerDidUpdate?(self)
  651. }
  652. func componentTextareaTextDidEndEditing(_ view: ComponentTextarea) {
  653. watermarkData.text = view.properties.text
  654. delegate?.watermarkPropertyControllerDidUpdate?(self)
  655. }
  656. }
  657. //MARK: - ComponentCColorDelegate
  658. extension KMNWatermarkPropertyController: ComponentCColorDelegate {
  659. func componentCColorDidChooseColor(_ view: NSView, _ color: NSColor?) {
  660. if view == fontColorGroup {
  661. watermarkData.textColor = color ?? NSColor.clear
  662. }
  663. reloadData()
  664. delegate?.watermarkPropertyControllerDidUpdate?(self)
  665. }
  666. func componentCColorGroupColorsUpdates(_ view: NSView, _ colors: [NSColor]) {
  667. KMAnnotationPropertiesColorManager.manager.updateDefaultColors(colors, forKey: KM_Watermark_default_ColorsKey)
  668. }
  669. }
  670. //MARK: - ComponentSelectDelegate
  671. extension KMNWatermarkPropertyController: ComponentSelectDelegate {
  672. func componentSelectDidSelect(view: ComponentSelect?, menuItemProperty: ComponentMenuitemProperty?) {
  673. if view == fontNameSelect {
  674. watermarkData.fontName = menuItemProperty?.text ?? "Helvetica"
  675. watermarkData.fontStyle = nil
  676. } else if view == fontStyleSelect {
  677. watermarkData.fontStyle = menuItemProperty?.text
  678. } else if view == fontSizeSelect {
  679. if let text = menuItemProperty?.text {
  680. let result = text.stringByDeleteCharString(" pt")
  681. watermarkData.fontSize = result.stringToCGFloat()
  682. }
  683. } else if view == appearance_OpacitySelect {
  684. if let text = menuItemProperty?.text {
  685. let result = text.stringByDeleteCharString("%")
  686. watermarkData.opacity = result.stringToCGFloat()/100
  687. }
  688. } else if view == appearance_RotateSelect {
  689. if let text = menuItemProperty?.text {
  690. let result = text.stringByDeleteCharString("°")
  691. watermarkData.rotation = result.stringToCGFloat()
  692. }
  693. } else if view == appearanceScaleSelect {
  694. if let text = menuItemProperty?.text {
  695. let result = text.stringByDeleteCharString("%")
  696. watermarkData.scale = result.stringToCGFloat()/100
  697. }
  698. }
  699. reloadData()
  700. delegate?.watermarkPropertyControllerDidUpdate?(self)
  701. }
  702. func componentSelectTextDidChange(_ view: ComponentSelect) {
  703. if view == fontSizeSelect {
  704. if let string = view.properties.text {
  705. var size = string.stringToCGFloat()
  706. size = min(size, 288)
  707. watermarkData.fontSize = size
  708. delegate?.watermarkPropertyControllerDidUpdate?(self)
  709. }
  710. }
  711. }
  712. func componentSelectTextDidEndEditing(_ view: ComponentSelect) {
  713. if view == fontSizeSelect {
  714. if let string = view.properties.text {
  715. let result = string.stringByDeleteCharString(" pt")
  716. var size = result.stringToCGFloat()
  717. size = min(size, 288)
  718. watermarkData.fontSize = size
  719. }
  720. } else if view == appearance_OpacitySelect {
  721. if let text = appearance_OpacitySelect.properties.text {
  722. let result = text.stringByDeleteCharString("%")
  723. watermarkData.opacity = result.stringToCGFloat()/100
  724. }
  725. } else if view == appearance_RotateSelect {
  726. if let text = appearance_RotateSelect.properties.text {
  727. let result = text.stringByDeleteCharString("°")
  728. watermarkData.rotation = result.stringToCGFloat()
  729. }
  730. } else if view == appearanceScaleSelect {
  731. if let text = appearanceScaleSelect.properties.text {
  732. let result = text.stringByDeleteCharString("%")
  733. watermarkData.scale = result.stringToCGFloat()/100
  734. }
  735. }
  736. reloadData()
  737. delegate?.watermarkPropertyControllerDidUpdate?(self)
  738. }
  739. }
  740. //MARK: - ComponentCPositionDelegate
  741. extension KMNWatermarkPropertyController: ComponentCPositionDelegate {
  742. func componentCPositionDidChoose(_ view: NSView, _ row: Int, _ column: Int) {
  743. if view == positionItemView {
  744. if row == 0 {
  745. watermarkData.verticalPosition = .bottom
  746. } else if row == 1 {
  747. watermarkData.verticalPosition = .center
  748. } else if row == 2 {
  749. watermarkData.verticalPosition = .top
  750. }
  751. if column == 0 {
  752. watermarkData.horizontalPosition = .left
  753. } else if column == 1 {
  754. watermarkData.horizontalPosition = .center
  755. } else if column == 2 {
  756. watermarkData.horizontalPosition = .right
  757. }
  758. }
  759. reloadData()
  760. delegate?.watermarkPropertyControllerDidUpdate?(self)
  761. }
  762. }
  763. //MARK: - ComponentInputNumberDelegate
  764. extension KMNWatermarkPropertyController: ComponentInputNumberDelegate {
  765. func componentInputNumberDidValueChanged(inputNumber: ComponentInputNumber?) {
  766. if inputNumber == positionXInput {
  767. if let text = inputNumber?.properties.text {
  768. watermarkData.tx = text.stringToCGFloat()
  769. }
  770. } else if inputNumber == positionYInput {
  771. if let text = inputNumber?.properties.text {
  772. watermarkData.ty = text.stringToCGFloat()
  773. }
  774. } else if inputNumber == positionTileHoriInput {
  775. if let text = inputNumber?.properties.text {
  776. watermarkData.horizontalSpacing = text.stringToCGFloat()
  777. }
  778. } else if inputNumber == positionTileVertInput {
  779. if let text = inputNumber?.properties.text {
  780. watermarkData.verticalSpacing = text.stringToCGFloat()
  781. }
  782. }
  783. reloadData()
  784. delegate?.watermarkPropertyControllerDidUpdate?(self)
  785. }
  786. }
  787. //MARK: - ComponentInputDelegate
  788. extension KMNWatermarkPropertyController: ComponentInputDelegate {
  789. func componentInputDidCoverButtonClicked(inputView: ComponentInput) {
  790. if let fileURL = self.chooseFileURL, inputView.properties.creatable == false {
  791. self.chooseURLAction(inputView, fileURL.path)
  792. }
  793. }
  794. }