KMPDFEditViewController.swift 57 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438
  1. //
  2. // KMPDFEditViewController.swift
  3. // PDF Reader Pro
  4. //
  5. // Created by tangchao on 2023/1/6.
  6. //
  7. typealias KMPDFEditViewControllerSelectionDidChange = (_ selectedIndexs: Set<IndexPath>) -> ()
  8. class KMPDFEditViewController: KMPDFThumbViewBaseController {
  9. @IBOutlet weak var topBarBox: NSBox!
  10. @IBOutlet weak var contentBox: NSBox!
  11. @IBOutlet weak var thumbnailSizeView: NSView!
  12. @IBOutlet weak var lineBoxView: NSView!
  13. var toolBar = KMPDFEditToolbar(frame: NSZeroRect)
  14. override var thumbnailView: KMPDFThumbnailView {
  15. get {
  16. if (super.thumbnailView is KMPageEditThumbnailView) {
  17. return super.thumbnailView
  18. }
  19. super.thumbnailView = KMPageEditThumbnailView()
  20. return super.thumbnailView
  21. }
  22. set {
  23. super.thumbnailView = newValue
  24. }
  25. }
  26. var pageEditThumbnailView: KMPageEditThumbnailView {
  27. get {
  28. return self.thumbnailView as! KMPageEditThumbnailView
  29. }
  30. }
  31. override var menuItems: [[KMPDFThumbViewMenu.Item]] {
  32. get {
  33. return [[.insert ,.extract], [.leftRotate, .rightRotate, .copy, .paste, .reverse],
  34. [.delete], [.share]]
  35. }
  36. }
  37. private var pdfDocument: CPDFDocument?
  38. var listView: CPDFView?
  39. var pageIsUpdate: Bool = true
  40. var selectedPages: [Int] = []
  41. private let defaultItemSize = NSMakeSize(215, 250)
  42. var itemSize: NSSize = NSMakeSize(215, 250)
  43. // index=1 双击退出;index=2 打印
  44. var itemClick: KMCommonClickBlock?
  45. var documentEditedCallback: KMCommonBlock?
  46. var selectionDidChange: KMPDFEditViewControllerSelectionDidChange?
  47. internal var viewFirstAppear = true
  48. private var _listViewCurrentIndex: Int = 0
  49. var listViewCurrentIndex: Int {
  50. get {
  51. return self._listViewCurrentIndex
  52. }
  53. }
  54. override var canPayFunction: Bool {
  55. didSet {
  56. self.thumbnailView.limit = !self.canPayFunction
  57. }
  58. }
  59. required init?(coder: NSCoder) {
  60. super.init(coder: coder)
  61. }
  62. init(_ document: CPDFDocument) {
  63. super.init(nibName: "KMPDFEditViewController", bundle: nil)
  64. self.pdfDocument = document
  65. self.kmUndoManager = UndoManager()
  66. }
  67. override func viewDidAppear() {
  68. super.viewDidAppear()
  69. if (self.pdfDocument?.isLocked == false && self.viewFirstAppear) {
  70. self.viewFirstAppear = false
  71. self.thumbnailView.document = self.pdfDocument!
  72. self.thumbnailView.reloadData()
  73. }
  74. if let view = self.listView { // 存储阅读视图的当前页面索引
  75. self._listViewCurrentIndex = view.currentPageIndex
  76. }
  77. if (self.selectedPages.isEmpty) {
  78. var index: Int = 0
  79. if (self.listView != nil) {
  80. index = self.listView!.currentPageIndex
  81. }
  82. self.selectPages(indexs: IndexSet(integer: index))
  83. self.toolBar.setPageRangeString("\(index+1)")
  84. } else {
  85. var indexs: IndexSet = []
  86. for pageIndex in self.selectedPages {
  87. if (pageIndex < 0 || pageIndex >= (self.listView?.document.pageCount)!) {
  88. continue
  89. }
  90. indexs.insert(pageIndex)
  91. }
  92. self.selectPages(indexs: indexs)
  93. self.toolBar.setPageRangeString(KMPageRangeTools.newParseSelectedIndexs(selectedIndex: indexs.sorted()))
  94. }
  95. self.view.window?.makeFirstResponder(self)
  96. }
  97. override func viewDidLoad() {
  98. super.viewDidLoad()
  99. // 显示拖放移动动效
  100. // self.thumbnailView.dragMoveEffectAnimated = true
  101. // self.thumbnailView.collectionView.backgroundColor(NSColor.km_init(hex: "#CED0D4"))
  102. self.thumbnailView.collectionView.wantsLayer = true
  103. self.thumbnailView.collectionView.layer?.backgroundColor = KMAppearance.PDFViewBackgroundColor().cgColor
  104. self.thumbnailSizeView.wantsLayer = true
  105. self.thumbnailSizeView.layer?.backgroundColor = KMAppearance.Layout.b80Color().cgColor
  106. self.lineBoxView.wantsLayer = true
  107. self.lineBoxView.layer?.backgroundColor = KMAppearance.Layout.h2Color().cgColor
  108. var allowedFileTypes: [String] = []
  109. // if let _types = KMConvertPDFManager.supportFileType() {
  110. allowedFileTypes = KMTools.pdfExtensions + KMConvertPDFManager.supportFileType()
  111. // } else {
  112. // allowedFileTypes = KMTools.pdfExtensions + KMTools.imageExtensions
  113. // }
  114. self.thumbnailView.kmAllowedFileTypes = allowedFileTypes
  115. let toolBar = self.toolBar
  116. self.topBarBox.contentView = toolBar
  117. self.topBarBox.fillColor = NSColor.km_init(hex: "#F7F8FA")
  118. toolBar.itemClick = { [unowned self] item, index in
  119. guard let type = KMPageEditType(rawValue: index) else {
  120. if (index == KMPageEditType.insert_files) {
  121. self.insertFileAction()
  122. } else if (index == KMPageEditType.insert_blank_page) {
  123. self.insertBlankPageAction()
  124. } else if (index == KMPageEditType.insert_custom_page) {
  125. self.item_insertCustomPage(sender: nil)
  126. } else if (index == KMPageEditType.extract_selected_pages) {
  127. self.extractSelectPageItemAction()
  128. } else if (index == KMPageEditType.split_selected_pages) {
  129. self.splitMenuAction()
  130. }
  131. return
  132. }
  133. if (type == .leftRotate) {
  134. self.leftRotateAction()
  135. } else if (type == .rightRotate) {
  136. self.rightRotateAction()
  137. } else if (type == .zoomOut) {
  138. self.zoomOutButtonAction(sender: item?.zoomOutButton)
  139. } else if (type == .zoomIn) {
  140. self.zoomInButtonAction(sender: item?.zoomInButton)
  141. } else if (type == .extract) {
  142. self.extractAction()
  143. } else if (type == .replace) {
  144. self.replaceAction()
  145. } else if (type == .split) {
  146. self.splitAction()
  147. } else if (type == .reverse) {
  148. self.reverseAction()
  149. } else if (type == .delete) {
  150. self.deleteAction()
  151. } else if (type == .pageRange) {
  152. // self.km_comboBoxSelectionDidChange((item?.pageRangeView)!)
  153. } else if (type == .insert) {
  154. self.trackEvent(type: .insert)
  155. } else if (type == .copy) {
  156. self.copyAction()
  157. } else if (type == .paste) {
  158. self.pasteAction()
  159. }
  160. }
  161. toolBar.pageRangeitemClick = { [unowned self] type in
  162. self.km_comboBoxSelectionDidChange(type)
  163. }
  164. toolBar.pageRangeValueDidChange = { [unowned self] value, _ in
  165. self.km_controlTextDidEndEditing(self.toolBar.pageRangeView!)
  166. }
  167. let view = self.thumbnailView
  168. self.contentBox.contentView?.addSubview(view)
  169. view.frame = self.contentBox.contentView!.bounds
  170. view.autoresizingMask = [.width, .height]
  171. self.pageEditThumbnailView.delegate = self
  172. self.pageEditThumbnailView.selectionDidChange = { [weak self] selectedIndexs in
  173. guard let self = self else {
  174. return
  175. }
  176. self.selectionDidChange?(selectedIndexs)
  177. self.view.window?.makeFirstResponder(self)
  178. let enabled = (selectedIndexs.count > 0)
  179. for i in 0 ..< self.toolBar.toolBar.items.count {
  180. let item = self.toolBar.toolBar.items[i]
  181. if (item.itemIdentifier == KMToolbarPageEditExtractItemIdentifier ||
  182. item.itemIdentifier == KMToolbarPageEditLeftRotateItemIdentifier ||
  183. item.itemIdentifier == KMToolbarPageEditRightRotateItemIdentifier ||
  184. item.itemIdentifier == KMToolbarPageEditReplaceItemIdentifier) {
  185. item.unEnabled = !enabled
  186. } else if (item.itemIdentifier == KMToolbarPageEditReverseItemIdentifier) {
  187. item.unEnabled = !(selectedIndexs.count > 1)
  188. } else if (item.itemIdentifier == KMToolbarPageEditDeleteItemIdentifier) {
  189. item.unEnabled = !self.canDelete()
  190. }
  191. }
  192. if (self.pageIsUpdate == false) {
  193. self.pageIsUpdate = true
  194. return
  195. }
  196. var indexs: Array<Int> = []
  197. for indexPath in selectedIndexs {
  198. indexs.append(indexPath.item)
  199. }
  200. indexs.sort(){$0 < $1}
  201. self.toolBar.setPageRangeString(KMPageRangeTools.newParseSelectedIndexs(selectedIndex: indexs))
  202. if let view = self.listView { // 存储阅读视图的当前页面索引
  203. if indexs.count != 0 {
  204. self._listViewCurrentIndex = indexs.last!
  205. }
  206. }
  207. }
  208. }
  209. // MARK: Private Methods
  210. private func getSelecteIndex() -> Int {
  211. let index = self.getPasteIndex()
  212. if (index != NSNotFound) {
  213. return index
  214. }
  215. return 0
  216. }
  217. private func getSelectedPage() -> CPDFPage {
  218. return (self.pdfDocument!.page(at: UInt(getSelecteIndex())))!
  219. }
  220. private func selectPages(indexs: IndexSet) {
  221. var indexpaths: Set<IndexPath> = []
  222. for index in indexs {
  223. indexpaths.insert(IndexPath(item: index, section: 0))
  224. }
  225. self.selectPages(indexpaths: indexpaths)
  226. }
  227. private func km_becomeFirstResponder() {
  228. DispatchQueue.main.async {
  229. self.view.window?.makeFirstResponder(self)
  230. }
  231. }
  232. fileprivate func dealPdfDocumentDidEditCallback(type: KMPageEditType = .none) {
  233. // 提醒外面文档已编辑
  234. guard let callback = self.documentEditedCallback else {
  235. return
  236. }
  237. callback(type)
  238. }
  239. // MARK: - 清空选择
  240. private func clearSelectPages() {
  241. self.selectPages(indexs: IndexSet())
  242. }
  243. override func refreshUI(indexpaths: Set<IndexPath> = [], keepSelected: Bool = false) {
  244. super.refreshUI(indexpaths: indexpaths, keepSelected: keepSelected)
  245. self.listView?.layoutDocumentView()
  246. }
  247. override func cutAction() {
  248. if (self.thumbnailView.selectionIndexPaths.count == (self.listView?.document.pageCount)!) {
  249. let _ = CustomAlertView.alertView(message: NSLocalizedString("Unable to delete all pages", comment: ""), fromView: self.thumbnailView, withStyle: .blue)
  250. return
  251. }
  252. super.cutAction()
  253. }
  254. override func copyAction() {
  255. self.trackEvent_copy()
  256. super.copyAction()
  257. }
  258. override func deleteAction() {
  259. self.trackEvent(type: .delete)
  260. super.deleteAction()
  261. self.toolBar.unSelectItem(for: KMToolbarPageEditDeleteItemIdentifier)
  262. }
  263. override func deleteIndexPaths(indexpaths: Set<IndexPath>) {
  264. guard let pageCount = self.thumbnailView.document?.pageCount else {
  265. KMPrint("pageCount invalid.")
  266. return
  267. }
  268. if (indexpaths.count == pageCount) {
  269. let _ = CustomAlertView.alertView(message: NSLocalizedString("Unable to delete all pages", comment: ""), fromView: self.thumbnailView, withStyle: .blue)
  270. return
  271. }
  272. self.dealPdfDocumentDidEditCallback(type: .delete)
  273. super.deleteIndexPaths(indexpaths: indexpaths)
  274. }
  275. override func leftRotateAction() {
  276. self.trackEvent(type: .leftRotate)
  277. super.leftRotateAction()
  278. // self.toolBar.unSelectItem(for: KMToolbarPageEditLeftRotateItemIdentifier)
  279. }
  280. override func leftRotateIndexpaths(indexpaths: Set<IndexPath>) {
  281. if (!self.indexpathsIsValid(indexpaths: indexpaths) || !self.canRotate()) {
  282. return
  283. }
  284. super.leftRotateIndexpaths(indexpaths: indexpaths)
  285. self.dealPdfDocumentDidEditCallback(type: .leftRotate)
  286. }
  287. override func rightRotateAction() {
  288. self.trackEvent(type: .rightRotate)
  289. super.rightRotateAction()
  290. // self.toolBar.unSelectItem(for: KMToolbarPageEditRightRotateItemIdentifier)
  291. }
  292. override func rightRotateIndexpaths(indexpaths: Set<IndexPath>) {
  293. if (!self.indexpathsIsValid(indexpaths: indexpaths) || !self.canRotate()) {
  294. return
  295. }
  296. super.rightRotateIndexpaths(indexpaths: indexpaths)
  297. self.dealPdfDocumentDidEditCallback(type: .rightRotate)
  298. }
  299. override func insertPages(pages: Array<CPDFPage>, at indexs: IndexSet) {
  300. super.insertPages(pages: pages, at: indexs)
  301. if (!indexs.isEmpty) {
  302. self.selectPages(indexs: indexs)
  303. self.dealPdfDocumentDidEditCallback(type: .insert)
  304. }
  305. }
  306. override func pasteAction() {
  307. if (self.canPaste()) {
  308. super.pasteAction()
  309. self.dealPdfDocumentDidEditCallback(type: .paste)
  310. }
  311. }
  312. override func extractAction() {
  313. self.trackEvent(type: .extract)
  314. super.extractAction()
  315. DispatchQueue.main.async {
  316. self.toolBar.unSelectItem(for: KMToolbarPageEditExtractItemIdentifier)
  317. }
  318. }
  319. override func replaceAction() {
  320. self.trackEvent(type: .replace)
  321. super.replaceAction()
  322. DispatchQueue.main.async {
  323. self.toolBar.unSelectItem(for: KMToolbarPageEditReplaceItemIdentifier)
  324. }
  325. }
  326. override func getInsertIndex() -> Int {
  327. var index = 0
  328. if let pageCount = self.thumbnailView.document?.pageCount, pageCount > 0 {
  329. // 默认为最后一页
  330. index = Int(pageCount-1)
  331. }
  332. // 在当前选中item之后插入
  333. // 多选时,则取最后一页
  334. for indexpath in self.thumbnailView.selectionIndexPaths.sorted() {
  335. index = indexpath.item
  336. }
  337. return index
  338. }
  339. override func splitAction() {
  340. // self.trackEvent(type: .split)
  341. //
  342. // Task { @MainActor in
  343. // self.toolBar.unSelectItem(for: KMToolbarPageEditSplitItemIdentifier)
  344. //
  345. // self.showSplitWindow { [unowned self] splitWindowC, model in
  346. // if (!self.canPayFunction) {
  347. // _ = KMSubscribeWaterMarkWindowController.show(window: splitWindowC.window!, type: .split) { isSubscribeSuccess, isWaterMarkExport, isClose in
  348. // if (isClose) {
  349. // return
  350. // }
  351. // if (isSubscribeSuccess) {
  352. // self.km_endSheet()
  353. // self._splitMethod(windowC: splitWindowC, model: model)
  354. // return
  355. // }
  356. // if (isWaterMarkExport) {
  357. // self.km_endSheet()
  358. // self._splitMethod(windowC: splitWindowC, model: model, limit: true)
  359. // return
  360. // }
  361. // }
  362. // return
  363. // }
  364. //
  365. // /// 拆分
  366. // self._splitMethod(windowC: splitWindowC, model: model)
  367. // }
  368. // }
  369. let windowC = SplitWindowController(document: self.pdfDocument!)
  370. self.km_beginSheet(windowC: windowC)
  371. }
  372. func splitMethod(windowC: KMPageEditSplitWindowController, model: KMPageEditSplitSettingModel, limit: Bool = false) {
  373. self._splitMethod(windowC: windowC, model: model, limit: limit)
  374. }
  375. func splitMenuAction() {
  376. // if (![IAPProductsManager defaultManager].isAvailableAllFunction) {
  377. // #if VERSION_DMG
  378. // [[KMPurchaseCompareWindowController sharedInstance] showWindow:nil];
  379. // #else
  380. // KMToolCompareWindowController * vc = [KMToolCompareWindowController toolCompareWithType:KMCompareWithToolType_PageEdit setSelectIndex:1];
  381. // [vc startModal:nil];
  382. // #endif
  383. // return;
  384. // }
  385. let pdf = self.thumbnailView.document
  386. let pages = self.thumbnailView.selectionIndexPaths
  387. if (pages.isEmpty == false) {
  388. let fileAttribute = KMFileAttribute()
  389. fileAttribute.bAllPage = false
  390. fileAttribute.pagesType = .custom
  391. fileAttribute.filePath = pdf?.documentURL.path ?? ""
  392. fileAttribute.pdfDocument = pdf
  393. // fileAttribute.pagesString = [self fileNameWithSelectedPages];
  394. fileAttribute.pagesString = KMPageRangeTools.newParseSelectedIndexs(selectedIndex: self.indexpathsToIndexs(indexpaths: pages).sorted())
  395. if (fileAttribute.fetchSelectPages().isEmpty) {
  396. KMAlertTool.runModelForMainThread(message: KMLocalizedString("Invalid page range or the page number is out of range. Please try again.", nil))
  397. } else {
  398. let panel = NSOpenPanel()
  399. panel.canChooseFiles = false
  400. panel.canChooseDirectories = true
  401. panel.canCreateDirectories = true
  402. panel.beginSheetModal(for: self.view.window!) { result in
  403. if (result == .OK) {
  404. let outputURL = panel.urls.first!
  405. var tFolderPath = outputURL.path
  406. var index = 0
  407. var folderName = self.pdfDocument?.documentURL.deletingPathExtension().lastPathComponent
  408. if folderName == nil {
  409. folderName = KMLocalizedString("Untitled", nil)
  410. }
  411. var folderPath = "\(tFolderPath)/\(folderName!)"
  412. //创建目录
  413. while (FileManager.default.fileExists(atPath: folderPath)) {
  414. index += 1
  415. folderPath = "\(tFolderPath)/\(folderName!)_\(index)"
  416. // folderPath = [tFolderPath stringByAppendingPathComponent:[NSString stringWithFormat:@"%@_%ld",folderName,index]];
  417. }
  418. tFolderPath = folderPath
  419. let result = try?FileManager.default.createDirectory(atPath: tFolderPath, withIntermediateDirectories: true)
  420. if (result == nil) {
  421. // if (![[NSFileManager defaultManager] createDirectoryAtPath:tFolderPath withIntermediateDirectories:YES attributes:nil error:nil]) {
  422. KMAlertTool.runModelForMainThread(message: KMLocalizedString("Failed to split!", nil))
  423. return;
  424. }
  425. let vc = KMProgressWindowController()
  426. NSApp.beginSheet(vc.window!, modalFor: self.view.window!, modalDelegate: nil, didEnd: nil, contextInfo: nil)
  427. DispatchQueue.global().async {
  428. // NSArray * successArray = [self.pdfDocument splitByFileWithPages:fileAttribute folerPath:tFolderPath fileName:folderName];
  429. let successArray = self.pdfDocument?.splitByFileWithPages(fileAttribute, folerPath: tFolderPath, fileName: folderName ?? "")
  430. DispatchQueue.main.async {
  431. if let cnt = successArray?.count, cnt > 0 {
  432. let filePath = successArray!.first!
  433. let workSpace = NSWorkspace.shared
  434. workSpace.activateFileViewerSelecting([URL(fileURLWithPath: filePath)])
  435. } else {
  436. KMAlertTool.runModelForMainThread(message: KMLocalizedString("Failed to split!", nil))
  437. }
  438. NSApp.endSheet(vc.window!)
  439. vc.close()
  440. }
  441. }
  442. }
  443. }
  444. }
  445. } else{
  446. let contextString = KMLocalizedString("No pages selected. Please select one or more pages first to organize.",nil)
  447. CustomAlertView.alertView(message: contextString, fromView: self.thumbnailView, withStyle: .blue)
  448. }
  449. }
  450. override func reverseAction() {
  451. self.trackEvent(type: .reverse)
  452. // Task { @MainActor in
  453. self.toolBar.unSelectItem(for: KMToolbarPageEditReverseItemIdentifier)
  454. self.reverseMenuAction(sender: nil)
  455. // }
  456. }
  457. func showSplitWindow(callback: @escaping (KMPageEditSplitWindowController, KMPageEditSplitSettingModel)->Void) {
  458. let model = KMPageEditSplitSettingModel()
  459. model.documentURL = self.listView?.document.documentURL
  460. if let pagecnt = self.thumbnailView.document?.pageCount {
  461. model.pageCount = Int(pagecnt)
  462. }
  463. model.fileName = model.documentURL.lastPathComponent
  464. model.pathExtension = model.fileName.components(separatedBy: ".").last
  465. if (self.thumbnailView.selectionIndexPaths.count > 0) {
  466. model.type = 2
  467. model.pageRangeType = 2
  468. var selectedIndexs: Array<Int> = []
  469. for indexpath in self.thumbnailView.selectionIndexPaths {
  470. selectedIndexs.append(indexpath.item)
  471. }
  472. model.pageRangeString = KMPageRangeTools.newParseSelectedIndexs(selectedIndex: selectedIndexs)
  473. } else {
  474. model.type = 0
  475. }
  476. let windowController = KMPageEditSplitWindowController(model)
  477. self.km_beginSheet(windowC: windowController)
  478. windowController.itemClick = { [weak self] index, value in
  479. if (index == 1) { /// 取消
  480. self?.km_endSheet()
  481. return
  482. }
  483. guard let _windowC = self?.kmCurrentWindowC as? KMPageEditSplitWindowController else {
  484. self?.km_endSheet()
  485. return
  486. }
  487. callback(_windowC, model)
  488. }
  489. }
  490. private func _splitMethod(windowC: KMPageEditSplitWindowController, model: KMPageEditSplitSettingModel, limit: Bool = false) {
  491. let outputModel = windowC.model! as! KMPageEditSplitSettingModel
  492. DispatchQueue.main.async {
  493. let panel = NSOpenPanel()
  494. panel.canChooseFiles = false
  495. panel.canChooseDirectories = true
  496. panel.canCreateDirectories = true
  497. panel.beginSheetModal(for: NSWindow.currentWindow()) { response in
  498. if (response == .cancel) {
  499. return
  500. }
  501. let file_indexs = outputModel.getSplitIndexSets
  502. if (file_indexs == nil || file_indexs!.count <= 0) {
  503. return
  504. }
  505. DispatchQueue.main.async {
  506. self.showProgressWindow(message: NSLocalizedString("Spliting...", comment: ""))
  507. self.progressC?.maxValue = Double(file_indexs!.count)
  508. let _document: CPDFDocument = (self.listView?.document)!
  509. let filePath = "\(panel.url!.path)/\(_document.documentURL.deletingPathExtension().lastPathComponent)"
  510. let uniqueFilePath = KMTools.getUniqueFilePath(filePath: filePath)
  511. // Swift.debugPrint(uniqueFilePath)
  512. if (!FileManager.default.fileExists(atPath: uniqueFilePath)) {
  513. try?FileManager.default.createDirectory(atPath: uniqueFilePath, withIntermediateDirectories: true)
  514. }
  515. var filepaths: [String] = []
  516. for i in 0 ..< file_indexs!.count {
  517. let indexs = file_indexs![i]
  518. if (indexs.isEmpty) {
  519. continue
  520. }
  521. let filepath = String(format: "%@/%@.pdf", uniqueFilePath, model.getNewOutputFileName(index: i+1))
  522. filepaths.append(filepath)
  523. let newDocument = CPDFDocument()
  524. newDocument?.importPages(indexs, from: _document, at: 0)
  525. self.progressC?.increment(by: 1.0)
  526. if (limit) {
  527. let _ = KMTools.saveWatermarkDocument(document: newDocument!, to: URL(fileURLWithPath: filepath))
  528. } else {
  529. newDocument?.write(to: URL(fileURLWithPath: filepath))
  530. }
  531. }
  532. self.hiddenProgressWindow()
  533. self.km_endSheet()
  534. if let _urlString = filepaths.first?.deletingLastPathComponent {
  535. NSWorkspace.shared.activateFileViewerSelecting([URL(fileURLWithPath: _urlString)])
  536. }
  537. }
  538. }
  539. }
  540. }
  541. // MARK: Undo / Redo
  542. @objc private func reversePages(indexs: IndexSet) {
  543. KMPageEditTools.reverse((self.thumbnailView.document)!, indexs
  544. ) { result, error in
  545. if (result) {
  546. self.refreshUI(indexpaths: self.indexsToIndexpaths(indexs: indexs), keepSelected: true)
  547. self.kmUndoManager?.registerUndo(withTarget: self, selector: #selector(self.reversePages), object: indexs)
  548. }
  549. }
  550. }
  551. @objc private func dragPages(_ indexs: IndexSet, _ toIndex: Int, dp: NSCollectionView.DropOperation, reversed: Bool = false) {
  552. if (indexs.count <= 0) {
  553. return
  554. }
  555. self.listViewCurrentPageUpdateForDrag(indexs: indexs, toIndex: toIndex)
  556. // 插入位置偏移量
  557. var toIndexOffset: Int = 0
  558. // 需要移动的页面index数组
  559. var newPages: [CPDFPage] = []
  560. var currentPage: CPDFPage?
  561. var originalIndex = self.listViewCurrentIndex
  562. // 从后面往前删除
  563. for index in indexs.reversed() {
  564. let newPage = (self.listView?.document.page(at: UInt(index)))?.copy() as? CPDFPage
  565. if (newPage == nil) {
  566. continue
  567. }
  568. newPages.insert(newPage!, at: 0)
  569. self.listView?.document.removePage(at: UInt(index))
  570. if (index < toIndex) { /// 删除了插入位置前面的页面,需要改变 toIndex
  571. toIndexOffset += 1
  572. }
  573. if (currentPage == nil) && (originalIndex == index) {
  574. currentPage = newPage
  575. }
  576. }
  577. if (indexs.last! < toIndex && dp == .on) { // 往后拖拽
  578. toIndexOffset -= 1
  579. }
  580. let insertIndex = max(0, toIndex-toIndexOffset)
  581. var selectIndexs: IndexSet = []
  582. for (index, page) in newPages.enumerated() {
  583. self.listView?.document.insertPageObject(page, at: UInt(insertIndex+index))
  584. selectIndexs.insert(insertIndex+index)
  585. if (currentPage?.pageIndex() == page.pageIndex()) {
  586. // self._listViewCurrentIndex = insertIndex+index
  587. }
  588. }
  589. // 新增 undo
  590. self.kmUndoManager?.registerUndo(withTarget: self) { target in
  591. target.clearSelectPages()
  592. target.dragPages_reversed(indexs, toIndex, newPages, dp: dp)
  593. }
  594. // 刷新UI
  595. self.refreshUI()
  596. if (reversed == false) {
  597. self.selectPages(indexs: selectIndexs)
  598. }
  599. }
  600. @objc private func dragPages_reversed(_ indexs: IndexSet, _ toIndex: Int, _ pages: [CPDFPage], dp: NSCollectionView.DropOperation) {
  601. if (indexs.count <= 0) {
  602. return
  603. }
  604. // 插入位置偏移量
  605. var toIndexOffset: Int = 0
  606. for index in indexs {
  607. if (index < toIndex) { /// 删除了插入位置前面的页面,需要改变 toIndex
  608. toIndexOffset += 1
  609. }
  610. }
  611. if (indexs.last! < toIndex && dp == .on) { // 往后拖拽
  612. toIndexOffset -= 1
  613. }
  614. let insertIndex = max(0, toIndex-toIndexOffset)
  615. // 需要移动的页面index数组
  616. var datas: [Int] = []
  617. for index in insertIndex ..< (insertIndex + indexs.count) {
  618. datas.append(index)
  619. }
  620. // 从后面开始删除
  621. for index in datas.reversed() {
  622. self.listView?.document.removePage(at: UInt(index))
  623. }
  624. // 再插入到对应的位置
  625. for (index,value) in indexs.enumerated() {
  626. var page: CPDFPage?
  627. if (index < pages.count) {
  628. page = pages[index]
  629. }
  630. if (page != nil) {
  631. self.listView?.document.insertPageObject(page!, at: UInt(value))
  632. }
  633. }
  634. // 新增 undo
  635. self.kmUndoManager?.registerUndo(withTarget: self) { target in
  636. target.dragPages(indexs, toIndex, dp: dp, reversed: true)
  637. }
  638. // 刷新UI
  639. self.refreshUI()
  640. }
  641. // MARK: topBar 事件
  642. @objc func zoomOutButtonAction(sender: NSButton?) {
  643. if (self.itemSize.width > self.defaultItemSize.width) {
  644. return
  645. }
  646. if let _document = self.thumbnailView.document {
  647. for i in 0 ..< _document.pageCount {
  648. let page = _document.page(at: i)
  649. page?.pageEditImage = nil
  650. }
  651. }
  652. if (self.itemSize.width < self.defaultItemSize.width) {
  653. self.itemSize = self.defaultItemSize
  654. } else if (self.itemSize.width == self.defaultItemSize.width) {
  655. self.itemSize = NSSize(width: self.defaultItemSize.width*2, height: self.defaultItemSize.height*2)
  656. }
  657. // self.toolBar.zoomInButton?.isEnabled = true
  658. // if (self.itemSize.width > self.defaultItemSize.width) {
  659. // sender?.isEnabled = false
  660. // }
  661. self.pageEditThumbnailView.zoomOut()
  662. }
  663. @objc func zoomInButtonAction(sender: NSButton?) {
  664. if (self.itemSize.width < self.defaultItemSize.width) {
  665. return
  666. }
  667. if let _document = self.thumbnailView.document {
  668. for i in 0 ..< _document.pageCount {
  669. let page = _document.page(at: i)
  670. page?.pageEditImage = nil
  671. }
  672. }
  673. if (self.itemSize.width > self.defaultItemSize.width) {
  674. self.itemSize = self.defaultItemSize
  675. } else if (self.itemSize.width == self.defaultItemSize.width) {
  676. self.itemSize = NSSize(width: self.defaultItemSize.width*0.5, height: self.defaultItemSize.height*0.5)
  677. }
  678. // self.toolBar.zoomOutButton?.isEnabled = true
  679. // if (self.itemSize.width < self.defaultItemSize.width) {
  680. // sender?.isEnabled = false
  681. // }
  682. self.pageEditThumbnailView.zoomIn()
  683. }
  684. // MARK: menu 菜单事件
  685. @objc func item_insertCustomPage(sender: NSMenuItem?) {
  686. let windowC = KMPDFEditInsertBlankPageWindow(document: self.pdfDocument!)
  687. windowC.callback = { [weak self] pdfDoc, _, pages, insertI in
  688. if let _pages = pages {
  689. // self?.insertPages(pages: _pages, at: IndexSet(integer: insertI+1))
  690. // mm 单位的大小
  691. guard let _winC = self?.kmCurrentWindowC as? KMPDFEditInsertBlankPageWindow else {
  692. self?.km_endSheet()
  693. return
  694. }
  695. var pageSize = _winC.pageSize
  696. let direction = windowC.pageRotation == 0 ? 0 : 1
  697. // var _pageSize = KMPageEditTools.sizeUnitToBoundSize(unit: .mm, pageSize: pageSize)
  698. if (direction == 0) { // 纵向
  699. if (pageSize.width > pageSize.height) { // 需要交换
  700. let tmp = pageSize.width
  701. pageSize.width = pageSize.height
  702. pageSize.height = tmp
  703. } else {
  704. // no things.
  705. }
  706. } else { // 横向
  707. if (pageSize.width > pageSize.height) {
  708. // no things.
  709. } else { // 需要交换
  710. let tmp = pageSize.width
  711. pageSize.width = pageSize.height
  712. pageSize.height = tmp
  713. }
  714. }
  715. /// 插入位置
  716. let document = CPDFDocument()
  717. document?.insertPage(pageSize, at: 0)
  718. if let page: CPDFPage = (document?.page(at: 0)) {
  719. self?.insertPages(pages: [page], at: IndexSet(integer: insertI))
  720. }
  721. }
  722. self?.km_endSheet()
  723. }
  724. self.km_beginSheet(windowC: windowC)
  725. #if false
  726. Task { @MainActor in
  727. let windowController = KMPageEditInsertCustomPageWindowController()
  728. if (self.thumbnailView.selectionIndexPaths.count > 0) {
  729. let page = self.getSelectedPage()
  730. let width = page.bounds.size.width * 210 / 595
  731. let height = page.bounds.size.height * 297 / 842
  732. windowController.selectedPageSize = NSSize(width: width, height: height)
  733. // windowController.selectedPageSize = self.getSelectedPage().bounds.size
  734. }
  735. self.km_beginSheet(windowC: windowController)
  736. windowController.itemClick = { [weak self] index in
  737. if (index == 1) { /// 取消
  738. self?.km_endSheet()
  739. return
  740. }
  741. /// 插入
  742. guard let windowC = self?.kmCurrentWindowC as? KMPageEditInsertCustomPageWindowController else {
  743. self?.km_endSheet()
  744. return
  745. }
  746. guard let _insertIndex = self?.getInsertIndex() else {
  747. self?.km_endSheet()
  748. return
  749. }
  750. /// 样式
  751. // let type = windowC.typeIndex
  752. // mm 单位的大小
  753. let pageSize = windowC.pageSize
  754. let direction = windowC.direction
  755. var _pageSize = KMPageEditTools.sizeUnitToBoundSize(unit: .mm, pageSize: pageSize)
  756. if (direction == 0) { // 纵向
  757. if (_pageSize.width > _pageSize.height) { // 需要交换
  758. let tmp = _pageSize.width
  759. _pageSize.width = _pageSize.height
  760. _pageSize.height = tmp
  761. } else {
  762. // no things.
  763. }
  764. } else { // 横向
  765. if (_pageSize.width > _pageSize.height) {
  766. // no things.
  767. } else { // 需要交换
  768. let tmp = _pageSize.width
  769. _pageSize.width = _pageSize.height
  770. _pageSize.height = tmp
  771. }
  772. }
  773. /// 插入位置
  774. let document = CPDFDocument()
  775. document?.insertPage(_pageSize, at: 0)
  776. if let page: CPDFPage = (document?.page(at: 0)) {
  777. self?.insertPages(pages: [page], at: IndexSet(integer: _insertIndex+1))
  778. }
  779. self?.km_endSheet()
  780. }
  781. }
  782. #endif
  783. }
  784. func extractSelectPageItemAction() {
  785. let indexs = self.thumbnailView.selectionIndexPaths
  786. if indexs.isEmpty {
  787. CustomAlertView.alertView(message: KMLocalizedString("No pages selected. Please select one or more pages first to organize.",nil), fromView: self.thumbnailView, withStyle: .blue)
  788. return
  789. }
  790. self.extractPages(indexpaths: indexs, oneDocumentPerPage: false) { result, params in }
  791. }
  792. // MARK: toolBar 菜单事件
  793. @objc func reverseMenuAction(sender: AnyObject?) {
  794. let selectedIndexs = self.thumbnailView.selectionIndexPaths
  795. if (selectedIndexs.count < 2 ) {
  796. let alertView = CustomAlertView.alertView(message: NSLocalizedString("Please select at least two pages for editing first", comment: ""), fromView: self.thumbnailView, withStyle: .blue)
  797. alertView?.layer?.backgroundColor = NSColor.km_init(hex: "#36383B").cgColor
  798. alertView?.layer?.cornerRadius = 4
  799. return
  800. }
  801. var indexs: IndexSet = []
  802. for indexpath in selectedIndexs {
  803. indexs.insert(indexpath.item)
  804. }
  805. self.reversePages(indexs: indexs)
  806. self.dealPdfDocumentDidEditCallback(type: .reverse)
  807. }
  808. override func addMenuItem(menu: NSMenu, item: KMPDFThumbViewMenu.Item) -> NSMenuItem? {
  809. var item_: NSMenuItem?
  810. if (item.sel == KMPDFThumbViewMenu.insertSelector) {
  811. // if (self.thumbnailView.selectionIndexPaths.count == 1) {
  812. // item_ = menu.addItem(withTitle: NSLocalizedString("Insert", comment: ""), action: nil, target: self)
  813. item_ = menu.addItem(withTitle: NSLocalizedString("Blank Page", comment: ""), action: #selector(insertPageItemAction), target: self)
  814. // let subMenu = NSMenu()
  815. // subMenu.addItem(withTitle: NSLocalizedString("Insert a Blank Page", comment: ""), action: #selector(insertPageItemAction), target: self, tag:0)
  816. // subMenu.addItem(withTitle: NSLocalizedString("Insert Custom Page", comment: ""), action: #selector(insertPageItemAction), target: self, tag:1)
  817. // subMenu.addItem(withTitle: NSLocalizedString("Insert File", comment: ""), action: #selector(insertPageItemAction), target: self, tag:2)
  818. // item_?.submenu = subMenu
  819. // }
  820. } else if (item.sel == KMPDFThumbViewMenu.splitSelector) {
  821. if (self.thumbnailView.selectionIndexPaths.count > 1) {
  822. item_ = menu.addItem(withTitle: NSLocalizedString("Split", comment: ""), action: KMPDFThumbViewMenu.splitSelector, target: self)
  823. }
  824. } else if (item.sel == KMPDFThumbViewMenu.reverseSelector) {
  825. if (self.thumbnailView.selectionIndexPaths.count > 1) {
  826. item_ = menu.addItem(withTitle: NSLocalizedString("Reverse", comment: ""), action: KMPDFThumbViewMenu.reverseSelector, target: self)
  827. }
  828. } else {
  829. return super.addMenuItem(menu: menu, item: item)
  830. }
  831. return item_
  832. }
  833. // MARK: - Zoom Methods
  834. @IBAction func buttonClicked_Bigger(_ sender: AnyObject?) {
  835. // if (self.thumbnailView.thumbnailSize.width == 100.) {
  836. // self.thumbnailView.thumbnailSize = CGSizeMake(200, 200);
  837. // } else if (self.thumbnailView.thumbnailSize.width == 200.) {
  838. // self.thumbnailView.thumbnailSize = CGSizeMake(400, 400);
  839. // }
  840. self.zoomOutButtonAction(sender: sender as? NSButton)
  841. }
  842. @IBAction func buttonClicked_Smaller(_ sender: AnyObject?) {
  843. // if (self.thumbnailView.thumbnailSize.width == 400.) {
  844. // self.thumbnailView.thumbnailSize = CGSizeMake(200, 200);
  845. // } else if (self.thumbnailView.thumbnailSize.width == 200.) {
  846. // self.thumbnailView.thumbnailSize = CGSizeMake(100, 100);
  847. // }
  848. self.zoomInButtonAction(sender: sender as? NSButton)
  849. }
  850. }
  851. extension KMPDFEditViewController {
  852. //MARK: menu Action
  853. @objc func insertPageItemAction(menu:NSMenuItem) {
  854. if (menu.tag == 0) {
  855. self.insertBlankPageAction()
  856. } else if (menu.tag == 1) {
  857. self.item_insertCustomPage(sender: nil)
  858. } else if (menu.tag == 2) {
  859. self.insertFileAction()
  860. }
  861. }
  862. @objc func sharePageItemAction(menu:NSMenuItem) {
  863. let item = menu.parent!
  864. let index = (item.representedObject as! IndexSet).first ?? -1
  865. if Int(index) >= 0 {
  866. let doucument = self.pdfDocument!
  867. let page = doucument.page(at: UInt(index))
  868. let filename : String = doucument.documentURL.lastPathComponent
  869. let folderPath = (NSSearchPathForDirectoriesInDomains(FileManager.SearchPathDirectory.applicationSupportDirectory, FileManager.SearchPathDomainMask.userDomainMask, true).last?.stringByAppendingPathComponent(filename))!
  870. try? FileManager.default.removeItem(atPath: folderPath)
  871. let pdfdocument = CPDFDocument()
  872. let ret = pdfdocument!.importPages((item.representedObject as! IndexSet), from: self.thumbnailView.document, at: 0)
  873. let url = URL(fileURLWithPath: folderPath)
  874. if ret {
  875. let success = pdfdocument!.write(to:url)
  876. if success {
  877. let workspace = NSWorkspace.shared
  878. workspace.activateFileViewerSelecting([url])
  879. }
  880. }
  881. let represent : NSSharingService = menu.representedObject as! NSSharingService
  882. represent.perform(withItems: [url])
  883. }
  884. }
  885. @objc func showPageSizeItemAction(menu: NSMenuItem) {
  886. self.thumbnailView.isShowPageSize = !self.thumbnailView.isShowPageSize
  887. self.refreshUIForKeepSelecteds()
  888. }
  889. @objc func printItemAction(menu:NSMenuItem) {
  890. var datas: [Int] = []
  891. for indexpath in self.thumbnailView.selectionIndexPaths {
  892. datas.append(indexpath.item)
  893. }
  894. guard let callback = self.itemClick else {
  895. return
  896. }
  897. callback(2, datas.sorted())
  898. }
  899. }
  900. extension KMPDFEditViewController {
  901. override func menuItemAction_print(_ sender: Any?) {
  902. self.printItemAction(menu: sender as! NSMenuItem)
  903. }
  904. override func menuItemAction_showPageSize(_ sender: Any?) {
  905. self.showPageSizeItemAction(menu: sender as! NSMenuItem)
  906. }
  907. override func menuItemAction_share(_ sender: Any?) {
  908. self.sharePageItemAction(menu: sender as! NSMenuItem)
  909. }
  910. }
  911. // MARK: - KMSelectPopButtonDelegate
  912. extension KMPDFEditViewController {
  913. func km_comboBoxSelectionDidChange(_ obj: KMPageRange) {
  914. self.pageIsUpdate = false
  915. // let index: Int = obj.indexOfSelectedItem
  916. let index = obj.rawValue
  917. var indexpaths: Set<IndexPath> = []
  918. var type: KMPageRange? = obj
  919. // if (index <= 0) { /// 全部页面
  920. // type = .all
  921. // } else if (index == 1) { /// 奇数页
  922. // type = .odd
  923. // } else if (index == 2) { /// 偶数页
  924. // type = .even
  925. // } else if (index == 3) { /// 横向页
  926. // type = .horizontal
  927. // } else if (index == 4) { /// 纵向页
  928. // type = .vertical
  929. // } else { /// 自定义
  930. // type = .custom
  931. // }
  932. for i in 0 ..< (self.pdfDocument?.pageCount)! {
  933. if (type == .all) {
  934. indexpaths.insert(IndexPath(item: Int(i), section: 0))
  935. } else if (type == .odd) {
  936. if (i % 2 == 1) {
  937. continue
  938. }
  939. indexpaths.insert(IndexPath(item: Int(i), section: 0))
  940. } else if (type == .even) {
  941. if (i % 2 == 0) {
  942. continue
  943. }
  944. indexpaths.insert(IndexPath(item: Int(i), section: 0))
  945. } else if (type == .horizontal) {
  946. let page = self.pdfDocument?.page(at: i)
  947. guard let isVertical = page?.isVertical(), !isVertical else {
  948. continue
  949. }
  950. indexpaths.insert(IndexPath(item: Int(i), section: 0))
  951. } else if (type == .vertical) {
  952. let page = self.pdfDocument?.page(at: i)
  953. guard let isHorizontal = page?.isHorizontal(), !isHorizontal else {
  954. continue
  955. }
  956. indexpaths.insert(IndexPath(item: Int(i), section: 0))
  957. }
  958. }
  959. // obj.editable = type == .custom
  960. if (type == .custom) {
  961. let windowC = KMPDFEditPageRangeWindowController()
  962. windowC.pdfDocument = self.pdfDocument
  963. windowC.pageSelectCallBack = { [weak self] pagesArray in
  964. var indexpaths = Set<IndexPath>()
  965. for page in pagesArray {
  966. indexpaths.insert(IndexPath(item: page-1, section: 0))
  967. }
  968. self?.selectPages(indexpaths: indexpaths)
  969. self?.km_endSheet()
  970. }
  971. self.km_beginSheet(windowC: windowC)
  972. } else {
  973. self.pageEditThumbnailView.selectPages(at: indexpaths)
  974. }
  975. }
  976. func km_controlTextDidEndEditing(_ obj: KMDesignSelect) {
  977. self.view.window?.makeFirstResponder(self)
  978. for index in 0 ..< obj.numberOfItems {
  979. if (obj.items[index] == obj.stringValue) {
  980. return
  981. }
  982. }
  983. let pages = KMPageRangeTools.findSelectPage(pageRangeString: self.toolBar.pageRangeView!.stringValue, pageCount: Int((self.listView?.document.pageCount)!))
  984. if (pages.isEmpty) {
  985. let alert = NSAlert()
  986. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  987. alert.runModal()
  988. self.toolBar.setPageRangeString("")
  989. DispatchQueue.main.async {
  990. self.view.window?.makeFirstResponder(self.toolBar.pageRangeView?.textField)
  991. }
  992. return
  993. }
  994. var indexpaths: Set<IndexPath> = []
  995. for page in pages {
  996. if (page >= 1) {
  997. indexpaths.insert(IndexPath(item: page-1, section: 0))
  998. }
  999. }
  1000. self.pageIsUpdate = false
  1001. self.pageEditThumbnailView.selectPages(at: indexpaths)
  1002. }
  1003. }
  1004. // MARK: -
  1005. // MARK: KMThumbnailViewDelegate
  1006. extension KMPDFEditViewController: KMThumbnailViewDelegate {
  1007. func thumbnailView(thumbanView: KMThumbnailView, shouldAcceptDrop draggingInfo: NSDraggingInfo, indexPath: IndexPath, dropOperation: NSCollectionView.DropOperation) -> Bool {
  1008. return true
  1009. }
  1010. func thumbnailView(thumbanView: KMThumbnailView, didDragAddFiles files: [URL], indexpath: IndexPath) {
  1011. self.km_add_file_multi(fileUrls: files) { index, params in
  1012. if (self.fetchProgressBlockParamsIsPasswordFile(params: params)) { // 加密文档进度回调
  1013. return
  1014. }
  1015. let _fileUrl = self.fetchProgressBlockParamsForFileUrl(params: params)
  1016. if let exn = _fileUrl?.pathExtension, KMTools.isPDFType(exn) {
  1017. if (_fileUrl!.path.isPDFValid() == false) {
  1018. let alert = NSAlert()
  1019. alert.alertStyle = .critical
  1020. alert.messageText = NSLocalizedString("An error occurred while opening this document. The file is damaged and could not be repaired.", comment: "")
  1021. alert.runModal()
  1022. }
  1023. }
  1024. } completionBlock: { [unowned self] documents in
  1025. var insertIndex: Int = indexpath.item
  1026. var pages: Array<CPDFPage> = []
  1027. var indexs = IndexSet()
  1028. for document in documents {
  1029. for i in 0 ..< document.pageCount {
  1030. let page = document.page(at: i)
  1031. pages.append(page!)
  1032. indexs.insert(insertIndex)
  1033. insertIndex += 1
  1034. }
  1035. }
  1036. self.insertPages(pages: pages, at: indexs)
  1037. }
  1038. }
  1039. func thumbnailView(thumbanView: KMThumbnailView, didDrag dragedIndexPaths: [IndexPath], indexpath: IndexPath, dragInfo: [KMThumbnailViewDragInfoKey.RawValue : Any]) {
  1040. let toIndex = max(0, indexpath.item)
  1041. var indexs = IndexSet()
  1042. for indexpath in dragedIndexPaths {
  1043. indexs.insert(indexpath.item)
  1044. }
  1045. var dp = NSCollectionView.DropOperation.on
  1046. if let _dp = dragInfo[KMThumbnailViewDragInfoKey.dropOperation.rawValue] as? NSCollectionView.DropOperation {
  1047. dp = _dp
  1048. }
  1049. self.dragPages(indexs, toIndex, dp: dp)
  1050. if (dragedIndexPaths.count > 0) {
  1051. self.dealPdfDocumentDidEditCallback()
  1052. }
  1053. }
  1054. func thumbnailView(thumbanView: KMThumbnailView, shouldPasteboardWriterForItemAt indexPath: IndexPath) -> Bool {
  1055. if (!self.canPayFunction) {
  1056. return false
  1057. }
  1058. return true
  1059. }
  1060. func thumbnailView(thumbanView: KMThumbnailView, insetForSectionAt section: Int) -> NSEdgeInsets {
  1061. return NSEdgeInsets(top: 6, left: 6, bottom: 0, right: 6)
  1062. }
  1063. func thumbnailView(thumbanView: KMThumbnailView, sizeForItemAt indexpath: IndexPath) -> NSSize {
  1064. self.thumbnailView.thumbnailSzie = CGSize(width: 120, height: 155)
  1065. if (!self.thumbnailView.isShowPageSize) {
  1066. return NSMakeSize(self.itemSize.width, self.itemSize.height+26)
  1067. }
  1068. return self.itemSize
  1069. }
  1070. func thumbnailView(thumbanView: KMThumbnailView, numberOfItemsInSection section: Int) -> Int {
  1071. if let count = self.pdfDocument?.pageCount {
  1072. return Int(count)
  1073. }
  1074. return 0
  1075. }
  1076. func thumbnailView(thumbanView: KMThumbnailView, itemForRepresentedObjectAt indexpath: IndexPath) -> NSCollectionViewItem {
  1077. let cellView: KMPageEditThumbnailItem = thumbanView.collectionView.km.dequeueReusableCell(for: indexpath)
  1078. cellView.isShowPageSize = self.thumbnailView.isShowPageSize
  1079. cellView.setPage(page: (self.pdfDocument?.page(at: UInt(indexpath.item)))!)
  1080. cellView.doubleClickAction = { [unowned self] _ in
  1081. guard let callback = self.itemClick else {
  1082. return
  1083. }
  1084. callback(1, indexpath.item)
  1085. }
  1086. cellView.mouseDownAction = { [unowned self] _ in
  1087. let selectionIndexPaths = self.thumbnailView.selectionIndexPaths
  1088. if (selectionIndexPaths.count == 1 && selectionIndexPaths.first?.item == indexpath.item) {
  1089. return
  1090. }
  1091. self.selectPages(indexs: IndexSet.init(integer: indexpath.item))
  1092. }
  1093. cellView.rightMouseDownAction = { [unowned self] _ in
  1094. self.selectPages(indexs: IndexSet.init(integer: indexpath.item))
  1095. }
  1096. return cellView
  1097. }
  1098. }
  1099. // MARK: - 系统菜单栏
  1100. extension KMPDFEditViewController {
  1101. override func undo(_ sender: Any?) {
  1102. super.undo(sender)
  1103. self.km_becomeFirstResponder()
  1104. }
  1105. override func redo(_ sender: Any?) {
  1106. super.redo(sender)
  1107. self.km_becomeFirstResponder()
  1108. }
  1109. }
  1110. // MARK: - KMExtensions
  1111. extension KMPDFEditViewController {
  1112. override func deleteAfter(indexpaths: Set<IndexPath>) {
  1113. self.clearSelectPages()
  1114. self.km_becomeFirstResponder()
  1115. }
  1116. override func copyAfter(indexpaths: Set<IndexPath>) {
  1117. self.km_becomeFirstResponder()
  1118. }
  1119. override func pasteAfter(at index: Int) {
  1120. self.km_becomeFirstResponder()
  1121. }
  1122. override func cutAfter(indexpaths: Set<IndexPath>) {
  1123. self.clearSelectPages()
  1124. self.km_becomeFirstResponder()
  1125. }
  1126. override func leftRotateAfter(indexpaths: Set<IndexPath>) {
  1127. NotificationCenter.default.post(name: NSNotification.Name(rawValue: "KMPDFViewRotatePage"), object: self.thumbnailView.document)
  1128. self.km_becomeFirstResponder()
  1129. }
  1130. override func rightRotateAfter(indexpaths: Set<IndexPath>) {
  1131. NotificationCenter.default.post(name: NSNotification.Name(rawValue: "KMPDFViewRotatePage"), object: self.thumbnailView.document)
  1132. self.km_becomeFirstResponder()
  1133. }
  1134. override func replaceAfter(of targetIndexs: IndexSet, with pages: [CPDFPage], at index: Int) {
  1135. var indexpaths = Set<IndexPath>()
  1136. var insertIndex = index
  1137. for _ in pages {
  1138. indexpaths.insert(IndexPath(item: insertIndex, section: 0))
  1139. insertIndex += 1
  1140. }
  1141. self.thumbnailView.selectionIndexPaths = indexpaths
  1142. self.dealPdfDocumentDidEditCallback(type: .replace)
  1143. }
  1144. override func insertBlankPageAfter(size: NSSize, at index: Int) {
  1145. self.selectPages(indexs: IndexSet(integer: index))
  1146. self.dealPdfDocumentDidEditCallback(type: .insert)
  1147. }
  1148. func listViewCurrentPageUpdateForDrag(indexs: IndexSet, toIndex: Int) {
  1149. if (indexs.isEmpty) {
  1150. return
  1151. }
  1152. // listView 当前页面 被拖拽
  1153. if (indexs.contains(self._listViewCurrentIndex)) {
  1154. // let idx = indexs.firstIndex(of: self._listViewCurrentIndex)
  1155. var idx: Int = 0
  1156. for index in indexs {
  1157. if (index == self._listViewCurrentIndex) {
  1158. break
  1159. }
  1160. idx += 1
  1161. }
  1162. // 插入位置偏移量
  1163. var toIndexOffset: Int = 0
  1164. // 从后面往前删除
  1165. for index in indexs.reversed() {
  1166. let newPage = (self.listView?.document.page(at: UInt(index)))?.copy() as? CPDFPage
  1167. if (newPage == nil) {
  1168. continue
  1169. }
  1170. if (index < toIndex) { /// 删除了插入位置前面的页面,需要改变 toIndex
  1171. toIndexOffset += 1
  1172. }
  1173. }
  1174. if (indexs.last! < toIndex) { // 往后拖拽
  1175. toIndexOffset -= 1
  1176. }
  1177. let insertIndex = max(0, toIndex-toIndexOffset)
  1178. self._listViewCurrentIndex = insertIndex+idx
  1179. } else {
  1180. var forwardCnt = 0
  1181. var nextCnt = 0
  1182. for index in indexs {
  1183. if (index > self._listViewCurrentIndex) {
  1184. nextCnt += 1
  1185. } else {
  1186. forwardCnt += 1
  1187. }
  1188. }
  1189. if (self._listViewCurrentIndex > toIndex) { // 插入到目标前面, 前面不管,只管后面的格式
  1190. self._listViewCurrentIndex += nextCnt
  1191. } else if (self._listViewCurrentIndex < toIndex) { // 插入到目标后面, 后面不管,只管前面的格式
  1192. self._listViewCurrentIndex = max(0, self._listViewCurrentIndex - forwardCnt)
  1193. } else { // 插入到目标位置 [跟插入到目标后面一样]
  1194. self._listViewCurrentIndex = max(0, self._listViewCurrentIndex - forwardCnt)
  1195. }
  1196. }
  1197. }
  1198. }
  1199. // MARK: - Analytics (埋点)
  1200. extension KMPDFEditViewController {
  1201. func trackEvent(type: KMPageEditType) -> Void {
  1202. if (type == .insert) {
  1203. KMAnalytics.trackEvent(eventName: "Btn_SubTbr_Insert", parameters: [
  1204. KMAnalytics.Parameter.categoryKey : KMAnalytics.Category.subTbr_PageEdit,
  1205. KMAnalytics.Parameter.labelKey : KMAnalytics.Label.subTbr_Btn], platform: .AppCenter, appTarget: .all)
  1206. } else if (type == .extract) {
  1207. KMAnalytics.trackEvent(eventName: "Btn_SubTbr_Extract", parameters: [
  1208. KMAnalytics.Parameter.categoryKey : KMAnalytics.Category.subTbr_PageEdit,
  1209. KMAnalytics.Parameter.labelKey : KMAnalytics.Label.subTbr_Btn], platform: .AppCenter, appTarget: .all)
  1210. } else if (type == .leftRotate || type == .rightRotate) {
  1211. KMAnalytics.trackEvent(eventName: "Btn_SubTbr_Rotate", parameters: [
  1212. KMAnalytics.Parameter.categoryKey : KMAnalytics.Category.subTbr_PageEdit,
  1213. KMAnalytics.Parameter.labelKey : KMAnalytics.Label.subTbr_Btn], platform: .AppCenter, appTarget: .all)
  1214. } else if (type == .replace) {
  1215. KMAnalytics.trackEvent(eventName: "Btn_SubTbr_Replace", parameters: [
  1216. KMAnalytics.Parameter.categoryKey : KMAnalytics.Category.subTbr_PageEdit,
  1217. KMAnalytics.Parameter.labelKey : KMAnalytics.Label.subTbr_Btn], platform: .AppCenter, appTarget: .all)
  1218. } else if (type == .split) {
  1219. KMAnalytics.trackEvent(eventName: "Btn_SubTbr_Split", parameters: [
  1220. KMAnalytics.Parameter.categoryKey : KMAnalytics.Category.subTbr_PageEdit,
  1221. KMAnalytics.Parameter.labelKey : KMAnalytics.Label.subTbr_Btn], platform: .AppCenter, appTarget: .all)
  1222. } else if (type == .reverse) {
  1223. KMAnalytics.trackEvent(eventName: "Btn_SubTbr_Reverse", parameters: [
  1224. KMAnalytics.Parameter.categoryKey : KMAnalytics.Category.subTbr_PageEdit,
  1225. KMAnalytics.Parameter.labelKey : KMAnalytics.Label.subTbr_Btn], platform: .AppCenter, appTarget: .all)
  1226. } else if (type == .delete) {
  1227. KMAnalytics.trackEvent(eventName: "Btn_SubTbr_Delete", parameters: [
  1228. KMAnalytics.Parameter.categoryKey : KMAnalytics.Category.subTbr_PageEdit,
  1229. KMAnalytics.Parameter.labelKey : KMAnalytics.Label.subTbr_Btn], platform: .AppCenter, appTarget: .all)
  1230. }
  1231. }
  1232. func trackEvent_copy() -> Void {
  1233. KMAnalytics.trackEvent(eventName: "Btn_SubTbr_Copy", parameters: [
  1234. KMAnalytics.Parameter.categoryKey : KMAnalytics.Category.subTbr_PageEdit,
  1235. KMAnalytics.Parameter.labelKey : KMAnalytics.Label.subTbr_Btn], platform: .AppCenter, appTarget: .all)
  1236. }
  1237. }