KMMainDocument.swift 62 KB

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