KMConvertBaseWindowController.swift 45 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227
  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. self.pdfPreView?.reloadUI()
  118. }
  119. override func initSubViews() {
  120. super.initSubViews()
  121. let view = self.initSettingView()
  122. // view?.frame = self.rightScrollView.bounds
  123. // view?.autoresizingMask = [.width, .height]
  124. self.settingView = view
  125. self.rightScrollView.documentView = view
  126. self.rightScrollView.verticalScrollElasticity = .none
  127. self.rightScrollView.horizontalScrollElasticity = .none
  128. view?.km_add_inset_constraint()
  129. view?.currentLanguage = self.getCurrentLanguage()
  130. view?.lanugageDidSelected = { [weak self] value, _ in
  131. guard let index = value as? Int else {
  132. return
  133. }
  134. self?.saveLanugageSelectedIndex(index: index)
  135. }
  136. view?.pageRangeDidChange = { [weak self] pageRange in
  137. self?.pageRange = pageRange
  138. if (pageRange != .custom) {
  139. self?.updatePreView(pageRange: pageRange)
  140. }
  141. }
  142. view?.pageRangeDidInputFinishCallback = { [unowned self] string in
  143. self.pageRangeString = string
  144. let array = self.findSelectPage(pageRangeString: string)
  145. if array.count == 0 {
  146. let alert = NSAlert()
  147. alert.alertStyle = .warning
  148. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  149. alert.runModal()
  150. self.settingView?.pageRangeItemView?.pageSizeVC?.stringValue = ""
  151. } else {
  152. self.updatePreView(pageRange: .custom, pageString: string)
  153. }
  154. }
  155. view?.tipView?.itemClick = { [unowned self] _, _ in
  156. #if VERSION_DMG
  157. KMTools.openPurchaseProductWebsite()
  158. #else
  159. Task { @MainActor in
  160. if await (KMLightMemberManager.manager.canPayFunction() == false) {
  161. let _ = KMSubscribeWaterMarkWindowController.show(window: self.window!) { isSubscribeSuccess, isWaterMarkExport, isClose in
  162. if (isSubscribeSuccess) {
  163. self.refreshUI()
  164. }
  165. }
  166. }
  167. }
  168. #endif
  169. }
  170. self.settingView?.tipView?.isHidden = true
  171. Task { @MainActor in
  172. if await (KMLightMemberManager.manager.canUseAdvanced() == false) {
  173. self.limitCount = true
  174. } else {
  175. self.limitCount = false
  176. }
  177. if await (KMLightMemberManager.manager.canPayFunction() == false) {
  178. #if VERSION_DMG
  179. if (self.limitCount) { // 未登录
  180. self.settingView?.tipView?.isHidden = true
  181. } else {
  182. self.settingView?.tipView?.isHidden = false
  183. }
  184. #else
  185. self.settingView?.tipView?.isHidden = false
  186. #endif
  187. } else {
  188. self.limit = false
  189. }
  190. }
  191. self.backButtonVC = KMDesignButton.init(withType: .Image)
  192. self.backBox.contentView = self.backButtonVC!.view
  193. self.nextButtonVC = KMDesignButton.init(withType: .Image)
  194. self.nextBox.contentView = self.nextButtonVC!.view
  195. cancelButton_ = ComponentButton()
  196. cancelButton_?.properties = ComponentButtonProperty(type: .default_tertiary, size: .s, state: .normal, buttonText: KMLocalizedString("Cancel"))
  197. cancelButton_?.setTarget(self, action: #selector(cancelButtonAction))
  198. self.canelBox.contentView = self.cancelButton_
  199. convertButton_ = ComponentButton()
  200. convertBox.contentView = convertButton_
  201. batchButton_ = ComponentButton()
  202. batchButton_?.properties = ComponentButtonProperty(type: .default_tertiary, size: .s, state: .normal, buttonText: KMLocalizedString("Batch"))
  203. batchButton_?.setTarget(self, action: #selector(batchButtonAction))
  204. self.batchButton.addSubview(self.batchButton_!)
  205. self.batchButton_?.frame = self.batchButton.bounds
  206. self.batchButton_?.autoresizingMask = [.width, .height]
  207. }
  208. override func initDefaultValue() {
  209. super.initDefaultValue()
  210. self.totalNumberLabel.lineBreakMode = .byTruncatingTail
  211. // self.prePDFView?.setDisplay(.singlePage)
  212. // self.prePDFView?.layoutDocumentView()
  213. // self.prePDFView?.autoScales = true
  214. // self.prePDFView?.delegate = self
  215. self.backBox.fillColor = .clear
  216. self.backButtonVC?.target = self
  217. self.backButtonVC?.action = #selector(backButtonAction)
  218. self.backButtonVC?.image = NSImage(named: "KMImageNamePreviousPageImage")!
  219. self.backButtonVC?.image_disabled = NSImage(named: "KMImageNamePreviousPageImageDisable")!
  220. self.backButtonVC?.pagination()
  221. self.numberBox.wantsLayer = true
  222. self.numberBox.layer?.backgroundColor = NSColor.white.cgColor
  223. self.numberBox.layer?.borderWidth = 1
  224. self.numberBox.layer?.cornerRadius = 4
  225. self.numberTextField.focusRingType = .none
  226. self.numberTextField.delegate = self
  227. self.numberTextField.isBordered = false
  228. self.nextBox.fillColor = .clear
  229. self.nextButtonVC?.target = self
  230. self.nextButtonVC?.action = #selector(nextButtonAction)
  231. self.nextButtonVC?.image = NSImage(named: "KMImageNameNextPageImage")!
  232. self.nextButtonVC?.image_disabled = NSImage(named: "KMImageNameNextPageImageDisable")!
  233. self.nextButtonVC?.pagination()
  234. self.canelBox.fillColor = .clear
  235. self.cancelButton_?.keyEquivalent = KMKeyEquivalent.esc.string()
  236. self.convertBox.fillColor = .clear
  237. convertButton_?.properties = ComponentButtonProperty(type: .primary,
  238. size: .s,
  239. state: .normal,
  240. isDisable: false,
  241. buttonText: KMLocalizedString("Convert"))
  242. convertButton_?.setTarget(self, action: #selector(convertButtonAction))
  243. convertButton_?.keyEquivalent = KMKeyEquivalent.enter
  244. self.rightScrollView.hasVerticalScroller = false
  245. self.rightScrollView.hasHorizontalScroller = false
  246. self.initUIProperty()
  247. }
  248. private func initUIProperty() {
  249. self.titleLabel.textColor = NSColor.titleColor()
  250. self.titleLabel.font = .SFProTextSemiboldFont(16)
  251. self.leftBox.fillColor = NSColor.km_init(hex: "#F7F8FA")
  252. // self.prePDFView?.backgroundColor = NSColor.km_init(hex: "#F7F8FA")
  253. self.numberBox.layer?.borderColor = NSColor.km_init(hex: "#DFE1E5").cgColor
  254. self.perLabel.textColor = NSColor.titleColor()
  255. self.perLabel.font = .SFProTextRegularFont(12)
  256. self.totalNumberLabel.textColor = NSColor.titleColor()
  257. self.totalNumberLabel.font = .SFProTextRegularFont(12)
  258. // self.batchButton.setTitleColor(NSColor.buttonTitleColor())
  259. // self.batchButton.layer?.borderColor = NSColor.buttonBorderColor().cgColor
  260. // self.batchButton.font = .SFProTextRegularFont(14)
  261. }
  262. func initSettingView() -> KMConvertSettingView? {
  263. let settingView = KMConvertSettingView.createFromNib()
  264. return settingView
  265. }
  266. public func initConvert() -> KMPDFConvert {
  267. return KMPDFConvert()
  268. }
  269. func setupContentHeight(height: CGFloat, animated: Bool = false) {
  270. if (animated) {
  271. self.rightBoxHeightConst.animator().constant = height
  272. } else {
  273. self.rightBoxHeightConst.constant = height
  274. }
  275. }
  276. func refreshUI() {
  277. }
  278. @objc func backButtonAction() {
  279. // if let _ = self.prePDFView?.canGoToPreviousPage() {
  280. // self.currentPageIndex -= 1
  281. // self.prePDFView?.goToPreviousPage(nil)
  282. //
  283. // self.numberTextField.stringValue = "\(self.prePDFView!.currentPageIndex+1)"
  284. // self.updateBackAndNextButtonState()
  285. // }
  286. }
  287. @objc func nextButtonAction() {
  288. // if let _ = self.prePDFView?.canGoToNextPage() {
  289. // self.currentPageIndex += 1
  290. // self.prePDFView?.goToNextPage(nil)
  291. //
  292. // self.numberTextField.stringValue = "\(self.prePDFView!.currentPageIndex+1)"
  293. // self.updateBackAndNextButtonState()
  294. // }
  295. }
  296. @objc func batchButtonAction() {
  297. // guard let callback = itemClick else {
  298. // return
  299. // }
  300. //
  301. // callback(1)
  302. self.own_closeEndSheet()
  303. }
  304. @objc func cancelButtonAction() {
  305. self._clearData()
  306. // guard let callback = itemClick else {
  307. // return
  308. // }
  309. //
  310. // callback(2)
  311. self.own_closeEndSheet()
  312. }
  313. @objc func convertButtonAction() {
  314. self.window?.makeFirstResponder(nil)
  315. if (self.pageRange == .custom) {
  316. let array = self.findSelectPage(pageRangeString: self.pageRangeString)
  317. if (array.count <= 0) {
  318. self.window?.makeFirstResponder(self.settingView?.pageRangeItemView?.pageSizeVC?.textField)
  319. return
  320. }
  321. }
  322. var pages: [Int] = []
  323. let filePath = self.getConvertFileSavePath()
  324. if (FileManager.default.fileExists(atPath: filePath)) {
  325. try?FileManager.default.removeItem(atPath: filePath)
  326. }
  327. let document = self.pdfPreView?.pdfPreView.pdfView.document
  328. if ((document?.writeDecrypt(to: URL(fileURLWithPath: filePath)))!) {
  329. for i in 0 ..< document!.pageCount {
  330. pages.append(Int(i)+1)
  331. }
  332. }
  333. if (pages.count <= 0) {
  334. let alert = NSAlert()
  335. alert.alertStyle = .warning
  336. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  337. alert.runModal()
  338. return
  339. }
  340. Task { @MainActor in
  341. #if VERSION_DMG
  342. if (await (KMLightMemberManager.manager.canUseAdvanced() == false)) {
  343. let _ = KMComparativeTableViewController.show(window: self.window!, .convert)
  344. return
  345. }
  346. #endif
  347. // if await (KMLightMemberManager.manager.canPayFunction() == false) {
  348. // var isContinue = false
  349. // if pages.count <= 10 {
  350. // isContinue = true
  351. // }
  352. //
  353. // var limit = self.convertCountIsReach()
  354. // if (!self.limitCount) { // 登录
  355. // limit = false
  356. // }
  357. //
  358. // if (limit) {
  359. // isContinue = false
  360. // }
  361. //
  362. // let _ = KMSubscribeWaterMarkWindowController.show(window: self.window!, isContinue: isContinue, limit: limit, type: self.subscribeWaterMarkType!) { isSubscribeSuccess, isWaterMarkExport, isClose in
  363. // if (isClose) {
  364. // return
  365. // }
  366. // if (isSubscribeSuccess) {
  367. // self._convertButtonAction()
  368. // return
  369. // }
  370. // if (isWaterMarkExport && !limit) {
  371. // self._convertButtonAction(limit: true)
  372. // return
  373. // }
  374. // }
  375. // return
  376. // }
  377. self._convertButtonAction()
  378. }
  379. }
  380. func convertModelAppendParams(convert: KMPDFConvert) -> () {
  381. if let isAllowOCR = self.settingView?.ocrEnabled {
  382. convert.isAllowOCR = isAllowOCR
  383. if let idx = self.settingView?.ocrItemView?.languageIndex {
  384. // convert.ocrLanguage = KMPDFConvertManager.defaultManager.languages_asc[idx]
  385. convert.ocrLanguage = self._getOCRLanguage(idx: idx)
  386. }
  387. } else {
  388. convert.isAllowOCR = false
  389. }
  390. }
  391. func getConvertFileSavePath() -> String {
  392. var path = NSSearchPathForDirectoriesInDomains(FileManager.SearchPathDirectory.applicationSupportDirectory, FileManager.SearchPathDomainMask.userDomainMask, true).last
  393. path?.append("/")
  394. path?.append(Bundle.main.bundleIdentifier!)
  395. if (FileManager.default.fileExists(atPath: path!) == false) {
  396. try?FileManager.default.createDirectory(atPath: path!, withIntermediateDirectories: false)
  397. }
  398. path?.append("/")
  399. path?.append("convert.pdf")
  400. return path!
  401. }
  402. func isValidPagesString(pagesString: String)-> Bool {
  403. var valid = false
  404. for ch in pagesString {
  405. if ch != "0" && ch != "1" && ch != "2" && ch != "3" && ch != "4" && ch != "5" && ch != "6" && ch != "7" && ch != "8" && ch != "9" && ch != "," && ch != "-" {
  406. valid = false
  407. break
  408. } else {
  409. valid = true
  410. }
  411. }
  412. return valid
  413. }
  414. func findSelectPage(pageRangeString: String) -> ([Int]) {
  415. if !isValidPagesString(pagesString: pageRangeString) {
  416. return []
  417. }
  418. var result: [Int] = []
  419. let array = pageRangeString.components(separatedBy: ",")
  420. for string in array {
  421. if string.isEmpty {
  422. return []
  423. } else {
  424. let pages = string .components(separatedBy: "-")
  425. if pages.count > 2 {
  426. return []
  427. } else if pages.count == 1 {
  428. let page = pages[0]
  429. if page.isEmpty || Int(page)! > self.documentModel!.pageCount || Int(page)! == 0 {
  430. return []
  431. } else {
  432. var hasSame: Bool = false
  433. for i in result {
  434. if i == Int(page)! {
  435. hasSame = true
  436. return []
  437. }
  438. }
  439. if !hasSame {
  440. result.append(Int(page)!)
  441. }
  442. }
  443. } else if pages.count == 2 {
  444. let page1 = pages[0]
  445. let page2 = pages[1]
  446. if page1.isEmpty || page2.isEmpty || Int(page1)! >= Int(page2)! || Int(page2)! > self.documentModel!.pageCount || Int(page1)! == 0 {
  447. return []
  448. } else {
  449. var hasSame: Bool = false
  450. for i in Int(page1)! ... Int(page2)! {
  451. for j in result {
  452. if j == i {
  453. hasSame = true
  454. return []
  455. }
  456. }
  457. }
  458. if !hasSame {
  459. for i in Int(page1)! ... Int(page2)! {
  460. result.append(i)
  461. }
  462. }
  463. }
  464. }
  465. }
  466. }
  467. return result
  468. }
  469. // MARK: -
  470. // MARK: Private Methods
  471. private func updateBackAndNextButtonState() {
  472. var nextEnabled = false
  473. // if let result = self.prePDFView?.canGoToNextPage() {
  474. // nextEnabled = result
  475. // }
  476. var backEnabled = false
  477. // if let result = self.prePDFView?.canGoToPreviousPage() {
  478. // backEnabled = result
  479. // }
  480. if (Thread.isMainThread) {
  481. self.backButtonVC?.state = backEnabled ? .Norm : .Disabled
  482. self.nextButtonVC?.state = nextEnabled ? .Norm : .Disabled
  483. } else {
  484. DispatchQueue.main.async {
  485. self.backButtonVC?.state = backEnabled ? .Norm : .Disabled
  486. self.nextButtonVC?.state = nextEnabled ? .Norm : .Disabled
  487. }
  488. }
  489. }
  490. internal func updatePreView(pageRange: KMPageRange, pageString: String = "") {
  491. let pages = self.getPages(pageRange, pageString: pageString)
  492. if (pages == nil || pages!.isEmpty) {
  493. let alert = NSAlert()
  494. alert.alertStyle = .warning
  495. alert.messageText = NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: "")
  496. alert.runModal()
  497. return
  498. }
  499. var indexs = IndexSet()
  500. let document = self.pdfPreView?.pdfPreView.pdfView.document
  501. for i in 0 ..< document!.pageCount {
  502. indexs.insert(IndexSet.Element(i))
  503. }
  504. document?.removePage(at: indexs)
  505. #if DEBUG
  506. let pageIndexs = self.getPageIndexs(pageRange, pageString: pageString)
  507. if (pageIndexs != nil) {
  508. for i in pageIndexs! {
  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. document?.insertPageObject(page, at: document?.pageCount ?? 0)
  516. }
  517. DispatchQueue.main.async {
  518. let prePDFView = self.pdfPreView?.pdfPreView.pdfView
  519. prePDFView?.layoutDocumentView()
  520. prePDFView?.setNeedsDisplayForVisiblePages()
  521. prePDFView?.goToFirstPage(nil)
  522. let number: Int = Int(prePDFView!.document!.pageCount)
  523. // self.totalNumberLabel.stringValue = "\(number)"
  524. self.currentPageIndex = 1
  525. // self.numberTextField.stringValue = "1"
  526. self.updateBackAndNextButtonState()
  527. self.pdfPreView?.reloadUI()
  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.pdfPreView?.pdfPreView.pdfView.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. }