KMConvertBaseWindowController.swift 45 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223
  1. //
  2. // KMConvertBaseWindowController.swift
  3. // PDF Reader Pro
  4. //
  5. // Created by tangchao on 2022/12/5.
  6. //
  7. import PDFKit
  8. import KMComponentLibrary
  9. typealias KMConvertBaseWindowControllerItemClick = (Int) -> ()
  10. class KMConvertBaseWindowController: KMNBaseWindowController {
  11. @IBOutlet weak var titleLabel: NSTextField!
  12. @IBOutlet weak var leftBox: NSBox!
  13. @IBOutlet weak var prePDFView: KMNPDFPreView?
  14. @IBOutlet weak var numberBox: NSView!
  15. @IBOutlet weak var numberTextField: NSTextField!
  16. @IBOutlet weak var perLabel: NSTextField!
  17. @IBOutlet weak var totalNumberLabel: NSTextField!
  18. @IBOutlet weak var backBox: NSBox!
  19. @IBOutlet weak var nextBox: NSBox!
  20. @IBOutlet weak var rightScrollView: NSScrollView!
  21. @IBOutlet weak var batchButton: NSButton!
  22. @IBOutlet weak var canelBox: NSBox!
  23. @IBOutlet weak var convertBox: NSBox!
  24. @IBOutlet weak var rightBoxHeightConst: NSLayoutConstraint!
  25. private var cancelButton_: ComponentButton?
  26. private var convertButton_: ComponentButton?
  27. private var batchButton_: ComponentButton?
  28. var backButtonVC: KMDesignButton?
  29. var nextButtonVC: KMDesignButton?
  30. var itemClick: KMConvertBaseWindowControllerItemClick?
  31. var documentModel: KMDocumentModel?
  32. var currentPageIndex: Int = 1
  33. var pageRange = KMPageRange.all
  34. var pageRangeString: String = ""
  35. var settingView: KMConvertSettingView?
  36. var subType: Int = 0
  37. var fileExtension: String {
  38. get {
  39. return ""
  40. }
  41. }
  42. var progressController: SKProgressController?
  43. var pdfPreView: KMNPreView?
  44. private var convert: KMPDFConvert?
  45. private var ocrLanguage_: COCRLanguage = .english
  46. var limit = true
  47. // 是否限制次数
  48. var limitCount = true
  49. var oriDocumentUrl: URL?
  50. var subscribeWaterMarkType: KMSubscribeWaterMarkType?
  51. deinit {
  52. KMPrint("KMConvertBaseWindowController 已释放")
  53. self.prePDFView?.removeFromSuperview()
  54. self.prePDFView = nil
  55. #if VERSION_DMG
  56. NotificationCenter.default.removeObserver(self)
  57. #endif
  58. }
  59. convenience init() {
  60. self.init(windowNibName: "KMConvertBaseWindowController")
  61. }
  62. override func awakeFromNib() {
  63. super.awakeFromNib()
  64. DispatchQueue.main.async {
  65. self.window?.makeFirstResponder(nil)
  66. }
  67. }
  68. override func windowDidLoad() {
  69. super.windowDidLoad()
  70. // if (self.convertCountIsReach()) {
  71. // Swift.debugPrint("转档次数已达到限制. count: \(self.getConvertCount())")
  72. // }
  73. // #if VERSION_DMG
  74. NotificationCenter.default.addObserver(self, selector: #selector(km_windowWillCloseNotifation), name: NSWindow.willCloseNotification, object: nil)
  75. // #endif
  76. let view = KMNPreView.createFromNib()
  77. leftBox.contentView = view
  78. self.pdfPreView = view
  79. if (self.documentModel != nil) {
  80. // let document = CPDFDocument(url: self.documentModel!.documentURL)
  81. // if (document!.isLocked) {
  82. // document?.unlock(withPassword: self.documentModel!.password)
  83. // }
  84. // self.prePDFView?.document = document!
  85. // self.prePDFView?.setFileUrl(self.documentModel!.documentURL, password: self.documentModel?.password)
  86. pdfPreView?.setFileUrl(self.documentModel!.documentURL, password: self.documentModel?.password)
  87. // let number: Int = Int((self.prePDFView?.document.pageCount)!)
  88. // let string = "\(number)"
  89. // totalNumberLabel.stringValue = string
  90. // self.prePDFView?.go(toPageIndex: self.documentModel!.currentIndex, animated: false)
  91. }
  92. // if (self.prePDFView?.documentView() != nil) {
  93. // self.prePDFView?.documentView().enclosingScrollView?.hasVerticalScroller = false
  94. // self.prePDFView?.documentView().enclosingScrollView?.hasHorizontalScroller = false
  95. // }
  96. // if let pageCount = self.prePDFView?.document.pageCount {
  97. // self.settingView?.pageCount = Int(pageCount)
  98. // }
  99. batchButton.title = NSLocalizedString("", comment: "")
  100. batchButton.isBordered = false
  101. // batchButton.wantsLayer = true
  102. // batchButton.layer?.borderWidth = 1
  103. // batchButton.layer?.cornerRadius = 4
  104. batchButton.target = self
  105. batchButton.action = #selector(batchButtonAction)
  106. // batchButton.isHidden = true
  107. self.prePDFView?.currentPageDidChanged = { [weak self] preView, idx in
  108. guard let pdfView = preView?.pdfView else {
  109. return
  110. }
  111. let currentIndex = pdfView.currentPageIndex
  112. self?.currentPageIndex = currentIndex + 1
  113. self?.numberTextField.stringValue = "\(currentIndex+1)"
  114. self?.updateBackAndNextButtonState()
  115. }
  116. self.updateBackAndNextButtonState()
  117. }
  118. override func initSubViews() {
  119. super.initSubViews()
  120. let view = self.initSettingView()
  121. // view?.frame = self.rightScrollView.bounds
  122. // view?.autoresizingMask = [.width, .height]
  123. self.settingView = view
  124. self.rightScrollView.documentView = view
  125. self.rightScrollView.verticalScrollElasticity = .none
  126. self.rightScrollView.horizontalScrollElasticity = .none
  127. view?.km_add_inset_constraint()
  128. view?.currentLanguage = self.getCurrentLanguage()
  129. view?.lanugageDidSelected = { [weak self] value, _ in
  130. guard let index = value as? Int else {
  131. return
  132. }
  133. self?.saveLanugageSelectedIndex(index: index)
  134. }
  135. view?.pageRangeDidChange = { [weak self] pageRange in
  136. self?.pageRange = pageRange
  137. if (pageRange != .custom) {
  138. self?.updatePreView(pageRange: pageRange)
  139. }
  140. }
  141. view?.pageRangeDidInputFinishCallback = { [unowned self] string in
  142. self.pageRangeString = string
  143. let array = self.findSelectPage(pageRangeString: string)
  144. if array.count == 0 {
  145. let alert = NSAlert()
  146. alert.alertStyle = .warning
  147. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  148. alert.runModal()
  149. self.settingView?.pageRangeItemView?.pageSizeVC?.stringValue = ""
  150. } else {
  151. self.updatePreView(pageRange: .custom, pageString: string)
  152. }
  153. }
  154. view?.tipView?.itemClick = { [unowned self] _, _ in
  155. #if VERSION_DMG
  156. KMTools.openPurchaseProductWebsite()
  157. #else
  158. Task { @MainActor in
  159. if await (KMLightMemberManager.manager.canPayFunction() == false) {
  160. let _ = KMSubscribeWaterMarkWindowController.show(window: self.window!) { isSubscribeSuccess, isWaterMarkExport, isClose in
  161. if (isSubscribeSuccess) {
  162. self.refreshUI()
  163. }
  164. }
  165. }
  166. }
  167. #endif
  168. }
  169. self.settingView?.tipView?.isHidden = true
  170. Task { @MainActor in
  171. if await (KMLightMemberManager.manager.canUseAdvanced() == false) {
  172. self.limitCount = true
  173. } else {
  174. self.limitCount = false
  175. }
  176. if await (KMLightMemberManager.manager.canPayFunction() == false) {
  177. #if VERSION_DMG
  178. if (self.limitCount) { // 未登录
  179. self.settingView?.tipView?.isHidden = true
  180. } else {
  181. self.settingView?.tipView?.isHidden = false
  182. }
  183. #else
  184. self.settingView?.tipView?.isHidden = false
  185. #endif
  186. } else {
  187. self.limit = false
  188. }
  189. }
  190. self.backButtonVC = KMDesignButton.init(withType: .Image)
  191. self.backBox.contentView = self.backButtonVC!.view
  192. self.nextButtonVC = KMDesignButton.init(withType: .Image)
  193. self.nextBox.contentView = self.nextButtonVC!.view
  194. cancelButton_ = ComponentButton()
  195. cancelButton_?.properties = ComponentButtonProperty(type: .default_tertiary, size: .s, state: .normal, buttonText: KMLocalizedString("Cancel"))
  196. cancelButton_?.setTarget(self, action: #selector(cancelButtonAction))
  197. self.canelBox.contentView = self.cancelButton_
  198. convertButton_ = ComponentButton()
  199. convertBox.contentView = convertButton_
  200. batchButton_ = ComponentButton()
  201. batchButton_?.properties = ComponentButtonProperty(type: .default_tertiary, size: .s, state: .normal, buttonText: KMLocalizedString("Batch"))
  202. batchButton_?.setTarget(self, action: #selector(batchButtonAction))
  203. self.batchButton.addSubview(self.batchButton_!)
  204. self.batchButton_?.frame = self.batchButton.bounds
  205. self.batchButton_?.autoresizingMask = [.width, .height]
  206. }
  207. override func initDefaultValue() {
  208. super.initDefaultValue()
  209. self.totalNumberLabel.lineBreakMode = .byTruncatingTail
  210. // self.prePDFView?.setDisplay(.singlePage)
  211. // self.prePDFView?.layoutDocumentView()
  212. // self.prePDFView?.autoScales = true
  213. // self.prePDFView?.delegate = self
  214. self.backBox.fillColor = .clear
  215. self.backButtonVC?.target = self
  216. self.backButtonVC?.action = #selector(backButtonAction)
  217. self.backButtonVC?.image = NSImage(named: "KMImageNamePreviousPageImage")!
  218. self.backButtonVC?.image_disabled = NSImage(named: "KMImageNamePreviousPageImageDisable")!
  219. self.backButtonVC?.pagination()
  220. self.numberBox.wantsLayer = true
  221. self.numberBox.layer?.backgroundColor = NSColor.white.cgColor
  222. self.numberBox.layer?.borderWidth = 1
  223. self.numberBox.layer?.cornerRadius = 4
  224. self.numberTextField.focusRingType = .none
  225. self.numberTextField.delegate = self
  226. self.numberTextField.isBordered = false
  227. self.nextBox.fillColor = .clear
  228. self.nextButtonVC?.target = self
  229. self.nextButtonVC?.action = #selector(nextButtonAction)
  230. self.nextButtonVC?.image = NSImage(named: "KMImageNameNextPageImage")!
  231. self.nextButtonVC?.image_disabled = NSImage(named: "KMImageNameNextPageImageDisable")!
  232. self.nextButtonVC?.pagination()
  233. self.canelBox.fillColor = .clear
  234. self.cancelButton_?.keyEquivalent = KMKeyEquivalent.esc.string()
  235. self.convertBox.fillColor = .clear
  236. convertButton_?.properties = ComponentButtonProperty(type: .primary,
  237. size: .s,
  238. state: .normal,
  239. isDisable: false,
  240. buttonText: KMLocalizedString("Convert"))
  241. convertButton_?.setTarget(self, action: #selector(convertButtonAction))
  242. convertButton_?.keyEquivalent = KMKeyEquivalent.enter
  243. self.rightScrollView.hasVerticalScroller = false
  244. self.rightScrollView.hasHorizontalScroller = false
  245. self.initUIProperty()
  246. }
  247. private func initUIProperty() {
  248. self.titleLabel.textColor = NSColor.titleColor()
  249. self.titleLabel.font = .SFProTextSemiboldFont(16)
  250. self.leftBox.fillColor = NSColor.km_init(hex: "#F7F8FA")
  251. // self.prePDFView?.backgroundColor = NSColor.km_init(hex: "#F7F8FA")
  252. self.numberBox.layer?.borderColor = NSColor.km_init(hex: "#DFE1E5").cgColor
  253. self.perLabel.textColor = NSColor.titleColor()
  254. self.perLabel.font = .SFProTextRegularFont(12)
  255. self.totalNumberLabel.textColor = NSColor.titleColor()
  256. self.totalNumberLabel.font = .SFProTextRegularFont(12)
  257. // self.batchButton.setTitleColor(NSColor.buttonTitleColor())
  258. // self.batchButton.layer?.borderColor = NSColor.buttonBorderColor().cgColor
  259. // self.batchButton.font = .SFProTextRegularFont(14)
  260. }
  261. func initSettingView() -> KMConvertSettingView? {
  262. let settingView = KMConvertSettingView.createFromNib()
  263. return settingView
  264. }
  265. public func initConvert() -> KMPDFConvert {
  266. return KMPDFConvert()
  267. }
  268. func setupContentHeight(height: CGFloat, animated: Bool = false) {
  269. if (animated) {
  270. self.rightBoxHeightConst.animator().constant = height
  271. } else {
  272. self.rightBoxHeightConst.constant = height
  273. }
  274. }
  275. func refreshUI() {
  276. }
  277. @objc func backButtonAction() {
  278. // if let _ = self.prePDFView?.canGoToPreviousPage() {
  279. // self.currentPageIndex -= 1
  280. // self.prePDFView?.goToPreviousPage(nil)
  281. //
  282. // self.numberTextField.stringValue = "\(self.prePDFView!.currentPageIndex+1)"
  283. // self.updateBackAndNextButtonState()
  284. // }
  285. }
  286. @objc func nextButtonAction() {
  287. // if let _ = self.prePDFView?.canGoToNextPage() {
  288. // self.currentPageIndex += 1
  289. // self.prePDFView?.goToNextPage(nil)
  290. //
  291. // self.numberTextField.stringValue = "\(self.prePDFView!.currentPageIndex+1)"
  292. // self.updateBackAndNextButtonState()
  293. // }
  294. }
  295. @objc func batchButtonAction() {
  296. // guard let callback = itemClick else {
  297. // return
  298. // }
  299. //
  300. // callback(1)
  301. self.own_closeEndSheet()
  302. }
  303. @objc func cancelButtonAction() {
  304. self._clearData()
  305. // guard let callback = itemClick else {
  306. // return
  307. // }
  308. //
  309. // callback(2)
  310. self.own_closeEndSheet()
  311. }
  312. @objc func convertButtonAction() {
  313. self.window?.makeFirstResponder(nil)
  314. if (self.pageRange == .custom) {
  315. let array = self.findSelectPage(pageRangeString: self.pageRangeString)
  316. if (array.count <= 0) {
  317. self.window?.makeFirstResponder(self.settingView?.pageRangeItemView?.pageSizeVC?.textField)
  318. return
  319. }
  320. }
  321. var pages: [Int] = []
  322. let filePath = self.getConvertFileSavePath()
  323. if (FileManager.default.fileExists(atPath: filePath)) {
  324. try?FileManager.default.removeItem(atPath: filePath)
  325. }
  326. let document = self.pdfPreView?.pdfPreView.pdfView.document
  327. if ((document?.writeDecrypt(to: URL(fileURLWithPath: filePath)))!) {
  328. for i in 0 ..< document!.pageCount {
  329. pages.append(Int(i)+1)
  330. }
  331. }
  332. if (pages.count <= 0) {
  333. let alert = NSAlert()
  334. alert.alertStyle = .warning
  335. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  336. alert.runModal()
  337. return
  338. }
  339. Task { @MainActor in
  340. #if VERSION_DMG
  341. if (await (KMLightMemberManager.manager.canUseAdvanced() == false)) {
  342. let _ = KMComparativeTableViewController.show(window: self.window!, .convert)
  343. return
  344. }
  345. #endif
  346. // if await (KMLightMemberManager.manager.canPayFunction() == false) {
  347. // var isContinue = false
  348. // if pages.count <= 10 {
  349. // isContinue = true
  350. // }
  351. //
  352. // var limit = self.convertCountIsReach()
  353. // if (!self.limitCount) { // 登录
  354. // limit = false
  355. // }
  356. //
  357. // if (limit) {
  358. // isContinue = false
  359. // }
  360. //
  361. // let _ = KMSubscribeWaterMarkWindowController.show(window: self.window!, isContinue: isContinue, limit: limit, type: self.subscribeWaterMarkType!) { isSubscribeSuccess, isWaterMarkExport, isClose in
  362. // if (isClose) {
  363. // return
  364. // }
  365. // if (isSubscribeSuccess) {
  366. // self._convertButtonAction()
  367. // return
  368. // }
  369. // if (isWaterMarkExport && !limit) {
  370. // self._convertButtonAction(limit: true)
  371. // return
  372. // }
  373. // }
  374. // return
  375. // }
  376. self._convertButtonAction()
  377. }
  378. }
  379. func convertModelAppendParams(convert: KMPDFConvert) -> () {
  380. if let isAllowOCR = self.settingView?.ocrEnabled {
  381. convert.isAllowOCR = isAllowOCR
  382. if let idx = self.settingView?.ocrItemView?.languageIndex {
  383. // convert.ocrLanguage = KMPDFConvertManager.defaultManager.languages_asc[idx]
  384. convert.ocrLanguage = self._getOCRLanguage(idx: idx)
  385. }
  386. } else {
  387. convert.isAllowOCR = false
  388. }
  389. }
  390. func getConvertFileSavePath() -> String {
  391. var path = NSSearchPathForDirectoriesInDomains(FileManager.SearchPathDirectory.applicationSupportDirectory, FileManager.SearchPathDomainMask.userDomainMask, true).last
  392. path?.append("/")
  393. path?.append(Bundle.main.bundleIdentifier!)
  394. if (FileManager.default.fileExists(atPath: path!) == false) {
  395. try?FileManager.default.createDirectory(atPath: path!, withIntermediateDirectories: false)
  396. }
  397. path?.append("/")
  398. path?.append("convert.pdf")
  399. return path!
  400. }
  401. func isValidPagesString(pagesString: String)-> Bool {
  402. var valid = false
  403. for ch in pagesString {
  404. if ch != "0" && ch != "1" && ch != "2" && ch != "3" && ch != "4" && ch != "5" && ch != "6" && ch != "7" && ch != "8" && ch != "9" && ch != "," && ch != "-" {
  405. valid = false
  406. break
  407. } else {
  408. valid = true
  409. }
  410. }
  411. return valid
  412. }
  413. func findSelectPage(pageRangeString: String) -> ([Int]) {
  414. if !isValidPagesString(pagesString: pageRangeString) {
  415. return []
  416. }
  417. var result: [Int] = []
  418. let array = pageRangeString.components(separatedBy: ",")
  419. for string in array {
  420. if string.isEmpty {
  421. return []
  422. } else {
  423. let pages = string .components(separatedBy: "-")
  424. if pages.count > 2 {
  425. return []
  426. } else if pages.count == 1 {
  427. let page = pages[0]
  428. if page.isEmpty || Int(page)! > self.documentModel!.pageCount || Int(page)! == 0 {
  429. return []
  430. } else {
  431. var hasSame: Bool = false
  432. for i in result {
  433. if i == Int(page)! {
  434. hasSame = true
  435. return []
  436. }
  437. }
  438. if !hasSame {
  439. result.append(Int(page)!)
  440. }
  441. }
  442. } else if pages.count == 2 {
  443. let page1 = pages[0]
  444. let page2 = pages[1]
  445. if page1.isEmpty || page2.isEmpty || Int(page1)! >= Int(page2)! || Int(page2)! > self.documentModel!.pageCount || Int(page1)! == 0 {
  446. return []
  447. } else {
  448. var hasSame: Bool = false
  449. for i in Int(page1)! ... Int(page2)! {
  450. for j in result {
  451. if j == i {
  452. hasSame = true
  453. return []
  454. }
  455. }
  456. }
  457. if !hasSame {
  458. for i in Int(page1)! ... Int(page2)! {
  459. result.append(i)
  460. }
  461. }
  462. }
  463. }
  464. }
  465. }
  466. return result
  467. }
  468. // MARK: -
  469. // MARK: Private Methods
  470. private func updateBackAndNextButtonState() {
  471. var nextEnabled = false
  472. // if let result = self.prePDFView?.canGoToNextPage() {
  473. // nextEnabled = result
  474. // }
  475. var backEnabled = false
  476. // if let result = self.prePDFView?.canGoToPreviousPage() {
  477. // backEnabled = result
  478. // }
  479. if (Thread.isMainThread) {
  480. self.backButtonVC?.state = backEnabled ? .Norm : .Disabled
  481. self.nextButtonVC?.state = nextEnabled ? .Norm : .Disabled
  482. } else {
  483. DispatchQueue.main.async {
  484. self.backButtonVC?.state = backEnabled ? .Norm : .Disabled
  485. self.nextButtonVC?.state = nextEnabled ? .Norm : .Disabled
  486. }
  487. }
  488. }
  489. internal func updatePreView(pageRange: KMPageRange, pageString: String = "") {
  490. // let pages = self.getPages(pageRange, pageString: pageString)
  491. // if (pages == nil || pages!.isEmpty) {
  492. // let alert = NSAlert()
  493. // alert.alertStyle = .warning
  494. // alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  495. // alert.runModal()
  496. // return
  497. // }
  498. // var indexs = IndexSet()
  499. // for i in 0 ..< self.prePDFView!.document.pageCount {
  500. // indexs.insert(IndexSet.Element(i))
  501. // }
  502. // self.prePDFView?.document.removePage(at: indexs)
  503. //
  504. // #if DEBUG
  505. // let pageIndexs = self.getPageIndexs(pageRange, pageString: pageString)
  506. // if (pageIndexs != nil) {
  507. // for i in pageIndexs! {
  508. //
  509. // KMPrint(i)
  510. // }
  511. // }
  512. // #endif
  513. //// self.prePDFView.document?.importPages(pageIndexs!, from: self.documentModel.document, at: 0)
  514. // for page in pages! {
  515. // self.prePDFView?.document.insertPageObject(page, at: self.prePDFView!.document.pageCount)
  516. // }
  517. // DispatchQueue.main.async {
  518. // self.prePDFView?.layoutDocumentView()
  519. // self.prePDFView?.setNeedsDisplayForVisiblePages()
  520. // self.prePDFView?.goToFirstPage(nil)
  521. //
  522. // let number: Int = Int(self.prePDFView!.document!.pageCount)
  523. // self.totalNumberLabel.stringValue = "\(number)"
  524. // self.currentPageIndex = 1
  525. // self.numberTextField.stringValue = "1"
  526. //
  527. // self.updateBackAndNextButtonState()
  528. // }
  529. }
  530. private func getPageIndexs(_ pageRange: KMPageRange, pageString: String = "") -> IndexSet? {
  531. guard let pageCount = self.documentModel?.pageCount, pageCount > 0 else {
  532. return nil
  533. }
  534. var indexs = IndexSet()
  535. if (pageRange == .all) {
  536. for i in 0 ..< pageCount {
  537. indexs.insert(i)
  538. }
  539. } else if (pageRange == .odd) {
  540. for i in 0 ..< pageCount {
  541. if ((i % 2) == 1) {
  542. continue
  543. }
  544. indexs.insert(i)
  545. }
  546. } else if (pageRange == .even) {
  547. for i in 0 ..< pageCount {
  548. if ((i % 2) == 0) {
  549. continue
  550. }
  551. indexs.insert(i)
  552. }
  553. } else if (pageRange == .current) {
  554. if (self.currentPageIndex >= 1 && self.currentPageIndex <= pageCount) {
  555. indexs.insert(self.currentPageIndex-1)
  556. }
  557. } else if (pageRange == .custom) {
  558. let array = self.findSelectPage(pageRangeString: pageString)
  559. if array.count == 0 {
  560. } else {
  561. for i in 0 ..< array.count {
  562. let index = array[i]
  563. if (index > pageCount || index < 1) {
  564. continue
  565. }
  566. indexs.insert(index-1)
  567. }
  568. }
  569. }
  570. return indexs
  571. }
  572. private func getPages(_ pageRange: KMPageRange, pageString: String = "") -> [CPDFPage]? {
  573. guard let pageCount = self.documentModel?.pageCount, pageCount > 0 else {
  574. return nil
  575. }
  576. var pages: [CPDFPage] = []
  577. if (pageRange == .all) {
  578. for i in 0 ..< pageCount {
  579. let page = self.documentModel!.document?.page(at: UInt(i))
  580. if (page == nil) {
  581. continue
  582. }
  583. pages.append(page!)
  584. }
  585. } else if (pageRange == .odd) {
  586. for i in 0 ..< pageCount {
  587. if ((i % 2) == 1) {
  588. continue
  589. }
  590. let page = self.documentModel!.document?.page(at: UInt(i))
  591. if (page == nil) {
  592. continue
  593. }
  594. pages.append(page!)
  595. }
  596. } else if (pageRange == .even) {
  597. for i in 0 ..< pageCount {
  598. if ((i % 2) == 0) {
  599. continue
  600. }
  601. let page = self.documentModel!.document?.page(at: UInt(i))
  602. if (page == nil) {
  603. continue
  604. }
  605. pages.append(page!)
  606. }
  607. } else if (pageRange == .current) {
  608. // let page = self.prePDFView?.currentPage()
  609. // if (page != nil) {
  610. // pages.append(page!)
  611. // }
  612. } else if (pageRange == .custom) {
  613. let array = self.findSelectPage(pageRangeString: pageString)
  614. if array.count == 0 {
  615. } else {
  616. for i in 0 ..< array.count {
  617. let index = array[i]
  618. if (index > pageCount || index < 1) {
  619. continue
  620. }
  621. let page = self.documentModel?.document?.page(at: UInt(index-1))
  622. if (page == nil) {
  623. continue
  624. }
  625. pages.append(page!)
  626. }
  627. }
  628. }
  629. return pages
  630. }
  631. private func _convertButtonAction(limit: Bool = false) {
  632. var pages: [Int] = []
  633. let filePath = self.getConvertFileSavePath()
  634. if (FileManager.default.fileExists(atPath: filePath)) {
  635. try?FileManager.default.removeItem(atPath: filePath)
  636. }
  637. if (limit) {
  638. // if let _ = KMTools.saveWatermarkDocument(document: self.prePDFView!.document, to: URL(fileURLWithPath: filePath), removePWD: true) {
  639. // for i in 0 ..< self.prePDFView!.document.pageCount {
  640. // if (i >= 10) {
  641. // break
  642. // }
  643. // pages.append(Int(i)+1)
  644. // }
  645. // }
  646. } else {
  647. let document = self.pdfPreView?.pdfPreView.pdfView.document
  648. if ((document?.writeDecrypt(to: URL(fileURLWithPath: filePath)))!) {
  649. for i in 0 ..< document!.pageCount {
  650. pages.append(Int(i)+1)
  651. }
  652. }
  653. }
  654. if (pages.count <= 0) {
  655. let alert = NSAlert()
  656. alert.alertStyle = .warning
  657. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  658. alert.runModal()
  659. return
  660. }
  661. DispatchQueue.main.async {
  662. NSPanel.savePanel(self.window!) { panel in
  663. var url: URL = (self.pdfPreView?.pdfPreView.pdfView.document.documentURL)!
  664. if (self.oriDocumentUrl != nil) {
  665. url = self.oriDocumentUrl!
  666. }
  667. let name = url.deletingPathExtension().lastPathComponent
  668. panel.nameFieldStringValue = name
  669. panel.allowedFileTypes = [self.fileExtension]
  670. } completion: { response, url in
  671. if (response == .cancel) {
  672. return
  673. }
  674. let outputFolderPath = url!.deletingLastPathComponent().path
  675. let convert = self.initConvert()
  676. self.convert = convert
  677. convert.password = self.documentModel!.password
  678. convert.outputFolderPath = outputFolderPath
  679. convert.filePath = filePath
  680. convert.pages = pages
  681. var fileName = url!.deletingPathExtension().lastPathComponent
  682. if (self.fileExtension.isEmpty) {
  683. fileName = url!.lastPathComponent
  684. }
  685. // if (fileName.isEmpty) {
  686. // fileName = NSLocalizedString("Untitled", comment: "")
  687. // }
  688. convert.outputFileName = fileName
  689. self.convertModelAppendParams(convert: convert)
  690. DispatchQueue.main.async {
  691. self.showProgressWindow()
  692. self.progressController?.maxValue = Double(convert.pages.count)
  693. }
  694. #if VERSION_DMG
  695. if self.settingView?.ocrItemView?.checkVC?.state == .Checked {
  696. let bundlePath = kResourcePath?.stringByAppendingPathComponent("DocumentAI.bundle")
  697. CPDFConvertKit.setOCRModelPath(bundlePath)
  698. }
  699. #else
  700. let bundlePath = Bundle.main.path(forResource: "DocumentAI", ofType: "bundle")
  701. CPDFConvertKit.setOCRModelPath(bundlePath)
  702. #endif
  703. DispatchQueue.global().async {
  704. KMPDFConvertManager.defaultManager.convert(convert: convert, progress: { index in
  705. DispatchQueue.main.async {
  706. self.progressController?.increment(by: 1.0)
  707. }
  708. }) { [unowned self] finished, error in
  709. self.hiddenProgressWindow()
  710. // 清除临时文件
  711. if (FileManager.default.fileExists(atPath: filePath)) {
  712. try?FileManager.default.removeItem(atPath: filePath)
  713. }
  714. if finished {
  715. self._clearData()
  716. self.recordConvertCount()
  717. cancelButtonAction()
  718. if FileManager.default.fileExists(atPath: convert.outputFilePath) {
  719. NSWorkspace.shared.activateFileViewerSelecting([URL(fileURLWithPath: convert.outputFilePath)])
  720. }
  721. } else {
  722. var errorString = ""
  723. let myError: NSError? = error as? NSError
  724. if myError?.code == 1 {
  725. errorString = NSLocalizedString("Password required or incorrect password. Please re-enter your password and try again", comment: "")
  726. } else if myError?.code == 2 {
  727. errorString = NSLocalizedString("The license doesn't allow the permission", comment: "")
  728. } else if myError?.code == 3 {
  729. errorString = NSLocalizedString("Malloc failure", comment: "")
  730. } else if myError?.code == 4 {
  731. errorString = NSLocalizedString("Unknown error in processing conversion. Please try again later", comment: "")
  732. } else if myError?.code == 5 {
  733. errorString = NSLocalizedString("Unknown error in processing PDF. Please try again later", comment: "")
  734. } else if myError?.code == 6 {
  735. errorString = NSLocalizedString("File not found or could not be opened. Check if your file exists or choose another file to convert", comment: "")
  736. } else if myError?.code == 7 {
  737. errorString = NSLocalizedString("File not in PDF format or corruptead. Change a PDF file and try again", comment: "")
  738. } else if myError?.code == 8 {
  739. errorString = NSLocalizedString("Unsupported security scheme", comment: "")
  740. } else if myError?.code == 9 {
  741. errorString = NSLocalizedString("Page not found or content error", comment: "")
  742. } else {
  743. errorString = NSLocalizedString("Table not found", comment: "")
  744. }
  745. let alert = NSAlert()
  746. alert.alertStyle = .critical
  747. alert.messageText = NSLocalizedString("Conversion Failed", comment: "")
  748. alert.informativeText = errorString
  749. alert.addButton(withTitle: NSLocalizedString("OK", comment: ""))
  750. alert.runModal()
  751. }
  752. }
  753. }
  754. }
  755. }
  756. }
  757. private func _getOCRLanguage(idx: Int) -> COCRLanguage {
  758. switch (idx) {
  759. case 0: //chinese
  760. self.ocrLanguage_ = .chinese
  761. break
  762. case 1: //chinese zht
  763. self.ocrLanguage_ = .chineseTraditional
  764. break
  765. case 2: //English
  766. self.ocrLanguage_ = .english
  767. break
  768. case 3:
  769. //Japanese
  770. self.ocrLanguage_ = .japanese
  771. break
  772. case 4:
  773. //Kroean
  774. self.ocrLanguage_ = .korean
  775. break
  776. case 5:
  777. //Serbian
  778. self.ocrLanguage_ = .serbian
  779. break
  780. case 6:
  781. //Occitan
  782. self.ocrLanguage_ = .occitan
  783. break
  784. case 7:
  785. //Danish
  786. self.ocrLanguage_ = .danish
  787. break
  788. case 8:
  789. //German
  790. self.ocrLanguage_ = .german
  791. break
  792. case 9:
  793. //French
  794. self.ocrLanguage_ = .french
  795. break
  796. case 10:
  797. //Italian
  798. self.ocrLanguage_ = .italian
  799. break
  800. case 11:
  801. //Spanish
  802. self.ocrLanguage_ = .spanish
  803. break
  804. case 12:
  805. //Portuguese
  806. self.ocrLanguage_ = .portuguese
  807. break
  808. case 13:
  809. //Maori
  810. self.ocrLanguage_ = .maori
  811. break
  812. case 14:
  813. //Malay
  814. self.ocrLanguage_ = .malay
  815. break
  816. case 15:
  817. //Malay
  818. self.ocrLanguage_ = .maltese
  819. break
  820. case 16:
  821. //Dutch
  822. self.ocrLanguage_ = .dutch
  823. break
  824. case 17:
  825. //Norwegian
  826. self.ocrLanguage_ = .norwegian
  827. break
  828. case 18:
  829. //Polish
  830. self.ocrLanguage_ = .polish
  831. break
  832. case 19:
  833. //Romanian
  834. self.ocrLanguage_ = .romanian
  835. break
  836. case 20:
  837. //Slovak
  838. self.ocrLanguage_ = .slovak
  839. break
  840. case 21:
  841. //Slovenian
  842. self.ocrLanguage_ = .slovenian
  843. break
  844. case 22:
  845. //Albanian
  846. self.ocrLanguage_ = .albanian
  847. break
  848. case 23:
  849. //Swedish
  850. self.ocrLanguage_ = .swedish
  851. break
  852. case 24:
  853. //Swahili
  854. self.ocrLanguage_ = .swahili
  855. break
  856. case 25:
  857. //Tagalog
  858. self.ocrLanguage_ = .tagalog
  859. break
  860. case 26:
  861. //Turish
  862. self.ocrLanguage_ = .turish
  863. break
  864. case 27:
  865. //Uzbek
  866. self.ocrLanguage_ = .uzbek
  867. break
  868. case 28:
  869. //Vietnamese
  870. self.ocrLanguage_ = .vietnamese
  871. break
  872. case 29:
  873. //Afrikaans
  874. self.ocrLanguage_ = .afrikaans
  875. break
  876. case 30:
  877. //Azerbaijani
  878. self.ocrLanguage_ = .azerbaijani
  879. break
  880. case 31:
  881. //Bosnian
  882. self.ocrLanguage_ = .bosnian
  883. break
  884. case 32:
  885. //Czech
  886. self.ocrLanguage_ = .czech
  887. break
  888. case 33: //Welsh
  889. self.ocrLanguage_ = .welsh
  890. break
  891. case 34: //Estonian
  892. self.ocrLanguage_ = .estonian
  893. break
  894. case 35: //Irish
  895. self.ocrLanguage_ = .irish
  896. break
  897. case 36: //Croatian
  898. self.ocrLanguage_ = .croatian
  899. break
  900. case 37: //Hungarian
  901. self.ocrLanguage_ = .hungarian
  902. break
  903. case 38: //Indonesian
  904. self.ocrLanguage_ = .indonesian
  905. break
  906. case 39: //Icelandic
  907. self.ocrLanguage_ = .icelandic
  908. break
  909. case 40: //Kurdish
  910. self.ocrLanguage_ = .kurdish
  911. break
  912. case 41: //Lithuanian
  913. self.ocrLanguage_ = .lithuanian
  914. break
  915. case 42: //Latvian
  916. self.ocrLanguage_ = .latvian
  917. break
  918. case 43: //Marathi
  919. self.ocrLanguage_ = .marathi
  920. break
  921. case 44: //Nepali
  922. self.ocrLanguage_ = .nepali
  923. break
  924. case 45: //Latvia
  925. self.ocrLanguage_ = .latvian
  926. break
  927. case 46: //Bihari
  928. self.ocrLanguage_ = .bihari
  929. break
  930. case 47: //Maithili
  931. self.ocrLanguage_ = .maithili
  932. break
  933. case 48: //Angika
  934. self.ocrLanguage_ = .angika
  935. break
  936. case 49: //Bhojpuri
  937. self.ocrLanguage_ = .bhojpuri
  938. break
  939. case 50: //Magahi
  940. self.ocrLanguage_ = .magahi
  941. break
  942. case 51: //Nagpur
  943. self.ocrLanguage_ = .nagpur
  944. break
  945. case 52: //Newari
  946. self.ocrLanguage_ = .newari
  947. break
  948. case 53: //GoanKonkani
  949. self.ocrLanguage_ = .goanKonkani
  950. break
  951. case 54: //SaudiArabia
  952. self.ocrLanguage_ = .saudiArabia
  953. break
  954. default:
  955. self.ocrLanguage_ = .english
  956. break
  957. }
  958. return self.ocrLanguage_
  959. }
  960. // MARK: - Notifation Methods
  961. @objc private func km_windowWillCloseNotifation() {
  962. Task { @MainActor in
  963. if await (KMLightMemberManager.manager.canUseAdvanced() == false) {
  964. self.limitCount = true
  965. } else {
  966. self.limitCount = false
  967. }
  968. if await (KMLightMemberManager.manager.canPayFunction() == false) {
  969. #if VERSION_DMG
  970. if (self.limitCount) { // 未登录
  971. self.settingView?.tipView?.isHidden = true
  972. } else {
  973. self.settingView?.tipView?.isHidden = false
  974. }
  975. #else
  976. self.settingView?.tipView?.isHidden = false
  977. #endif
  978. } else {
  979. self.limit = false
  980. }
  981. }
  982. }
  983. // MARK: Progress
  984. func showProgressWindow() {
  985. self.hiddenProgressWindow()
  986. let progress = SKProgressController()
  987. progress.window?.backgroundColor = NSColor.km_init(hex: "#36383B")
  988. progress.window?.contentView?.wantsLayer = true
  989. progress.window?.contentView?.layer?.backgroundColor = NSColor.km_init(hex: "#36383B").cgColor
  990. progress.progressField.textColor = NSColor.white
  991. progress.message = NSLocalizedString("Converting...", comment: "")
  992. progress.closeBlock = { [weak self] in
  993. if let convert = self?.convert {
  994. KMPDFConvertManager.defaultManager.cancel(convert: convert)
  995. }
  996. }
  997. self.progressController = progress
  998. self.window?.beginSheet(progress.window!)
  999. }
  1000. func hiddenProgressWindow() {
  1001. if (self.progressController != nil) {
  1002. self.window?.endSheet((self.progressController?.window)!)
  1003. self.progressController = nil
  1004. }
  1005. }
  1006. func getCurrentLanguage() -> String {
  1007. // return Bundle.main.preferredLocalizations.first!
  1008. let array: [String] = UserDefaults.standard.object(forKey: "AppleLanguages") as! [String]
  1009. return array.first!
  1010. }
  1011. /// 存储用户的选择的语言
  1012. func saveLanugageSelectedIndex(index: Int) {
  1013. UserDefaults.standard.setValue("\(index)", forKey: kKMConvertLanugageSelectedIndex)
  1014. UserDefaults.standard.synchronize()
  1015. }
  1016. private func _clearData() {
  1017. if let _ = self.oriDocumentUrl {
  1018. // if let data = self.prePDFView?.document.documentURL.path, FileManager.default.fileExists(atPath: data) {
  1019. // try?FileManager.default.removeItem(atPath: data)
  1020. // }
  1021. }
  1022. }
  1023. }
  1024. extension KMConvertBaseWindowController: NSTextFieldDelegate {
  1025. func controlTextDidChange(_ obj: Notification) {
  1026. if numberTextField.isEqual(to: obj.object) {
  1027. let textField: NSTextField = obj.object as! NSTextField
  1028. var value: String = ""
  1029. for ch in textField.stringValue {
  1030. if ch != "0" && ch != "1" && ch != "2" && ch != "3" && ch != "4" && ch != "5" && ch != "6" && ch != "7" && ch != "8" && ch != "9" {
  1031. } else {
  1032. if value.isEmpty && ch == "0" {
  1033. } else {
  1034. value.append(ch)
  1035. }
  1036. }
  1037. }
  1038. if value.isEmpty {
  1039. value.append("1")
  1040. } else {
  1041. // if Int(value)! <= 0 {
  1042. // value = "1"
  1043. // } else if Int(value)! > self.prePDFView!.document!.pageCount {
  1044. // let number: Int = Int(self.prePDFView!.document!.pageCount)
  1045. // value = "\(number)"
  1046. // }
  1047. }
  1048. numberTextField.stringValue = value
  1049. }
  1050. }
  1051. func controlTextDidEndEditing(_ obj: Notification) {
  1052. if numberTextField.isEqual(to: obj.object) {
  1053. if (numberTextField.stringValue.isEmpty) {
  1054. numberTextField.stringValue = "1"
  1055. // self.prePDFView?.go(toPageIndex: 0, animated: false)
  1056. self.currentPageIndex = 1
  1057. self.updateBackAndNextButtonState()
  1058. return
  1059. }
  1060. let number: Int = Int(numberTextField.stringValue)!
  1061. // if number > 0 && number <= self.prePDFView!.document!.pageCount {
  1062. // guard let page = self.prePDFView?.document?.page(at: UInt(number-1)) else {
  1063. // return
  1064. // }
  1065. // self.prePDFView?.go(to: page)
  1066. // currentPageIndex = number
  1067. //
  1068. // self.updateBackAndNextButtonState()
  1069. // }
  1070. }
  1071. }
  1072. }
  1073. let kKMConvertCountKey = "KMConvertCountKey"
  1074. let kKMConvertCountMax = 2
  1075. extension KMConvertBaseWindowController {
  1076. func convertCountIsReach() -> Bool {
  1077. #if VERSION_DMG
  1078. return false
  1079. #else
  1080. let cnt = self.getConvertCount()
  1081. if (cnt >= kKMConvertCountMax) {
  1082. return true
  1083. }
  1084. return false
  1085. #endif
  1086. }
  1087. func recordConvertCount() {
  1088. let cnt = self.getConvertCount()
  1089. UserDefaults.standard.setValue(cnt+1, forKey: kKMConvertCountKey)
  1090. UserDefaults.standard.synchronize()
  1091. }
  1092. func clearConvertRecordCount() {
  1093. UserDefaults.standard.setValue(0, forKey: kKMConvertCountKey)
  1094. UserDefaults.standard.synchronize()
  1095. }
  1096. private func getConvertCount() -> Int {
  1097. return UserDefaults.standard.integer(forKey: kKMConvertCountKey)
  1098. }
  1099. }