KMPDFEditViewController.swift 66 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685
  1. //
  2. // KMPDFEditViewController.swift
  3. // PDF Master
  4. //
  5. // Created by tangchao on 2023/1/6.
  6. //
  7. import Cocoa
  8. enum KMPageEditType: Int {
  9. case pageRange = 1
  10. case leftRotate = 2
  11. case rightRotate = 3
  12. case insert = 4
  13. case extract = 5
  14. case replace = 6
  15. case split = 7
  16. case reverse = 8
  17. case delete = 9
  18. }
  19. class KMPDFEditViewController: NSViewController {
  20. @IBOutlet weak var topBarBox: NSBox!
  21. @IBOutlet weak var contentBox: NSBox!
  22. var pageRangeView: KMDesignSelect?
  23. var pageRangeViewItems = [NSLocalizedString("All Pages", comment: ""),
  24. NSLocalizedString("Odd Pages", comment: ""),
  25. NSLocalizedString("Even Pages", comment: ""),
  26. NSLocalizedString("Horizontal Pages", comment: ""),
  27. NSLocalizedString("Vertical Pages", comment: ""),
  28. NSLocalizedString("Customized", comment: "")]
  29. var toolBar = KMToolbar()
  30. private var zoomOutButton: NSButton?
  31. private var zoomInButton: NSButton?
  32. lazy var rightItemView: NSView = {
  33. let view = NSView()
  34. let size: CGFloat = 28
  35. let y: CGFloat = 6
  36. let zoomOutButton = NSButton()
  37. view.addSubview(zoomOutButton)
  38. zoomOutButton.frame = NSMakeRect(0, y, size, size)
  39. zoomOutButton.isBordered = false
  40. zoomOutButton.image = NSImage(named: "KMImageNamePageEditZoomOut")
  41. zoomOutButton.target = self
  42. zoomOutButton.action = #selector(zoomOutButtonAction)
  43. self.zoomOutButton = zoomOutButton
  44. let zoomInButton = NSButton()
  45. view.addSubview(zoomInButton)
  46. zoomInButton.frame = NSMakeRect(size+7, y, size, size)
  47. zoomInButton.isBordered = false
  48. zoomInButton.image = NSImage(named: "KMImageNamePageEditZoomIn")
  49. zoomInButton.target = self
  50. zoomInButton.action = #selector(zoomInButtonAction)
  51. self.zoomInButton = zoomInButton
  52. return view
  53. }()
  54. var thumbnailView = KMPageEditThumbnailView()
  55. private var pdfDocument: CPDFDocument?
  56. var listView: CPDFView?
  57. var windowController: NSWindowController?
  58. var pageIsUpdate: Bool = true
  59. /// 临时存储 document
  60. private var documentCaches: [CPDFDocument] = []
  61. var selectedPages: [Int] = []
  62. var copysPages: Array<CPDFPage> = []
  63. private let defaultItemSize = NSMakeSize(218 + 12 * 2, 292 + 24)
  64. var itemSize: NSSize = NSMakeSize(218 + 12 * 2, 292 + 24)
  65. private var lockedFiles: Array<URL> = []
  66. private var undoRedoManager: UndoManager = UndoManager()
  67. // index=1 双击退出;index=2 打印
  68. var itemClick: KMCommonClickBlock?
  69. var documentEditedCallback: KMCommonBlock?
  70. internal var viewFirstAppear = true
  71. required init?(coder: NSCoder) {
  72. super.init(coder: coder)
  73. }
  74. init(_ document: CPDFDocument) {
  75. super.init(nibName: "KMPDFEditViewController", bundle: nil)
  76. self.pdfDocument = document
  77. }
  78. override func viewDidAppear() {
  79. super.viewDidAppear()
  80. if (self.pdfDocument?.isLocked == false && self.viewFirstAppear) {
  81. self.viewFirstAppear = false
  82. self.thumbnailView.document = self.pdfDocument!
  83. self.thumbnailView.reloadData()
  84. }
  85. if (self.selectedPages.isEmpty) {
  86. var index: Int = 0
  87. if (self.listView != nil) {
  88. index = self.listView!.currentPageIndex
  89. }
  90. self.selectPages(indexs: IndexSet(integer: index))
  91. self.pageRangeView?.stringValue = "\(index+1)"
  92. self.pageRangeView?.editable = true
  93. } else {
  94. var indexs: IndexSet = []
  95. for pageIndex in self.selectedPages {
  96. if (pageIndex < 0 || pageIndex >= (self.listView?.document.pageCount)!) {
  97. continue
  98. }
  99. indexs.insert(pageIndex)
  100. }
  101. self.selectPages(indexs: indexs)
  102. self.pageRangeView?.editable = true
  103. self.pageRangeView?.stringValue = KMPageRangeTools.newParseSelectedIndexs(selectedIndex: indexs.sorted())
  104. }
  105. self.view.window?.makeFirstResponder(self)
  106. }
  107. override func viewDidLoad() {
  108. super.viewDidLoad()
  109. self.thumbnailView.collectionView.backgroundColor(NSColor(hex: "#CED0D4"))
  110. let toolBar = self.toolBar
  111. self.topBarBox.contentView?.addSubview(toolBar)
  112. toolBar.frame = self.topBarBox.contentView!.bounds
  113. toolBar.autoresizingMask = [.width, .height]
  114. toolBar.delegate = self
  115. self.topBarBox.fillColor = NSColor(hex: "#F7F8FA")
  116. let rightItemView = self.rightItemView
  117. self.topBarBox.contentView?.addSubview(rightItemView)
  118. rightItemView.frame = NSMakeRect(NSWidth(self.topBarBox.contentView!.frame)-80, 0, 80, NSHeight(self.topBarBox.contentView!.frame))
  119. rightItemView.autoresizingMask = [.minXMargin]
  120. let view = self.thumbnailView
  121. self.contentBox.contentView?.addSubview(view)
  122. view.frame = self.contentBox.contentView!.bounds
  123. view.autoresizingMask = [.width, .height]
  124. self.thumbnailView.delegate = self
  125. self.thumbnailView.selectionDidChange = { [weak self] selectedIndexs in
  126. self?.view.window?.makeFirstResponder(self)
  127. var replaceEnable: Bool = false
  128. if (selectedIndexs.count == 1) { /// 替换item 置灰
  129. replaceEnable = true
  130. }
  131. var enabled: Bool = false
  132. if (selectedIndexs.count > 0) {
  133. enabled = true
  134. }
  135. if (self?.toolBar.items == nil) {
  136. return
  137. }
  138. var replaceItem: KMToolBoxItem?
  139. for i in 0 ..< (self?.toolBar.items.count)! {
  140. let item: KMToolBoxItem = (self?.toolBar.items[i])!
  141. if (item.itemIdentifier == KMToolbarPageEditReplaceItemIdentifier) {
  142. replaceItem = item
  143. }
  144. if (item.itemIdentifier == KMToolbarPageEditExtractItemIdentifier ||
  145. item.itemIdentifier == KMToolbarPageEditReverseItemIdentifier ||
  146. item.itemIdentifier == KMToolbarPageEditDeleteItemIdentifier ||
  147. item.itemIdentifier == KMToolbarPageEditLeftRotateItemIdentifier ||
  148. item.itemIdentifier == KMToolbarPageEditRightRotateItemIdentifier) {
  149. item.unEnabled = !enabled
  150. }
  151. }
  152. if (replaceItem != nil) {
  153. replaceItem?.unEnabled = !replaceEnable
  154. }
  155. if (self!.pageIsUpdate == false) {
  156. self?.pageIsUpdate = true
  157. return
  158. }
  159. var indexs: Array<Int> = []
  160. for indexPath in selectedIndexs {
  161. indexs.append(indexPath.item)
  162. }
  163. indexs.sort(){$0 < $1}
  164. self?.pageRangeView?.editable = true;
  165. self?.pageRangeView?.stringValue = KMPageRangeTools.newParseSelectedIndexs(selectedIndex: indexs)
  166. }
  167. let menu = NSMenu()
  168. menu.delegate = self
  169. self.view.menu = menu
  170. }
  171. // MARK: Private Methods
  172. private func copyPages(with indexs: IndexSet) {
  173. Task { @MainActor in
  174. if await (KMLightMemberManager.manager.canUseAdvanced() == false) {
  175. let _ = KMComparativeTableViewController.show(window: self.view.window!)
  176. return
  177. }
  178. if indexs.count > 0 {
  179. self.copysPages = []
  180. for i in indexs {
  181. let page = self.pdfDocument?.page(at: UInt(i)).copy() as? CPDFPage
  182. if (page == nil) {
  183. continue
  184. }
  185. self.copysPages.append(page!)
  186. }
  187. }
  188. }
  189. }
  190. private func cutPages(with indexs: IndexSet) {
  191. Task { @MainActor in
  192. if await (KMLightMemberManager.manager.canUseAdvanced() == false) {
  193. let _ = KMComparativeTableViewController.show(window: self.view.window!)
  194. return
  195. }
  196. if indexs.count > 0 {
  197. if (indexs.count == (self.listView?.document.pageCount)!) {
  198. let _ = CustomAlertView(message: NSLocalizedString("Unable to delete all pages", comment: ""), from: self.thumbnailView, with: .blue)
  199. return
  200. }
  201. self.copysPages = []
  202. for i in indexs {
  203. let page = self.pdfDocument?.page(at: UInt(i)).copy() as? CPDFPage
  204. if (page == nil) {
  205. continue
  206. }
  207. self.copysPages.append(page!)
  208. }
  209. self.clearSelectPages()
  210. self.deletePages(indexs: indexs)
  211. // 提醒外面文档已编辑
  212. if (self.documentEditedCallback != nil) {
  213. self.documentEditedCallback!()
  214. }
  215. }
  216. }
  217. }
  218. private func pastePages(at index: Int) {
  219. Task { @MainActor in
  220. if await (KMLightMemberManager.manager.canUseAdvanced() == false) {
  221. let _ = KMComparativeTableViewController.show(window: self.view.window!)
  222. return
  223. }
  224. if (index >= 0 && self.copysPages.count > 0) {
  225. var indexs: IndexSet = IndexSet()
  226. for (i, _) in self.copysPages.enumerated() {
  227. indexs.insert(index+i+1)
  228. }
  229. self.insertPagesForDescSort(pages: self.copysPages, indexs: indexs)
  230. // 提醒外面文档已编辑
  231. if (self.documentEditedCallback != nil) {
  232. self.documentEditedCallback!()
  233. }
  234. var indexpaths: Set<IndexPath> = []
  235. indexpaths.insert(IndexPath(item: index, section: 0))
  236. self.thumbnailView.collectionView.animator().scrollToItems(at: indexpaths, scrollPosition: .top)
  237. }
  238. }
  239. }
  240. private func getSelecteIndex() -> Int {
  241. var index: Int = 0
  242. for indexpath in self.thumbnailView.selectionIndexPaths {
  243. index = indexpath.item
  244. }
  245. return index
  246. }
  247. private func getSelectedPage() -> CPDFPage {
  248. return (self.pdfDocument!.page(at: UInt(getSelecteIndex())))!
  249. }
  250. private func showPasswordWindow(_ insertIndex: Int) {
  251. if (self.lockedFiles.count > 0) {
  252. let fileURL = self.lockedFiles.first
  253. KMPasswordInputWindow.openWindow(window: self.view.window!, type: .open, url: fileURL!) { result, password in
  254. if (result == .cancel) { /// 取消
  255. self.lockedFiles.removeFirst()
  256. if (self.lockedFiles.count > 0) {
  257. self.showPasswordWindow(insertIndex)
  258. }
  259. return
  260. }
  261. /// 解密成功
  262. let document = CPDFDocument(url: fileURL)
  263. document?.unlock(withPassword: password)
  264. var newInsertIndex: Int = insertIndex
  265. var pages: [CPDFPage] = []
  266. var indexs: IndexSet = IndexSet()
  267. for i in 0 ..< document!.pageCount {
  268. let page = document?.page(at: i)
  269. pages.append(page!)
  270. indexs.insert(newInsertIndex)
  271. newInsertIndex += 1
  272. }
  273. self.insertPagesForDescSort(pages: pages, indexs: indexs)
  274. self.lockedFiles.removeFirst()
  275. if (self.lockedFiles.count > 0) {
  276. self.showPasswordWindow(newInsertIndex)
  277. }
  278. }
  279. }
  280. }
  281. @objc private func insertPagesForDescSort(pages: Array<CPDFPage>, indexs: IndexSet) {
  282. self.insertPages(pages: pages, indexs: indexs)
  283. /// 选中插入的页面
  284. self.selectPages(indexs: indexs)
  285. }
  286. private func selectPages(indexs: IndexSet) {
  287. var indexpaths: Set<IndexPath> = []
  288. for index in indexs {
  289. indexpaths.insert(IndexPath(item: index, section: 0))
  290. }
  291. self.thumbnailView.selectPages(at: indexpaths)
  292. }
  293. private func km_becomeFirstResponder() {
  294. DispatchQueue.main.async {
  295. self.view.window?.makeFirstResponder(self)
  296. }
  297. }
  298. // MARK: - 清空选择
  299. private func clearSelectPages() {
  300. self.selectPages(indexs: IndexSet())
  301. }
  302. private func refreshUI(indexs: IndexSet = []) {
  303. if (Thread.isMainThread) {
  304. self.listView?.layoutDocumentView()
  305. var indexpaths: Set<IndexPath> = []
  306. for index in indexs {
  307. indexpaths.insert(IndexPath(item: index, section: 0))
  308. }
  309. self.thumbnailView.refreshUI(indexs: indexpaths)
  310. } else {
  311. DispatchQueue.main.async {
  312. self.listView?.layoutDocumentView()
  313. var indexpaths: Set<IndexPath> = []
  314. for index in indexs {
  315. indexpaths.insert(IndexPath(item: index, section: 0))
  316. }
  317. self.thumbnailView.refreshUI(indexs: indexpaths)
  318. }
  319. }
  320. }
  321. private func refreshUIForKeepSelecteds(indexs: IndexSet = []) {
  322. let indexpaths = self.thumbnailView.selectionIndexPaths
  323. self.refreshUI(indexs: indexs)
  324. self.thumbnailView.selectPages(at: indexpaths)
  325. }
  326. // MARK: Undo / Redo
  327. @objc private func leftRotate(indexs: IndexSet) {
  328. Task { @MainActor in
  329. if await (KMLightMemberManager.manager.canUseAdvanced() == false) {
  330. let _ = KMComparativeTableViewController.show(window: self.view.window!)
  331. return
  332. }
  333. if (indexs.count <= 0) {
  334. return
  335. }
  336. // 提醒外面文档已编辑
  337. if (self.documentEditedCallback != nil) {
  338. self.documentEditedCallback!()
  339. }
  340. for i in 0 ..< (self.listView?.document.pageCount)! {
  341. if (indexs.contains(IndexSet.Element(i))) {
  342. let page = self.listView?.document.page(at: i)
  343. if (page == nil) {
  344. continue
  345. }
  346. page!.rotation = (page!.rotation-90)%360
  347. }
  348. }
  349. self.undoRedoManager.registerUndo(withTarget: self, selector: #selector(rightRotate), object: indexs)
  350. self.refreshUIForKeepSelecteds(indexs: indexs)
  351. NotificationCenter.default.post(name: NSNotification.Name(rawValue: "KMPDFViewRotatePage"), object: self.listView?.document)
  352. }
  353. }
  354. @objc private func rightRotate(indexs: IndexSet) {
  355. Task { @MainActor in
  356. if await (KMLightMemberManager.manager.canUseAdvanced() == false) {
  357. let _ = KMComparativeTableViewController.show(window: self.view.window!)
  358. return
  359. }
  360. if (indexs.count <= 0) {
  361. return
  362. }
  363. // 提醒外面文档已编辑
  364. if (self.documentEditedCallback != nil) {
  365. self.documentEditedCallback!()
  366. }
  367. for i in 0 ..< (self.listView?.document.pageCount)! {
  368. if (indexs.contains(IndexSet.Element(i))) {
  369. let page = self.listView?.document.page(at: i)
  370. if (page == nil) {
  371. continue
  372. }
  373. page!.rotation = (page!.rotation+90)%360
  374. }
  375. }
  376. self.undoRedoManager.registerUndo(withTarget: self, selector: #selector(leftRotate), object: indexs)
  377. self.refreshUIForKeepSelecteds(indexs: indexs)
  378. NotificationCenter.default.post(name: NSNotification.Name(rawValue: "KMPDFViewRotatePage"), object: self.listView?.document)
  379. }
  380. }
  381. @objc private func insertPages(pages: Array<CPDFPage>, indexs: IndexSet) {
  382. var count: Int = 0
  383. for index in indexs {
  384. if (count >= pages.count) {
  385. break
  386. }
  387. if (index > (self.listView?.document.pageCount)!) {
  388. break
  389. }
  390. let newPage = pages[count]
  391. self.listView?.document.insertPageObject(newPage, at: UInt(index))
  392. count += 1
  393. }
  394. self.undoRedoManager.registerUndo(withTarget: self, selector: #selector(deletePages), object: indexs)
  395. self.refreshUI()
  396. }
  397. @objc private func deletePages(indexs: IndexSet) {
  398. if (indexs.count <= 0) {
  399. return
  400. }
  401. if (indexs.count == (self.listView?.document.pageCount)!) {
  402. let _ = CustomAlertView(message: NSLocalizedString("Unable to delete all pages", comment: ""), from: self.thumbnailView, with: .blue)
  403. return
  404. }
  405. // 提醒外面文档已编辑
  406. if (self.documentEditedCallback != nil) {
  407. self.documentEditedCallback!()
  408. }
  409. var array: Array<CPDFPage> = []
  410. for i in indexs {
  411. let page = self.listView?.document.page(at: UInt(i)).copy() as? CPDFPage
  412. if (page != nil) {
  413. array.append(page!)
  414. }
  415. }
  416. self.listView?.document.removePage(at: indexs)
  417. self.undoRedoManager.registerUndo(withTarget: self) { target in
  418. target.insertPages(pages: array, indexs: indexs)
  419. }
  420. self.refreshUI()
  421. }
  422. @objc private func reversePages(indexs: IndexSet) {
  423. KMPageEditTools.reverse((self.listView?.document)!, indexs
  424. ) { result, error in
  425. if (result) {
  426. self.refreshUI()
  427. self.undoRedoManager.registerUndo(withTarget: self, selector: #selector(self.reversePages), object: indexs)
  428. }
  429. }
  430. }
  431. @objc private func dragPages(_ indexs: IndexSet, _ toIndex: Int, reversed: Bool = false) {
  432. if (indexs.count <= 0) {
  433. return
  434. }
  435. // 插入位置偏移量
  436. var toIndexOffset: Int = 0
  437. // 需要移动的页面index数组
  438. var newPages: [CPDFPage] = []
  439. // 从后面往前删除
  440. for index in indexs.reversed() {
  441. let newPage = (self.listView?.document.page(at: UInt(index)))?.copy() as? CPDFPage
  442. if (newPage == nil) {
  443. continue
  444. }
  445. newPages.insert(newPage!, at: 0)
  446. self.listView?.document.removePage(at: UInt(index))
  447. if (index < toIndex) { /// 删除了插入位置前面的页面,需要改变 toIndex
  448. toIndexOffset += 1
  449. }
  450. }
  451. if (indexs.last! < toIndex) { // 往后拖拽
  452. toIndexOffset -= 1
  453. }
  454. let insertIndex = max(0, toIndex-toIndexOffset)
  455. var selectIndexs: IndexSet = []
  456. for (index, page) in newPages.enumerated() {
  457. self.listView?.document.insertPageObject(page, at: UInt(insertIndex+index))
  458. selectIndexs.insert(insertIndex+index)
  459. }
  460. // 新增 undo
  461. self.undoRedoManager.registerUndo(withTarget: self) { target in
  462. target.clearSelectPages()
  463. target.dragPages_reversed(indexs, toIndex, newPages)
  464. }
  465. // 刷新UI
  466. self.refreshUI()
  467. if (reversed == false) {
  468. self.selectPages(indexs: selectIndexs)
  469. }
  470. }
  471. @objc private func dragPages_reversed(_ indexs: IndexSet, _ toIndex: Int, _ pages: [CPDFPage]) {
  472. if (indexs.count <= 0) {
  473. return
  474. }
  475. // 插入位置偏移量
  476. var toIndexOffset: Int = 0
  477. for index in indexs {
  478. if (index < toIndex) { /// 删除了插入位置前面的页面,需要改变 toIndex
  479. toIndexOffset += 1
  480. }
  481. }
  482. if (indexs.last! < toIndex) { // 往后拖拽
  483. toIndexOffset -= 1
  484. }
  485. let insertIndex = max(0, toIndex-toIndexOffset)
  486. // 需要移动的页面index数组
  487. var datas: [Int] = []
  488. for index in insertIndex ..< (insertIndex + indexs.count) {
  489. datas.append(index)
  490. }
  491. // 从后面开始删除
  492. for index in datas.reversed() {
  493. self.listView?.document.removePage(at: UInt(index))
  494. }
  495. // 再插入到对应的位置
  496. for (index,value) in indexs.enumerated() {
  497. var page: CPDFPage?
  498. if (index < pages.count) {
  499. page = pages[index]
  500. }
  501. if (page != nil) {
  502. self.listView?.document.insertPageObject(page!, at: UInt(value))
  503. }
  504. }
  505. // 新增 undo
  506. self.undoRedoManager.registerUndo(withTarget: self) { target in
  507. target.dragPages(indexs, toIndex, reversed: true)
  508. }
  509. // 刷新UI
  510. self.refreshUI()
  511. }
  512. // MARK: topBar 事件
  513. @objc func zoomOutButtonAction(sender: NSButton) {
  514. if (self.itemSize.width > self.defaultItemSize.width) {
  515. return
  516. }
  517. if (self.itemSize.width < self.defaultItemSize.width) {
  518. self.itemSize = self.defaultItemSize
  519. } else if (self.itemSize.width == self.defaultItemSize.width) {
  520. self.itemSize = NSSize(width: self.defaultItemSize.width*1.5, height: self.defaultItemSize.height*1.5)
  521. }
  522. self.zoomInButton?.isEnabled = true
  523. if (self.itemSize.width > self.defaultItemSize.width) {
  524. sender.isEnabled = false
  525. }
  526. self.thumbnailView.zoomOut()
  527. }
  528. @objc func zoomInButtonAction(sender: NSButton) {
  529. if (self.itemSize.width < self.defaultItemSize.width) {
  530. return
  531. }
  532. if (self.itemSize.width > self.defaultItemSize.width) {
  533. self.itemSize = self.defaultItemSize
  534. } else if (self.itemSize.width == self.defaultItemSize.width) {
  535. self.itemSize = NSSize(width: self.defaultItemSize.width*0.5, height: self.defaultItemSize.height*0.5)
  536. }
  537. self.zoomOutButton?.isEnabled = true
  538. if (self.itemSize.width < self.defaultItemSize.width) {
  539. sender.isEnabled = false
  540. }
  541. self.thumbnailView.zoomIn()
  542. }
  543. // MARK: menu 菜单事件
  544. @objc func item_insertFile(sender: NSMenuItem?) {
  545. let panel = NSOpenPanel()
  546. panel.allowedFileTypes = ["pdf"]
  547. panel.allowsMultipleSelection = true
  548. panel.beginSheetModal(for: self.view.window!) { [weak self] response in
  549. if (response == .cancel) {
  550. return
  551. }
  552. /// 处理page
  553. var insertIndex: Int = (self?.getSelecteIndex())!
  554. self!.lockedFiles.removeAll()
  555. var pages: Array<CPDFPage> = []
  556. self?.documentCaches.removeAll()
  557. var indexs = IndexSet()
  558. for fileURL in panel.urls {
  559. let document = CPDFDocument(url: fileURL)
  560. if ((document?.isLocked)!) {
  561. self?.lockedFiles.append(fileURL)
  562. continue
  563. }
  564. self?.documentCaches.append(document!)
  565. for i in 0 ..< document!.pageCount {
  566. let page = document?.page(at: i)
  567. pages.append(page!)
  568. insertIndex += 1
  569. indexs.insert(insertIndex)
  570. }
  571. }
  572. self?.insertPagesForDescSort(pages: pages, indexs: indexs)
  573. if ((self?.lockedFiles.count)! <= 0) {
  574. self?.refreshUI()
  575. return
  576. }
  577. self?.refreshUI()
  578. self!.showPasswordWindow(insertIndex)
  579. }
  580. }
  581. @objc func item_insertBlankPage(sender: NSMenuItem?) {
  582. let index = self.getSelecteIndex()
  583. let page = self.listView?.document.page(at: UInt(index))
  584. self.documentCaches.removeAll()
  585. let document = CPDFDocument()
  586. document?.insertPage((page?.bounds.size)!, at: 0)
  587. self.documentCaches.append(document!)
  588. let newPage: CPDFPage = (document?.page(at: 0))!
  589. self.insertPagesForDescSort(pages: [newPage], indexs: IndexSet(integer: index+1))
  590. }
  591. @objc func item_insertCustomPage(sender: NSMenuItem?) {
  592. let windowController = KMPageEditInsertCustomPageWindowController(windowNibName: "KMPageEditInsertCustomPageWindowController")
  593. if (self.thumbnailView.selectionIndexPaths.count > 0) {
  594. windowController.selectedPageSize = self.getSelectedPage().bounds.size
  595. }
  596. self.windowController = windowController
  597. self.view.window?.beginSheet(windowController.window!, completionHandler: { response in})
  598. windowController.itemClick = {
  599. [weak self] (index: Int) in
  600. if (index == 1) { /// 取消
  601. self?.view.window?.endSheet((self!.windowController?.window)!)
  602. self?.windowController = nil
  603. return
  604. }
  605. /// 插入
  606. /// 样式
  607. let windowController_Insert: KMPageEditInsertCustomPageWindowController = self?.windowController as! KMPageEditInsertCustomPageWindowController
  608. let type = windowController_Insert.typeIndex
  609. /// 页面大小
  610. let pageSize = windowController_Insert.pageSize
  611. /// 方向
  612. let direction = windowController_Insert.direction
  613. /// 插入位置
  614. let insertIndex: Int = (self?.getSelecteIndex())!
  615. if (type == 1) { /// 空白页
  616. let document = CPDFDocument()
  617. document?.insertPage(pageSize, at: 0)
  618. let page: CPDFPage = (document?.page(at: 0))!
  619. if (direction == 0) { /// 纵向
  620. page.rotation = 90
  621. }
  622. self?.insertPagesForDescSort(pages: [page], indexs: IndexSet(integer: insertIndex+1))
  623. } else {
  624. let document = CPDFDocument()
  625. var imageName = "plaid"
  626. if (type == 2) {
  627. imageName = "horizontal_line"
  628. } else if (type == 3) {
  629. imageName = "five_line_score"
  630. }
  631. let imagePath = Bundle.main.pathForImageResource(imageName)
  632. document?.insertPage(pageSize, withImage: imagePath, at: 0)
  633. let page: CPDFPage = (document?.page(at: 0))!
  634. if (direction == 0) { /// 纵向
  635. page.rotation = 90
  636. }
  637. self?.insertPagesForDescSort(pages: [page], indexs: IndexSet(integer: insertIndex+1))
  638. }
  639. self?.view.window?.endSheet((self?.windowController?.window)!)
  640. self?.windowController = nil
  641. }
  642. }
  643. // MARK: toolBar 菜单事件
  644. @objc func pageRangeMenuAction(sender: KMToolbarClickButton) {
  645. }
  646. @objc func leftRotateMenuAction(sender: KMToolbarClickButton?) {
  647. let selectedIndexs = self.thumbnailView.selectionIndexPaths
  648. var indexs: IndexSet = []
  649. for indexpath in selectedIndexs {
  650. indexs.insert(indexpath.item)
  651. }
  652. self.leftRotate(indexs: indexs)
  653. }
  654. @objc func rightRotateMenuAction(sender: KMToolbarClickButton?) {
  655. let selectedIndexs = self.thumbnailView.selectionIndexPaths
  656. var indexs: IndexSet = []
  657. for indexpath in selectedIndexs {
  658. indexs.insert(indexpath.item)
  659. }
  660. self.rightRotate(indexs: indexs)
  661. }
  662. @objc func insertMenuAction(sender: KMToolbarClickButton) {
  663. let menu = NSMenu()
  664. let insertFileItem = NSMenuItem(title: "Insert File", action: #selector(item_insertFile), target: self)
  665. menu.addItem(insertFileItem!)
  666. let insertBlankPageItem = NSMenuItem(title: "Insert a blank page", action: #selector(item_insertBlankPage), target: self)
  667. menu.addItem(insertBlankPageItem!)
  668. let insertCustomPageItem = NSMenuItem(title: "Insert custom page", action: #selector(item_insertCustomPage), target: self)
  669. menu.addItem(insertCustomPageItem!)
  670. menu.popUp(positioning: nil, at: NSPoint(x: 0, y: 40), in: sender)
  671. }
  672. @objc func extractMenuAction(sender: KMToolbarClickButton?) {
  673. if (self.thumbnailView.selectionIndexPaths.count == 0) {
  674. let alert = NSAlert()
  675. alert.messageText = "请选择需要提取的页面"
  676. alert.runModal()
  677. return
  678. }
  679. let windowController = KMPageEditExtractWindowController(windowNibName: "KMPageEditSettingBaseWindowController")
  680. self.view.window?.beginSheet(windowController.window!)
  681. self.windowController = windowController
  682. windowController.itemClick = { [weak self] index, value in
  683. if (index == 1) { /// 取消
  684. self?.view.window?.endSheet((self?.windowController?.window)!)
  685. self?.windowController = nil
  686. return
  687. }
  688. let windowController_extract: KMPageEditExtractWindowController = self?.windowController as! KMPageEditExtractWindowController
  689. /// 提取设置项
  690. let settingView: KMPageEditExtractSettingView = windowController_extract.settingView! as! KMPageEditExtractSettingView
  691. let oneDocumentPerPage: Bool = settingView.independentPerPageCheck.state != .on
  692. let extractAfterDelete: Bool = settingView.extractAfterDeletePageCheck.state == .on
  693. self?.view.window?.endSheet((self?.windowController!.window)!)
  694. self?.windowController = nil
  695. // /// 提取的页面
  696. let indexpaths = self?.thumbnailView.selectionIndexPaths
  697. var indexs: IndexSet = []
  698. var extractPages: Array<CPDFPage> = []
  699. for indexpath in indexpaths! {
  700. let page = self?.listView?.document.page(at: UInt(indexpath.item))
  701. if (page != nil) {
  702. extractPages.append(page!)
  703. }
  704. indexs.insert(indexpath.item)
  705. }
  706. KMPageEditTools.extract((self?.listView?.document)!, indexs, oneDocumentPerPage, (self?.view.window)!) { result, urls, error in
  707. if (result == false) {
  708. return
  709. }
  710. if (extractAfterDelete) {
  711. var indexs: IndexSet = IndexSet()
  712. for page in extractPages {
  713. let indexPage: Int = Int((self?.listView?.document.index(for: page))!)
  714. indexs.insert(indexPage)
  715. }
  716. self?.deletePages(indexs: indexs)
  717. }
  718. }
  719. }
  720. }
  721. @objc func replaceMenuAction(sender: KMToolbarClickButton?) {
  722. let panel = NSOpenPanel()
  723. panel.allowedFileTypes = ["pdf"]
  724. panel.allowsMultipleSelection = true
  725. panel.beginSheetModal(for: self.view.window!) { [weak self] response in
  726. if (response == .cancel) {
  727. return
  728. }
  729. /// 处理page
  730. var pages: Array<CPDFPage> = []
  731. let selectedIndex: Int = (self?.getSelecteIndex())!
  732. self?.listView?.document.removePage(at: UInt(selectedIndex))
  733. var insertIndex: Int = selectedIndex
  734. self?.lockedFiles.removeAll()
  735. for fileURL in panel.urls {
  736. let document = CPDFDocument(url: fileURL)
  737. if ((document?.isLocked)!) {
  738. self?.lockedFiles.append(fileURL)
  739. continue
  740. }
  741. for i in 0 ..< document!.pageCount {
  742. let page = document?.page(at: i)
  743. pages.append(page!)
  744. self!.listView?.document.insertPageObject(page, at:UInt(insertIndex))
  745. insertIndex += 1
  746. }
  747. }
  748. if ((self?.lockedFiles.count)! <= 0) {
  749. self?.refreshUI()
  750. return
  751. }
  752. self?.refreshUI()
  753. self!.showPasswordWindow(insertIndex)
  754. }
  755. }
  756. @objc func splitMenuAction(sender: KMToolbarClickButton) {
  757. let model = KMPageEditSplitSettingModel()
  758. model.documentURL = self.listView?.document.documentURL
  759. model.fileName = model.documentURL.lastPathComponent
  760. model.pathExtension = model.fileName.components(separatedBy: ".").last
  761. if (self.thumbnailView.selectionIndexPaths.count > 0) {
  762. model.type = 2
  763. model.pageRangeType = 2
  764. var selectedIndexs: Array<Int> = []
  765. for indexpath in self.thumbnailView.selectionIndexPaths {
  766. selectedIndexs.append(indexpath.item)
  767. }
  768. model.pageRangeString = KMPageRangeTools.newParseSelectedIndexs(selectedIndex: selectedIndexs)
  769. } else {
  770. model.type = 0
  771. }
  772. let windowController = KMPageEditSplitWindowController(model)
  773. self.view.window?.beginSheet(windowController.window!)
  774. self.windowController = windowController
  775. windowController.itemClick = { [weak self] index, value in
  776. if (index == 1) { /// 取消
  777. self?.view.window?.endSheet((self?.windowController!.window)!)
  778. self?.windowController = nil
  779. return
  780. }
  781. /// 拆分
  782. let windowController_split: KMPageEditSplitWindowController = self?.windowController as! KMPageEditSplitWindowController
  783. let outputModel: KMPageEditSplitSettingModel = windowController_split.model! as! KMPageEditSplitSettingModel
  784. self?.view.window?.endSheet((self?.windowController!.window)!)
  785. self?.windowController = nil
  786. let panel = NSOpenPanel()
  787. panel.canChooseFiles = false
  788. panel.canChooseDirectories = true
  789. panel.canCreateDirectories = true
  790. let button = NSButton.init(checkboxWithTitle: "保存后打开文档", target: nil, action: nil)
  791. button.state = .off
  792. panel.accessoryView = button
  793. panel.beginSheetModal(for: (self?.view.window)!) { response in
  794. if (response == .cancel) {
  795. return
  796. }
  797. KMPageEditTools.split((self?.listView?.document)!, outputModel, panel.url!.path, outputModel.outputFileNameDeletePathExtension) { result, outputFilepaths, error in
  798. if (result) {
  799. if (button.state == .on) {
  800. for filepath in outputFilepaths! {
  801. if !filepath.isPDFValid() {
  802. let alert = NSAlert()
  803. alert.alertStyle = .critical
  804. alert.messageText = NSLocalizedString("An error occurred while opening this document. The file is damaged and could not be repaired.", comment: "")
  805. alert.runModal()
  806. } else {
  807. NSDocumentController.shared.openDocument(withContentsOf: URL(fileURLWithPath: filepath), display: true) { document, alreadyOpen, error in
  808. }
  809. }
  810. }
  811. } else {
  812. var urls: [URL] = []
  813. for filepath in outputFilepaths! {
  814. urls.append(URL(fileURLWithPath: filepath))
  815. }
  816. NSWorkspace.shared.activateFileViewerSelecting(urls)
  817. }
  818. }
  819. }
  820. }
  821. }
  822. }
  823. @objc func reverseMenuAction(sender: KMToolbarClickButton) {
  824. let selectedIndexs = self.thumbnailView.selectionIndexPaths
  825. if (selectedIndexs.count < 2 ) {
  826. let _ = CustomAlertView(message: "请先选择至少两个页面进行编辑操作", from: self.thumbnailView, with: .blue)
  827. return
  828. }
  829. var indexs: IndexSet = []
  830. for indexpath in selectedIndexs {
  831. indexs.insert(indexpath.item)
  832. }
  833. self.reversePages(indexs: indexs)
  834. }
  835. @objc func deleteMenuAction(sender: KMToolbarClickButton) {
  836. Task { @MainActor in
  837. if await (KMLightMemberManager.manager.canUseAdvanced() == false) {
  838. let _ = KMComparativeTableViewController.show(window: self.view.window!)
  839. return
  840. }
  841. var indexs: IndexSet = IndexSet()
  842. for indexPath in self.thumbnailView.selectionIndexPaths {
  843. indexs.insert(indexPath.item)
  844. }
  845. self.clearSelectPages()
  846. self.deletePages(indexs: indexs)
  847. }
  848. }
  849. }
  850. // MARK: -
  851. // MARK: KMNewToolbarDelegate
  852. extension KMPDFEditViewController: KMNewToolbarDelegate {
  853. func toolbarDefaultItemIdentifiers(_ toolbar: KMToolbar!) -> [String]! {
  854. return [KMToolbarPageEditPageRangeItemIdentifier,
  855. KMToolbarPageEditLeftRotateItemIdentifier,
  856. KMToolbarPageEditRightRotateItemIdentifier,
  857. // KMToolbarDividerItemIdentifier,
  858. // KMToolbarPageEditInsetItemIdentifier,
  859. // KMToolbarPageEditExtractItemIdentifier,
  860. // KMToolbarPageEditReplaceItemIdentifier,
  861. // KMToolbarPageEditSplitItemIdentifier,
  862. // KMToolbarPageEditReverseItemIdentifier,
  863. KMToolbarDividerItemIdentifier,
  864. KMToolbarPageEditDeleteItemIdentifier]
  865. }
  866. func toolbar(_ toolbar: KMToolbar!, itemForItemIdentifier identifier: String!) -> KMToolBoxItem! {
  867. let item = KMToolBoxItem(itemIdentifier: identifier)
  868. if (identifier == KMToolbarPageEditPageRangeItemIdentifier) {
  869. // item?.image = NSImage(named: "KMImageNameMainToolbarrRight")
  870. item?.alternateImage = NSImage(named: "KMImageNameMainToolbarrRight")
  871. item?.target = self
  872. item?.btnTag = KMPageEditType.pageRange.rawValue
  873. // item?.toolTip = NSLocalizedString("pageRange", comment: "")
  874. item?.titleName = NSLocalizedString("pageRange ", comment: "")
  875. item?.boxImagePosition = KMToolBoxImagePosition.left
  876. item?.btnAction = #selector(pageRangeMenuAction)
  877. let view = NSView()
  878. item?.addSubview(view)
  879. view.frame = item!.bounds
  880. view.autoresizingMask = [.width,.height]
  881. view.wantsLayer = true
  882. view.layer?.backgroundColor = NSColor(hex: "#F7F8FA").cgColor
  883. let comBoxView = KMDesignSelect(withType: .Combox)
  884. view.addSubview(comBoxView.view)
  885. comBoxView.view.frame = NSMakeRect(0, 8, NSWidth(view.frame), 24)
  886. comBoxView.view.autoresizingMask = [.width]
  887. comBoxView.removeAllItems()
  888. comBoxView.addItems(withObjectValues: self.pageRangeViewItems)
  889. comBoxView.selectItem(at: 0)
  890. comBoxView.placeholderString = NSLocalizedString("eg .1,3-5,10", comment: "")
  891. comBoxView.editable = false
  892. comBoxView.delete = self
  893. pageRangeView = comBoxView
  894. let menuItem = NSMenuItem.init(title: "pageRange", action: #selector(pageRangeMenuAction), target: self)
  895. item?.menuFormRepresentation = menuItem
  896. } else if (identifier == KMToolbarPageEditLeftRotateItemIdentifier) {
  897. item?.image = NSImage(named: "KMImageNameMainToolPageLeftRotate")
  898. item?.alternateImage = NSImage(named: "KMImageNameMainToolPageLeftRotate")
  899. item?.target = self
  900. item?.btnTag = KMPageEditType.leftRotate.rawValue
  901. item?.toolTip = NSLocalizedString("Rotate Left", comment: "")
  902. item?.titleName = NSLocalizedString("", comment: "")
  903. item?.boxImagePosition = KMToolBoxImagePosition.left
  904. item?.btnAction = #selector(leftRotateMenuAction)
  905. let menuItem = NSMenuItem.init(title: "", action: #selector(leftRotateMenuAction), target: self)
  906. item?.menuFormRepresentation = menuItem
  907. } else if (identifier == KMToolbarPageEditRightRotateItemIdentifier) {
  908. item?.image = NSImage(named: "KMImageNameMainToolPageRightRotate")
  909. item?.alternateImage = NSImage(named: "KMImageNameMainToolPageRightRotate")
  910. item?.target = self
  911. item?.btnTag = KMPageEditType.rightRotate.rawValue
  912. item?.toolTip = NSLocalizedString("Rotate Right", comment: "")
  913. item?.titleName = NSLocalizedString("", comment: "")
  914. item?.boxImagePosition = KMToolBoxImagePosition.left
  915. item?.btnAction = #selector(rightRotateMenuAction)
  916. let menuItem = NSMenuItem.init(title: "", action: #selector(rightRotateMenuAction), target: self)
  917. item?.menuFormRepresentation = menuItem
  918. } else if (identifier == KMToolbarPageEditInsetItemIdentifier) {
  919. item?.image = NSImage(named: "KMImageNameMainToolPageInsert")
  920. item?.alternateImage = NSImage(named: "KMImageNameMainToolPageInsert")
  921. item?.target = self
  922. item?.btnTag = KMPageEditType.insert.rawValue
  923. item?.toolTip = NSLocalizedString("Insert", comment: "")
  924. item?.titleName = NSLocalizedString("Insert", comment: "")
  925. item?.boxImagePosition = KMToolBoxImagePosition.left
  926. item?.btnAction = #selector(insertMenuAction)
  927. let menuItem = NSMenuItem.init(title: "Insert", action: #selector(insertMenuAction), target: self)
  928. item?.menuFormRepresentation = menuItem
  929. } else if (identifier == KMToolbarPageEditExtractItemIdentifier) {
  930. item?.image = NSImage(named: "KMImageNameMainToolPageExtract")
  931. item?.alternateImage = NSImage(named: "KMImageNameMainToolPageExtract")
  932. item?.target = self
  933. item?.btnTag = KMPageEditType.extract.rawValue
  934. item?.toolTip = NSLocalizedString("Extract", comment: "")
  935. item?.titleName = NSLocalizedString("Extract", comment: "")
  936. item?.boxImagePosition = KMToolBoxImagePosition.left
  937. item?.btnAction = #selector(extractMenuAction)
  938. let menuItem = NSMenuItem.init(title: "Extract", action: #selector(extractMenuAction), target: self)
  939. item?.menuFormRepresentation = menuItem
  940. } else if (identifier == KMToolbarPageEditReplaceItemIdentifier) {
  941. item?.image = NSImage(named: "KMImageNameMainToolPageReplace")
  942. item?.alternateImage = NSImage(named: "KMImageNameMainToolPageReplace")
  943. item?.target = self
  944. item?.btnTag = KMPageEditType.replace.rawValue
  945. item?.toolTip = NSLocalizedString("Replace", comment: "")
  946. item?.titleName = NSLocalizedString("Replace", comment: "")
  947. item?.boxImagePosition = KMToolBoxImagePosition.left
  948. item?.btnAction = #selector(replaceMenuAction)
  949. let menuItem = NSMenuItem.init(title: "Replace", action: #selector(replaceMenuAction), target: self)
  950. item?.menuFormRepresentation = menuItem
  951. } else if (identifier == KMToolbarPageEditSplitItemIdentifier) {
  952. item?.image = NSImage(named: "KMImageNameMainToolPageSplit")
  953. item?.alternateImage = NSImage(named: "KMImageNameMainToolPageSplit")
  954. item?.target = self
  955. item?.btnTag = KMPageEditType.split.rawValue
  956. item?.toolTip = NSLocalizedString("Split the PDF into multiple files", comment: "")
  957. item?.titleName = NSLocalizedString("Split", comment: "")
  958. item?.boxImagePosition = KMToolBoxImagePosition.left
  959. item?.btnAction = #selector(splitMenuAction)
  960. let menuItem = NSMenuItem.init(title: "Split", action: #selector(splitMenuAction), target: self)
  961. item?.menuFormRepresentation = menuItem
  962. } else if (identifier == KMToolbarPageEditReverseItemIdentifier) {
  963. item?.image = NSImage(named: "KMImageNameMainToolPageReverse")
  964. item?.alternateImage = NSImage(named: "KMImageNameMainToolPageReverse")
  965. item?.target = self
  966. item?.btnTag = KMPageEditType.reverse.rawValue
  967. item?.toolTip = NSLocalizedString("Reverse", comment: "")
  968. item?.titleName = NSLocalizedString("Reverse", comment: "")
  969. item?.boxImagePosition = KMToolBoxImagePosition.left
  970. item?.btnAction = #selector(reverseMenuAction)
  971. let menuItem = NSMenuItem.init(title: "Reverse", action: #selector(reverseMenuAction), target: self)
  972. item?.menuFormRepresentation = menuItem
  973. } else if (identifier == KMToolbarPageEditDeleteItemIdentifier) {
  974. item?.image = NSImage(named: "KMImageNameMainToolPageDelete")
  975. item?.alternateImage = NSImage(named: "KMImageNameMainToolPageDelete")
  976. item?.target = self
  977. item?.btnTag = KMPageEditType.delete.rawValue
  978. item?.toolTip = NSLocalizedString("Delete", comment: "")
  979. item?.titleName = NSLocalizedString("", comment: "")
  980. item?.boxImagePosition = KMToolBoxImagePosition.left
  981. item?.btnAction = #selector(deleteMenuAction)
  982. let menuItem = NSMenuItem.init(title: "Delete", action: #selector(deleteMenuAction), target: self)
  983. item?.menuFormRepresentation = menuItem
  984. }
  985. return item
  986. }
  987. //MARK: menu Action
  988. @objc func insertPageItemAction(menu:NSMenuItem) {
  989. if (menu.tag == 0) {
  990. self.item_insertFile(sender: nil)
  991. } else if (menu.tag == 1) {
  992. self.item_insertBlankPage(sender: nil)
  993. } else if (menu.tag == 2) {
  994. self.item_insertCustomPage(sender: nil)
  995. }
  996. }
  997. @objc func rotatePageItemAction(menu:NSMenuItem) {
  998. rightRotateMenuAction(sender: nil)
  999. }
  1000. @objc func leftRotatePageItemAction(menu:NSMenuItem) {
  1001. leftRotateMenuAction(sender: nil)
  1002. }
  1003. @objc func extractPageItemAction(menu:NSMenuItem) {
  1004. extractMenuAction(sender: nil)
  1005. }
  1006. @objc func replacePageItemAction(menu:NSMenuItem) {
  1007. replaceMenuAction(sender: nil)
  1008. }
  1009. @objc func deletePageItemAction(menu:NSMenuItem) {
  1010. Task { @MainActor in
  1011. if await (KMLightMemberManager.manager.canUseAdvanced() == false) {
  1012. let _ = KMComparativeTableViewController.show(window: self.view.window!)
  1013. return
  1014. }
  1015. let indexs = menu.representedObject as! IndexSet
  1016. if indexs.count > 0 {
  1017. self.clearSelectPages()
  1018. self.deletePages(indexs: indexs)
  1019. }
  1020. }
  1021. }
  1022. @objc func sharePageItemAction(menu:NSMenuItem) {
  1023. let item = menu.parent!
  1024. let index = (item.representedObject as! IndexSet).first ?? -1
  1025. if Int(index) >= 0 {
  1026. let doucument = self.pdfDocument!
  1027. let page = doucument.page(at: UInt(index))
  1028. let filename : String = doucument.documentURL.lastPathComponent
  1029. let folderPath = (NSSearchPathForDirectoriesInDomains(FileManager.SearchPathDirectory.applicationSupportDirectory, FileManager.SearchPathDomainMask.userDomainMask, true).last?.stringByAppendingPathComponent(filename))!
  1030. try? FileManager.default.removeItem(atPath: folderPath)
  1031. let pdfdocument = CPDFDocument()
  1032. let ret = pdfdocument!.importPages((item.representedObject as! IndexSet), from: self.thumbnailView.document, at: 0)
  1033. let url = URL(fileURLWithPath: folderPath)
  1034. if ret {
  1035. let success = pdfdocument!.write(to:url)
  1036. if success {
  1037. let workspace = NSWorkspace.shared
  1038. workspace.activateFileViewerSelecting([url])
  1039. }
  1040. }
  1041. let represent : NSSharingService = menu.representedObject as! NSSharingService
  1042. represent.perform(withItems: [url])
  1043. }
  1044. }
  1045. @objc func copyItemAction(menu:NSMenuItem) {
  1046. let indexs = menu.representedObject as! IndexSet
  1047. self.copyPages(with: indexs)
  1048. }
  1049. @objc func cutItemAction(menu:NSMenuItem) {
  1050. let indexs = menu.representedObject as! IndexSet
  1051. self.cutPages(with: indexs)
  1052. }
  1053. @objc func pastePageItemAction(menu:NSMenuItem) {
  1054. var index = (menu.representedObject as! IndexSet).last ?? -1
  1055. self.pastePages(at: index)
  1056. }
  1057. @objc func showPageSizeItemAction(menu: NSMenuItem) {
  1058. self.thumbnailView.isShowPageSize = !self.thumbnailView.isShowPageSize
  1059. self.refreshUIForKeepSelecteds()
  1060. }
  1061. @objc func printItemAction(menu:NSMenuItem) {
  1062. var datas: [Int] = []
  1063. for indexpath in self.thumbnailView.selectionIndexPaths {
  1064. datas.append(indexpath.item)
  1065. }
  1066. guard let callback = self.itemClick else {
  1067. return
  1068. }
  1069. callback(2, datas.sorted())
  1070. }
  1071. }
  1072. // MARK: -
  1073. // MARK: NSMenuDelegate,NSMenuItemValidation
  1074. extension KMPDFEditViewController : NSMenuDelegate,NSMenuItemValidation {
  1075. func validateMenuItem(_ menuItem: NSMenuItem) -> Bool {
  1076. let action = menuItem.action
  1077. if action == #selector(insertPageItemAction) ||
  1078. action == #selector(insertPageItemAction) ||
  1079. action == #selector(deletePageItemAction) ||
  1080. action == #selector(replacePageItemAction) ||
  1081. action == #selector(rotatePageItemAction) ||
  1082. action == #selector(leftRotatePageItemAction) ||
  1083. action == #selector(copyItemAction) ||
  1084. action == #selector(cutItemAction) ||
  1085. action == #selector(pastePageItemAction) ||
  1086. action == #selector(printItemAction) ||
  1087. action == #selector(sharePageItemAction) ||
  1088. action == #selector(extractPageItemAction) ||
  1089. action == #selector(showPageSizeItemAction) {
  1090. let row = self.thumbnailView.collectionView.selectionIndexes
  1091. if row.count > 0 {
  1092. if (row.count > 1 && action == #selector(replacePageItemAction)) {
  1093. return false
  1094. }
  1095. if (action == #selector(pastePageItemAction)) {
  1096. return self.copysPages.count > 0
  1097. }
  1098. return true
  1099. }
  1100. return false
  1101. }
  1102. if (action == #selector(cut)) {
  1103. return self.thumbnailView.collectionView.selectionIndexes.count > 0
  1104. }
  1105. if (action == NSSelectorFromString("copy:")) {
  1106. return self.thumbnailView.collectionView.selectionIndexes.count > 0
  1107. }
  1108. if (action == #selector(paste)) {
  1109. return self.copysPages.count > 0
  1110. }
  1111. if (action == #selector(undo)) {
  1112. return self.undoRedoManager.canUndo
  1113. }
  1114. if (action == #selector(redo)) {
  1115. return self.undoRedoManager.canRedo
  1116. }
  1117. if (action == #selector(delete)) {
  1118. return self.thumbnailView.collectionView.selectionIndexes.count > 0
  1119. }
  1120. return true
  1121. }
  1122. func menuNeedsUpdate(_ menu: NSMenu) {
  1123. menu.removeAllItems()
  1124. let row = self.thumbnailView.collectionView.selectionIndexes
  1125. // let point = self.thumbnailView.collectionView.convert(NSEvent.mouseLocation, to: nil)
  1126. // let indexpaths = self.thumbnailView.collectionView.indexPathForItem(at: point)
  1127. if (row.isEmpty) {
  1128. return
  1129. }
  1130. var item = NSMenuItem()
  1131. // if (row.count == 1) {
  1132. // item = menu.addItem(withTitle: NSLocalizedString("Insert", comment: ""), action: nil, target: self)
  1133. // item.representedObject = row
  1134. //
  1135. // var subMenu = NSMenu()
  1136. // var subitem = NSMenuItem()
  1137. // subitem = subMenu.addItem(withTitle: NSLocalizedString("Insert File", comment: ""), action: #selector(insertPageItemAction), target: self, tag:0)
  1138. // subitem.representedObject = row
  1139. // subitem = subMenu.addItem(withTitle: NSLocalizedString("Insert blank page", comment: ""), action: #selector(insertPageItemAction), target: self, tag:1)
  1140. // subitem.representedObject = row
  1141. // subitem = subMenu.addItem(withTitle: NSLocalizedString("Insert page", comment: ""), action: #selector(insertPageItemAction), target: self, tag:2)
  1142. // subitem.representedObject = row
  1143. // item.submenu = subMenu
  1144. // }
  1145. // item = menu.addItem(withTitle: NSLocalizedString("Extract", comment: ""), action: #selector(extractPageItemAction), target: self)
  1146. // item.representedObject = row
  1147. // if (row.count == 1) {
  1148. // item = menu.addItem(withTitle: NSLocalizedString("Replace Page", comment: ""), action: #selector(replacePageItemAction), target: self)
  1149. // item.representedObject = row
  1150. // }
  1151. item = menu.addItem(withTitle: NSLocalizedString("Delete Page", comment: ""), action: #selector(deletePageItemAction), target: self)
  1152. item.keyEquivalent = String(Unicode.Scalar(NSBackspaceCharacter)!)
  1153. item.keyEquivalentModifierMask = []
  1154. item.representedObject = row
  1155. menu.addItem(NSMenuItem.separator())
  1156. item = menu.addItem(withTitle: NSLocalizedString("Rotate Clockwise", comment: ""), action: #selector(rotatePageItemAction), target: self)
  1157. item.representedObject = row
  1158. item = menu.addItem(withTitle: NSLocalizedString("Rotate Counterclockwise", comment: ""), action: #selector(leftRotatePageItemAction), target: self)
  1159. item.representedObject = row
  1160. menu.addItem(NSMenuItem.separator())
  1161. item = menu.addItem(withTitle: NSLocalizedString("Copy", comment: ""), action: #selector(copyItemAction), target: self)
  1162. item.representedObject = row
  1163. item.keyEquivalent = "c"
  1164. item.keyEquivalentModifierMask = [.command]
  1165. item = menu.addItem(withTitle: NSLocalizedString("Cut", comment: ""), action: #selector(cutItemAction), target: self)
  1166. item.representedObject = row
  1167. item.keyEquivalent = "x"
  1168. item = menu.addItem(withTitle: NSLocalizedString("Paste", comment: ""), action: #selector(pastePageItemAction), target: self)
  1169. item.representedObject = row
  1170. item.keyEquivalent = "v"
  1171. menu.addItem(NSMenuItem.separator())
  1172. if (!self.thumbnailView.isShowPageSize) {
  1173. item = menu.addItem(withTitle: NSLocalizedString("Hide Page Size", comment: ""), action: #selector(showPageSizeItemAction), target: self)
  1174. item.representedObject = row
  1175. } else {
  1176. item = menu.addItem(withTitle: NSLocalizedString("Display Page Size", comment: ""), action: #selector(showPageSizeItemAction), target: self)
  1177. item.representedObject = row
  1178. }
  1179. menu.addItem(NSMenuItem.separator())
  1180. item = menu.addItem(withTitle: NSLocalizedString("Print", comment: ""), action: #selector(printItemAction), target: self)
  1181. item.representedObject = row
  1182. // if (row.count <= 1) {
  1183. // item = menu.addItem(withTitle: NSLocalizedString("Share", comment: ""), action: nil, target: self)
  1184. // item.representedObject = row
  1185. // item.submenu = NSSharingServicePicker.menu(forSharingItems: [self.listView!.document.documentURL ?? ""], subjectContext: "", withTarget: self, selector: #selector(sharePageItemAction), serviceDelegate: nil)
  1186. // }
  1187. }
  1188. }
  1189. // MARK: -
  1190. // MARK: KMSelectPopButtonDelegate
  1191. extension KMPDFEditViewController: KMSelectPopButtonDelegate {
  1192. func km_comboBoxSelectionDidChange(_ obj: KMDesignSelect) {
  1193. let index: Int = self.pageRangeView!.indexOfSelectedItem
  1194. var indexpaths: Set<IndexPath> = []
  1195. if (index <= 0) { /// 全部页面
  1196. self.pageRangeView?.editable = false
  1197. for i in 0 ..< (self.listView?.document.pageCount)! {
  1198. indexpaths.insert(IndexPath(item: Int(i), section: 0))
  1199. }
  1200. self.pageIsUpdate = false
  1201. self.thumbnailView.selectPages(at: indexpaths)
  1202. } else if (index == 1) { /// 奇数页
  1203. self.pageRangeView?.editable = false
  1204. for i in 0 ..< (self.listView?.document.pageCount)! {
  1205. if (i % 2 == 1) {
  1206. continue
  1207. }
  1208. indexpaths.insert(IndexPath(item: Int(i), section: 0))
  1209. }
  1210. self.pageIsUpdate = false
  1211. self.thumbnailView.selectPages(at: indexpaths)
  1212. } else if (index == 2) { /// 偶数页
  1213. self.pageRangeView?.editable = false
  1214. for i in 0 ..< (self.listView?.document.pageCount)! {
  1215. if (i % 2 == 0) {
  1216. continue
  1217. }
  1218. indexpaths.insert(IndexPath(item: Int(i), section: 0))
  1219. }
  1220. self.pageIsUpdate = false
  1221. self.thumbnailView.selectPages(at: indexpaths)
  1222. } else if (index == 3) { /// 横向页
  1223. self.pageRangeView?.editable = false
  1224. for i in 0 ..< (self.listView?.document.pageCount)! {
  1225. let page = self.listView?.document.page(at: i)
  1226. if (page?.rotation == 0 || page?.rotation == 180 || page?.rotation == -180) {
  1227. continue
  1228. }
  1229. indexpaths.insert(IndexPath(item: Int(i), section: 0))
  1230. }
  1231. self.pageIsUpdate = false
  1232. self.thumbnailView.selectPages(at: indexpaths)
  1233. } else if (index == 4) { /// 纵向页
  1234. self.pageRangeView?.editable = false
  1235. for i in 0 ..< (self.listView?.document.pageCount)! {
  1236. let page = self.listView?.document.page(at: i)
  1237. if (page?.rotation == 90 || page?.rotation == 270 || page?.rotation == -90 || page?.rotation == -270) {
  1238. continue
  1239. }
  1240. indexpaths.insert(IndexPath(item: Int(i), section: 0))
  1241. }
  1242. self.pageIsUpdate = false
  1243. self.thumbnailView.selectPages(at: indexpaths)
  1244. } else { /// 自定义
  1245. self.pageIsUpdate = false
  1246. self.pageRangeView?.editable = true
  1247. self.clearSelectPages()
  1248. DispatchQueue.main.async {
  1249. self.pageRangeView?.stringValue = ""
  1250. }
  1251. }
  1252. }
  1253. func km_controlTextDidChange(_ obj: KMDesignSelect) {
  1254. }
  1255. func km_controlTextDidEndEditing(_ obj: KMDesignSelect) {
  1256. self.view.window?.makeFirstResponder(self)
  1257. for index in 0 ..< self.pageRangeView!.numberOfItems {
  1258. if (self.pageRangeViewItems[index] == self.pageRangeView!.stringValue) {
  1259. return
  1260. }
  1261. }
  1262. let pages = KMPageRangeTools.findSelectPage(pageRangeString: self.pageRangeView!.stringValue, pageCount: Int((self.listView?.document.pageCount)!))
  1263. if (pages.isEmpty) {
  1264. let alert = NSAlert()
  1265. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  1266. alert.runModal()
  1267. return
  1268. }
  1269. var indexpaths: Set<IndexPath> = []
  1270. for page in pages {
  1271. if (page >= 1) {
  1272. indexpaths.insert(IndexPath(item: page-1, section: 0))
  1273. }
  1274. }
  1275. self.pageIsUpdate = false
  1276. self.thumbnailView.selectPages(at: indexpaths)
  1277. }
  1278. }
  1279. // MARK: -
  1280. // MARK: KMPDFThumbnailViewDelegate
  1281. extension KMPDFEditViewController: KMPDFThumbnailViewDelegate {
  1282. func thumbnailView(thumbanView: KMPDFThumbnailView, didDragAddFiles files: [URL], indexpath: IndexPath) {
  1283. /// 处理page
  1284. // var insertIndex: Int = indexpath.item
  1285. // self.lockedFiles.removeAll()
  1286. // var pages: Array<CPDFPage> = []
  1287. // self.documentCaches.removeAll()
  1288. // var indexs = IndexSet()
  1289. // for fileURL in files {
  1290. // let document = CPDFDocument(url: fileURL)
  1291. // if ((document?.isLocked)!) {
  1292. // self.lockedFiles.append(fileURL)
  1293. // continue
  1294. // }
  1295. //
  1296. // self.documentCaches.append(document!)
  1297. // for i in 0 ..< document!.pageCount {
  1298. // let page = document?.page(at: i)
  1299. // pages.append(page!)
  1300. // indexs.insert(insertIndex)
  1301. // insertIndex += 1
  1302. // }
  1303. // }
  1304. // self.insertPagesForDescSort(pages: pages, indexs: indexs)
  1305. //
  1306. // if (self.lockedFiles.count <= 0) {
  1307. // self.refreshUI()
  1308. // return
  1309. // }
  1310. //
  1311. // self.refreshUI()
  1312. // self.showPasswordWindow(insertIndex)
  1313. }
  1314. func thumbnailView(thumbanView: KMPDFThumbnailView, didDragPages pages: [Int], indexpath: IndexPath) {
  1315. let toIndex = max(0, indexpath.item)
  1316. var indexs = IndexSet()
  1317. for page in pages {
  1318. indexs.insert(page)
  1319. }
  1320. self.dragPages(indexs, toIndex)
  1321. if (pages.count > 0) {
  1322. // 提醒外面文档已编辑
  1323. if (self.documentEditedCallback != nil) {
  1324. self.documentEditedCallback!()
  1325. }
  1326. }
  1327. }
  1328. // MARK: thumbanView layout
  1329. func thumbnailView(thumbanView: KMPDFThumbnailView, minimumLineSpacingForSectionAt section: Int) -> CGFloat {
  1330. return 0.01
  1331. }
  1332. func thumbnailView(thumbanView: KMPDFThumbnailView, minimumInteritemSpacingForSectionAt section: Int) -> CGFloat {
  1333. return 0.01
  1334. }
  1335. func thumbnailView(thumbanView: KMPDFThumbnailView, insetForSectionAt section: Int) -> NSEdgeInsets {
  1336. return NSEdgeInsets(top: 16, left: 16, bottom: 0, right: 32)
  1337. }
  1338. func thumbnailView(thumbanView: KMPDFThumbnailView, sizeForItemAt indexpath: IndexPath) -> NSSize {
  1339. self.thumbnailView.thumbnailSzie = CGSize(width: 120, height: 155)
  1340. if (!self.thumbnailView.isShowPageSize) {
  1341. return NSMakeSize(self.itemSize.width, self.itemSize.height+26)
  1342. }
  1343. return self.itemSize
  1344. }
  1345. func thumbnailView(thumbanView: KMPDFThumbnailView, itemForRepresentedObjectAt indexpath: IndexPath) -> NSCollectionViewItem {
  1346. let cellView: KMPageEditThumbnailItem = thumbanView.collectionView.makeItem(withIdentifier: NSUserInterfaceItemIdentifier(rawValue: "KMPageEditThumbnailItem"), for: indexpath) as! KMPageEditThumbnailItem
  1347. cellView.isShowPageSize = self.thumbnailView.isShowPageSize
  1348. cellView.page = (self.pdfDocument?.page(at: UInt(indexpath.item)))!
  1349. cellView.setPage(page: (self.pdfDocument?.page(at: UInt(indexpath.item)))!)
  1350. cellView.doubleClickAction = { [unowned self] _ in
  1351. guard let callback = self.itemClick else {
  1352. return
  1353. }
  1354. callback(1, indexpath.item)
  1355. }
  1356. cellView.mouseDownAction = { [unowned self] _ in
  1357. let selectionIndexPaths = self.thumbnailView.selectionIndexPaths
  1358. if (selectionIndexPaths.count == 1 && selectionIndexPaths.first?.item == indexpath.item) {
  1359. return
  1360. }
  1361. self.selectPages(indexs: IndexSet.init(integer: indexpath.item))
  1362. }
  1363. cellView.rightMouseDownAction = { [unowned self] _ in
  1364. self.selectPages(indexs: IndexSet.init(integer: indexpath.item))
  1365. }
  1366. return cellView
  1367. }
  1368. }
  1369. // MARK: -
  1370. // MARK: 系统菜单栏
  1371. extension KMPDFEditViewController: KMEditMenuProtocol {
  1372. @IBAction func cut(_ sender: Any) {
  1373. self.cutPages(with: self.thumbnailView.collectionView.selectionIndexes)
  1374. self.km_becomeFirstResponder()
  1375. }
  1376. @IBAction func copy(_ sender: Any) {
  1377. self.copyPages(with: self.thumbnailView.collectionView.selectionIndexes)
  1378. self.km_becomeFirstResponder()
  1379. }
  1380. @IBAction func paste(_ sender: Any) {
  1381. var index: Int = Int(self.pdfDocument!.pageCount-1)
  1382. let indexSet = self.thumbnailView.collectionView.selectionIndexes
  1383. if (indexSet.isEmpty == false) {
  1384. index = indexSet.last!
  1385. }
  1386. self.pastePages(at: index)
  1387. self.km_becomeFirstResponder()
  1388. }
  1389. @IBAction func delete(_ sender: Any) {
  1390. Task { @MainActor in
  1391. if await (KMLightMemberManager.manager.canUseAdvanced() == false) {
  1392. let _ = KMComparativeTableViewController.show(window: self.view.window!)
  1393. return
  1394. }
  1395. let selectedIndexs = self.thumbnailView.collectionView.selectionIndexPaths
  1396. var indexs: IndexSet = IndexSet()
  1397. for indexPath in selectedIndexs {
  1398. indexs.insert(indexPath.item)
  1399. }
  1400. self.clearSelectPages()
  1401. self.deletePages(indexs: indexs)
  1402. self.km_becomeFirstResponder()
  1403. }
  1404. }
  1405. @IBAction func undo(_ sender: Any) {
  1406. if (self.undoRedoManager.canUndo) {
  1407. self.undoRedoManager.undo()
  1408. }
  1409. self.km_becomeFirstResponder()
  1410. }
  1411. @IBAction func redo(_ sender: Any) {
  1412. if (self.undoRedoManager.canRedo) {
  1413. self.undoRedoManager.redo()
  1414. }
  1415. self.km_becomeFirstResponder()
  1416. }
  1417. }