KMPDFEditViewController.swift 65 KB

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