KMConvertBaseWindowController.swift 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877
  1. //
  2. // KMConvertBaseWindowController.swift
  3. // PDF Master
  4. //
  5. // Created by tangchao on 2022/12/5.
  6. //
  7. import PDFKit
  8. typealias KMConvertBaseWindowControllerItemClick = (Int) -> ()
  9. class KMConvertBaseWindowController: KMBaseWindowController {
  10. @IBOutlet weak var titleLabel: NSTextField!
  11. @IBOutlet weak var leftBox: NSBox!
  12. @IBOutlet weak var prePDFView: KMCustomPDFView?
  13. @IBOutlet weak var numberBox: NSView!
  14. @IBOutlet weak var numberTextField: NSTextField!
  15. @IBOutlet weak var perLabel: NSTextField!
  16. @IBOutlet weak var totalNumberLabel: NSTextField!
  17. @IBOutlet weak var backBox: NSBox!
  18. @IBOutlet weak var nextBox: NSBox!
  19. @IBOutlet weak var rightScrollView: NSScrollView!
  20. @IBOutlet weak var batchButton: NSButton!
  21. @IBOutlet weak var canelBox: NSBox!
  22. @IBOutlet weak var convertBox: NSBox!
  23. @IBOutlet weak var rightBoxHeightConst: NSLayoutConstraint!
  24. var canelButtonVC: KMDesignButton?
  25. var convertButtonVC: KMDesignButton?
  26. var backButtonVC: KMDesignButton?
  27. var nextButtonVC: KMDesignButton?
  28. var itemClick: KMConvertBaseWindowControllerItemClick?
  29. var documentModel: KMDocumentModel?
  30. var currentPageIndex: Int = 1
  31. var pageRange = KMPageRange.all
  32. var pageRangeString: String = ""
  33. var settingView: KMConvertSettingView?
  34. var subType: Int = 0
  35. var fileExtension: String {
  36. get {
  37. return ""
  38. }
  39. }
  40. var progressController: SKProgressController?
  41. private var convert: KMPDFConvert?
  42. var limit = true
  43. var oriDocumentUrl: URL?
  44. var subscribeWaterMarkType: KMSubscribeWaterMarkType?
  45. deinit {
  46. KMPrint("KMConvertBaseWindowController 已释放")
  47. self.prePDFView?.removeFromSuperview()
  48. self.prePDFView = nil
  49. }
  50. convenience init() {
  51. self.init(windowNibName: "KMConvertBaseWindowController")
  52. }
  53. override func awakeFromNib() {
  54. super.awakeFromNib()
  55. DispatchQueue.main.async {
  56. self.window?.makeFirstResponder(nil)
  57. }
  58. }
  59. override func windowDidLoad() {
  60. super.windowDidLoad()
  61. if (self.documentModel != nil) {
  62. let document = CPDFDocument(url: self.documentModel!.documentURL)
  63. if (document!.isLocked) {
  64. document?.unlock(withPassword: self.documentModel!.password)
  65. }
  66. self.prePDFView?.document = document!
  67. let number: Int = Int((self.prePDFView?.document.pageCount)!)
  68. let string = "\(number)"
  69. totalNumberLabel.stringValue = string
  70. self.prePDFView?.go(toPageIndex: self.documentModel!.currentIndex, animated: false)
  71. }
  72. if (self.prePDFView?.documentView() != nil) {
  73. self.prePDFView?.documentView().enclosingScrollView?.hasVerticalScroller = false
  74. self.prePDFView?.documentView().enclosingScrollView?.hasHorizontalScroller = false
  75. }
  76. if let pageCount = self.prePDFView?.document.pageCount {
  77. self.settingView?.pageCount = Int(pageCount)
  78. }
  79. batchButton.title = NSLocalizedString("Batch", comment: "")
  80. batchButton.isBordered = false
  81. batchButton.wantsLayer = true
  82. batchButton.layer?.borderWidth = 1
  83. batchButton.layer?.cornerRadius = 4
  84. batchButton.target = self
  85. batchButton.action = #selector(batchButtonAction)
  86. batchButton.isHidden = true
  87. self.updateBackAndNextButtonState()
  88. }
  89. override func initSubViews() {
  90. super.initSubViews()
  91. let view = self.initSettingView()
  92. view?.frame = self.rightScrollView.bounds
  93. view?.autoresizingMask = [.width, .height]
  94. self.settingView = view
  95. self.rightScrollView.documentView = view
  96. self.rightScrollView.verticalScrollElasticity = .none
  97. self.rightScrollView.horizontalScrollElasticity = .none
  98. view?.currentLanguage = self.getCurrentLanguage()
  99. view?.lanugageDidSelected = { [unowned self] value, _ in
  100. guard let index = value as? Int else {
  101. return
  102. }
  103. self.saveLanugageSelectedIndex(index: index)
  104. }
  105. view?.pageRangeDidChange = { [unowned self] pageRange in
  106. self.pageRange = pageRange
  107. if (pageRange != .custom) {
  108. self.updatePreView(pageRange: pageRange)
  109. }
  110. }
  111. view?.pageRangeDidInputFinishCallback = { [unowned self] string in
  112. self.pageRangeString = string
  113. let array = self.findSelectPage(pageRangeString: string)
  114. if array.count == 0 {
  115. let alert = NSAlert()
  116. alert.alertStyle = .warning
  117. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  118. alert.runModal()
  119. self.settingView?.pageRangeItemView?.pageSizeVC?.stringValue = ""
  120. } else {
  121. self.updatePreView(pageRange: .custom, pageString: string)
  122. }
  123. }
  124. view?.tipView?.itemClick = { [unowned self] _, _ in
  125. #if VERSION_DMG
  126. KMTools.openPurchaseProductWebsite()
  127. #else
  128. Task { @MainActor in
  129. if await (KMLightMemberManager.manager.canPayFunction() == false) {
  130. let _ = KMSubscribeWaterMarkWindowController.show(window: self.window!) { isSubscribeSuccess, isWaterMarkExport, isClose in
  131. if (isSubscribeSuccess) {
  132. self.refreshUI()
  133. }
  134. }
  135. }
  136. }
  137. #endif
  138. }
  139. self.settingView?.tipView?.isHidden = true
  140. Task { @MainActor in
  141. if await (KMLightMemberManager.manager.canPayFunction() == false) {
  142. self.settingView?.tipView?.isHidden = false
  143. } else {
  144. self.limit = false
  145. }
  146. }
  147. self.backButtonVC = KMDesignButton.init(withType: .Image)
  148. self.backBox.contentView = self.backButtonVC!.view
  149. self.nextButtonVC = KMDesignButton.init(withType: .Image)
  150. self.nextBox.contentView = self.nextButtonVC!.view
  151. self.canelButtonVC = KMDesignButton.init(withType: .Text)
  152. self.canelBox.contentView = self.canelButtonVC!.view
  153. self.convertButtonVC = KMDesignButton.init(withType: .Text)
  154. self.convertBox.contentView = self.convertButtonVC!.view
  155. }
  156. override func initDefaultValue() {
  157. super.initDefaultValue()
  158. self.totalNumberLabel.lineBreakMode = .byTruncatingTail
  159. self.prePDFView?.setDisplay(.singlePage)
  160. self.prePDFView?.layoutDocumentView()
  161. self.prePDFView?.autoScales = true
  162. self.prePDFView?.delegate = self
  163. self.backBox.fillColor = .clear
  164. self.backButtonVC?.target = self
  165. self.backButtonVC?.action = #selector(backButtonAction)
  166. self.backButtonVC?.image = NSImage(named: "KMImageNamePreviousPageImage")!
  167. self.backButtonVC?.image_disabled = NSImage(named: "KMImageNamePreviousPageImageDisable")!
  168. self.backButtonVC?.pagination()
  169. self.numberBox.wantsLayer = true
  170. self.numberBox.layer?.backgroundColor = NSColor.white.cgColor
  171. self.numberBox.layer?.borderWidth = 1
  172. self.numberBox.layer?.cornerRadius = 4
  173. self.numberTextField.focusRingType = .none
  174. self.numberTextField.delegate = self
  175. self.numberTextField.isBordered = false
  176. self.nextBox.fillColor = .clear
  177. self.nextButtonVC?.target = self
  178. self.nextButtonVC?.action = #selector(nextButtonAction)
  179. self.nextButtonVC?.image = NSImage(named: "KMImageNameNextPageImage")!
  180. self.nextButtonVC?.image_disabled = NSImage(named: "KMImageNameNextPageImageDisable")!
  181. self.nextButtonVC?.pagination()
  182. self.canelBox.fillColor = .clear
  183. self.canelButtonVC?.target = self
  184. self.canelButtonVC?.action = #selector(cancelButtonAction)
  185. self.canelButtonVC?.stringValue = NSLocalizedString("Cancel", comment: "")
  186. self.canelButtonVC?.button(type: .Sec, size: .m)
  187. self.canelButtonVC?.button.keyEquivalent = KMKeyEquivalent.esc.string()
  188. self.convertBox.fillColor = .clear
  189. self.convertButtonVC?.target = self
  190. self.convertButtonVC?.action = #selector(convertButtonAction)
  191. self.convertButtonVC?.stringValue = NSLocalizedString("Convert", comment: "")
  192. self.convertButtonVC?.button(type: .Cta, size: .m)
  193. // convertButtonVC.button.keyEquivalent = KMKeyEquivalent.enter
  194. self.rightScrollView.hasVerticalScroller = false
  195. self.rightScrollView.hasHorizontalScroller = false
  196. self.initUIProperty()
  197. }
  198. private func initUIProperty() {
  199. self.titleLabel.textColor = NSColor.titleColor()
  200. self.titleLabel.font = .SFProTextSemibold(16)
  201. self.leftBox.fillColor = NSColor(hex: "#F7F8FA")
  202. self.prePDFView?.backgroundColor = NSColor(hex: "#F7F8FA")
  203. self.numberBox.layer?.borderColor = NSColor(hex: "#DFE1E5").cgColor
  204. self.perLabel.textColor = NSColor.titleColor()
  205. self.perLabel.font = .SFProTextRegular(12)
  206. self.totalNumberLabel.textColor = NSColor.titleColor()
  207. self.totalNumberLabel.font = .SFProTextRegular(12)
  208. self.batchButton.setTitleColor(NSColor.buttonTitleColor())
  209. self.batchButton.layer?.borderColor = NSColor.buttonBorderColor().cgColor
  210. self.batchButton.font = .SFProTextRegular(14)
  211. }
  212. func initSettingView() -> KMConvertSettingView? {
  213. let settingView = KMConvertSettingView.createFromNib()
  214. return settingView
  215. }
  216. public func initConvert() -> KMPDFConvert {
  217. return KMPDFConvert()
  218. }
  219. func setupContentHeight(height: CGFloat) {
  220. self.rightBoxHeightConst.constant = height
  221. }
  222. func refreshUI() {
  223. }
  224. @objc func backButtonAction() {
  225. if let _ = self.prePDFView?.canGoToPreviousPage() {
  226. self.currentPageIndex -= 1
  227. self.prePDFView?.goToPreviousPage(nil)
  228. self.numberTextField.stringValue = "\(self.prePDFView!.currentPageIndex+1)"
  229. self.updateBackAndNextButtonState()
  230. }
  231. }
  232. @objc func nextButtonAction() {
  233. if let _ = self.prePDFView?.canGoToNextPage() {
  234. self.currentPageIndex += 1
  235. self.prePDFView?.goToNextPage(nil)
  236. self.numberTextField.stringValue = "\(self.prePDFView!.currentPageIndex+1)"
  237. self.updateBackAndNextButtonState()
  238. }
  239. }
  240. @objc func batchButtonAction() {
  241. guard let callback = itemClick else {
  242. return
  243. }
  244. callback(1)
  245. }
  246. @objc func cancelButtonAction() {
  247. self._clearData()
  248. guard let callback = itemClick else {
  249. return
  250. }
  251. callback(2)
  252. }
  253. @objc func convertButtonAction() {
  254. self.window?.makeFirstResponder(nil)
  255. if (self.pageRange == .custom) {
  256. let array = self.findSelectPage(pageRangeString: self.pageRangeString)
  257. if (array.count <= 0) {
  258. self.window?.makeFirstResponder(self.settingView?.pageRangeItemView?.pageSizeVC?.textField)
  259. return
  260. }
  261. }
  262. var pages: [Int] = []
  263. let filePath = self.getConvertFileSavePath()
  264. if (FileManager.default.fileExists(atPath: filePath)) {
  265. try?FileManager.default.removeItem(atPath: filePath)
  266. }
  267. if ((self.prePDFView?.document?.writeDecrypt(to: URL(fileURLWithPath: filePath)))!) {
  268. for i in 0 ..< self.prePDFView!.document.pageCount {
  269. pages.append(Int(i)+1)
  270. }
  271. }
  272. if (pages.count <= 0) {
  273. let alert = NSAlert()
  274. alert.alertStyle = .warning
  275. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  276. alert.runModal()
  277. return
  278. }
  279. Task { @MainActor in
  280. if await (KMLightMemberManager.manager.canPayFunction() == false) {
  281. let _ = KMSubscribeWaterMarkWindowController.show(window: self.window!, type: self.subscribeWaterMarkType!) { isSubscribeSuccess, isWaterMarkExport, isClose in
  282. if (isClose) {
  283. return
  284. }
  285. if (isSubscribeSuccess) {
  286. self._convertButtonAction()
  287. return
  288. }
  289. if (isWaterMarkExport) {
  290. self._convertButtonAction(limit: true)
  291. return
  292. }
  293. }
  294. return
  295. }
  296. self._convertButtonAction()
  297. }
  298. }
  299. func convertModelAppendParams(convert: KMPDFConvert) -> () {
  300. if let isAllowOCR = self.settingView?.ocrEnabled {
  301. convert.isAllowOCR = isAllowOCR
  302. if let idx = self.settingView?.ocrItemView?.languageIndex {
  303. convert.ocrLanguage = KMPDFConvertManager.defaultManager.languages_asc[idx]
  304. }
  305. } else {
  306. convert.isAllowOCR = false
  307. }
  308. }
  309. func getConvertFileSavePath() -> String {
  310. var path = NSSearchPathForDirectoriesInDomains(FileManager.SearchPathDirectory.applicationSupportDirectory, FileManager.SearchPathDomainMask.userDomainMask, true).last
  311. path?.append("/")
  312. path?.append(Bundle.main.bundleIdentifier!)
  313. if (FileManager.default.fileExists(atPath: path!) == false) {
  314. try?FileManager.default.createDirectory(atPath: path!, withIntermediateDirectories: false)
  315. }
  316. path?.append("/")
  317. path?.append("convert.pdf")
  318. return path!
  319. }
  320. func isValidPagesString(pagesString: String)-> Bool {
  321. var valid = false
  322. for ch in pagesString {
  323. if ch != "0" && ch != "1" && ch != "2" && ch != "3" && ch != "4" && ch != "5" && ch != "6" && ch != "7" && ch != "8" && ch != "9" && ch != "," && ch != "-" {
  324. valid = false
  325. break
  326. } else {
  327. valid = true
  328. }
  329. }
  330. return valid
  331. }
  332. func findSelectPage(pageRangeString: String) -> ([Int]) {
  333. if !isValidPagesString(pagesString: pageRangeString) {
  334. return []
  335. }
  336. var result: [Int] = []
  337. let array = pageRangeString.components(separatedBy: ",")
  338. for string in array {
  339. if string.isEmpty {
  340. return []
  341. } else {
  342. let pages = string .components(separatedBy: "-")
  343. if pages.count > 2 {
  344. return []
  345. } else if pages.count == 1 {
  346. let page = pages[0]
  347. if page.isEmpty || Int(page)! > self.documentModel!.pageCount || Int(page)! == 0 {
  348. return []
  349. } else {
  350. var hasSame: Bool = false
  351. for i in result {
  352. if i == Int(page)! {
  353. hasSame = true
  354. return []
  355. }
  356. }
  357. if !hasSame {
  358. result.append(Int(page)!)
  359. }
  360. }
  361. } else if pages.count == 2 {
  362. let page1 = pages[0]
  363. let page2 = pages[1]
  364. if page1.isEmpty || page2.isEmpty || Int(page1)! >= Int(page2)! || Int(page2)! > self.documentModel!.pageCount || Int(page1)! == 0 {
  365. return []
  366. } else {
  367. var hasSame: Bool = false
  368. for i in Int(page1)! ... Int(page2)! {
  369. for j in result {
  370. if j == i {
  371. hasSame = true
  372. return []
  373. }
  374. }
  375. }
  376. if !hasSame {
  377. for i in Int(page1)! ... Int(page2)! {
  378. result.append(i)
  379. }
  380. }
  381. }
  382. }
  383. }
  384. }
  385. return result
  386. }
  387. // MARK: -
  388. // MARK: Private Methods
  389. private func updateBackAndNextButtonState() {
  390. var nextEnabled = false
  391. if let result = self.prePDFView?.canGoToNextPage() {
  392. nextEnabled = result
  393. }
  394. var backEnabled = false
  395. if let result = self.prePDFView?.canGoToPreviousPage() {
  396. backEnabled = result
  397. }
  398. if (Thread.isMainThread) {
  399. self.backButtonVC?.state = backEnabled ? .Norm : .Disabled
  400. self.nextButtonVC?.state = nextEnabled ? .Norm : .Disabled
  401. } else {
  402. DispatchQueue.main.async {
  403. self.backButtonVC?.state = backEnabled ? .Norm : .Disabled
  404. self.nextButtonVC?.state = nextEnabled ? .Norm : .Disabled
  405. }
  406. }
  407. }
  408. internal func updatePreView(pageRange: KMPageRange, pageString: String = "") {
  409. let pages = self.getPages(pageRange, pageString: pageString)
  410. if (pages == nil || pages!.isEmpty) {
  411. let alert = NSAlert()
  412. alert.alertStyle = .warning
  413. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  414. alert.runModal()
  415. return
  416. }
  417. var indexs = IndexSet()
  418. for i in 0 ..< self.prePDFView!.document.pageCount {
  419. indexs.insert(IndexSet.Element(i))
  420. }
  421. self.prePDFView?.document.removePage(at: indexs)
  422. #if DEBUG
  423. let pageIndexs = self.getPageIndexs(pageRange, pageString: pageString)
  424. if (pageIndexs != nil) {
  425. for i in pageIndexs! {
  426. KMPrint(i)
  427. }
  428. }
  429. #endif
  430. // self.prePDFView.document?.importPages(pageIndexs!, from: self.documentModel.document, at: 0)
  431. for page in pages! {
  432. self.prePDFView?.document.insertPageObject(page, at: self.prePDFView!.document.pageCount)
  433. }
  434. DispatchQueue.main.async {
  435. self.prePDFView?.layoutDocumentView()
  436. self.prePDFView?.setNeedsDisplayForVisiblePages()
  437. self.prePDFView?.goToFirstPage(nil)
  438. let number: Int = Int(self.prePDFView!.document!.pageCount)
  439. self.totalNumberLabel.stringValue = "\(number)"
  440. self.currentPageIndex = 1
  441. self.numberTextField.stringValue = "1"
  442. self.updateBackAndNextButtonState()
  443. }
  444. }
  445. private func getPageIndexs(_ pageRange: KMPageRange, pageString: String = "") -> IndexSet? {
  446. guard let pageCount = self.documentModel?.pageCount, pageCount > 0 else {
  447. return nil
  448. }
  449. var indexs = IndexSet()
  450. if (pageRange == .all) {
  451. for i in 0 ..< pageCount {
  452. indexs.insert(i)
  453. }
  454. } else if (pageRange == .odd) {
  455. for i in 0 ..< pageCount {
  456. if ((i % 2) == 1) {
  457. continue
  458. }
  459. indexs.insert(i)
  460. }
  461. } else if (pageRange == .even) {
  462. for i in 0 ..< pageCount {
  463. if ((i % 2) == 0) {
  464. continue
  465. }
  466. indexs.insert(i)
  467. }
  468. } else if (pageRange == .current) {
  469. if (self.currentPageIndex >= 1 && self.currentPageIndex <= pageCount) {
  470. indexs.insert(self.currentPageIndex-1)
  471. }
  472. } else if (pageRange == .custom) {
  473. let array = self.findSelectPage(pageRangeString: pageString)
  474. if array.count == 0 {
  475. } else {
  476. for i in 0 ..< array.count {
  477. let index = array[i]
  478. if (index > pageCount || index < 1) {
  479. continue
  480. }
  481. indexs.insert(index-1)
  482. }
  483. }
  484. }
  485. return indexs
  486. }
  487. private func getPages(_ pageRange: KMPageRange, pageString: String = "") -> [CPDFPage]? {
  488. guard let pageCount = self.documentModel?.pageCount, pageCount > 0 else {
  489. return nil
  490. }
  491. var pages: [CPDFPage] = []
  492. if (pageRange == .all) {
  493. for i in 0 ..< pageCount {
  494. let page = self.documentModel!.document?.page(at: UInt(i))
  495. if (page == nil) {
  496. continue
  497. }
  498. pages.append(page!)
  499. }
  500. } else if (pageRange == .odd) {
  501. for i in 0 ..< pageCount {
  502. if ((i % 2) == 1) {
  503. continue
  504. }
  505. let page = self.documentModel!.document?.page(at: UInt(i))
  506. if (page == nil) {
  507. continue
  508. }
  509. pages.append(page!)
  510. }
  511. } else if (pageRange == .even) {
  512. for i in 0 ..< pageCount {
  513. if ((i % 2) == 0) {
  514. continue
  515. }
  516. let page = self.documentModel!.document?.page(at: UInt(i))
  517. if (page == nil) {
  518. continue
  519. }
  520. pages.append(page!)
  521. }
  522. } else if (pageRange == .current) {
  523. let page = self.prePDFView?.currentPage()
  524. if (page != nil) {
  525. pages.append(page!)
  526. }
  527. } else if (pageRange == .custom) {
  528. let array = self.findSelectPage(pageRangeString: pageString)
  529. if array.count == 0 {
  530. } else {
  531. for i in 0 ..< array.count {
  532. let index = array[i]
  533. if (index > pageCount || index < 1) {
  534. continue
  535. }
  536. let page = self.documentModel?.document?.page(at: UInt(index-1))
  537. if (page == nil) {
  538. continue
  539. }
  540. pages.append(page!)
  541. }
  542. }
  543. }
  544. return pages
  545. }
  546. private func _convertButtonAction(limit: Bool = false) {
  547. var pages: [Int] = []
  548. let filePath = self.getConvertFileSavePath()
  549. if (FileManager.default.fileExists(atPath: filePath)) {
  550. try?FileManager.default.removeItem(atPath: filePath)
  551. }
  552. if (limit) {
  553. if let _ = KMTools.saveWatermarkDocument(document: self.prePDFView!.document, to: URL(fileURLWithPath: filePath), removePWD: true) {
  554. for i in 0 ..< self.prePDFView!.document.pageCount {
  555. if (i >= 10) {
  556. break
  557. }
  558. pages.append(Int(i)+1)
  559. }
  560. }
  561. } else {
  562. if ((self.prePDFView?.document?.writeDecrypt(to: URL(fileURLWithPath: filePath)))!) {
  563. for i in 0 ..< self.prePDFView!.document.pageCount {
  564. pages.append(Int(i)+1)
  565. }
  566. }
  567. }
  568. if (pages.count <= 0) {
  569. let alert = NSAlert()
  570. alert.alertStyle = .warning
  571. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  572. alert.runModal()
  573. return
  574. }
  575. DispatchQueue.main.async {
  576. NSPanel.savePanel(self.window!) { panel in
  577. var url: URL = self.prePDFView!.document.documentURL
  578. if (self.oriDocumentUrl != nil) {
  579. url = self.oriDocumentUrl!
  580. }
  581. let name = url.deletingPathExtension().lastPathComponent
  582. panel.nameFieldStringValue = name
  583. panel.allowedFileTypes = [self.fileExtension]
  584. } completion: { response, url in
  585. if (response == .cancel) {
  586. return
  587. }
  588. let outputFolderPath = url!.deletingLastPathComponent().path
  589. let convert = self.initConvert()
  590. self.convert = convert
  591. convert.password = self.documentModel!.password
  592. convert.outputFolderPath = outputFolderPath
  593. convert.filePath = filePath
  594. convert.pages = pages
  595. var fileName = url!.deletingPathExtension().lastPathComponent
  596. if (self.fileExtension.isEmpty) {
  597. fileName = url!.lastPathComponent
  598. }
  599. // if (fileName.isEmpty) {
  600. // fileName = NSLocalizedString("Untitled", comment: "")
  601. // }
  602. convert.outputFileName = fileName
  603. self.convertModelAppendParams(convert: convert)
  604. DispatchQueue.main.async {
  605. self.showProgressWindow()
  606. self.progressController?.maxValue = Double(convert.pages.count)
  607. }
  608. DispatchQueue.global().async {
  609. KMPDFConvertManager.defaultManager.convert(convert: convert, progress: { index in
  610. DispatchQueue.main.async {
  611. self.progressController?.increment(by: 1.0)
  612. }
  613. }) { [unowned self] finished, error in
  614. self.hiddenProgressWindow()
  615. // 清除临时文件
  616. if (FileManager.default.fileExists(atPath: filePath)) {
  617. try?FileManager.default.removeItem(atPath: filePath)
  618. }
  619. if finished {
  620. self._clearData()
  621. cancelButtonAction()
  622. if FileManager.default.fileExists(atPath: convert.outputFilePath) {
  623. NSWorkspace.shared.activateFileViewerSelecting([URL(fileURLWithPath: convert.outputFilePath)])
  624. }
  625. } else {
  626. var errorString = ""
  627. let myError: NSError? = error as? NSError
  628. if myError?.code == 1 {
  629. errorString = NSLocalizedString("Password required or incorrect password. Please re-enter your password and try again", comment: "")
  630. } else if myError?.code == 2 {
  631. errorString = NSLocalizedString("The license doesn't allow the permission", comment: "")
  632. } else if myError?.code == 3 {
  633. errorString = NSLocalizedString("Malloc failure", comment: "")
  634. } else if myError?.code == 4 {
  635. errorString = NSLocalizedString("Unknown error in processing conversion. Please try again later", comment: "")
  636. } else if myError?.code == 5 {
  637. errorString = NSLocalizedString("Unknown error in processing PDF. Please try again later", comment: "")
  638. } else if myError?.code == 6 {
  639. errorString = NSLocalizedString("File not found or could not be opened. Check if your file exists or choose another file to convert", comment: "")
  640. } else if myError?.code == 7 {
  641. errorString = NSLocalizedString("File not in PDF format or corruptead. Change a PDF file and try again", comment: "")
  642. } else if myError?.code == 8 {
  643. errorString = NSLocalizedString("Unsupported security scheme", comment: "")
  644. } else if myError?.code == 9 {
  645. errorString = NSLocalizedString("Page not found or content error", comment: "")
  646. } else {
  647. errorString = NSLocalizedString("Table not found", comment: "")
  648. }
  649. let alert = NSAlert()
  650. alert.alertStyle = .critical
  651. alert.messageText = NSLocalizedString("Conversion Failed", comment: "")
  652. alert.informativeText = errorString
  653. alert.addButton(withTitle: NSLocalizedString("OK", comment: ""))
  654. alert.runModal()
  655. }
  656. }
  657. }
  658. }
  659. }
  660. }
  661. // MARK: Progress
  662. func showProgressWindow() {
  663. self.hiddenProgressWindow()
  664. let progress = SKProgressController()
  665. progress.message = NSLocalizedString("Converting...", comment: "")
  666. progress.window?.backgroundColor = NSColor(hex: "#36383B")
  667. progress.window?.contentView?.wantsLayer = true
  668. progress.window?.contentView?.layer?.backgroundColor = NSColor(hex: "#36383B").cgColor
  669. progress.progressField.textColor = NSColor.white
  670. progress.closeBlock = { [weak self] in
  671. if let convert = self?.convert {
  672. KMPDFConvertManager.defaultManager.cancel(convert: convert)
  673. }
  674. }
  675. self.progressController = progress
  676. self.window?.beginSheet(progress.window!)
  677. }
  678. func hiddenProgressWindow() {
  679. if (self.progressController != nil) {
  680. self.window?.endSheet((self.progressController?.window)!)
  681. self.progressController = nil
  682. }
  683. }
  684. func getCurrentLanguage() -> String {
  685. // return Bundle.main.preferredLocalizations.first!
  686. let array: [String] = UserDefaults.standard.object(forKey: "AppleLanguages") as! [String]
  687. return array.first!
  688. }
  689. /// 存储用户的选择的语言
  690. func saveLanugageSelectedIndex(index: Int) {
  691. UserDefaults.standard.setValue("\(index)", forKey: kKMConvertLanugageSelectedIndex)
  692. UserDefaults.standard.synchronize()
  693. }
  694. private func _clearData() {
  695. if let _ = self.oriDocumentUrl {
  696. if let data = self.prePDFView?.document.documentURL.path, FileManager.default.fileExists(atPath: data) {
  697. try?FileManager.default.removeItem(atPath: data)
  698. }
  699. }
  700. }
  701. }
  702. extension KMConvertBaseWindowController: NSTextFieldDelegate {
  703. func controlTextDidChange(_ obj: Notification) {
  704. if numberTextField.isEqual(to: obj.object) {
  705. let textField: NSTextField = obj.object as! NSTextField
  706. var value: String = ""
  707. for ch in textField.stringValue {
  708. if ch != "0" && ch != "1" && ch != "2" && ch != "3" && ch != "4" && ch != "5" && ch != "6" && ch != "7" && ch != "8" && ch != "9" {
  709. } else {
  710. if value.isEmpty && ch == "0" {
  711. } else {
  712. value.append(ch)
  713. }
  714. }
  715. }
  716. if value.isEmpty {
  717. value.append("1")
  718. } else {
  719. if Int(value)! <= 0 {
  720. value = "1"
  721. } else if Int(value)! > self.prePDFView!.document!.pageCount {
  722. let number: Int = Int(self.prePDFView!.document!.pageCount)
  723. value = "\(number)"
  724. }
  725. }
  726. numberTextField.stringValue = value
  727. }
  728. }
  729. func controlTextDidEndEditing(_ obj: Notification) {
  730. if numberTextField.isEqual(to: obj.object) {
  731. if (numberTextField.stringValue.isEmpty) {
  732. numberTextField.stringValue = "1"
  733. self.prePDFView?.go(toPageIndex: 0, animated: false)
  734. self.currentPageIndex = 1
  735. self.updateBackAndNextButtonState()
  736. return
  737. }
  738. let number: Int = Int(numberTextField.stringValue)!
  739. if number > 0 && number <= self.prePDFView!.document!.pageCount {
  740. guard let page = self.prePDFView?.document?.page(at: UInt(number-1)) else {
  741. return
  742. }
  743. self.prePDFView?.go(to: page)
  744. currentPageIndex = number
  745. self.updateBackAndNextButtonState()
  746. }
  747. }
  748. }
  749. }
  750. extension KMConvertBaseWindowController: CPDFViewDelegate {
  751. func pdfViewCurrentPageDidChanged(_ pdfView: CPDFView!) {
  752. let currentIndex = pdfView.currentPageIndex
  753. currentPageIndex = currentIndex + 1
  754. numberTextField.stringValue = "\(currentIndex+1)"
  755. self.updateBackAndNextButtonState()
  756. }
  757. }