KMMainDocument.swift 65 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506
  1. //
  2. // KMMainDocument.swift
  3. // PDF Reader Pro
  4. //
  5. // Created by wanjun on 2022/12/6.
  6. //
  7. import Cocoa
  8. import CoreFoundation
  9. @objc enum KMArchiveMask: Int {
  10. case diskImage = 1
  11. case email = 2
  12. }
  13. @objc enum KMExportOption: Int {
  14. case `default` = 0
  15. case withoutNotes
  16. case withEmbeddedNotes
  17. }
  18. typealias KMMainDocumentCloudUploadHanddler = (@escaping(Bool, String)->()) -> ()
  19. @objcMembers class KMMainDocument: CTTabContents {
  20. struct MDFlags {
  21. var exportOption: UInt32 // assuming this is a 2-bit field, change to appropriate data type
  22. var exportUsingPanel: UInt32 // assuming this is a 1-bit field, change to appropriate data type
  23. var gettingFileType: UInt32 // assuming this is a 1-bit field, change to appropriate data type
  24. var convertingNotes: UInt32 // assuming this is a 1-bit field, change to appropriate data type
  25. var needsPasswordToConvert: UInt32 // assuming this is a 1-bit field, change to appropriate data type
  26. }
  27. static let kLastExportedTypeKey = "SKLastExportedType"
  28. static let kLastExportedOptionKey = "SKLastExportedOption"
  29. static let kBundleDataFilename = "data"
  30. var mainViewController: KMMainViewController?
  31. var homeViewController: KMNHomeViewController?
  32. var bookmarkSheetController: KMBookmarkSheetController?
  33. var bookmarkController: KMBookmarkController?
  34. var isNewCreated: Bool = false
  35. var closedByUserGestureFlag: Bool = false // 标记 closedByUserGesture 这个状态需要延后存储(如果需要)
  36. var cloud: Bool = false
  37. var cloudUploadHanddler: KMMainDocumentCloudUploadHanddler?
  38. var isUnlockFromKeychain: Bool = false
  39. private var _saveAsing = false
  40. var fileUpdateChecker: SKFileUpdateChecker?
  41. var mdFlags: MDFlags?
  42. var currentDocumentSetup: [String: Any] {
  43. get {
  44. var tempSetup: [String: Any] = [:]
  45. let tempMainSetup: [String: Any] = mainViewController?.currentSetup() ?? [:]
  46. let filePath = fileURL?.path ?? ""
  47. if filePath.count > 0{
  48. tempSetup.updateValue(filePath, forKey: KMDocumentSetupFileNameKey)
  49. }else {
  50. return tempSetup
  51. }
  52. if let alias = SKAlias.init(url: fileURL){
  53. if let data = alias.data{
  54. tempSetup.updateValue(data as Any, forKey: KMDocumentSetupAliasKey)
  55. }
  56. }
  57. if tempSetup.count > 0 {
  58. tempSetup.merge(tempMainSetup) { (_, new) in new }
  59. }
  60. return tempSetup
  61. }
  62. set {
  63. }
  64. }
  65. private var _saveToURL: URL?
  66. var saveToURL: URL? {
  67. get {
  68. return self._saveToURL
  69. }
  70. }
  71. var exportAccessoryC: SKExportAccessoryController?
  72. var pdfData: Data?
  73. weak var watermarkSaveDelegate: AnyObject?
  74. private var _trackEvents = IndexSet()
  75. override func save(to url: URL, ofType typeName: String, for saveOperation: NSDocument.SaveOperationType, delegate: Any?, didSave didSaveSelector: Selector?, contextInfo: UnsafeMutableRawPointer?) {
  76. if (self.isNewCreated) {
  77. // if let data = self.mainViewController, !data.isPDFDocumentEdited && !data.needSave && !self.isDocumentEdited {
  78. self._km_save(to: url, ofType: typeName, for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  79. return
  80. // }
  81. }
  82. if (!self.needSaveWatermark()) {
  83. self._km_save(to: url, ofType: typeName, for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  84. return
  85. }
  86. var openAccessoryView = self.watermarkSaveDelegate != nil
  87. if (openAccessoryView) {
  88. if let _browser = self.watermarkSaveDelegate as? KMBrowser, _browser.isCloseAllTabViewItem {
  89. openAccessoryView = false
  90. }
  91. }
  92. self._km_saveForWatermark(openAccessoryView: openAccessoryView) { [unowned self] in
  93. self.trackEvents()
  94. } callback: { [unowned self] needSave, params in
  95. if (needSave) {
  96. self._km_save(to: url, ofType: typeName, for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  97. } else { // 水印保存
  98. if (self.watermarkSaveDelegate == nil) {
  99. if let data = params.first as? KMResult, data == .cancel {
  100. if let shouldClose = params.last as? Bool, shouldClose {
  101. DispatchQueue.main.async {
  102. self.mainViewController?.browserWindowController?.browser.windowDidBeginToClose()
  103. }
  104. }
  105. } else {
  106. DispatchQueue.main.async {
  107. self.mainViewController?.browserWindowController?.browser.windowDidBeginToClose()
  108. }
  109. }
  110. return
  111. }
  112. if let data = params.first as? KMResult, data == .cancel {
  113. if var shouldClose = params.last as? Bool {
  114. if let _browser = self.watermarkSaveDelegate as? KMBrowser, _browser.isCloseAllTabViewItem {
  115. shouldClose = true
  116. }
  117. (self.watermarkSaveDelegate as? KMBrowser)?.document(self, shouldClose: shouldClose, contextInfo: nil)
  118. }
  119. } else {
  120. (self.watermarkSaveDelegate as? KMBrowser)?.document(self, shouldClose: true, contextInfo: nil)
  121. }
  122. self.watermarkSaveDelegate = nil
  123. }
  124. }
  125. }
  126. override func makeWindowControllers() {
  127. // Returns the storyboard that contains your document window.
  128. if ((self.fileURL?.path) != nil) {
  129. if !self.fileURL!.path.isPDFValid() {
  130. let alert = NSAlert()
  131. alert.alertStyle = .critical
  132. alert.messageText = NSLocalizedString("An error occurred while opening this document. The file is damaged and could not be repaired.", comment: "")
  133. alert.runModal()
  134. return
  135. }
  136. }
  137. let mainWindow = NSApp.mainWindow
  138. var currentWindowController: KMBrowserWindowController?
  139. if mainWindow != nil {
  140. let windowController = mainWindow!.windowController
  141. if windowController is KMBrowserWindowController {
  142. currentWindowController = (windowController as! KMBrowserWindowController)
  143. } else {
  144. for window in NSApp.windows {
  145. let windowController = window.windowController
  146. if windowController is KMBrowserWindowController {
  147. currentWindowController = (windowController as! KMBrowserWindowController)
  148. break
  149. }
  150. }
  151. }
  152. } else {
  153. for window in NSApp.windows {
  154. let windowController = window.windowController
  155. if windowController is KMBrowserWindowController {
  156. currentWindowController = (windowController as! KMBrowserWindowController)
  157. break
  158. }
  159. }
  160. }
  161. // if (currentWindowController == nil) && (self.fileURL != nil) {
  162. // let browser = KMBrowser.init() as KMBrowser
  163. // browser.addHomeTabContents()
  164. // browser.windowController = KMBrowserWindowController.init(browser: browser)
  165. // currentWindowController = browser.windowController as? KMBrowserWindowController
  166. // }
  167. if currentWindowController?.browser == nil && (self.fileURL != nil) {
  168. let browser: KMBrowser = KMBrowser.init()
  169. browser.windowController = KMBrowserWindowController.init(browser: browser)
  170. browser.addHomeTabContents()
  171. currentWindowController = browser.windowController as? KMBrowserWindowController
  172. browser.windowController.showWindow(self)
  173. }
  174. mainViewController = KMMainViewController.init()
  175. mainViewController?.myDocument = self
  176. self.mdFlags = MDFlags(exportOption: 0, exportUsingPanel: 0, gettingFileType: 0, convertingNotes: 0, needsPasswordToConvert: 0)
  177. if ((self.fileURL?.path) != nil) {
  178. let pdfDocument = CPDFDocument.init(url: URL(fileURLWithPath: self.fileURL!.path))
  179. mainViewController?.document = pdfDocument
  180. }
  181. if mainViewController?.document == nil {
  182. return
  183. }
  184. self.view = mainViewController?.view
  185. if let currentBrowser = currentWindowController?.browser {
  186. let activeBrowser = currentBrowser.activeTabContents()
  187. let activeIndex = currentBrowser.activeTabIndex()
  188. let ishome = activeBrowser?.isHome ?? false
  189. let isfirstTab = (activeIndex == 0)
  190. if ishome && !isfirstTab { // 替换 【标签需要被替换】
  191. self.addWindowController(currentWindowController!)
  192. self.mainViewController?.browserWindowController = currentWindowController
  193. // 替换 document
  194. currentWindowController?.browser.replaceTabContents(at: Int32(activeIndex), with: self)
  195. // 刷新标签
  196. currentWindowController?.browser.updateTabState(at: Int32(activeIndex))
  197. // 刷新 home icon
  198. if let tabStripController = currentWindowController?.tabStripController {
  199. if let view = tabStripController.view(at: UInt(activeIndex)) as? CTTabView {
  200. view.controller().isHome = self.isHome
  201. view.controller().isNewTab = self.isNewTab
  202. view.controller().updateUI()
  203. }
  204. }
  205. } else {
  206. if currentWindowController?.browser.tabCount() ?? 0 > 1 && (!IAPProductsManager.default().isAvailableAllFunction() || KMPreference.shared.openDocumentType == .newWindow){
  207. // 开启新窗口
  208. let browser = KMBrowser.init() as KMBrowser
  209. browser.windowController = KMBrowserWindowController.init(browser: browser)
  210. browser.addHomeTabContents()
  211. browser.windowController.showWindow(self)
  212. browser.add(self, at: Int32()-1, inForeground: true)
  213. self.addWindowController(browser.windowController)
  214. self.mainViewController?.browserWindowController = browser.windowController as? KMBrowserWindowController
  215. }else { // 正常拼接到后面
  216. self.addWindowController(currentWindowController!)
  217. self.mainViewController?.browserWindowController = currentWindowController
  218. currentWindowController?.browser.add(self, at: Int32()-1, inForeground: true)
  219. }
  220. }
  221. }
  222. }
  223. override func showWindows() {
  224. super.showWindows()
  225. self.setDataFromTmpData()
  226. }
  227. override func windowControllerDidLoadNib(_ aController: NSWindowController) {
  228. super.windowControllerDidLoadNib(aController)
  229. self.setDataFromTmpData()
  230. fileUpdateChecker = SKFileUpdateChecker.init(for: self)
  231. fileUpdateChecker?.isEnabled = true
  232. }
  233. override func save(to url: URL, ofType typeName: String, for saveOperation: NSDocument.SaveOperationType) async throws {
  234. do {
  235. try await super.save(to: url, ofType: typeName, for: saveOperation)
  236. } catch let outError {
  237. Swift.print(outError)
  238. }
  239. if saveOperation == .saveToOperation {
  240. if let data = self.mdFlags?.exportUsingPanel, data == 1 {
  241. if FileManager.default.fileExists(atPath: url.path) {
  242. let ws = NSWorkspace.shared
  243. ws.activateFileViewerSelecting([url])
  244. }
  245. }
  246. }
  247. self.mdFlags?.exportUsingPanel = 0
  248. self.mdFlags?.exportOption = UInt32(KMExportOption.default.rawValue)
  249. }
  250. override func write(to url: URL, ofType typeName: String, for saveOperation: NSDocument.SaveOperationType, originalContentsURL absoluteOriginalContentsURL: URL?) throws {
  251. try self._km_write(to: url, ofType: typeName, for: saveOperation, originalContentsURL: absoluteOriginalContentsURL)
  252. }
  253. override func canClose(withDelegate delegate: Any, shouldClose shouldCloseSelector: Selector?, contextInfo: UnsafeMutableRawPointer?) {
  254. let isPrompt = KMPreferenceManager.shared.closeFileIsPrompt()
  255. if (isPrompt) {
  256. super.canClose(withDelegate: delegate, shouldClose: shouldCloseSelector, contextInfo: contextInfo)
  257. return
  258. }
  259. if (self.isNewCreated) {
  260. self.save(nil)
  261. } else if (self.isDocumentEdited) {
  262. self.save(nil)
  263. } else if (mainViewController != nil) {
  264. if self.mainViewController!.isPDFDocumentEdited || self.mainViewController!.needSave {
  265. self.save(nil)
  266. }
  267. }
  268. super.canClose(withDelegate: delegate, shouldClose: shouldCloseSelector, contextInfo: contextInfo)
  269. }
  270. override func saveAs(_ sender: Any?) {
  271. if (!self.needSaveWatermark()) {
  272. self._km_saveAs(sender)
  273. return
  274. }
  275. self._km_saveForWatermark { [weak self] needSave, _ in
  276. if (needSave) {
  277. self?._km_saveAs(sender)
  278. }
  279. }
  280. }
  281. override func saveTo(_ sender: Any?) {
  282. guard let pdfDoc = self.mainViewController?.listView.document else {
  283. NSSound.beep()
  284. return
  285. }
  286. if pdfDoc.allowsPrinting == false || pdfDoc.allowsCopying == false {
  287. Task {
  288. _ = await KMAlertTool.runModel(message: NSLocalizedString("This is a secured document. Editing is not permitted.", comment: ""))
  289. }
  290. return
  291. }
  292. let idx = (sender as? NSMenuItem)?.tag ?? 0
  293. var typeName = KMPDFDocumentType
  294. if idx == 0 {
  295. typeName = KMPDFDocumentType
  296. } else if idx == 1 {
  297. typeName = KMPDFBundleDocumentType
  298. } else if idx == 2 {
  299. typeName = KMNotesDocumentType
  300. } else if idx == 3 {
  301. typeName = KMNotesTextDocumentType
  302. } else if idx == 4 {
  303. typeName = KMNotesRTFDocumentType
  304. } else if idx == 5 {
  305. typeName = KMNotesRTFDDocumentType
  306. } else if idx == 6 {
  307. typeName = KMNotesDocumentType
  308. }
  309. KMDataManager.ud_set(typeName, forKey: Self.kLastExportedTypeKey)
  310. super.saveTo(sender)
  311. }
  312. override func prepareSavePanel(_ savePanel: NSSavePanel) -> Bool {
  313. let success = super.prepareSavePanel(savePanel)
  314. let exportUsingPanel = self.mdFlags?.exportUsingPanel ?? 0
  315. if success && exportUsingPanel > 0 {
  316. // *formatPopup = [[savePanel accessoryView] subviewOfClass:[NSPopUpButton class]];
  317. var formatPopup: NSPopUpButton?
  318. let svs = savePanel.accessoryView?.subviews.first?.subviews ?? []
  319. for sv in svs {
  320. if let data = sv as? NSPopUpButton {
  321. formatPopup = data
  322. break
  323. }
  324. }
  325. self._removeSavePanelOfFormatPopupItems(savePanel)
  326. if (formatPopup != nil) {
  327. let lastExportedType = KMDataManager.ud_string(forKey: Self.kLastExportedTypeKey)
  328. var lastExportedOption = KMDataManager.ud_integer(forKey: Self.kLastExportedOptionKey)
  329. if lastExportedOption == 0 {
  330. lastExportedOption = KMExportOption.withEmbeddedNotes.rawValue
  331. }
  332. // if (lastExportedType != nil) {
  333. // let idx = formatPopup?.indexOfItem(withRepresentedObject: lastExportedType) ?? -1
  334. // let selectedIdx = formatPopup?.indexOfSelectedItem ?? 0
  335. // if idx != -1 && idx != selectedIdx {
  336. // formatPopup?.selectItem(at: idx)
  337. // formatPopup?.sendAction(formatPopup?.action, to: formatPopup?.target)
  338. // [savePanel setAllowedFileTypes:[NSArray arrayWithObjects:[self fileNameExtensionForType:lastExportedType saveOperation:NSSaveToOperation], nil]];
  339. // if let data = self.fileNameExtension(forType: lastExportedType!, saveOperation: .saveToOperation) {
  340. // savePanel.allowedFileTypes = [data]
  341. // }
  342. // }
  343. // }
  344. self.mdFlags?.exportOption = UInt32(lastExportedOption)
  345. //
  346. // self.exportAccessoryC = SKExportAccessoryController()
  347. // self.exportAccessoryC?.addFormatPopUpButton(formatPopup)
  348. // self.exportAccessoryC?.matrix.target = self
  349. // self.exportAccessoryC?.matrix.action = #selector(changeExportOption)
  350. // savePanel.accessoryView = self.exportAccessoryC?.view
  351. // self._updateExportAccessoryView()
  352. }
  353. } else if success {
  354. self._removeSavePanelOfFormatPopupItems(savePanel)
  355. }
  356. return success
  357. }
  358. override func runModalSavePanel(for saveOperation: NSDocument.SaveOperationType, delegate: Any?, didSave didSaveSelector: Selector?, contextInfo: UnsafeMutableRawPointer?) {
  359. self.mdFlags?.exportUsingPanel = saveOperation == .saveToOperation ? 1 : 0
  360. self.mdFlags?.exportOption = UInt32(KMExportOption.default.rawValue)
  361. if (self.isNewCreated) {
  362. // if let data = self.mainViewController, !data.isPDFDocumentEdited && !data.needSave && !self.isDocumentEdited {
  363. self._km_runModalSavePanel(for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  364. return
  365. // }
  366. }
  367. if (!self.needSaveWatermark()) {
  368. self._km_runModalSavePanel(for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  369. return
  370. }
  371. self._km_saveForWatermark { [weak self] needSave, _ in
  372. if (needSave) {
  373. self?._km_runModalSavePanel(for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  374. }
  375. }
  376. }
  377. override func save(_ sender: Any?) {
  378. if (!self.needSaveWatermark()) {
  379. self._km_save(sender)
  380. return
  381. }
  382. self._km_saveForWatermark { [weak self] in
  383. self?.trackEvents()
  384. } callback: { [weak self] needSave, _ in
  385. if (needSave) {
  386. self?._km_save(sender)
  387. }
  388. }
  389. }
  390. func systemInteractionMode() -> KMInteractionMode {
  391. let mainWindow = NSApp.mainWindow
  392. if mainWindow != nil {
  393. let windowController = mainWindow!.windowController
  394. if windowController?.window?.screen?.isEqual(NSScreen.screens[0]) ?? false{
  395. return mainViewController?.interactionMode ?? .normal
  396. }
  397. }
  398. return .normal
  399. }
  400. func saveForWatermark() {
  401. if (!self.needSaveWatermark()) {
  402. self._km_save(nil)
  403. return
  404. }
  405. self._km_saveForWatermark { [weak self] in
  406. self?.trackEvents()
  407. } callback: { [unowned self] needSave, params in
  408. if (needSave) {
  409. self._km_save(nil)
  410. } else { // 水印保存
  411. if (self.watermarkSaveDelegate == nil) {
  412. if let data = params.first as? KMResult, data == .cancel {
  413. if let shouldClose = params.last as? Bool, shouldClose {
  414. DispatchQueue.main.async {
  415. self.mainViewController?.browserWindowController?.browser.windowDidBeginToClose()
  416. }
  417. }
  418. } else {
  419. DispatchQueue.main.async {
  420. self.mainViewController?.browserWindowController?.browser.windowDidBeginToClose()
  421. }
  422. }
  423. return
  424. }
  425. if let data = params.first as? KMResult, data == .cancel {
  426. if var shouldClose = params.last as? Bool {
  427. if let _browser = self.watermarkSaveDelegate as? KMBrowser, _browser.isCloseAllTabViewItem {
  428. shouldClose = true
  429. }
  430. (self.watermarkSaveDelegate as? KMBrowser)?.document(self, shouldClose: shouldClose, contextInfo: nil)
  431. }
  432. } else {
  433. (self.watermarkSaveDelegate as? KMBrowser)?.document(self, shouldClose: true, contextInfo: nil)
  434. }
  435. self.watermarkSaveDelegate = nil
  436. }
  437. }
  438. }
  439. override func read(from absoluteURL: URL, ofType typeName: String) throws {
  440. do {
  441. try super.read(from: absoluteURL, ofType: typeName)
  442. updateChangeCount(.changeCleared)
  443. } catch let outError {
  444. Swift.print(outError)
  445. }
  446. }
  447. override func read(from data: Data, ofType typeName: String) throws {
  448. // Insert code here to read your document from the given data of the specified type, throwing an error in case of failure.
  449. // 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.
  450. let pdfDocument = CPDFDocument.init(data: data)
  451. self.pdfData = data
  452. if pdfDocument == nil {
  453. throw NSError(domain: NSOSStatusErrorDomain, code: unimpErr, userInfo: nil)
  454. }
  455. }
  456. // MARK: Autosaving
  457. override func close() {
  458. if self.isActive {
  459. if browser != nil {
  460. var activeIndex = 0
  461. let dex = browser.index(of: self)
  462. if dex == browser.tabCount() - 1 {
  463. activeIndex = Int(browser.tabCount()-2)
  464. } else {
  465. activeIndex = Int(dex + 1)
  466. }
  467. let activeContents = browser.tabContents(at: Int32(activeIndex))
  468. activeContents?.addWindowController(browser.windowController)
  469. }
  470. }
  471. super.close()
  472. }
  473. // MARK: init
  474. override init() {
  475. super.init()
  476. // Add your subclass-specific initialization here.
  477. NotificationCenter.default.addObserver(self, selector: #selector(pdfChangedNotification(_:)), name: NSNotification.Name.init(rawValue: "CPDFListViewAnnotationsAttributeHasChangeNotification"), object: nil)
  478. // NotificationCenter.default.addObserver(self, selector: #selector(pdfChangedNotification(_:)), name: NSNotification.Name.init(rawValue: "CPDFViewDocumentChangedNotification"), object: nil)
  479. // NotificationCenter.default.addObserver(self, selector: #selector(pdfChangedNotification(_:)), name: NSNotification.Name.init(rawValue: "CPDFViewPageChangedNotification"), object: nil)
  480. NotificationCenter.default.addObserver(self, selector: #selector(pdfChangedNotification(_:)), name: NSNotification.Name.init(rawValue: "CPDFListViewDidAddAnnotationNotification"), object: nil)
  481. NotificationCenter.default.addObserver(self, selector: #selector(pdfChangedNotification(_:)), name: NSNotification.Name.CPDFViewPageChanged, object: nil)
  482. }
  483. override init?(baseTabContents baseContents: CTTabContents?) {
  484. super.init(baseTabContents: baseContents)
  485. if isHome {
  486. homeViewController = KMNHomeViewController.init()
  487. self.view = homeViewController?.view
  488. }
  489. }
  490. // MARK: Handling User Actions
  491. override var title: String? {
  492. get {
  493. if isHome {
  494. if (self.isNewTab) {
  495. return NSLocalizedString("New Tab", comment: "")
  496. } else {
  497. return NSLocalizedString("Home", comment: "")
  498. }
  499. } else {
  500. return fileURL?.lastPathComponent
  501. }
  502. }
  503. set {
  504. super.title = newValue
  505. }
  506. }
  507. func needSaveWatermark() -> Bool {
  508. if let need = self.mainViewController?.saveWatermarkFlag {
  509. return need
  510. }
  511. return false
  512. }
  513. func changeExportOption(_ sender: NSMatrix?) {
  514. self.mdFlags?.exportOption = UInt32(sender?.selectedCell()?.tag ?? 0)
  515. }
  516. // MARK: Private Methods
  517. func pdfChangedNotification(_ notification: Notification) -> Void {
  518. if !isHome {
  519. let mainViewController = mainViewController
  520. var document: CPDFDocument!
  521. let dic = notification.object as? NSDictionary
  522. if dic?["object"] is CPDFAnnotation {
  523. guard let annotation = dic?["object"] as? CPDFAnnotation else { return }
  524. document = annotation.page.document
  525. } else if dic?["object"] is CPDFListView {
  526. let pdflistView = notification.object as? CPDFListView
  527. document = pdflistView?.document
  528. }
  529. if mainViewController != nil {
  530. if document == mainViewController!.document {
  531. updateChangeCount(.changeDone)
  532. }
  533. }
  534. }
  535. }
  536. override func updateChangeCount(_ change: NSDocument.ChangeType) {
  537. let mainWindow = NSApp.mainWindow
  538. var currentWindowController: KMBrowserWindowController?
  539. if mainWindow != nil {
  540. let windowController = mainWindow!.windowController
  541. if windowController is KMBrowserWindowController {
  542. currentWindowController = (windowController as! KMBrowserWindowController)
  543. } else {
  544. for window in NSApp.windows {
  545. let windowController = window.windowController
  546. if windowController is KMBrowserWindowController {
  547. currentWindowController = (windowController as! KMBrowserWindowController)
  548. break
  549. }
  550. }
  551. }
  552. } else {
  553. for window in NSApp.windows {
  554. let windowController = window.windowController
  555. if windowController is KMBrowserWindowController {
  556. currentWindowController = (windowController as! KMBrowserWindowController)
  557. break
  558. }
  559. }
  560. }
  561. if let currentBroser = currentWindowController?.browser {
  562. if self.isEqual(to: currentBroser.activeTabContents()) {
  563. super.updateChangeCount(change)
  564. return
  565. }
  566. }
  567. super.updateChangeCount(.changeCleared)
  568. }
  569. func uploadToCloud(_ callback: (@escaping(Bool, String)->())) {
  570. guard let handdler = self.cloudUploadHanddler else {
  571. return
  572. }
  573. handdler(callback)
  574. }
  575. func isPDFDocument() -> Bool {
  576. return true
  577. }
  578. func setDataFromTmpData() {
  579. guard let _document = self.mainViewController?.document else {
  580. return
  581. }
  582. // self.tryToUnlockDocument(document!)
  583. if (_document.permissionsStatus != .owner) {
  584. var password: NSString? = nil
  585. let fileId = self.fileId(for: _document)
  586. if (fileId.isEmpty) {
  587. return
  588. }
  589. self.getPassword(&password, fileId: fileId)
  590. if (password != nil) {
  591. self.isUnlockFromKeychain = true
  592. // document.unlock(withPassword: password! as String)
  593. self.mainViewController?.model.password = password as String?
  594. }
  595. }
  596. //如果已存在,开个存在页签
  597. var selectDocument: KMMainDocument? = self
  598. if selectDocument != nil {
  599. if let browser_ = selectDocument?.browser {
  600. let currentIndex = browser_.tabStripModel.index(of: selectDocument)
  601. browser_.tabStripModel.selectTabContents(at: Int32(currentIndex), userGesture: true)
  602. if browser_.window.isVisible {
  603. DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + 0.1) {
  604. browser_.window.orderFront(nil)
  605. }
  606. } else if browser_.window.isMiniaturized {
  607. DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + 0.1) {
  608. browser_.window.orderFront(nil)
  609. }
  610. }
  611. }
  612. }
  613. }
  614. func tryToUnlockDocument(_ document: CPDFDocument) {
  615. if (document.permissionsStatus != .owner) {
  616. var password: NSString? = nil
  617. let fileId = self.fileId(for: document)
  618. if (fileId.isEmpty) {
  619. return
  620. }
  621. self.getPassword(&password, fileId: fileId)
  622. if (password != nil) {
  623. self.isUnlockFromKeychain = true
  624. document.unlock(withPassword: password! as String)
  625. }
  626. }
  627. }
  628. func km_updateChangeCount(_ change: NSDocument.ChangeType) {
  629. super.updateChangeCount(change)
  630. }
  631. func trackEvents() {
  632. km_synchronized(self) {
  633. for i in self._trackEvents {
  634. if let type = KMSubscribeWaterMarkType(rawValue: i) {
  635. KMTools.trackEvent(type: type)
  636. }
  637. }
  638. }
  639. self.clearTrackEvents()
  640. }
  641. func recordTrackEvent(type: KMSubscribeWaterMarkType) {
  642. if (type == .none) {
  643. return
  644. }
  645. km_synchronized(self) {
  646. self._trackEvents.insert(type.rawValue)
  647. }
  648. }
  649. func clearTrackEvents() {
  650. km_synchronized(self) {
  651. self._trackEvents.removeAll()
  652. }
  653. }
  654. @IBAction func saveArchive(_ sender: Any?) {
  655. guard let item = sender as? NSMenuItem else {
  656. NSSound.beep()
  657. return
  658. }
  659. guard let fileURL = self.fileURL else {
  660. NSSound.beep()
  661. return
  662. }
  663. let check = try?fileURL.checkResourceIsReachable()
  664. if check == false || self.isDocumentEdited {
  665. let msg = KMLocalizedString("You must save this file first")
  666. let inf = KMLocalizedString("The document has unsaved changes, or has not previously been saved to disk.")
  667. Task {
  668. _ = await KMAlertTool.runModel(message: msg, informative: inf)
  669. }
  670. return
  671. }
  672. // NSString *ext = ([sender tag] | SKArchiveDiskImageMask) ? @"dmg" : @"tgz";
  673. let idx = item.tag
  674. let ext = (idx == 1 || idx == 3) ? "dmg" : "tgz"
  675. let isEmail = (idx == 2 || idx == 3)
  676. if isEmail {
  677. // if (([sender tag] | SKArchiveEmailMask)) {
  678. let tmpDirURL = FileManager.default.uniqueChewableItemsDirectoryURL()
  679. let tmpFileURL = tmpDirURL.appendingPathComponent(fileURL.lastPathComponentReplacingPathExtension(ext))
  680. self.newSaveArchive(to: tmpFileURL, email: true)
  681. } else {
  682. let sp = NSSavePanel()
  683. sp.allowedFileTypes = [ext]
  684. sp.canCreateDirectories = true
  685. sp.nameFieldStringValue = fileURL.lastPathComponentReplacingPathExtension(ext)
  686. sp.beginSheetModal(for: self.windowForSheet!) { result in
  687. if result == .OK {
  688. self.newSaveArchive(to: sp.url!, email: false)
  689. }
  690. }
  691. }
  692. }
  693. // func saveArchiveToURL(to fileURL: URL, email: Bool) {
  694. // NSTask *task = [[[NSTask alloc] init] autorelease];
  695. // let task = Task()
  696. // if fileURL.pathExtension == "dmg" {
  697. // [task setLaunchPath:@""];
  698. // task.launchPath = "/usr/bin/hdiutil"
  699. // [task setArguments:[NSArray arrayWithObjects:@"create", @"-srcfolder", [[self fileURL] path], @"-format", @"UDZO", @"-volname", [[fileURL lastPathComponent] stringByDeletingPathExtension], [fileURL path], nil]];
  700. // } else {
  701. // [task setLaunchPath:@"/usr/bin/tar"];
  702. // [task setArguments:[NSArray arrayWithObjects:@"-czf", [fileURL path], [[self fileURL] lastPathComponent], nil]];
  703. // }
  704. // [task setCurrentDirectoryPath:[[[self fileURL] URLByDeletingLastPathComponent] path]];
  705. // [task setStandardOutput:[NSFileHandle fileHandleWithNullDevice]];
  706. // [task setStandardError:[NSFileHandle fileHandleWithNullDevice]];
  707. //
  708. // SKAttachmentEmailer *emailer = nil;
  709. // if (email)
  710. // emailer = [SKAttachmentEmailer attachmentEmailerWithFileURL:fileURL subject:[self displayName] waitingForTask:task];
  711. //
  712. // @try {
  713. // [task launch];
  714. // }
  715. // @catch (id exception) {
  716. // [emailer taskFailed];
  717. // }
  718. // }
  719. @IBAction func readNotes(_ sender: Any?) {
  720. KMPrint("readNotes")
  721. }
  722. @IBAction func convertNotes(_ sender: Any?) {
  723. KMPrint("convertNotes")
  724. }
  725. @IBAction func batchRemovePassWord(_ sender: Any?) {
  726. self.mainViewController?.clickChildTool(type: .secure, index: 2)
  727. }
  728. @IBAction func batchRemovPrivatySecurity(_ sender: Any?) {
  729. self.mainViewController?.removeOwnerPassword()
  730. }
  731. @IBAction func printPDFDocument(_ sender: Any?) {
  732. self.mainViewController?.saveDocument()
  733. KMPrintWindowController.showNewPrintWindowControll(inputDocument: self.mainViewController?.document, inputPageRange: KMPrintPageRange())
  734. }
  735. @IBAction func performFindPanelAction(_ sender: Any?) {
  736. self.mainViewController?.toolbarController.showFindBar()
  737. }
  738. @IBAction func addBookmark(_ sender: Any?) {
  739. guard let item = sender as? NSMenuItem else {
  740. return
  741. }
  742. // let bookmarkSheetController = SKBookmarkSheetController(windowNibName: "BookmarkSheet")
  743. // NSWindow.currentWindow().beginSheet(bookmarkSheetController.window!) { resoponse in
  744. //
  745. // }
  746. // bookmarkSheetController.textField.stringValue = self.displayName
  747. // bookmarkSheetController.beginSheetModal(for: self.windowForSheet) { [unowned self] result in
  748. // if (result == NSApplication.ModalResponse.OK.rawValue) {
  749. // let label = bookmarkSheetController.textField.stringValue;
  750. // let folder: SKBookmark = bookmarkSheetController.selectedFolder ?? SKBookmarkController.shared().bookmarkRoot
  751. // var bookmark = SKBookmark()
  752. // switch (item.tag) {
  753. // case 0:
  754. // let mainViewController = self.mainViewController
  755. // let page = mainViewController?.listView.currentPage()
  756. // bookmark = SKBookmark.bookmark(with: self.fileURL, pageIndex: (page?.pageIndex())!, label: label) as! SKBookmark
  757. // case 1:
  758. // let setup = self.currentDocumntSetup()
  759. // bookmark = SKBookmark.bookmark(withSetup: setup, label: label) as! SKBookmark
  760. //
  761. // case 2:
  762. // let setups = NSApp.orderedDocuments.map { $0.value(forKey: "currentDocumentSetup") }
  763. // bookmark = SKBookmark.bookmarkSession(withSetups: setups as [Any], label: label) as! SKBookmark
  764. //
  765. // default:
  766. // break;
  767. // }
  768. // folder.mutableArrayValue(forKey: "children").add(bookmark)
  769. // }
  770. // }
  771. if item.tag == 3 {
  772. KMPrint("Edit Bookmark")
  773. bookmarkController = KMBookmarkController.showBookmarkController()
  774. } else if item.tag == 2 {
  775. KMPrint("session Bookmark")
  776. bookmarkSheetController = KMBookmarkSheetController.showBookmarkSheetController(type: .session)
  777. } else if item.tag == 0 {
  778. KMPrint("add Bookmark")
  779. bookmarkSheetController = KMBookmarkSheetController.showBookmarkSheetController(type: .bookmark)
  780. }
  781. bookmarkSheetController?.stringValue = self.displayName
  782. bookmarkSheetController?.cancelAction = { [weak self] controller, type in
  783. }
  784. //
  785. bookmarkSheetController?.doneAction = { [unowned self] controller, type, label in
  786. let folder = controller.selectedFolder
  787. var bookmark: KMBookmark?
  788. switch type {
  789. case .bookmark:
  790. let mainViewController = mainViewController
  791. if let page = mainViewController?.listView.currentPage() {
  792. let index: UInt = page.pageIndex()
  793. bookmark = KMBookmark.bookmark(url: self.fileURL!, pageIndex: index, label: label)
  794. }
  795. case .setup: break
  796. let setup = currentDocumentSetup
  797. bookmark = KMBookmark.bookmark(setup: setup, label: label)
  798. case .session:
  799. let setups = NSApp.orderedDocuments.compactMap { $0.value(forKey:"currentDocumentSetup") }
  800. bookmark = KMSessionBookmark.bookmarkSession(setups: setups as NSArray, label: label)
  801. default:
  802. break
  803. }
  804. if let bookmark = bookmark {
  805. folder?.children.append(bookmark)
  806. }
  807. KMBookmarkManager.manager.saveData()
  808. }
  809. }
  810. func currentDocumntSetup() -> [String: Any] {
  811. var setup: [String: Any] = [:]
  812. let data = SKAlias.init(url: fileURL).data
  813. if (data != nil) {
  814. setup.updateValue(data as Any, forKey: "_BDAlias")
  815. } else {
  816. setup.updateValue(fileURL?.path as Any, forKey: "fileName")
  817. }
  818. return setup;
  819. }
  820. @IBAction func showWindow(_ sender: Any?) {
  821. KMPrint("showWindow")
  822. }
  823. // MARK: - Private Methods
  824. private func _PDFBundleFileWrapper(for name: String) -> FileWrapper {
  825. var aName = name
  826. if name.isCaseInsensitiveEqual(Self.kBundleDataFilename) {
  827. aName = aName + "1"
  828. }
  829. // var data: Data?
  830. let fileWrapper = FileWrapper(directoryWithFileWrappers: [:])
  831. let info = KMInfoWindowController.shared.info(for: self)
  832. if let data = self.pdfData {
  833. fileWrapper.addRegularFile(withContents: data, preferredFilename: aName + ".pdf")
  834. }
  835. if let data = self.mainViewController?.document?.string()?.data(using: .utf8) {
  836. fileWrapper.addRegularFile(withContents: data, preferredFilename: Self.kBundleDataFilename + ".txt")
  837. }
  838. if let data = try?PropertyListSerialization.data(fromPropertyList: info, format: .xml, options: PropertyListSerialization.WriteOptions(0)) {
  839. fileWrapper.addRegularFile(withContents: data, preferredFilename: Self.kBundleDataFilename + ".plist")
  840. }
  841. // if ((data = [NSPropertyListSerialization dataWithPropertyList:info format:NSPropertyListXMLFormat_v1_0 options:0 error:NULL]))
  842. // [fileWrapper addRegularFileWithContents:data preferredFilename:[BUNDLE_DATA_FILENAME stringByAppendingPathExtension:@"plist"]];
  843. // if ([[self notes] count] > 0) {
  844. // if ((data = [self notesData]))
  845. // [fileWrapper addRegularFileWithContents:data preferredFilename:[name stringByAppendingPathExtension:@"skim"]];
  846. // if ((data = [[self notesString] dataUsingEncoding:NSUTF8StringEncoding]))
  847. // [fileWrapper addRegularFileWithContents:data preferredFilename:[name stringByAppendingPathExtension:@"txt"]];
  848. // if ((data = [self notesRTFData]))
  849. // [fileWrapper addRegularFileWithContents:data preferredFilename:[name stringByAppendingPathExtension:@"rtf"]];
  850. // if ((data = [self notesFDFDataForFile:[name stringByAppendingPathExtension:@"pdf"] fileIDStrings:[[self pdfDocument] fileIDStrings]]))
  851. // [fileWrapper addRegularFileWithContents:data preferredFilename:[name stringByAppendingPathExtension:@"fdf"]];
  852. // }
  853. return fileWrapper
  854. }
  855. private func _km_write(to url: URL, ofType typeName: String, for saveOperation: NSDocument.SaveOperationType, originalContentsURL absoluteOriginalContentsURL: URL?) throws {
  856. if typeName == KMPDFBundleDocumentType {
  857. // NSFileWrapper *fileWrapper = [self PDFBundleFileWrapperForName:[[absoluteURL lastPathComponent] stringByDeletingPathExtension]];
  858. // if (fileWrapper)
  859. // didWrite = [fileWrapper writeToURL:absoluteURL options:0 originalContentsURL:nil error:&error];
  860. // else
  861. // error = [NSError writeFileErrorWithLocalizedDescription:NSLocalizedString(@"Unable to write file", @"Error description")];
  862. let fileWrapper = self._PDFBundleFileWrapper(for: url.deletingPathExtension().lastPathComponent)
  863. do {
  864. try fileWrapper.write(to: url, options: FileWrapper.WritingOptions(rawValue: 0), originalContentsURL: nil)
  865. } catch {
  866. NSApp.presentError(error)
  867. }
  868. return
  869. }
  870. var success = true
  871. if !self.isHome {
  872. if mainViewController != nil {
  873. mainViewController?.savePdfAlertView()
  874. if mainViewController?.document != nil {
  875. self.mainViewController?.commitEditingIfNeed()
  876. // if mainViewController!.document!.isEncrypted {
  877. // success = mainViewController!.document!.write(to: url)
  878. // } else {
  879. if (mainViewController!.needSave) {
  880. if let options = self.mainViewController?.secureOptions, !options.isEmpty {
  881. self.mainViewController!.document?.setDocumentAttributes(self.mainViewController?.documentAttribute)
  882. success = self.mainViewController!.document!.write(to: url, withOptions: options)
  883. } else if let flag = self.mainViewController?.removeSecureFlag, flag {
  884. success = self.mainViewController!.document!.writeDecrypt(to: url)
  885. } else {
  886. success = mainViewController!.document!.write(to: url)
  887. }
  888. } else {
  889. success = mainViewController!.document!.write(to: url)
  890. }
  891. // }
  892. self.mainViewController?.needSave = false
  893. self.mainViewController?.clearSecureOptions()
  894. self.mainViewController?.clearRemoveSecureFlag()
  895. }
  896. }
  897. } else {
  898. success = false
  899. }
  900. if (success && self._saveAsing) {
  901. if let tabView = self.browser?.windowController?.tabStripController?.activeTabView() as? CTTabView {
  902. tabView.controller()?.title = url.lastPathComponent
  903. }
  904. self._saveAsing = false
  905. }
  906. if success && isNewCreated && NSDocument.SaveOperationType.saveAsOperation == saveOperation {
  907. isNewCreated = false
  908. }
  909. if mainViewController != nil {
  910. mainViewController?.savePdfFinishAlertView()
  911. }
  912. }
  913. private func _km_saveForWatermark(openAccessoryView: Bool = true, subscribeDidClick: (()->Void)? = nil, callback:@escaping (_ needSave: Bool, _ param: Any...)->Void) {
  914. Task { @MainActor in
  915. if await (KMLightMemberManager.manager.canPayFunction() == false) {
  916. let _ = KMSubscribeWaterMarkWindowController.show(window: NSApp.mainWindow!, isContinue:false, type: .save) {
  917. if let _callback = subscribeDidClick {
  918. _callback()
  919. }
  920. } completion: { isSubscribeSuccess, isWaterMarkExport, isClose in
  921. if (isClose) {
  922. callback(false, KMResult.cancel, false)
  923. return
  924. }
  925. if (isSubscribeSuccess) {
  926. callback(true)
  927. return
  928. }
  929. if (isWaterMarkExport) {
  930. guard let _document = self.mainViewController?.document else {
  931. callback(false, KMResult.failure)
  932. return
  933. }
  934. // 提交文本编辑的内容
  935. self.mainViewController?.commitEditingIfNeed()
  936. DispatchQueue.main.async {
  937. NSPanel.savePanel(NSApp.mainWindow!, openAccessoryView, panel:{ panel in
  938. if (!self.isNewCreated) {
  939. panel.directoryURL = _document.documentURL.deletingLastPathComponent()
  940. }
  941. panel.nameFieldStringValue = _document.documentURL.lastPathComponent
  942. }) { response, url, isOpen in
  943. if (response == .cancel) {
  944. callback(false, KMResult.cancel, true)
  945. return
  946. }
  947. guard let _url = KMTools.saveWatermarkDocument(document: _document, to: url!, secureOptions: self.mainViewController?.secureOptions, documentAttribute: self.mainViewController?.documentAttribute,removePWD: self.mainViewController!.removeSecureFlag) else {
  948. callback(false, KMResult.failure)
  949. return
  950. }
  951. callback(false, KMResult.success)
  952. if (isOpen) {
  953. NSDocumentController.shared.km_safe_openDocument(withContentsOf: _url, display: true) { _, _, _ in
  954. }
  955. } else {
  956. NSWorkspace.shared.activateFileViewerSelecting([_url])
  957. }
  958. }
  959. }
  960. return
  961. }
  962. callback(false, KMResult.cancel, false)
  963. }
  964. return
  965. }
  966. callback(true)
  967. }
  968. }
  969. private func _km_save(_ sender: Any?) {
  970. super.save(sender)
  971. }
  972. private func _km_save(to url: URL, ofType typeName: String, for saveOperation: NSDocument.SaveOperationType, delegate: Any?, didSave didSaveSelector: Selector?, contextInfo: UnsafeMutableRawPointer?) {
  973. self._saveToURL = url
  974. super.save(to: url, ofType: typeName, for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  975. // self.mdFlags?.exportUsingPanel = 0
  976. // self.mdFlags?.exportOption = UInt32(KMExportOption.default.rawValue)
  977. }
  978. private func _km_saveAs(_ sender: Any?) {
  979. super.saveAs(sender)
  980. self._saveAsing = true
  981. }
  982. private func _km_runModalSavePanel(for saveOperation: NSDocument.SaveOperationType, delegate: Any?, didSave didSaveSelector: Selector?, contextInfo: UnsafeMutableRawPointer?) {
  983. super.runModalSavePanel(for: saveOperation, delegate: delegate, didSave: didSaveSelector, contextInfo: contextInfo)
  984. }
  985. private func _updateExportAccessoryView() {
  986. let typeName = self.fileTypeFromLastRunSavePanel ?? ""
  987. let matrix = self.exportAccessoryC?.matrix
  988. matrix?.selectCell(withTag: Int(self.mdFlags?.exportOption ?? 0))
  989. if self._canAttachNotesForType(typeName) {
  990. matrix?.isHidden = false
  991. let ws = NSWorkspace.shared
  992. let isLocked = self.mainViewController?.listView.document.isLocked ?? false
  993. let allowsPrinting = self.mainViewController?.listView.document.allowsPrinting ?? false
  994. if ws.type(typeName, conformsToType: KMPDFDocumentType) && isLocked == false && allowsPrinting {
  995. (matrix?.cell(withTag: KMExportOption.withEmbeddedNotes.rawValue))?.isEnabled = true
  996. } else {
  997. (matrix?.cell(withTag: KMExportOption.withEmbeddedNotes.rawValue))?.isEnabled = false
  998. if let data = self.mdFlags?.exportOption, data == KMExportOption.withEmbeddedNotes.rawValue {
  999. self.mdFlags?.exportOption = UInt32(KMExportOption.default.rawValue)
  1000. matrix?.selectCell(withTag: KMExportOption.default.rawValue)
  1001. }
  1002. }
  1003. } else {
  1004. matrix?.isHidden = true
  1005. }
  1006. }
  1007. private func _canAttachNotesForType(_ typeName: String) -> Bool {
  1008. let ws = NSWorkspace.shared
  1009. return ws.type(typeName, conformsToType: KMPDFDocumentType) || ws.type(typeName, conformsToType: KMPostScriptDocumentType) || ws.type(typeName, conformsToType: KMDVIDocumentType) || ws.type(typeName, conformsToType: KMXDVDocumentType)
  1010. }
  1011. private func _removeSavePanelOfFormatPopupItems(_ savePanel: NSSavePanel) {
  1012. var formatPopup: NSPopUpButton?
  1013. let svs = savePanel.accessoryView?.subviews.first?.subviews ?? []
  1014. for sv in svs {
  1015. if let data = sv as? NSPopUpButton {
  1016. formatPopup = data
  1017. break
  1018. }
  1019. }
  1020. if let item = formatPopup?.item(withTitle: "Notes as Text") {
  1021. formatPopup?.removeItem(at: formatPopup!.index(of: item))
  1022. }
  1023. if let item = formatPopup?.item(withTitle: "Notes as FDF") {
  1024. formatPopup?.removeItem(at: formatPopup!.index(of: item))
  1025. }
  1026. if let item = formatPopup?.item(withTitle: "Notes as RTFD") {
  1027. formatPopup?.removeItem(at: formatPopup!.index(of: item))
  1028. }
  1029. if let item = formatPopup?.item(withTitle: NSLocalizedString("Text", comment: "")) {
  1030. formatPopup?.removeItem(at: formatPopup!.index(of: item))
  1031. }
  1032. if let item = formatPopup?.item(withTitle: NSLocalizedString("text", comment: "")) {
  1033. formatPopup?.removeItem(at: formatPopup!.index(of: item))
  1034. }
  1035. if let item = formatPopup?.item(withTitle: "PDF Reader Pro Edition Notes") {
  1036. formatPopup?.removeItem(at: formatPopup!.index(of: item))
  1037. }
  1038. if let item = formatPopup?.item(withTitle: "XDV") {
  1039. formatPopup?.removeItem(at: formatPopup!.index(of: item))
  1040. }
  1041. if let item = formatPopup?.item(withTitle: "DVI") {
  1042. formatPopup?.removeItem(at: formatPopup!.index(of: item))
  1043. }
  1044. if let item = formatPopup?.item(withTitle: "Images") {
  1045. formatPopup?.removeItem(at: formatPopup!.index(of: item))
  1046. }
  1047. if let item = formatPopup?.item(withTitle: "Encapsulated PostScript") {
  1048. formatPopup?.removeItem(at: formatPopup!.index(of: item))
  1049. }
  1050. }
  1051. // MARK: - Printing
  1052. override func printOperation(withSettings printSettings: [NSPrintInfo.AttributeKey : Any]) throws -> NSPrintOperation {
  1053. let printInfo = self.printInfo.copy() as! NSPrintInfo
  1054. printInfo.dictionary().addEntries(from: printSettings)
  1055. var printOperation: NSPrintOperation?
  1056. if self.isHome {
  1057. return NSPrintOperation()
  1058. }
  1059. let documentURL = self.mainViewController?.document?.documentURL
  1060. if documentURL == nil {
  1061. return NSPrintOperation()
  1062. }
  1063. guard let pdfDoc = PDFDocument(url: documentURL!) else {
  1064. return NSPrintOperation()
  1065. }
  1066. if pdfDoc.responds(to: #selector(PDFDocument.printOperation(for:scalingMode:autoRotate:))) {
  1067. printOperation = pdfDoc.printOperation(for: printInfo, scalingMode: .pageScaleNone, autoRotate: true)
  1068. } else if pdfDoc.responds(to: #selector(PDFDocument.getPrintOperation(for:autoRotate:))) {
  1069. printOperation = pdfDoc.getPrintOperation(for: printInfo, autoRotate: true)
  1070. }
  1071. // NSPrintProtected is a private key that disables the items in the PDF popup of the Print panel, and is set for encrypted documents
  1072. if pdfDoc.isEncrypted {
  1073. printOperation?.printInfo.dictionary().setValue(false, forKey: "NSPrintProtected")
  1074. }
  1075. let printPanel = printOperation?.printPanel
  1076. printPanel?.options = [.showsCopies, .showsPageRange, .showsPaperSize, .showsOrientation, .showsScaling, .showsPreview]
  1077. if printOperation == nil {
  1078. throw NSError.printDocumentError(withLocalizedDescription: "")
  1079. }
  1080. return printOperation!
  1081. }
  1082. override func runModalPrintOperation(_ printOperation: NSPrintOperation, delegate: Any?, didRun didRunSelector: Selector?, contextInfo: UnsafeMutableRawPointer?) {
  1083. printOperation.run()
  1084. }
  1085. }
  1086. extension PDFDocument {
  1087. @objc func getPrintOperation(for printInfo: NSPrintInfo, autoRotate: Bool) -> NSPrintOperation {
  1088. // 在此处实现方法的具体逻辑
  1089. return NSPrintOperation()
  1090. }
  1091. }
  1092. extension KMMainDocument {
  1093. override func validateMenuItem(_ menuItem: NSMenuItem) -> Bool {
  1094. if (menuItem.action == #selector(save(_ :))) {
  1095. if (self.isHome) {
  1096. return false
  1097. }
  1098. if (self.isDocumentEdited) {
  1099. return self.isDocumentEdited
  1100. }
  1101. guard let mainVC = self.mainViewController else {
  1102. return false
  1103. }
  1104. return mainVC.isPDFDocumentEdited || mainVC.needSave
  1105. } else if (menuItem.action == #selector(saveAs(_ :))) {
  1106. return !self.isHome
  1107. } else if menuItem.action == #selector(batchRemovPrivatySecurity) {
  1108. if self.isHome {
  1109. return false
  1110. }
  1111. guard let doc = self.mainViewController?.listView?.document else {
  1112. return false
  1113. }
  1114. let allowsPrinting = doc.allowsPrinting
  1115. let allowsCopying = doc.allowsCopying
  1116. if allowsCopying && allowsPrinting {
  1117. return false
  1118. }
  1119. return true
  1120. } else if menuItem.action == #selector(saveArchive) {
  1121. return !self.isHome
  1122. } else if (menuItem.action == #selector(saveTo(_ :))) {
  1123. return !self.isHome
  1124. } else if (menuItem.action == #selector(batchRemovePassWord)) {
  1125. return !self.isHome
  1126. } else if (menuItem.action == #selector(addBookmark)) {
  1127. if menuItem.tag == 3 {
  1128. return true
  1129. }else {
  1130. return !self.isHome
  1131. }
  1132. }
  1133. return super.validateMenuItem(menuItem)
  1134. }
  1135. }
  1136. extension NSDocument {
  1137. @objc class func isDamage(url: URL) -> Bool {
  1138. // 文件路径是否存在
  1139. if (FileManager.default.fileExists(atPath: url.path) == false) {
  1140. return true
  1141. }
  1142. /// PDF 格式文件
  1143. if (url.pathExtension.lowercased() == "pdf") {
  1144. let document = PDFDocument(url: url)
  1145. if (document == nil) {
  1146. return true
  1147. }
  1148. if (document!.isLocked) { // 加锁文件不在这里判断
  1149. return false
  1150. }
  1151. if (document!.pageCount <= 0) {
  1152. return true
  1153. }
  1154. return false
  1155. }
  1156. // 支持的图片格式
  1157. let imageExts = ["jpg","cur","bmp","jpeg","gif","png","tiff","tif","ico","icns","tga","psd","eps","hdr","jp2","jpc","pict","sgi","heic"]
  1158. let isImage = imageExts.contains(url.pathExtension.lowercased())
  1159. if (isImage == false) { // 其他格式目前返回没损坏,后续再补充(如果有需求)
  1160. return false
  1161. }
  1162. // 图片格式
  1163. let image = NSImage(contentsOf: url)
  1164. let data = image?.tiffRepresentation
  1165. if (data == nil) {
  1166. return true
  1167. }
  1168. let imageRep = NSBitmapImageRep(data: data!)
  1169. imageRep!.size = image!.size
  1170. var imageData: NSData?
  1171. if (url.pathExtension.lowercased() == "png") {
  1172. imageData = imageRep?.representation(using: .png, properties: [:]) as NSData?
  1173. } else {
  1174. imageData = imageRep?.representation(using: .jpeg, properties: [:]) as NSData?
  1175. }
  1176. if (imageData == nil) {
  1177. return true
  1178. }
  1179. let path = NSSearchPathForDirectoriesInDomains(.applicationSupportDirectory, .userDomainMask, true).last?.stringByAppendingPathComponent(Bundle.main.bundleIdentifier!)
  1180. if (FileManager.default.fileExists(atPath: path!) == false) {
  1181. try?FileManager.default.createDirectory(atPath: path!, withIntermediateDirectories: false)
  1182. }
  1183. var tagString: String = ""
  1184. let dateFormatter = DateFormatter()
  1185. dateFormatter.dateFormat = "yyMMddHHmmss"
  1186. tagString.append(dateFormatter.string(from: Date()))
  1187. tagString = tagString.appendingFormat("%04d", arc4random()%10000)
  1188. let filePath = path?.appending("/\(tagString).png")
  1189. if (imageData!.write(toFile: filePath!, atomically: true) == false) {
  1190. return true
  1191. }
  1192. // 删除临时图片
  1193. try?FileManager.default.removeItem(atPath: filePath!)
  1194. return false
  1195. }
  1196. @objc class func isDamage(url: URL, needAlertIfDamage need: Bool) -> Bool {
  1197. let result = self.isDamage(url: url)
  1198. if (result == false) {
  1199. return false
  1200. }
  1201. if (need == false) {
  1202. return true
  1203. }
  1204. let alert = NSAlert()
  1205. alert.messageText = NSLocalizedString("An error occurred while opening this document. The file is damaged and could not be repaired.", comment: "")
  1206. alert.runModal()
  1207. return true
  1208. }
  1209. }
  1210. // MARK: -
  1211. // MARK: 保存密码
  1212. extension NSDocument {
  1213. func savePasswordInKeychain(_ password: String, _ document: CPDFDocument) {
  1214. if (document.isLocked || password.isEmpty) {
  1215. return
  1216. }
  1217. let fileId = self.fileId(for: document)
  1218. if (fileId.isEmpty) {
  1219. return
  1220. }
  1221. // let status: SKPasswordStatus =
  1222. let label = "PDF Reader Pro: \(self.displayName!)"
  1223. SKKeychain.setPassword(password, item: nil, forService: self.passwordServiceName(), account: fileId, label: label, comment: self.fileURL?.path)
  1224. }
  1225. func getPassword(_ password: AutoreleasingUnsafeMutablePointer<NSString?>, fileId: String) {
  1226. let status = SKKeychain.getPassword(password, item: nil, forService: self.passwordServiceName(), account: fileId)
  1227. // if (status == .found) {
  1228. // }
  1229. }
  1230. fileprivate func fileId(for document: CPDFDocument) -> String {
  1231. return "\(document.documentURL.path.hash)"
  1232. }
  1233. private func passwordServiceName() -> String {
  1234. return "PDF Reader Pro password"
  1235. }
  1236. }
  1237. extension KMMainDocument: SKPDFSynchronizerDelegate {
  1238. func synchronizer(_ synchronizer: SKPDFSynchronizer!, foundLine line: Int, inFile file: String!) {
  1239. if FileManager.default.fileExists(atPath: file) {
  1240. let defaults = UserDefaults.standard
  1241. var editorPreset = defaults.string(forKey: SKTeXEditorPresetKey) ?? ""
  1242. var editorCmd: String?
  1243. var editorArgs: String?
  1244. var cmdString: String?
  1245. if !KMSyncPreferences.getTeXEditorCommand(command: &editorCmd, arguments: &editorArgs, forPreset: editorPreset) {
  1246. editorCmd = defaults.string(forKey: SKTeXEditorCommandKey)
  1247. editorArgs = defaults.string(forKey: SKTeXEditorArgumentsKey)
  1248. }
  1249. if var cmdString = editorArgs {
  1250. if !editorCmd!.hasPrefix("/") {
  1251. var searchPaths = ["/usr/bin", "/usr/local/bin"]
  1252. var toolPath: String?
  1253. let fm = FileManager.default
  1254. if !(editorPreset.isEmpty) {
  1255. if let path = NSWorkspace.shared.fullPath(forApplication: editorPreset) {
  1256. if let appBundle = Bundle(path: path) {
  1257. if let contentsPath = appBundle.path(forResource: "Contents", ofType: nil) {
  1258. searchPaths.insert(contentsPath, at: 0)
  1259. }
  1260. if editorPreset != "BBEdit", let execPath = appBundle.executablePath {
  1261. searchPaths.insert(execPath, at: 0)
  1262. }
  1263. if let resourcePath = appBundle.resourcePath {
  1264. searchPaths.insert(resourcePath, at: 0)
  1265. }
  1266. if let sharedSupportPath = appBundle.sharedSupportPath {
  1267. searchPaths.insert(sharedSupportPath, at: 0)
  1268. }
  1269. }
  1270. }
  1271. } else {
  1272. let appSupportDirs = FileManager.default.urls(for: .applicationSupportDirectory, in: .userDomainMask)
  1273. let appSupportPaths = appSupportDirs.map { $0.path }
  1274. searchPaths.append(contentsOf: appSupportPaths)
  1275. }
  1276. for path in searchPaths {
  1277. toolPath = (path as NSString).appendingPathComponent(editorCmd!)
  1278. if fm.isExecutableFile(atPath: toolPath!) {
  1279. editorCmd = toolPath
  1280. break
  1281. }
  1282. toolPath = ((path as NSString).appendingPathComponent("bin") as NSString).appendingPathComponent(editorCmd!)
  1283. if fm.isExecutableFile(atPath: toolPath!) {
  1284. editorCmd = toolPath
  1285. break
  1286. }
  1287. }
  1288. }
  1289. cmdString = cmdString.replacingOccurrences(of: "%line", with: "\(line + 1)")
  1290. cmdString = cmdString.replacingOccurrences(of: "%file", with: file)
  1291. cmdString = cmdString.replacingOccurrences(of: "%output", with: fileURL?.path ?? "")
  1292. cmdString.insert(contentsOf: "\" ", at: cmdString.startIndex)
  1293. cmdString.insert(contentsOf: editorCmd!, at: cmdString.startIndex)
  1294. cmdString.insert("\"", at: cmdString.startIndex)
  1295. let ws = NSWorkspace.shared
  1296. if let theUTI = try? ws.type(ofFile: editorCmd!) {
  1297. if ws.type(theUTI, conformsToType: "com.apple.applescript.script") || ws.type(theUTI, conformsToType: "com.apple.applescript.text") {
  1298. cmdString.insert(contentsOf: "/usr/bin/osascript ", at: cmdString.startIndex)
  1299. }
  1300. }
  1301. let task = Process()
  1302. task.launchPath = "/bin/sh"
  1303. task.currentDirectoryPath = (file as NSString).deletingLastPathComponent
  1304. task.arguments = ["-c", cmdString]
  1305. task.standardOutput = FileHandle.nullDevice
  1306. task.standardError = FileHandle.nullDevice
  1307. do {
  1308. try task.run()
  1309. } catch let error {
  1310. Swift.print("command failed: \(cmdString ?? ""): \(error)")
  1311. }
  1312. }
  1313. }
  1314. }
  1315. func synchronizer(_ synchronizer: SKPDFSynchronizer!, foundLocation point: NSPoint, atPageIndex pageIndex: UInt, options: Int) {
  1316. guard let pdfDoc = self.mainViewController?.document else { return }
  1317. if pageIndex < pdfDoc.pageCount {
  1318. if let page = pdfDoc.page(at: pageIndex) {
  1319. var adjustedPoint = point
  1320. if options & SKPDFSynchronizerFlippedMask != 0 {
  1321. let mediaBox = page.bounds(for: .mediaBox)
  1322. adjustedPoint.y = NSMaxY(mediaBox) - adjustedPoint.y
  1323. }
  1324. self.mainViewController?.listView.displayLine(at: adjustedPoint, inPageAtIndex: Int(pageIndex), showReadingBar: options & SKPDFSynchronizerShowReadingBarMask != 0)
  1325. }
  1326. }
  1327. }
  1328. }
  1329. extension KMMainDocument {
  1330. }