KMWatermarkView.swift 42 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968
  1. //
  2. // KMWatermarkView.swift
  3. // PDF Master
  4. //
  5. // Created by lizhe on 2023/11/14.
  6. //
  7. import Cocoa
  8. enum KMWatermarkManagerType {
  9. case add
  10. case edit
  11. case use
  12. }
  13. typealias KMWatermarkViewOperateCallBack = (_ watermark: KMWatermarkModel, _ countType: Int) -> ()
  14. typealias KMWatermarkViewCancelAction = (_ view: KMWatermarkView) -> Void
  15. class KMWatermarkView: KMBaseXibView, NSTextFieldDelegate, NSComboBoxDelegate, NSTextViewDelegate, CPDFViewDelegate {
  16. //顶部box
  17. @IBOutlet var typeBox: NSBox!
  18. @IBOutlet var txtButton: NSButton!
  19. @IBOutlet var fileButton: NSButton!
  20. @IBOutlet var filePathLabel: NSTextField!
  21. @IBOutlet var browsebutton: NSButton!
  22. @IBOutlet var ratioLabel: NSTextField!
  23. @IBOutlet var ratioTextField: NSTextField!
  24. @IBOutlet var ratioStepper: NSStepper!
  25. @IBOutlet var fontLabel: NSTextField!
  26. @IBOutlet var fontComboBox: NSComboBox!
  27. @IBOutlet var fontSizeLabel: NSTextField!
  28. @IBOutlet var fontSizeCombobox: NSComboBox!
  29. @IBOutlet var colorLabel: NSTextField!
  30. @IBOutlet var colorWell: NSColorWell!
  31. @IBOutlet var watermarkStringTextView: NSTextView!
  32. var maskView: KMBookletMaskView?
  33. //底部box
  34. @IBOutlet var appearanceBox: NSBox!
  35. @IBOutlet var angleLabel: NSTextField!
  36. @IBOutlet var angleTextF: NSTextField!
  37. @IBOutlet var angleStepper: NSStepper!
  38. @IBOutlet var left45IndicateView: KMAngleIndicateView!
  39. @IBOutlet var horizontalIndicateView: KMAngleIndicateView!
  40. @IBOutlet var right45IndicateView: KMAngleIndicateView!
  41. @IBOutlet var alphaLabel: NSTextField!
  42. @IBOutlet var alphaSlider: NSSlider!
  43. @IBOutlet var alphaTextField: NSTextField!
  44. @IBOutlet var alphaStepper: NSStepper!
  45. @IBOutlet var watermarkPositionLabel: NSTextField!
  46. @IBOutlet var OnPageButton: NSButton!
  47. @IBOutlet var backPageButton: NSButton!
  48. @IBOutlet var positionView: KMPostionIndicateView!
  49. @IBOutlet var horizentalGapLabel: NSTextField!
  50. @IBOutlet var verticalGapLabel: NSTextField!
  51. @IBOutlet var horizentalGapTextField: NSTextField!
  52. @IBOutlet var verticalFapTextField: NSTextField!
  53. @IBOutlet var horizentalGapStepper: NSStepper!
  54. @IBOutlet var verticalGapStepper: NSStepper!
  55. @IBOutlet var pageRangeLabel: NSTextField!
  56. @IBOutlet var pageRangeComboBox: NSComboBox!
  57. //保存到模版按钮
  58. @IBOutlet var saveToTemplateButton: NSButton!
  59. @IBOutlet var templateNameLabel: NSTextField!
  60. @IBOutlet var templateNameTextField: NSTextField!
  61. @IBOutlet var isTileWater: NSButton!
  62. @IBOutlet var spacingHorTextField: NSTextField!
  63. @IBOutlet var spacingHorStepper: NSStepper!
  64. @IBOutlet var horizentalIconButton: NSButton!
  65. @IBOutlet var verticalIconButton: NSButton!
  66. @IBOutlet var spacingVerTextField: NSTextField!
  67. @IBOutlet var spacingStepper: NSStepper!
  68. //底部按钮
  69. @IBOutlet var batchButton: NSButton!
  70. @IBOutlet var cancelButton: NSButton!
  71. @IBOutlet var applyButton: NSButton!
  72. //左侧
  73. @IBOutlet var previewPageButton: NSButton!
  74. @IBOutlet var nextPageButton: NSButton!
  75. @IBOutlet var currentPageIndexTextF: NSTextField!
  76. @IBOutlet var totalPageCountLabel: NSTextField!
  77. @IBOutlet var pdfView: KMWatermarkPDFView!
  78. var fontModified: Bool = false
  79. var currentType: Int = 0
  80. var filePath: String = Bundle.main.path(forResource: "Quick Start Guide", ofType: "pdf") ?? ""
  81. var password: String = ""
  82. var originalWatermark: KMWatermarkModel = KMWatermarkModel()
  83. var watermark: KMWatermarkModel = KMWatermarkModel()
  84. var type: KMWatermarkManagerType = .add
  85. var pdfDocument: CPDFDocument? {
  86. didSet {
  87. self.pdfView.document = pdfDocument
  88. self.reloadData()
  89. }
  90. }
  91. var onlyManagerTemplate: Bool = false
  92. var cancelAction: KMWatermarkViewCancelAction?
  93. var operateCallBack: KMWatermarkViewOperateCallBack?
  94. convenience init(baseFile filePath: String, watermark: KMWatermarkModel, password: String, type: KMWatermarkManagerType, fileType countType: Int) {
  95. self.init()
  96. self.filePath = filePath
  97. self.password = password
  98. self.originalWatermark = watermark.copy() as! KMWatermarkModel
  99. self.watermark = watermark.copy() as! KMWatermarkModel
  100. self.type = type
  101. self.pdfDocument = CPDFDocument(url: URL(fileURLWithPath: self.filePath))
  102. self.currentType = countType
  103. if pdfDocument?.isLocked ?? false {
  104. pdfDocument?.unlock(withPassword: password)
  105. }
  106. }
  107. override func setup() {
  108. pdfView.watermark = watermark
  109. pdfView.document = pdfDocument
  110. pdfView.autoScales = true
  111. // pdfView.documentView?.enclosingScrollView?.hasVerticalScroller = false
  112. // pdfView.documentView?.enclosingScrollView?.hasHorizontalScroller = false
  113. for i in 0..<3 {
  114. for j in 0..<3 {
  115. if i == Int(watermark.horizontalMode) && j == Int(watermark.verticalMode) {
  116. positionView.style = KMPositionIndicateViewStyle(rawValue: i + 3 * j)!
  117. }
  118. }
  119. }
  120. positionView.styleChangedCallBack = { [weak self] in
  121. guard let self = self else { return }
  122. self.watermark.horizontalMode = self.positionView.style.rawValue % 3
  123. self.watermark.verticalMode = self.positionView.style.rawValue / 3
  124. self.pdfView.needsDisplay = true
  125. }
  126. currentPageIndexTextF.stringValue = "1"
  127. // let numberFormatter = currentPageIndexTextF.formatter as! NumberFormatter
  128. // numberFormatter.maximum = NSNumber(value: pdfDocument!.pageCount)
  129. pdfView.delegate = self
  130. left45IndicateView.style = .left45
  131. left45IndicateView.touchCallBack = { [weak self] in
  132. guard let self = self else { return }
  133. self.watermark.rotation = -45
  134. self.angleStepper.doubleValue = self.watermark.rotation
  135. self.angleTextF.stringValue = "\(self.angleStepper.intValue)"
  136. self.checkAngle()
  137. self.pdfView.needsDisplay = true
  138. }
  139. horizontalIndicateView.style = .horizontal
  140. horizontalIndicateView.touchCallBack = { [weak self] in
  141. guard let self = self else { return }
  142. self.watermark.rotation = 0
  143. self.angleStepper.doubleValue = 0
  144. self.angleTextF.stringValue = "\(0)"
  145. self.checkAngle()
  146. self.pdfView.needsDisplay = true
  147. }
  148. right45IndicateView.style = .right45
  149. right45IndicateView.touchCallBack = { [weak self] in
  150. guard let self = self else { return }
  151. self.watermark.rotation = 45
  152. self.angleStepper.doubleValue = self.watermark.rotation
  153. self.angleTextF.stringValue = "\(self.angleStepper.intValue)"
  154. self.checkAngle()
  155. self.pdfView.needsDisplay = true
  156. }
  157. if !onlyManagerTemplate {
  158. saveToTemplateButton.isEnabled = false
  159. }
  160. if type == .use {
  161. saveToTemplateButton.isHidden = true
  162. saveToTemplateButton.state = .off
  163. } else {
  164. saveToTemplateButton.isHidden = false
  165. saveToTemplateButton.state = .on
  166. }
  167. if type == .add {
  168. applyButton.title = NSLocalizedString("Apply", comment: "")
  169. batchButton.isHidden = true
  170. } else if type == .edit {
  171. applyButton.title = NSLocalizedString("Apply", comment: "")
  172. batchButton.isHidden = true
  173. } else if type == .use {
  174. applyButton.title = NSLocalizedString("Save & Apply", comment: "")
  175. }
  176. filePathLabel.stringValue = ""
  177. filePathLabel.placeholderString = NSLocalizedString("Select a File", comment: "")
  178. fontSizeCombobox.removeAllItems()
  179. fontSizeCombobox.addItems(withObjectValues: ["9","10","11","12","13","14","18","24","36","48","64","72","96","144","288"])
  180. fontSizeCombobox.formatter = TextFieldFormatter()
  181. watermarkStringTextView.string = watermark.text
  182. pageRangeComboBox.removeAllItems()
  183. pageRangeComboBox.addItems(withObjectValues: [
  184. NSLocalizedString("All Pages", comment: ""),
  185. NSLocalizedString("Odd Pages Only", comment: ""),
  186. NSLocalizedString("Even Pages Only", comment: ""),
  187. NSLocalizedString("e.g. 1,3-5,10", comment: "")
  188. ])
  189. pageRangeComboBox.placeholderString = NSLocalizedString("e.g. 1,3-5,10", comment: "")
  190. pageRangeComboBox.delegate = nil
  191. pageRangeComboBox.selectItem(at: 0)
  192. pageRangeComboBox.isEditable = false
  193. pageRangeComboBox.delegate = self
  194. fontSizeCombobox.wantsLayer = true
  195. pageRangeComboBox.wantsLayer = true
  196. fontSizeCombobox.layer?.cornerRadius = 3.0
  197. pageRangeComboBox.layer?.cornerRadius = 3.0
  198. }
  199. override func updateLanguage() {
  200. typeBox.title = NSLocalizedString("Source", comment: "")
  201. typeBox.titleFont = NSFont.systemFont(ofSize: 13)
  202. txtButton.title = NSLocalizedString("Text", comment: "")
  203. fileButton.title = NSLocalizedString("File", comment: "")
  204. browsebutton.title = NSLocalizedString("Choose...", comment: "")
  205. ratioLabel.stringValue = "\(NSLocalizedString("Ratio", comment: "")):"
  206. fontLabel.stringValue = NSLocalizedString("Font", comment: "")
  207. horizentalIconButton.toolTip = NSLocalizedString("Horizontal Distance", comment: "")
  208. verticalIconButton.toolTip = NSLocalizedString("Vertical Distance", comment: "")
  209. fontSizeLabel.stringValue = "\(NSLocalizedString("Font Size", comment: "")):"
  210. colorLabel.stringValue = "\(NSLocalizedString("Color", comment: "")):"
  211. colorWell.color = watermark.waterTextColor
  212. templateNameTextField.stringValue = watermark.watermarkID ?? ""
  213. appearanceBox.title = NSLocalizedString("Appearance", comment: "")
  214. appearanceBox.titleFont = NSFont.systemFont(ofSize: 13)
  215. angleLabel.stringValue = "\(NSLocalizedString("Rotation", comment: "")):"
  216. alphaLabel.stringValue = "\(NSLocalizedString("Opacity", comment: "")):"
  217. watermarkPositionLabel.stringValue = "\(NSLocalizedString("Position", comment: "")):"
  218. OnPageButton.title = NSLocalizedString("Appear on top of page", comment: "")
  219. backPageButton.title = NSLocalizedString("Appear behind page", comment: "")
  220. pageRangeLabel.stringValue = "\(NSLocalizedString("Page Range", comment: "")):"
  221. isTileWater.title = NSLocalizedString("Tile", comment: "")
  222. OnPageButton.toolTip = NSLocalizedString("Appear on top of page", comment: "")
  223. backPageButton.toolTip = NSLocalizedString("Appear behind page", comment: "")
  224. pageRangeLabel.toolTip = "\(NSLocalizedString("Page Range", comment: "")):"
  225. horizentalGapLabel.stringValue = "X:"
  226. verticalGapLabel.stringValue = "Y:"
  227. saveToTemplateButton.title = NSLocalizedString("Add to Template", comment: "")
  228. templateNameLabel.stringValue = NSLocalizedString("Name:", comment: "")
  229. batchButton.title = NSLocalizedString("Batch", comment: "")
  230. cancelButton.title = NSLocalizedString("Cancel", comment: "")
  231. }
  232. override func reloadData() {
  233. guard let pdfDocument = pdfDocument else { return }
  234. let opacity = round(watermark.opacity * 100) / 100
  235. totalPageCountLabel.stringValue = "/ \(pdfDocument.pageCount)"
  236. fontSizeCombobox.stringValue = "\(watermark.textFontSize)"
  237. angleStepper.doubleValue = Double(watermark.rotation)
  238. angleTextF.stringValue = "\(angleStepper.doubleValue)"
  239. OnPageButton.state = watermark.isFront ? .on : .off
  240. backPageButton.state = watermark.isFront ? .off : .on
  241. alphaSlider.doubleValue = watermark.opacity
  242. alphaStepper.doubleValue = watermark.opacity
  243. alphaTextField.stringValue = "\(Int(opacity * 100))%"
  244. ratioStepper.doubleValue = watermark.scale
  245. ratioTextField.stringValue = "\(Int(opacity * 100))%"
  246. fontComboBox.stringValue = "\(watermark.textFontSize )"
  247. filePathLabel.stringValue = watermark.imagePath ?? ""
  248. if currentType == 0 {
  249. changeTypeBoxState(isTextWatermark: true)
  250. } else {
  251. changeTypeBoxState(isTextWatermark: false)
  252. }
  253. switch watermark.pageRangeType {
  254. case .all:
  255. pageRangeComboBox.isEditable = false
  256. pageRangeComboBox.selectItem(at: 0)
  257. case .odd:
  258. pageRangeComboBox.isEditable = false
  259. pageRangeComboBox.selectItem(at: 1)
  260. case .even:
  261. pageRangeComboBox.isEditable = false
  262. pageRangeComboBox.selectItem(at: 2)
  263. case .other:
  264. pageRangeComboBox.isEditable = true
  265. pageRangeComboBox.selectItem(at: 3)
  266. pageRangeComboBox.stringValue = watermark.pagesString
  267. }
  268. verticalGapStepper.doubleValue = Double(watermark.verticalSpace)
  269. verticalFapTextField.stringValue = "\(verticalGapStepper.doubleValue)"
  270. horizentalGapStepper.doubleValue = Double(watermark.horizontalSpace)
  271. horizentalGapTextField.stringValue = "\(horizentalGapStepper.doubleValue)"
  272. spacingHorTextField.stringValue = "\(watermark.tileHorizontalSpace)"
  273. spacingVerTextField.stringValue = "\(watermark.tileVerticalSpace)"
  274. spacingHorStepper.doubleValue = Double(watermark.tileHorizontalSpace)
  275. spacingStepper.doubleValue = Double(watermark.tileVerticalSpace)
  276. if watermark.isTilePage {
  277. isTileWater.state = .on
  278. spacingHorTextField.isEnabled = true
  279. spacingHorStepper.isEnabled = true
  280. spacingVerTextField.isEnabled = true
  281. spacingStepper.isEnabled = true
  282. } else {
  283. isTileWater.state = .off
  284. spacingHorTextField.isEnabled = false
  285. spacingHorStepper.isEnabled = false
  286. spacingVerTextField.isEnabled = false
  287. spacingStepper.isEnabled = false
  288. }
  289. pageRangeComboxAction(Any.self)
  290. checkAngle()
  291. }
  292. func checkAngle() {
  293. left45IndicateView.isSelcted = false
  294. horizontalIndicateView.isSelcted = false
  295. right45IndicateView.isSelcted = false
  296. switch watermark.rotation {
  297. case -45:
  298. left45IndicateView.isSelcted = true
  299. case 0:
  300. horizontalIndicateView.isSelcted = true
  301. case 45:
  302. right45IndicateView.isSelcted = true
  303. default:
  304. break
  305. }
  306. }
  307. func changeTypeBoxState(isTextWatermark: Bool) {
  308. if isTextWatermark {
  309. txtButton.state = .on
  310. watermarkStringTextView.isEditable = true
  311. watermarkStringTextView.isSelectable = true
  312. fontSizeCombobox.isEnabled = true
  313. colorWell.isEnabled = true
  314. fileButton.state = .off
  315. browsebutton.isEnabled = false
  316. ratioTextField.isEnabled = false
  317. ratioStepper.isEnabled = false
  318. watermark.text = watermarkStringTextView.string
  319. watermarkStringTextView.string = watermarkStringTextView.string
  320. watermark.image = nil
  321. if watermarkStringTextView.string.isEmpty == false {
  322. applyButton.isEnabled = true
  323. batchButton.isEnabled = true
  324. } else {
  325. applyButton.isEnabled = false
  326. batchButton.isEnabled = false
  327. }
  328. } else {
  329. txtButton.state = .off
  330. watermarkStringTextView.isEditable = false
  331. watermarkStringTextView.isSelectable = false
  332. fontSizeCombobox.isEnabled = false
  333. colorWell.isEnabled = false
  334. fileButton.state = .on
  335. browsebutton.isEnabled = true
  336. ratioTextField.isEnabled = true
  337. ratioStepper.isEnabled = true
  338. watermark.text = ""
  339. watermark.image = NSImage(contentsOfFile: watermark.imagePath ?? "")
  340. if filePathLabel.stringValue.isEmpty == false {
  341. applyButton.isEnabled = true
  342. batchButton.isEnabled = true
  343. } else {
  344. applyButton.isEnabled = false
  345. batchButton.isEnabled = false
  346. }
  347. }
  348. pdfView.needsDisplay = true
  349. }
  350. func checkPageRangeValidate(_ pageRangeString: String) -> Bool {
  351. let fileAttribute = KMFileAttribute()
  352. fileAttribute.filePath = (self.pdfDocument?.documentURL.path)!
  353. fileAttribute.bAllPage = false
  354. fileAttribute.pagesString = self.pageRangeComboBox.stringValue
  355. if fileAttribute.fetchSelectPages().count == 0 {
  356. return false
  357. }
  358. return true
  359. }
  360. @objc func controlTextDidEndEditing(_ obj: Notification) {
  361. if obj.object as? NSComboBox == self.fontSizeCombobox {
  362. self.watermark.textFontSize = CGFloat(Int(self.fontSizeCombobox.stringValue) ?? 0)
  363. if self.watermark.textFontSize == 0 {
  364. self.watermark.textFontSize = 48
  365. if self.watermark.isTilePage {
  366. self.watermark.textFontSize = 18
  367. }
  368. self.fontSizeCombobox.stringValue = "\(self.watermark.textFontSize)"
  369. }
  370. self.pdfView.needsDisplay = true
  371. } else if obj.object as? NSTextField == self.ratioTextField {
  372. let formatter = self.ratioTextField.formatter as? NumberFormatter
  373. let floatValue = formatter?.number(from: self.ratioTextField.stringValue)?.floatValue ?? 0
  374. self.ratioStepper.doubleValue = Double(self.watermark.scale)
  375. self.pdfView.needsDisplay = true
  376. } else if obj.object as? NSTextField == self.angleTextF {
  377. self.watermark.rotation = CGFloat(Int(self.angleTextF.stringValue) ?? 0)
  378. self.angleStepper.doubleValue = self.watermark.rotation
  379. self.checkAngle()
  380. self.pdfView.needsDisplay = true
  381. } else if obj.object as? NSTextField == self.alphaTextField {
  382. let formatter = self.alphaTextField.formatter as? NumberFormatter
  383. let floatValue = formatter?.number(from: self.alphaTextField.stringValue)?.floatValue ?? 0
  384. self.alphaSlider.doubleValue = Double(self.watermark.opacity)
  385. self.alphaStepper.doubleValue = Double(self.watermark.opacity)
  386. self.pdfView.needsDisplay = true
  387. } else if obj.object as? NSComboBox == self.pageRangeComboBox {
  388. if self.pageRangeComboBox.indexOfSelectedItem == -1 {
  389. if !self.checkPageRangeValidate(self.pageRangeComboBox.stringValue) {
  390. let alert = NSAlert()
  391. alert.alertStyle = .critical
  392. alert.messageText = "\(String(describing: self.pdfDocument?.documentURL.lastPathComponent)) \(NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: ""))"
  393. alert.runModal()
  394. self.window?.makeFirstResponder(self.pageRangeComboBox)
  395. return
  396. } else {
  397. self.watermark.pagesString = self.pageRangeComboBox.stringValue
  398. self.pdfView.needsDisplay = true
  399. }
  400. }
  401. } else if obj.object as? NSTextField == self.verticalFapTextField {
  402. self.watermark.verticalSpace = CGFloat(Int(self.verticalFapTextField.stringValue) ?? 0)
  403. self.verticalGapStepper.doubleValue = Double(self.watermark.verticalSpace)
  404. self.pdfView.needsDisplay = true
  405. } else if obj.object as? NSTextField == self.horizentalGapTextField {
  406. self.watermark.horizontalSpace = CGFloat(Int(self.horizentalGapTextField.stringValue) ?? 0)
  407. self.horizentalGapStepper.doubleValue = Double(self.watermark.horizontalSpace)
  408. self.pdfView.needsDisplay = true
  409. } else if obj.object as? NSTextField == self.currentPageIndexTextF {
  410. self.pdfView.go(toPageIndex: (Int(self.currentPageIndexTextF.stringValue) ?? 0 - 1), animated: false)
  411. } else if obj.object as? NSTextField == self.spacingHorTextField {
  412. self.watermark.tileHorizontalSpace = CGFloat(Int(self.spacingHorTextField.stringValue) ?? 0)
  413. self.spacingHorStepper.doubleValue = Double(self.watermark.tileHorizontalSpace)
  414. self.pdfView.needsDisplay = true
  415. } else if obj.object as? NSTextField == self.spacingVerTextField {
  416. self.watermark.tileVerticalSpace = CGFloat(Int(self.spacingVerTextField.stringValue) ?? 0)
  417. self.spacingStepper.doubleValue = Double(self.watermark.tileVerticalSpace)
  418. self.pdfView.needsDisplay = true
  419. }
  420. }
  421. @objc func comboBoxSelectionDidChange(_ notification: Notification) {
  422. if notification.object as? NSComboBox == self.pageRangeComboBox {
  423. self.pageRangeComboBox.isEditable = false
  424. if self.pageRangeComboBox.indexOfSelectedItem == 0 {
  425. self.watermark.pageRangeType = .all
  426. } else if self.pageRangeComboBox.indexOfSelectedItem == 1 {
  427. self.watermark.pageRangeType = .odd
  428. } else if self.pageRangeComboBox.indexOfSelectedItem == 2 {
  429. self.watermark.pageRangeType = .even
  430. } else {
  431. self.watermark.pageRangeType = .other
  432. self.pageRangeComboBox.stringValue = ""
  433. self.pageRangeComboBox.isEditable = true
  434. self.window?.makeFirstResponder(self.pageRangeComboBox)
  435. }
  436. }
  437. }
  438. @objc func textDidChange(_ notification: Notification) {
  439. if notification.object as? NSTextView == self.watermarkStringTextView {
  440. self.watermark.text = self.watermarkStringTextView.string
  441. self.applyButton.isEnabled = !self.watermark.text.isEmpty
  442. self.batchButton.isEnabled = !self.watermark.text.isEmpty
  443. self.pdfView.needsDisplay = true
  444. }
  445. }
  446. func isDamageImage(_ image: NSImage?, imagePath path: String) -> Bool {
  447. let addImageAnnotation = (NSSearchPathForDirectoriesInDomains(.applicationSupportDirectory, .userDomainMask, true).last! as NSString).appendingPathComponent(Bundle.main.bundleIdentifier!)
  448. if !FileManager.default.fileExists(atPath: addImageAnnotation) {
  449. try? FileManager.default.createDirectory(atPath: addImageAnnotation, withIntermediateDirectories: false, attributes: nil)
  450. }
  451. let data = image?.tiffRepresentation
  452. let imageRep = NSBitmapImageRep(data: data!)
  453. imageRep?.size = image?.size ?? NSSize.zero
  454. var imageData: Data?
  455. if path.lowercased() == "png" {
  456. imageData = imageRep?.representation(using: .png, properties: [:])
  457. } else {
  458. imageData = imageRep?.representation(using: .jpeg, properties: [:])
  459. }
  460. let rPath = (addImageAnnotation as NSString).appendingPathComponent("waterAnnotation.png")
  461. if ((try? imageData?.write(to: URL(fileURLWithPath: rPath), options: .atomic)) == nil) {
  462. return true
  463. } else {
  464. return false
  465. }
  466. }
  467. func tagString() -> String {
  468. let dateFormatter = DateFormatter()
  469. dateFormatter.dateFormat = "yyMMddHHmmss"
  470. return "\(dateFormatter.string(from: Date()))\(arc4random() % 10000)"
  471. }
  472. // MARK: - Action
  473. // NSNotification
  474. @objc func pageChangeNotification(_ notification: Notification) {
  475. currentPageIndexTextF.stringValue = pdfView.currentPage().label ?? ""
  476. }
  477. // 开始转圈,隐藏转圈提示
  478. func showWaitting() {
  479. if maskView == nil {
  480. maskView = KMBookletMaskView(frame: NSRect(x: 0, y: 0, width: window?.frame.size.width ?? 0, height: window?.frame.size.height ?? 0))
  481. }
  482. window?.contentView?.addSubview(maskView!)
  483. }
  484. func hideWaitting() {
  485. maskView?.removeFromSuperview()
  486. }
  487. override func mouseDown(with event: NSEvent) {
  488. super.mouseDown(with: event)
  489. window?.makeFirstResponder(nil)
  490. }
  491. }
  492. extension KMWatermarkView {
  493. @IBAction func buttonClicked_SwitchTextWatermark(_ sender: Any) {
  494. currentType = 0
  495. changeTypeBoxState(isTextWatermark: true)
  496. }
  497. @IBAction func buttonClicked_SwitchFileWaterMark(_ sender: Any) {
  498. currentType = 1
  499. changeTypeBoxState(isTextWatermark: false)
  500. }
  501. @IBAction func buttonClicked_BrowseFile(_ sender: Any) {
  502. let openPanel = NSOpenPanel()
  503. openPanel.canChooseDirectories = false
  504. openPanel.canChooseFiles = true
  505. openPanel.allowsMultipleSelection = false
  506. openPanel.allowedFileTypes = ["jpg", "jpeg", "png", "pdf"]
  507. openPanel.beginSheetModal(for: window!) { (result) in
  508. if result == NSApplication.ModalResponse.OK {
  509. guard let url = openPanel.url else { return }
  510. let filePath = url.path
  511. if filePath.extension.lowercased() == "pdf" {
  512. let pdf = CPDFDocument(url: url)
  513. guard !pdf!.isEncrypted else { return }
  514. self.showWaitting()
  515. let convert = KMPDFConvert()
  516. convert.convertType = .png
  517. convert.filePath = url.path
  518. convert.pages = [1]
  519. convert.outputFolderPath = (NSSearchPathForDirectoriesInDomains(.applicationSupportDirectory, .userDomainMask, true).last! as NSString).appendingPathComponent(Bundle.main.bundleIdentifier!)
  520. convert.outputFileName = "pdfConvertPng"
  521. KMPDFConvertManager.defaultManager.convert(convert: convert, completion: { (finished, error) in
  522. self.hideWaitting()
  523. if finished {
  524. let outputFilePath = convert.outputFilePath
  525. if FileManager.default.fileExists(atPath: outputFilePath) {
  526. let fimagePath = (try? FileManager.default.subpaths(atPath: outputFilePath))?.first
  527. let pdfPath = convert.outputFilePath + "/" + fimagePath!
  528. var newPath = (convert.outputFolderPath as NSString).appendingPathComponent(self.tagString())
  529. newPath = (newPath as NSString).appendingPathComponent(((url.path as NSString).lastPathComponent as NSString).deletingPathExtension + ".png")
  530. do {
  531. try FileManager.default.copyItem(atPath: pdfPath, toPath: newPath)
  532. try FileManager.default.removeItem(atPath: convert.outputFilePath)
  533. } catch {
  534. print(error.localizedDescription)
  535. }
  536. let image = NSImage(contentsOfFile: pdfPath)
  537. if self.isDamageImage(image, imagePath: pdfPath) {
  538. let alert = NSAlert()
  539. alert.alertStyle = .critical
  540. alert.messageText = "The file \"\(url.lastPathComponent)\" could not be opened."
  541. alert.informativeText = "It may be damaged or use a file format that PDF Reader Pro doesn’t recognize."
  542. alert.addButton(withTitle: "Cancel")
  543. alert.beginSheetModal(for: NSApp.mainWindow!) { (response) in
  544. if response == NSApplication.ModalResponse.alertFirstButtonReturn {
  545. // Cancel button clicked
  546. }
  547. }
  548. return
  549. }
  550. self.filePathLabel.stringValue = url.path
  551. self.watermark.imagePath = pdfPath
  552. self.watermark.image = NSImage(contentsOfFile: pdfPath)
  553. self.watermark.text = ""
  554. self.templateNameTextField.stringValue = self.watermark.watermarkID
  555. self.applyButton.isEnabled = true
  556. self.batchButton.isEnabled = true
  557. self.pdfView.setNeedsDisplay(NSRect.zero)
  558. }
  559. }
  560. })
  561. } else {
  562. let image = NSImage(contentsOfFile: url.path)
  563. if self.isDamageImage(image, imagePath: url.path) {
  564. let alert = NSAlert()
  565. alert.alertStyle = .critical
  566. alert.messageText = "The file \"\(url.lastPathComponent)\" could not be opened."
  567. alert.informativeText = "It may be damaged or use a file format that PDF Reader Pro doesn’t recognize."
  568. alert.addButton(withTitle: "Cancel")
  569. alert.beginSheetModal(for: NSApp.mainWindow!) { (response) in
  570. if response == NSApplication.ModalResponse.alertFirstButtonReturn {
  571. // Cancel button clicked
  572. }
  573. }
  574. return
  575. }
  576. self.filePathLabel.stringValue = url.path
  577. self.watermark.imagePath = url.path
  578. self.watermark.image = NSImage(contentsOfFile: url.path)
  579. self.watermark.text = ""
  580. self.templateNameTextField.stringValue = self.watermark.watermarkID
  581. self.applyButton.isEnabled = true
  582. self.batchButton.isEnabled = true
  583. self.pdfView.setNeedsDisplay(NSRect.zero)
  584. }
  585. }
  586. }
  587. }
  588. @IBAction func buttonClicked_Batch(_ sender: Any) {
  589. let needSave = saveToTemplateButton.state == .on
  590. if needSave {
  591. if templateNameTextField.stringValue.isEmpty {
  592. self.watermark.watermarkID = templateNameTextField.stringValue
  593. } else {
  594. self.watermark.watermarkID = KMWatermarkManager.defaultManager.fetchAvailableName()
  595. }
  596. let result = KMWatermarkManager.defaultManager.addWatermark(watermark: self.watermark)
  597. if !result {
  598. return
  599. }
  600. }
  601. if let operateCallBack = self.operateCallBack {
  602. operateCallBack(self.watermark, currentType)
  603. }
  604. self.cancelAction?(self)
  605. }
  606. @IBAction func buttonClicked_Cancel(_ sender: Any) {
  607. self.cancelAction?(self)
  608. }
  609. @IBAction func buttonClicked_Done(_ sender: Any) {
  610. if checkPageRangeValidate(pageRangeComboBox.stringValue) {
  611. self.watermark.pagesString = pageRangeComboBox.stringValue
  612. pdfView.needsDisplay = true
  613. }
  614. if self.watermark.pagesString == nil || self.watermark.pagesString!.isEmpty {
  615. let alert = NSAlert()
  616. alert.alertStyle = .critical
  617. alert.messageText = "\(self.pdfDocument?.documentURL.lastPathComponent) Invalid page range or the page number is out of range. Please try again."
  618. alert.runModal()
  619. return
  620. } else {
  621. window?.makeFirstResponder(nil)
  622. }
  623. let needSave = saveToTemplateButton.state == .on
  624. if self.watermark.imagePath == nil {
  625. if self.watermark.text.isEmpty {
  626. return
  627. }
  628. }
  629. if self.type == .edit || self.type == .add {
  630. if needSave {
  631. if self.templateNameTextField.stringValue.isEmpty {
  632. self.watermark.watermarkID = KMWatermarkManager.defaultManager.fetchAvailableName()
  633. } else {
  634. self.watermark.watermarkID = self.templateNameTextField.stringValue
  635. }
  636. if KMWatermarkManager.defaultManager.watermarks.contains(self.originalWatermark) {
  637. KMWatermarkManager.defaultManager.removeWatermark(watermark: self.originalWatermark)
  638. KMWatermarkManager.defaultManager.addWatermark(watermark: self.watermark)
  639. } else {
  640. KMWatermarkManager.defaultManager.addWatermark(watermark: self.watermark)
  641. }
  642. }
  643. if let operateCallBack = self.operateCallBack {
  644. operateCallBack(self.watermark, currentType)
  645. }
  646. self.cancelAction?(self)
  647. } else if self.type == .use {
  648. let fileName = "\(self.pdfDocument?.documentURL.lastPathComponent.deletingPathExtension ?? "Untitled")_Watermark"
  649. let savePanelAccessoryViewController = KMSavePanelAccessoryController()
  650. let savePanel = NSSavePanel()
  651. savePanel.nameFieldStringValue = fileName
  652. savePanel.allowedFileTypes = ["pdf"]
  653. savePanel.accessoryView = savePanelAccessoryViewController.view
  654. // savePanel.beginSheetModal(for: self.window!) { (result) in
  655. // if result == NSApplication.ModalResponse.OK {
  656. // savePanelAccessoryViewController.retain()
  657. // let document = CPDFDocument(url: self.pdfdocument?.documentURL)
  658. // if let password = self.password {
  659. // document?.unlock(withPassword: password)
  660. // }
  661. //
  662. // KMWatermarkManager.defaultManager
  663. // let obj = KMPDFWatermarkObject(pdfDocument: document)
  664. // obj?.addWatermark(self.watermark, toPath: savePanel.url!.path, completionHandler: { (isSuccessfully) in
  665. // DispatchQueue.main.async {
  666. // savePanelAccessoryViewController.release()
  667. //
  668. // if isSuccessfully {
  669. // if savePanelAccessoryViewController.openAutomaticButton.state == .on {
  670. // NSDocumentController.shared.openDocument(withContentsOf: savePanel.url!, display: true) { (document, documentWasAlreadyOpen, error) in
  671. // // Handle the completion if needed
  672. // }
  673. // self.dismissSheet(sender)
  674. // } else {
  675. // NSWorkspace.shared.activateFileViewer(selecting: [savePanel.url!])
  676. // self.dismissSheet(sender)
  677. // }
  678. // }
  679. // }
  680. // })
  681. //
  682. // if needSave {
  683. // if self.templateNameTextField.stringValue.isEmpty {
  684. // self.watermark.watermarkID = KMWatermarkManager.defaultManager().fetchAvailableName()
  685. // } else {
  686. // self.watermark.watermarkID = self.templateNameTextField.stringValue
  687. // }
  688. //
  689. // if KMWatermarkManager.defaultManager().watermarks.contains(self.originalWatermark) {
  690. // KMWatermarkManager.defaultManager().removeWatermark(withData: self.originalWatermark)
  691. // KMWatermarkManager.defaultManager().addWatermark(withData: self.watermark)
  692. // } else {
  693. // KMWatermarkManager.defaultManager().addWatermark(withData: self.watermark)
  694. // }
  695. //
  696. // NotificationCenter.default.post(name: NSNotification.Name("KMBatchOperateWatermarksNotification"), object: self)
  697. // }
  698. // }
  699. //
  700. // savePanelAccessoryViewController.release()
  701. // }
  702. }
  703. }
  704. @IBAction func colorWellColorChanged(_ sender: NSColorWell) {
  705. self.watermark.waterTextColor = sender.color
  706. pdfView.needsDisplay = true
  707. }
  708. @IBAction func radioStepperAction(_ sender: NSStepper) {
  709. self.ratioTextField.stringValue = "\(Int(sender.doubleValue * 100))%"
  710. self.watermark.scale = sender.doubleValue
  711. pdfView.needsDisplay = true
  712. }
  713. @IBAction func angleStepperAction(_ sender: NSStepper) {
  714. self.angleTextF.stringValue = "\(sender.doubleValue)"
  715. self.watermark.rotation = sender.doubleValue
  716. checkAngle()
  717. pdfView.needsDisplay = true
  718. }
  719. @IBAction func alphaSliderAction(_ sender: NSSlider) {
  720. self.watermark.opacity = sender.doubleValue
  721. self.alphaStepper.doubleValue = sender.doubleValue
  722. self.alphaTextField.stringValue = "\(Int(sender.doubleValue * 100))%"
  723. pdfView.needsDisplay = true
  724. }
  725. @IBAction func verticalStepperAction(_ sender: NSStepper) {
  726. self.verticalFapTextField.stringValue = "\(sender.doubleValue)"
  727. self.watermark.verticalSpace = sender.doubleValue
  728. pdfView.needsDisplay = true
  729. }
  730. @IBAction func tileVerticalStepperAction(_ sender: NSStepper) {
  731. self.watermark.tileVerticalSpace = sender.doubleValue
  732. self.spacingVerTextField.stringValue = "\(sender.doubleValue)"
  733. self.watermark.tileVerticalSpace = sender.doubleValue
  734. pdfView.needsDisplay = true
  735. }
  736. @IBAction func horizentalStepperAction(_ sender: NSStepper) {
  737. self.watermark.horizontalSpace = sender.doubleValue
  738. self.horizentalGapTextField.stringValue = "\(sender.doubleValue)"
  739. self.watermark.horizontalSpace = sender.doubleValue
  740. pdfView.needsDisplay = true
  741. }
  742. @IBAction func tileHorizentalStepperAction(_ sender: NSStepper) {
  743. self.watermark.tileHorizontalSpace = sender.doubleValue
  744. self.spacingHorTextField.stringValue = "\(sender.doubleValue)"
  745. self.watermark.tileHorizontalSpace = sender.doubleValue
  746. pdfView.needsDisplay = true
  747. }
  748. @IBAction func alphaSteperAction(_ sender: NSStepper) {
  749. self.alphaTextField.stringValue = "\(Int(sender.doubleValue * 100))%"
  750. self.watermark.opacity = sender.doubleValue
  751. self.alphaSlider.doubleValue = sender.doubleValue
  752. pdfView.needsDisplay = true
  753. }
  754. @IBAction func onPageButtonAction(_ sender: Any) {
  755. self.backPageButton.state = .off
  756. self.watermark.isFront = true
  757. pdfView.needsDisplay = true
  758. }
  759. @IBAction func backPageButtonAction(_ sender: Any) {
  760. self.OnPageButton.state = .off
  761. self.watermark.isFront = false
  762. pdfView.needsDisplay = true
  763. }
  764. @IBAction func onPageTileButtonAction(_ sender: Any) {
  765. self.watermark.isTilePage = !self.watermark.isTilePage
  766. if self.type != .edit && !self.fontModified && self.txtButton.state == .on {
  767. if self.watermark.isTilePage {
  768. self.isTileWater.state = .on
  769. self.fontSizeCombobox.stringValue = "\(self.watermark.textFontSize == 48 ? 18 : self.watermark.textFontSize)"
  770. self.watermark.textFontSize = CGFloat(Int(self.fontSizeCombobox.stringValue) ?? 0)
  771. } else {
  772. self.isTileWater.state = .off
  773. self.fontSizeCombobox.stringValue = "\(self.watermark.textFontSize == 18 ? 48 : self.watermark.textFontSize)"
  774. self.watermark.textFontSize = CGFloat(Int(self.fontSizeCombobox.stringValue) ?? 0)
  775. }
  776. }
  777. self.spacingHorTextField.isEnabled = self.watermark.isTilePage
  778. self.spacingHorStepper.isEnabled = self.watermark.isTilePage
  779. self.spacingVerTextField.isEnabled = self.watermark.isTilePage
  780. self.spacingStepper.isEnabled = self.watermark.isTilePage
  781. pdfView.needsDisplay = true
  782. }
  783. @IBAction func goPrevious(_ sender: Any) {
  784. if pdfView.canGoToPreviousPage() {
  785. pdfView.goToPreviousPage(sender)
  786. }
  787. let index = pdfDocument?.index(for: pdfView.currentPage())
  788. currentPageIndexTextF.stringValue = "\(index! + 1)"
  789. }
  790. @IBAction func goNext(_ sender: Any) {
  791. if pdfView.canGoToNextPage() {
  792. pdfView.goToNextPage(sender)
  793. }
  794. let index = pdfDocument!.index(for: pdfView.currentPage())
  795. currentPageIndexTextF.stringValue = "\(index + 1)"
  796. }
  797. @IBAction func fontSizeComboxAction(_ sender: NSComboBox) {
  798. watermark.textFontSize = CGFloat(Int(fontSizeCombobox.stringValue) ?? 0)
  799. fontModified = true
  800. pdfView.needsDisplay = true
  801. }
  802. @IBAction func pageRangeComboxAction(_ sender: Any) {
  803. pageRangeComboBox.isEditable = false
  804. var pages = [Int]()
  805. switch pageRangeComboBox.indexOfSelectedItem {
  806. case 0:
  807. for i in 0..<pdfDocument!.pageCount {
  808. pages.append(Int(i))
  809. }
  810. watermark.pageRangeType = .all
  811. case 1:
  812. for i in 0..<pdfDocument!.pageCount where i % 2 == 0 {
  813. pages.append(Int(i))
  814. }
  815. watermark.pageRangeType = .odd
  816. case 2:
  817. for i in 0..<pdfDocument!.pageCount where i % 2 != 0 {
  818. pages.append(Int(i))
  819. }
  820. watermark.pageRangeType = .even
  821. default:
  822. let fileAttribute = KMFileAttribute()
  823. fileAttribute.filePath = pdfDocument?.documentURL?.path ?? ""
  824. fileAttribute.bAllPage = false
  825. fileAttribute.pagesString = pageRangeComboBox.stringValue
  826. watermark.pageRangeType = .other
  827. pageRangeComboBox.isEditable = true
  828. window?.makeFirstResponder(pageRangeComboBox)
  829. let selectPages = fileAttribute.fetchSelectPages()
  830. pages = selectPages.map { $0.intValue - 1 }
  831. }
  832. if !pages.isEmpty {
  833. watermark.pagesString = pages.map { "\($0)" }.joined(separator: ",")
  834. } else {
  835. watermark.pagesString = nil
  836. }
  837. pdfView.needsDisplay = true
  838. }
  839. }