KMMainDocument.swift 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741
  1. //
  2. // KMMainDocument.swift
  3. // PDF Master
  4. //
  5. // Created by wanjun on 2022/12/6.
  6. //
  7. import Cocoa
  8. typealias KMMainDocumentCloudUploadHanddler = (@escaping(Bool, String)->()) -> ()
  9. @objcMembers class KMMainDocument: CTTabContents {
  10. var mainViewController: KMMainViewController?
  11. var homeWindowController: KMHomeWindowController?
  12. var homeViewController: KMHomeViewController?
  13. var isNewCreated: Bool = false
  14. var closedByUserGestureFlag: Bool = false // 标记 closedByUserGesture 这个状态需要延后存储(如果需要)
  15. var cloud: Bool = false
  16. var cloudUploadHanddler: KMMainDocumentCloudUploadHanddler?
  17. var isUnlockFromKeychain: Bool = false
  18. private var _saveAsing = false
  19. private var _saveToURL: URL?
  20. var saveToURL: URL? {
  21. get {
  22. return self._saveToURL
  23. }
  24. }
  25. weak var watermarkSaveDelegate: AnyObject?
  26. private var _trackEvents = IndexSet()
  27. override func save(to url: URL, ofType typeName: String, for saveOperation: NSDocument.SaveOperationType, delegate: Any?, didSave didSaveSelector: Selector?, contextInfo: UnsafeMutableRawPointer?) {
  28. if (self.isNewCreated) {
  29. // if let data = self.mainViewController, !data.isPDFDocumentEdited && !data.needSave && !self.isDocumentEdited {
  30. self._km_save(to: url, ofType: typeName, for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  31. return
  32. // }
  33. }
  34. if (!self.needSaveWatermark()) {
  35. self._km_save(to: url, ofType: typeName, for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  36. return
  37. }
  38. var openAccessoryView = self.watermarkSaveDelegate != nil
  39. if (openAccessoryView) {
  40. if let _browser = self.watermarkSaveDelegate as? KMBrowser, _browser.isCloseAllTabViewItem {
  41. openAccessoryView = false
  42. }
  43. }
  44. self._km_saveForWatermark(openAccessoryView: openAccessoryView) { [unowned self] in
  45. self.trackEvents()
  46. } callback: { [unowned self] needSave, params in
  47. if (needSave) {
  48. self._km_save(to: url, ofType: typeName, for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  49. } else { // 水印保存
  50. if (self.watermarkSaveDelegate == nil) {
  51. if let data = params.first as? KMResult, data == .cancel {
  52. if let shouldClose = params.last as? Bool, shouldClose {
  53. DispatchQueue.main.async {
  54. self.mainViewController?.browserWindowController?.browser.windowDidBeginToClose()
  55. }
  56. }
  57. } else {
  58. DispatchQueue.main.async {
  59. self.mainViewController?.browserWindowController?.browser.windowDidBeginToClose()
  60. }
  61. }
  62. return
  63. }
  64. if let data = params.first as? KMResult, data == .cancel {
  65. if var shouldClose = params.last as? Bool {
  66. if let _browser = self.watermarkSaveDelegate as? KMBrowser, _browser.isCloseAllTabViewItem {
  67. shouldClose = true
  68. }
  69. (self.watermarkSaveDelegate as? KMBrowser)?.document(self, shouldClose: shouldClose, contextInfo: nil)
  70. }
  71. } else {
  72. (self.watermarkSaveDelegate as? KMBrowser)?.document(self, shouldClose: true, contextInfo: nil)
  73. }
  74. self.watermarkSaveDelegate = nil
  75. }
  76. }
  77. }
  78. override func makeWindowControllers() {
  79. // Returns the storyboard that contains your document window.
  80. if ((self.fileURL?.path) != nil) {
  81. if !self.fileURL!.path.isPDFValid() {
  82. let alert = NSAlert()
  83. alert.alertStyle = .critical
  84. alert.messageText = NSLocalizedString("An error occurred while opening this document. The file is damaged and could not be repaired.", comment: "")
  85. alert.runModal()
  86. return
  87. }
  88. }
  89. let mainWindow = NSApp.mainWindow
  90. var currentWindowController: KMBrowserWindowController?
  91. if mainWindow != nil {
  92. let windowController = mainWindow!.windowController
  93. if windowController is KMBrowserWindowController {
  94. currentWindowController = (windowController as! KMBrowserWindowController)
  95. } else {
  96. for window in NSApp.windows {
  97. let windowController = window.windowController
  98. if windowController is KMBrowserWindowController {
  99. currentWindowController = (windowController as! KMBrowserWindowController)
  100. break
  101. }
  102. }
  103. }
  104. } else {
  105. for window in NSApp.windows {
  106. let windowController = window.windowController
  107. if windowController is KMBrowserWindowController {
  108. currentWindowController = (windowController as! KMBrowserWindowController)
  109. break
  110. }
  111. }
  112. }
  113. if (currentWindowController == nil) && (self.fileURL != nil) {
  114. let browser = KMBrowser.init() as KMBrowser
  115. browser.addHomeTabContents()
  116. browser.windowController = KMBrowserWindowController.init(browser: browser)
  117. currentWindowController = browser.windowController as? KMBrowserWindowController
  118. }
  119. mainViewController = KMMainViewController.init()
  120. mainViewController?.myDocument = self
  121. if ((self.fileURL?.path) != nil) {
  122. let pdfDocument = CPDFDocument.init(url: URL(fileURLWithPath: self.fileURL!.path))
  123. mainViewController?.document = pdfDocument
  124. }
  125. self.view = mainViewController?.view
  126. if currentWindowController != nil {
  127. if currentWindowController?.browser != nil {
  128. // currentWindowController?.browser.add(self, at: Int32()-1, inForeground: true)
  129. // self.addWindowController(currentWindowController!)
  130. // mainViewController.browserWindowController = currentWindowController
  131. let activeBrowser = (currentWindowController?.browser.activeTabContents())! as CTTabContents
  132. let activeIndex = Int((currentWindowController?.browser.activeTabIndex())!)
  133. self.addWindowController(currentWindowController!)
  134. self.mainViewController?.browserWindowController = currentWindowController
  135. let ishome = activeBrowser.isHome as Bool
  136. let isfirstTab = (activeIndex == 0)
  137. if ishome && !isfirstTab {
  138. // 替换 document
  139. currentWindowController?.browser.replaceTabContents(at: Int32(activeIndex), with: self)
  140. // 刷新标签
  141. currentWindowController?.browser.updateTabState(at: Int32(activeIndex))
  142. // 刷新 home icon
  143. if let tabStripController = currentWindowController?.tabStripController {
  144. if let view = tabStripController.view(at: UInt(activeIndex)) as? CTTabView {
  145. view.controller().isHome = self.isHome
  146. view.controller().isNewTab = self.isNewTab
  147. view.controller().updateUI()
  148. }
  149. }
  150. } else {
  151. currentWindowController?.browser.add(self, at: Int32()-1, inForeground: true)
  152. }
  153. }
  154. }
  155. }
  156. override func showWindows() {
  157. super.showWindows()
  158. self.setDataFromTmpData()
  159. }
  160. override func windowControllerDidLoadNib(_ aController: NSWindowController) {
  161. super.windowControllerDidLoadNib(aController)
  162. self.setDataFromTmpData()
  163. }
  164. override func save(to url: URL, ofType typeName: String, for saveOperation: NSDocument.SaveOperationType) async throws {
  165. do {
  166. try await super.save(to: url, ofType: typeName, for: saveOperation)
  167. } catch let outError {
  168. Swift.print(outError)
  169. }
  170. }
  171. override func write(to url: URL, ofType typeName: String, for saveOperation: NSDocument.SaveOperationType, originalContentsURL absoluteOriginalContentsURL: URL?) throws {
  172. try self._km_write(to: url, ofType: typeName, for: saveOperation, originalContentsURL: absoluteOriginalContentsURL)
  173. }
  174. override func canClose(withDelegate delegate: Any, shouldClose shouldCloseSelector: Selector?, contextInfo: UnsafeMutableRawPointer?) {
  175. let isPrompt = KMPreferenceManager.shared.closeFileIsPrompt()
  176. if (isPrompt) {
  177. super.canClose(withDelegate: delegate, shouldClose: shouldCloseSelector, contextInfo: contextInfo)
  178. return
  179. }
  180. if (self.isNewCreated) {
  181. self.save(nil)
  182. } else if (self.isDocumentEdited) {
  183. self.save(nil)
  184. } else if (mainViewController != nil) {
  185. if self.mainViewController!.isPDFDocumentEdited || self.mainViewController!.needSave {
  186. self.save(nil)
  187. }
  188. }
  189. super.canClose(withDelegate: delegate, shouldClose: shouldCloseSelector, contextInfo: contextInfo)
  190. }
  191. override func saveAs(_ sender: Any?) {
  192. if (!self.needSaveWatermark()) {
  193. self._km_saveAs(sender)
  194. return
  195. }
  196. self._km_saveForWatermark { [unowned self] needSave, _ in
  197. if (needSave) {
  198. self._km_saveAs(sender)
  199. }
  200. }
  201. }
  202. override func runModalSavePanel(for saveOperation: NSDocument.SaveOperationType, delegate: Any?, didSave didSaveSelector: Selector?, contextInfo: UnsafeMutableRawPointer?) {
  203. if (self.isNewCreated) {
  204. // if let data = self.mainViewController, !data.isPDFDocumentEdited && !data.needSave && !self.isDocumentEdited {
  205. self._km_runModalSavePanel(for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  206. return
  207. // }
  208. }
  209. if (!self.needSaveWatermark()) {
  210. self._km_runModalSavePanel(for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  211. return
  212. }
  213. self._km_saveForWatermark { [unowned self] needSave, _ in
  214. if (needSave) {
  215. self._km_runModalSavePanel(for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  216. }
  217. }
  218. }
  219. override func save(_ sender: Any?) {
  220. if (!self.needSaveWatermark()) {
  221. self._km_save(sender)
  222. return
  223. }
  224. self._km_saveForWatermark { [unowned self] in
  225. self.trackEvents()
  226. } callback: { [unowned self] needSave, _ in
  227. if (needSave) {
  228. self._km_save(sender)
  229. }
  230. }
  231. }
  232. override func read(from absoluteURL: URL, ofType typeName: String) throws {
  233. do {
  234. try super.read(from: absoluteURL, ofType: typeName)
  235. updateChangeCount(.changeCleared)
  236. } catch let outError {
  237. Swift.print(outError)
  238. }
  239. }
  240. override func read(from data: Data, ofType typeName: String) throws {
  241. // Insert code here to read your document from the given data of the specified type, throwing an error in case of failure.
  242. // Alternatively, you could remove this method and override read(from:ofType:) instead. If you do, you should also override isEntireFileLoaded to return false if the contents are lazily loaded.
  243. let pdfDocument = CPDFDocument.init(data: data)
  244. if pdfDocument == nil {
  245. throw NSError(domain: NSOSStatusErrorDomain, code: unimpErr, userInfo: nil)
  246. }
  247. }
  248. // MARK: Autosaving
  249. override func close() {
  250. if self.isActive {
  251. if browser != nil {
  252. var activeIndex = 0
  253. let dex = browser.index(of: self)
  254. if dex == browser.tabCount() - 1 {
  255. activeIndex = Int(browser.tabCount()-2)
  256. } else {
  257. activeIndex = Int(dex + 1)
  258. }
  259. let activeContents = browser.tabContents(at: Int32(activeIndex))
  260. activeContents?.addWindowController(browser.windowController)
  261. }
  262. }
  263. super.close()
  264. }
  265. // MARK: init
  266. override init() {
  267. super.init()
  268. // Add your subclass-specific initialization here.
  269. NotificationCenter.default.addObserver(self, selector: #selector(pdfChangedNotification(_:)), name: NSNotification.Name.init(rawValue: "CPDFListViewAnnotationsAttributeHasChangeNotification"), object: nil)
  270. // NotificationCenter.default.addObserver(self, selector: #selector(pdfChangedNotification(_:)), name: NSNotification.Name.init(rawValue: "CPDFViewDocumentChangedNotification"), object: nil)
  271. // NotificationCenter.default.addObserver(self, selector: #selector(pdfChangedNotification(_:)), name: NSNotification.Name.init(rawValue: "CPDFViewPageChangedNotification"), object: nil)
  272. NotificationCenter.default.addObserver(self, selector: #selector(pdfChangedNotification(_:)), name: NSNotification.Name.init(rawValue: "CPDFListViewDidAddAnnotationNotification"), object: nil)
  273. NotificationCenter.default.addObserver(self, selector: #selector(pdfChangedNotification(_:)), name: NSNotification.Name.init(rawValue: "CPDFListViewDidRemoveAnnotationNotification"), object: nil)
  274. }
  275. override init?(baseTabContents baseContents: CTTabContents?) {
  276. super.init(baseTabContents: baseContents)
  277. if isHome {
  278. homeViewController = KMHomeViewController.init()
  279. homeViewController?.myDocument = self
  280. self.view = homeViewController?.view
  281. }
  282. }
  283. // MARK: Handling User Actions
  284. override var title: String? {
  285. get {
  286. if isHome {
  287. if (self.isNewTab) {
  288. return NSLocalizedString("New Tab", comment: "")
  289. } else {
  290. return NSLocalizedString("Home", comment: "")
  291. }
  292. } else {
  293. return fileURL?.lastPathComponent
  294. }
  295. }
  296. set {
  297. super.title = newValue
  298. }
  299. }
  300. func needSaveWatermark() -> Bool {
  301. if let need = self.mainViewController?.saveWatermarkFlag {
  302. return need
  303. }
  304. return false
  305. }
  306. // MARK: Private Methods
  307. func pdfChangedNotification(_ notification: Notification) -> Void {
  308. if !isHome {
  309. let mainViewController = mainViewController
  310. var document: CPDFDocument!
  311. let dic = notification.object as? NSDictionary
  312. if dic?["object"] is CPDFAnnotation {
  313. let annotation : CPDFAnnotation = dic?["object"] as? CPDFAnnotation ?? CPDFAnnotation()
  314. document = annotation.page.document
  315. } else if dic?["object"] is CPDFListView {
  316. let pdflistView = notification.object as? CPDFListView
  317. document = pdflistView?.document
  318. }
  319. if mainViewController != nil {
  320. if document == mainViewController!.document {
  321. updateChangeCount(.changeDone)
  322. }
  323. }
  324. }
  325. }
  326. override func updateChangeCount(_ change: NSDocument.ChangeType) {
  327. let mainWindow = NSApp.mainWindow
  328. var currentWindowController: KMBrowserWindowController?
  329. if mainWindow != nil {
  330. let windowController = mainWindow!.windowController
  331. if windowController is KMBrowserWindowController {
  332. currentWindowController = (windowController as! KMBrowserWindowController)
  333. } else {
  334. for window in NSApp.windows {
  335. let windowController = window.windowController
  336. if windowController is KMBrowserWindowController {
  337. currentWindowController = (windowController as! KMBrowserWindowController)
  338. break
  339. }
  340. }
  341. }
  342. } else {
  343. for window in NSApp.windows {
  344. let windowController = window.windowController
  345. if windowController is KMBrowserWindowController {
  346. currentWindowController = (windowController as! KMBrowserWindowController)
  347. break
  348. }
  349. }
  350. }
  351. if currentWindowController != nil {
  352. if currentWindowController?.browser != nil {
  353. let activeBrowser = (currentWindowController?.browser.activeTabContents())! as CTTabContents
  354. let activeIndex = Int((currentWindowController?.browser.activeTabIndex())!)
  355. if self == activeBrowser {
  356. super.updateChangeCount(change)
  357. return
  358. }
  359. }
  360. }
  361. super.updateChangeCount(.changeCleared)
  362. }
  363. func uploadToCloud(_ callback: (@escaping(Bool, String)->())) {
  364. guard let handdler = self.cloudUploadHanddler else {
  365. return
  366. }
  367. handdler(callback)
  368. }
  369. func isPDFDocument() -> Bool {
  370. return true
  371. }
  372. func setDataFromTmpData() {
  373. guard let _document = self.mainViewController?.document else {
  374. return
  375. }
  376. // self.tryToUnlockDocument(document!)
  377. if (_document.permissionsStatus != .owner) {
  378. var password: NSString? = nil
  379. let fileId = self.fileId(for: _document)
  380. if (fileId.isEmpty) {
  381. return
  382. }
  383. self.getPassword(&password, fileId: fileId)
  384. if (password != nil) {
  385. self.isUnlockFromKeychain = true
  386. // document.unlock(withPassword: password! as String)
  387. self.mainViewController?.password = password as String?
  388. }
  389. }
  390. }
  391. func tryToUnlockDocument(_ document: CPDFDocument) {
  392. if (document.permissionsStatus != .owner) {
  393. var password: NSString? = nil
  394. let fileId = self.fileId(for: document)
  395. if (fileId.isEmpty) {
  396. return
  397. }
  398. self.getPassword(&password, fileId: fileId)
  399. if (password != nil) {
  400. self.isUnlockFromKeychain = true
  401. document.unlock(withPassword: password! as String)
  402. }
  403. }
  404. }
  405. func km_updateChangeCount(_ change: NSDocument.ChangeType) {
  406. super.updateChangeCount(change)
  407. }
  408. func trackEvents() {
  409. km_synchronized(self) {
  410. for i in self._trackEvents {
  411. if let type = KMSubscribeWaterMarkType(rawValue: i) {
  412. KMTools.trackEvent(type: type)
  413. }
  414. }
  415. }
  416. self.clearTrackEvents()
  417. }
  418. func recordTrackEvent(type: KMSubscribeWaterMarkType) {
  419. if (type == .none) {
  420. return
  421. }
  422. km_synchronized(self) {
  423. self._trackEvents.insert(type.rawValue)
  424. }
  425. }
  426. func clearTrackEvents() {
  427. km_synchronized(self) {
  428. self._trackEvents.removeAll()
  429. }
  430. }
  431. // MARK: - Private Methods
  432. private func _km_write(to url: URL, ofType typeName: String, for saveOperation: NSDocument.SaveOperationType, originalContentsURL absoluteOriginalContentsURL: URL?) throws {
  433. var success = true
  434. if !self.isHome {
  435. if mainViewController != nil {
  436. if mainViewController?.document != nil {
  437. self.mainViewController?.commitEditingIfNeed()
  438. // if mainViewController!.document!.isEncrypted {
  439. // success = mainViewController!.document!.write(to: url)
  440. // } else {
  441. if (mainViewController!.needSave) {
  442. if let options = self.mainViewController?.secureOptions, !options.isEmpty {
  443. success = self.mainViewController!.document!.write(to: url, withOptions: options)
  444. } else if let flag = self.mainViewController?.removeSecureFlag, flag {
  445. success = self.mainViewController!.document!.writeDecrypt(to: url)
  446. } else {
  447. success = mainViewController!.document!.write(to: url)
  448. }
  449. } else {
  450. success = mainViewController!.document!.write(to: url)
  451. }
  452. // }
  453. self.mainViewController?.needSave = false
  454. self.mainViewController?.clearSecureOptions()
  455. self.mainViewController?.clearRemoveSecureFlag()
  456. }
  457. }
  458. } else {
  459. success = false
  460. }
  461. if (success && self._saveAsing) {
  462. if let tabView = self.browser?.windowController?.tabStripController?.activeTabView() as? CTTabView {
  463. tabView.controller()?.title = url.lastPathComponent
  464. }
  465. self._saveAsing = false
  466. }
  467. if success && isNewCreated && NSDocument.SaveOperationType.saveAsOperation == saveOperation {
  468. isNewCreated = false
  469. }
  470. }
  471. private func _km_saveForWatermark(openAccessoryView: Bool = true, subscribeDidClick: (()->Void)? = nil, callback:@escaping (_ needSave: Bool, _ param: Any...)->Void) {
  472. Task { @MainActor in
  473. if await (KMLightMemberManager.manager.canPayFunction() == false) {
  474. let _ = KMSubscribeWaterMarkWindowController.show(window: NSApp.mainWindow!) {
  475. if let _callback = subscribeDidClick {
  476. _callback()
  477. }
  478. } completion: { isSubscribeSuccess, isWaterMarkExport, isClose in
  479. if (isClose) {
  480. callback(false, KMResult.cancel, false)
  481. return
  482. }
  483. if (isSubscribeSuccess) {
  484. callback(true)
  485. return
  486. }
  487. if (isWaterMarkExport) {
  488. guard let _document = self.mainViewController?.document else {
  489. callback(false, KMResult.failure)
  490. return
  491. }
  492. // 提交文本编辑的内容
  493. self.mainViewController?.commitEditingIfNeed()
  494. DispatchQueue.main.async {
  495. NSPanel.savePanel(NSApp.mainWindow!, openAccessoryView, panel:{ panel in
  496. if (!self.isNewCreated) {
  497. panel.directoryURL = _document.documentURL.deletingLastPathComponent()
  498. }
  499. panel.nameFieldStringValue = _document.documentURL.lastPathComponent
  500. }) { response, url, isOpen in
  501. if (response == .cancel) {
  502. callback(false, KMResult.cancel, true)
  503. return
  504. }
  505. guard let _url = KMTools.saveWatermarkDocument(document: _document, to: url!, secureOptions: self.mainViewController?.secureOptions, removePWD: self.mainViewController!.removeSecureFlag) else {
  506. callback(false, KMResult.failure)
  507. return
  508. }
  509. callback(false, KMResult.success)
  510. if (isOpen) {
  511. NSDocumentController.shared.km_safe_openDocument(withContentsOf: _url, display: true) { _, _, _ in
  512. }
  513. } else {
  514. NSWorkspace.shared.activateFileViewerSelecting([_url])
  515. }
  516. }
  517. }
  518. return
  519. }
  520. callback(false)
  521. }
  522. return
  523. }
  524. callback(true)
  525. }
  526. }
  527. private func _km_save(_ sender: Any?) {
  528. super.save(sender)
  529. }
  530. private func _km_save(to url: URL, ofType typeName: String, for saveOperation: NSDocument.SaveOperationType, delegate: Any?, didSave didSaveSelector: Selector?, contextInfo: UnsafeMutableRawPointer?) {
  531. self._saveToURL = url
  532. super.save(to: url, ofType: typeName, for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  533. }
  534. private func _km_saveAs(_ sender: Any?) {
  535. super.saveAs(sender)
  536. self._saveAsing = true
  537. }
  538. private func _km_runModalSavePanel(for saveOperation: NSDocument.SaveOperationType, delegate: Any?, didSave didSaveSelector: Selector?, contextInfo: UnsafeMutableRawPointer?) {
  539. super.runModalSavePanel(for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  540. }
  541. }
  542. extension NSDocument {
  543. @objc class func isDamage(url: URL) -> Bool {
  544. // 文件路径是否存在
  545. if (FileManager.default.fileExists(atPath: url.path) == false) {
  546. return true
  547. }
  548. /// PDF 格式文件
  549. if (url.pathExtension.lowercased() == "pdf") {
  550. let document = PDFDocument(url: url)
  551. if (document == nil) {
  552. return true
  553. }
  554. if (document!.isLocked) { // 加锁文件不在这里判断
  555. return false
  556. }
  557. if (document!.pageCount <= 0) {
  558. return true
  559. }
  560. return false
  561. }
  562. // 支持的图片格式
  563. let imageExts = ["jpg","cur","bmp","jpeg","gif","png","tiff","tif","ico","icns","tga","psd","eps","hdr","jp2","jpc","pict","sgi","heic"]
  564. let isImage = imageExts.contains(url.pathExtension.lowercased())
  565. if (isImage == false) { // 其他格式目前返回没损坏,后续再补充(如果有需求)
  566. return false
  567. }
  568. // 图片格式
  569. let image = NSImage(contentsOf: url)
  570. let data = image?.tiffRepresentation
  571. if (data == nil) {
  572. return true
  573. }
  574. let imageRep = NSBitmapImageRep(data: data!)
  575. imageRep!.size = image!.size
  576. var imageData: NSData?
  577. if (url.pathExtension.lowercased() == "png") {
  578. imageData = imageRep?.representation(using: .png, properties: [:]) as NSData?
  579. } else {
  580. imageData = imageRep?.representation(using: .jpeg, properties: [:]) as NSData?
  581. }
  582. if (imageData == nil) {
  583. return true
  584. }
  585. let path = NSSearchPathForDirectoriesInDomains(.applicationSupportDirectory, .userDomainMask, true).last?.stringByAppendingPathComponent(Bundle.main.bundleIdentifier!)
  586. if (FileManager.default.fileExists(atPath: path!) == false) {
  587. try?FileManager.default.createDirectory(atPath: path!, withIntermediateDirectories: false)
  588. }
  589. var tagString: String = ""
  590. let dateFormatter = DateFormatter()
  591. dateFormatter.dateFormat = "yyMMddHHmmss"
  592. tagString.append(dateFormatter.string(from: Date()))
  593. tagString = tagString.appendingFormat("%04d", arc4random()%10000)
  594. let filePath = path?.appending("/\(tagString).png")
  595. if (imageData!.write(toFile: filePath!, atomically: true) == false) {
  596. return true
  597. }
  598. // 删除临时图片
  599. try?FileManager.default.removeItem(atPath: filePath!)
  600. return false
  601. }
  602. @objc class func isDamage(url: URL, needAlertIfDamage need: Bool) -> Bool {
  603. let result = self.isDamage(url: url)
  604. if (result == false) {
  605. return false
  606. }
  607. if (need == false) {
  608. return true
  609. }
  610. let alert = NSAlert()
  611. alert.messageText = NSLocalizedString("An error occurred while opening this document. The file is damaged and could not be repaired.", comment: "")
  612. alert.runModal()
  613. return true
  614. }
  615. }
  616. // MARK: -
  617. // MARK: 保存密码
  618. extension NSDocument {
  619. func savePasswordInKeychain(_ password: String, _ document: CPDFDocument) {
  620. if (document.isLocked || password.isEmpty) {
  621. return
  622. }
  623. let fileId = self.fileId(for: document)
  624. if (fileId.isEmpty) {
  625. return
  626. }
  627. // let status: SKPasswordStatus =
  628. let label = "PDF Master: \(self.displayName!)"
  629. SKKeychain.setPassword(password, item: nil, forService: self.passwordServiceName(), account: fileId, label: label, comment: self.fileURL?.path)
  630. }
  631. func getPassword(_ password: AutoreleasingUnsafeMutablePointer<NSString?>, fileId: String) {
  632. let status = SKKeychain.getPassword(password, item: nil, forService: self.passwordServiceName(), account: fileId)
  633. // if (status == .found) {
  634. // }
  635. }
  636. fileprivate func fileId(for document: CPDFDocument) -> String {
  637. return "\(document.documentURL.path.hash)"
  638. }
  639. private func passwordServiceName() -> String {
  640. return "PDF Master password"
  641. }
  642. }