KMPDFEditViewController.swift 60 KB

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