KMTools.swift 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778
  1. //
  2. // KMTools.swift
  3. // PDF Reader Pro
  4. //
  5. // Created by tangchao on 2023/3/7.
  6. //
  7. import Cocoa
  8. @objc class KMTools: NSObject {
  9. // MARK: - 获取已打开的文件
  10. @objc class func getOpenDocumentURLs() -> [URL] {
  11. var files:[URL] = []
  12. for window in NSApp.windows {
  13. if ((window.windowController is KMBrowserWindowController) == false) {
  14. continue
  15. }
  16. let controller: KMBrowserWindowController = window.windowController as! KMBrowserWindowController
  17. let model = controller.browser?.tabStripModel
  18. guard let count = model?.count() else {
  19. continue
  20. }
  21. if (count <= 0) {
  22. continue
  23. }
  24. for i in 0 ..< count {
  25. let document = model?.tabContents(at: Int32(i))
  26. if (document?.fileURL == nil) {
  27. continue
  28. }
  29. if (document?.isHome == nil || document!.isHome) {
  30. continue
  31. }
  32. files.append((document?.fileURL)!)
  33. }
  34. }
  35. return files
  36. }
  37. // MARK: - 无法区分 [权限+开启] [开启] 这两种情况 请不要使用
  38. private class func isDocumentHasPermissionsPassword(_ url: URL) -> Bool {
  39. let document = PDFDocument(url: url)
  40. if (document == nil) {
  41. return false
  42. }
  43. if (document?.permissionsStatus == .user) {
  44. return true
  45. }
  46. // document?.permissionsStatus == .none
  47. if (document!.isLocked == false) { // 没有加锁
  48. return false
  49. }
  50. // 已加锁 [权限+开启] [开启]
  51. if (KMTools.hasPermissionsLimit(document!)) { // 有权限限制
  52. return true
  53. }
  54. return false
  55. }
  56. // MARK: - 暂时只处理了复制和打印两项(后续项目需求有新增时,可以再此方法里扩展)
  57. @objc class func hasPermissionsLimit(_ document: PDFDocument) -> Bool {
  58. if (document.allowsCopying == false) {
  59. return true
  60. }
  61. if (document.allowsPrinting == false) {
  62. return true
  63. }
  64. return false
  65. }
  66. // MARK: - 打开网页
  67. @objc class func openURL(url: URL?) {
  68. guard let _url = url else {
  69. KMPrint("url invalid.")
  70. return
  71. }
  72. NSWorkspace.shared.open(_url)
  73. }
  74. @objc class func openURL(urlString: String?) {
  75. guard let _urlString = urlString else {
  76. KMPrint("url invalid.")
  77. return
  78. }
  79. KMTools.openURL(url: URL(string: _urlString))
  80. }
  81. // MARK: - 查看文件
  82. @objc class func viewFile(at filepath: String) {
  83. let ws = NSWorkspace.shared
  84. let url = URL(fileURLWithPath: filepath)
  85. ws.activateFileViewerSelecting([url])
  86. }
  87. // MARK: - 获取 App 版本号
  88. @objc class func getAppVersion() -> String {
  89. let infoDictionary = Bundle.main.infoDictionary
  90. if (infoDictionary == nil) {
  91. return "1.0.0"
  92. }
  93. var version = infoDictionary!["CFBundleShortVersionString"]
  94. if (version != nil && (version is String) && (version as! String).isEmpty == false) {
  95. return version as! String
  96. }
  97. version = infoDictionary!["CFBundleVersion"]
  98. if (version != nil && (version is String) && (version as! String).isEmpty == false) {
  99. return version as! String
  100. }
  101. return "1.0.0"
  102. }
  103. class func getSystemVersion() -> (Int, Int, Int) {
  104. let versionInfo = ProcessInfo.processInfo.operatingSystemVersion
  105. return (versionInfo.majorVersion, versionInfo.minorVersion, versionInfo.patchVersion)
  106. }
  107. @objc class func isDefaultPDFReader() -> Bool {
  108. let app = LSCopyDefaultRoleHandlerForContentType("pdf" as CFString, LSRolesMask.all)?.takeUnretainedValue()
  109. if (app == nil) {
  110. return false
  111. }
  112. return (app! as String) == Bundle.main.bundleIdentifier!
  113. }
  114. @objc class func setDefaultPDFReader(_ isOrNo: Bool) -> Bool {
  115. var bid = "com.apple.Preview"
  116. if (isOrNo) {
  117. bid = Bundle.main.bundleIdentifier!
  118. }
  119. let status: OSStatus = LSSetDefaultRoleHandlerForContentType(KMTools.UTIforFileExtension("pdf") as CFString, LSRolesMask.all, bid as CFString)
  120. if (status == 0) {
  121. return true
  122. }
  123. return false
  124. }
  125. @objc class func UTIforFileExtension(_ exn: String) -> String {
  126. return (UTTypeCreatePreferredIdentifierForTag(kUTTagClassFilenameExtension, exn as CFString, nil)?.takeUnretainedValue())! as String
  127. }
  128. // MARK: - 是否全屏
  129. @objc class func isFullScreen(_ window: NSWindow) -> Bool {
  130. return window.styleMask.contains(.fullScreen)
  131. }
  132. // MARK: - 文件类型
  133. static let imageExtensions = ["jpg","cur","bmp","jpeg","gif","png","tiff","tif",/*@"pic",*/"ico","icns","tga","psd","eps","hdr","jp2","jpc","pict","sgi","heic"]
  134. static let pdfExtensions = ["pdf"]
  135. static let officeExtensions = ["doc", "docx", "xls", "xlsx", "ppt", "pptx"]
  136. @objc class func isImageType(_ exn: String) -> Bool {
  137. return KMTools.imageExtensions.contains(exn.lowercased())
  138. }
  139. @objc class func isPDFType(_ exn: String) -> Bool {
  140. return KMTools.pdfExtensions.contains(exn.lowercased())
  141. }
  142. @objc class func isOfficeType(_ exn: String) -> Bool {
  143. return KMTools.officeExtensions.contains(exn.lowercased())
  144. }
  145. @objc class func getUniqueFilePath(filePath: String) -> String {
  146. var isDirectory: ObjCBool = false
  147. var uniqueFilePath = filePath
  148. let fileManager = FileManager.default
  149. fileManager.fileExists(atPath: uniqueFilePath, isDirectory: &isDirectory)
  150. var i = 0
  151. if (isDirectory.boolValue) {
  152. while fileManager.fileExists(atPath: uniqueFilePath) {
  153. i += 1
  154. uniqueFilePath = "\(filePath)(\(i))"
  155. }
  156. } else {
  157. let fileURL = URL(fileURLWithPath: filePath)
  158. let path = fileURL.deletingPathExtension().path
  159. while fileManager.fileExists(atPath: uniqueFilePath) {
  160. i += 1
  161. uniqueFilePath = "\(path)(\(i)).\(fileURL.pathExtension)"
  162. }
  163. }
  164. return uniqueFilePath
  165. }
  166. @objc class func getTempFloderPath() -> String? {
  167. return self.getTempRootPath()?.stringByAppendingPathComponent("KMTemp")
  168. }
  169. @objc class func getTempRootPath() -> String? {
  170. return NSSearchPathForDirectoriesInDomains(FileManager.SearchPathDirectory.applicationSupportDirectory, FileManager.SearchPathDomainMask.userDomainMask, true).last?.stringByAppendingPathComponent(Bundle.main.bundleIdentifier!)
  171. }
  172. // MARK: - Document isDocumentEdited
  173. @objc class func setDocumentEditedState(window: NSWindow) {
  174. guard let _document = NSDocumentController.shared.document(for: window) else {
  175. return
  176. }
  177. self.setDocumentEditedState(document: _document)
  178. }
  179. @objc class func setDocumentEditedState(url: URL) {
  180. guard let _document = NSDocumentController.shared.document(for: url) else {
  181. return
  182. }
  183. self.setDocumentEditedState(document: _document)
  184. }
  185. @objc class func setDocumentEditedState(document: NSDocument) {
  186. km_synchronized(document) {
  187. document.updateChangeCount(.changeDone)
  188. }
  189. }
  190. @objc class func clearDocumentEditedState(window: NSWindow) {
  191. guard let _document = NSDocumentController.shared.document(for: window) else {
  192. return
  193. }
  194. self.clearDocumentEditedState(document: _document)
  195. }
  196. @objc class func clearDocumentEditedState(url: URL) {
  197. guard let _document = NSDocumentController.shared.document(for: url) else {
  198. return
  199. }
  200. self.clearDocumentEditedState(document: _document)
  201. }
  202. @objc class func clearDocumentEditedState(document: NSDocument) {
  203. km_synchronized(document) {
  204. document.updateChangeCount(.changeCleared)
  205. }
  206. }
  207. // MARK: - 解析 [1-3,5-7]
  208. @objc class func parseIndexSet(indexSet: IndexSet) -> String {
  209. return self.parseIndexs(indexs: indexSet.sorted())
  210. }
  211. @objc class func parseIndexs(indexs: [Int]) -> String {
  212. if (indexs.isEmpty) {
  213. return ""
  214. }
  215. if (indexs.count == 1) {
  216. return "\(indexs.first!+1)"
  217. }
  218. var sortArray: [Int] = []
  219. for i in indexs {
  220. sortArray.append(i)
  221. }
  222. /// 排序 (升序)
  223. sortArray.sort(){$0 < $1}
  224. var a: Int = 0
  225. var b: Int = 0
  226. var result: String?
  227. for i in sortArray {
  228. if (result == nil) {
  229. a = i
  230. b = i
  231. result = ""
  232. continue
  233. }
  234. if (i == b+1) {
  235. b = i
  236. if (i == sortArray.last) {
  237. result?.append(String(format: "%d-%d", a+1,b+1))
  238. }
  239. } else {
  240. if (a == b) {
  241. result?.append(String(format: "%d,", a+1))
  242. } else {
  243. result?.append(String(format: "%d-%d,", a+1,b+1))
  244. }
  245. a = i
  246. b = i
  247. if (i == sortArray.last) {
  248. result?.append(String(format: "%d", a+1))
  249. }
  250. }
  251. }
  252. return result ?? ""
  253. }
  254. }
  255. // MARK: - PDFReaderPro
  256. let KMNCompanyWebsite = NSLocalizedString("https://www.pdfreaderpro.com", comment: "")
  257. let APPMain_Link = NSLocalizedString("https://www.pdfreaderpro.com/mac", comment: "")
  258. //跳转Store界面
  259. let Store_Link = NSLocalizedString("https://www.pdfreaderpro.com/store", comment: "")
  260. //PDF to office pack
  261. let PDFToOfficePacket_Link = NSLocalizedString("https://www.pdfreaderpro.com/store/upgrade?utm_source=appdmg&utm_medium=toptbr&utm_campaign=pdftooffice", comment: "")
  262. //跨平台
  263. let CrossPlatform_Link = NSLocalizedString("https://www.pdfreaderpro.com/store/pdf-pro-permanent-cross-platform-package?utm_source=appdmg&utm_medium=toptbr&utm_campaign=crossplatform", comment: "")
  264. //AI包
  265. let AIProduct_Link = NSLocalizedString("https://www.pdfreaderpro.com/store/aitools?utm_source=appdmg&utm_medium=allapp&utm_campaign=aitool", comment: "")
  266. //教育折扣
  267. let Store_Edu_Link = NSLocalizedString("https://www.pdfreaderpro.com/store?mode=edu&utm_source=appdmg&utm_medium=toptbr&utm_campaign=edu", comment: "")
  268. let APP_Privacy_Policy_URL = NSLocalizedString("https://www.pdfreaderpro.com/privacy-policy", comment: "")
  269. let APP_Terms_Of_Service_URL = NSLocalizedString("https://www.pdfreaderpro.com/terms_of_service", comment: "")
  270. // MARK: - 快速工具
  271. extension KMTools {
  272. // 打开 [快速教学]
  273. @objc class func openQuickStartStudy() {
  274. var fileName = KMLocalizedString("Quick Start Guide.pdf")
  275. let fileType = "pdf"
  276. let path = Bundle.main.path(forResource: fileName, ofType: "")
  277. if (path == nil || FileManager.default.fileExists(atPath: path!) == false) {
  278. KMTools.openURL(url: URL(string: "https://www.pdfreaderpro.com/help"))
  279. return
  280. }
  281. fileName = String.deletingPathExtension(fileName)
  282. let version = KMTools.getAppVersion()
  283. fileName.append(" v\(version).\(fileType)")
  284. let folderPath = NSSearchPathForDirectoriesInDomains(.applicationSupportDirectory, .userDomainMask, true).last?.appending("/\(Bundle.main.bundleIdentifier!)")
  285. if (FileManager.default.fileExists(atPath: folderPath!) == false) {
  286. try?FileManager.default.createDirectory(atPath: folderPath!, withIntermediateDirectories: false)
  287. }
  288. let toPath = "\(folderPath!)/\(fileName)"
  289. if (FileManager.default.fileExists(atPath: toPath)) {
  290. try?FileManager.default.removeItem(atPath: toPath)
  291. }
  292. try?FileManager.default.copyItem(atPath: path!, toPath: toPath)
  293. NSDocumentController.shared.km_safe_openDocument(withContentsOf: URL(fileURLWithPath: toPath), display: true) { _, _, _ in
  294. }
  295. }
  296. // 打开 [FAQ] 网站
  297. @objc class func openFAQWebsite() {
  298. var tStrUrl: String?
  299. #if VERSION_FREE
  300. #if VERSION_DMG
  301. tStrUrl = "https://www.pdfreaderpro.com/help?utm_source=MacAppDmg&utm_campaign=HelpLink&utm_medium=PdfHelp"
  302. #else
  303. tStrUrl = "https://www.pdfreaderpro.com/help?utm_source=MacAppLite&utm_campaign=HelpLink&utm_medium=PdfHelp"
  304. #endif
  305. #else
  306. tStrUrl = "https://www.pdfreaderpro.com/help?utm_source=MacApp&utm_campaign=HelpLink&utm_medium=PdfHelp"
  307. #endif
  308. KMTools.openURL(urlString: tStrUrl)
  309. }
  310. // 打开 [更多产品] 网站
  311. @objc class func openMoreProductWebsite() {
  312. var tStrUrl: String?
  313. if KMKdanRemoteConfig.remoteConfig.showHelp_More_RecommendLink() {
  314. #if VERSION_FREE
  315. tStrUrl = "https://www.pdfreaderpro.com/product?utm_source=MacAppLite&utm_campaign=ProductLink&utm_medium=PdfProduct"
  316. #else
  317. tStrUrl = "https://www.pdfreaderpro.com/product?utm_source=MacApp&utm_campaign=ProductLink&utm_medium=PdfProduct"
  318. #endif
  319. } else {
  320. tStrUrl = "https://apps.apple.com/developer/pdf-technologies-inc/id1263126485"
  321. }
  322. #if VERSION_DMG
  323. tStrUrl = "https://www.pdfreaderpro.com/product?utm_source=MacAppDmg&utm_campaign=ProductLink&utm_medium=PdfProduct"
  324. #endif
  325. KMTools.openURL(urlString: tStrUrl)
  326. }
  327. // 打开 [免费 PDF 模板] 网站
  328. @objc class func openFreePDFTemplatesWebsite() {
  329. var tStrUrl: String?
  330. #if VERSION_FREE
  331. #if VERSION_DMG
  332. tStrUrl = "https://www.pdfreaderpro.com/templates?utm_source=MacAppDmg&utm_campaign=TemplatesLink&utm_medium=PdfTemplates"
  333. #else
  334. tStrUrl = "https://www.pdfreaderpro.com/templates?utm_source=MacAppLite&utm_campaign=TemplatesLink&utm_medium=PdfTemplates"
  335. #endif
  336. #else
  337. tStrUrl = "https://www.pdfreaderpro.com/templates?utm_source=MacApp&utm_campaign=TemplatesLink&utm_medium=PdfTemplates"
  338. #endif
  339. KMTools.openURL(urlString: tStrUrl)
  340. }
  341. // 打开 [ComPDFKit 授权] 网站
  342. @objc class func openComPDFKitPowerWebsite() {
  343. KMTools.openURL(url: URL(string: "https://www.compdf.com?utm_source=macapp&utm_medium=pdfmac&utm_campaign=compdfkit-promp"))
  344. }
  345. // 打开 [官网 下载页] 网站
  346. // 测试环境 http://test-pdf-pro.kdan.cn:3021/pdf-master-mac-download
  347. @objc class func openDownloadDMGWebsite() {
  348. KMTools.openURL(urlString: "https://www.pdfreaderpro.com/pdf-master-mac-download")
  349. }
  350. @objc class func openPurchaseProductWebsite() {
  351. KMTools.openURL(urlString: Store_Link)
  352. }
  353. @objc class func openVPPWebSite() {
  354. var tStrHttpUrl = "https://www.pdfreaderpro.com/vpp-purchase-program"
  355. #if VERSION_DMG
  356. tStrHttpUrl = KMNCompanyWebsite + "/vpp-purchase-program?utm_source=pdfdmg&utm_campaign=menubar_vpp"
  357. #else
  358. tStrHttpUrl = KMNCompanyWebsite + "/vpp-purchase-program?utm_source=pdfmacstore&utm_campaign=menubar_vpp"
  359. #endif
  360. KMTools.openURL(urlString: tStrHttpUrl)
  361. }
  362. @objc class func openEducateWebSite() {
  363. let tStrHttpUrl = KMLocalizedString("https://www.pdfreaderpro.com/store?mode=edu#utm_source=mac_app&utm_medium=pdf_pro_dmg&utm_content=menu_bar_click")
  364. KMTools.openURL(urlString: tStrHttpUrl)
  365. }
  366. @objc class func openPrivacyPolicyWebSite() {
  367. KMTools.openURL(urlString: APP_Privacy_Policy_URL)
  368. }
  369. @objc class func openTermOfServiceWebSite() {
  370. KMTools.openURL(urlString: APP_Terms_Of_Service_URL)
  371. }
  372. }
  373. extension KMTools {
  374. // 意见反馈
  375. @objc class func feekback() {
  376. let (major, minor, bugFix) = KMTools.getSystemVersion()
  377. let versionInfoString = "\(KMTools.getRawSystemInfo()) - \(major).\(minor).\(bugFix)"
  378. let appVersion = KMTools.getAppVersion()
  379. let appName = KMNAppName
  380. let subjects = "\(appName) - \(appVersion);\(NSLocalizedString("Feedback", comment: ""));\(versionInfoString)"
  381. let email = "support@pdfreaderpro.com"
  382. KMMailHelper.newEmail(withContacts: email, andSubjects: subjects)
  383. }
  384. //
  385. @objc class func reportBug() {
  386. let (major, minor, bugFix) = KMTools.getSystemVersion()
  387. let versionInfoString = "\(KMTools.getRawSystemInfo()) - \(major).\(minor).\(bugFix)"
  388. let appVersion = KMTools.getAppVersion()
  389. let appName = KMNAppName
  390. let subjects = "\(appName) - \(appVersion);\(NSLocalizedString("Report a Bug", comment: ""));\(versionInfoString)"
  391. let email = "support@pdfreaderpro.com"
  392. KMMailHelper.newEmail(withContacts: email, andSubjects: subjects)
  393. }
  394. //
  395. @objc class func proposeNewFeature() {
  396. let (major, minor, bugFix) = KMTools.getSystemVersion()
  397. let versionInfoString = "\(KMTools.getRawSystemInfo()) - \(major).\(minor).\(bugFix)"
  398. let appVersion = KMTools.getAppVersion()
  399. let appName = KMNAppName
  400. let subjects = "\(appName) - \(appVersion);\(NSLocalizedString("Propose a New Feature", comment: ""));\(versionInfoString)"
  401. let email = "support@pdfreaderpro.com"
  402. KMMailHelper.newEmail(withContacts: email, andSubjects: subjects)
  403. }
  404. @objc class func reportGeneralQuestions() {
  405. let (major, minor, bugFix) = KMTools.getSystemVersion()
  406. let versionInfoString = "\(KMTools.getRawSystemInfo()) - \(major).\(minor).\(bugFix)"
  407. let appVersion = KMTools.getAppVersion()
  408. let appName = KMNAppName
  409. let subjects = "\(appName) - \(appVersion);\(NSLocalizedString("General Questions", comment: ""));\(versionInfoString)"
  410. let email = "support@pdfreaderpro.com"
  411. KMMailHelper.newEmail(withContacts: email, andSubjects: subjects)
  412. }
  413. @objc class func rateUs() {
  414. #if VERSION_FREE
  415. iRate.sharedInstance().appStoreID = 919472673
  416. #else
  417. iRate.sharedInstance().appStoreID = 825459243
  418. #endif
  419. if UserDefaults.standard.bool(forKey: "kUserHaveClickRateUsKey") == false {
  420. UserDefaults.standard.set(true, forKey: "kUserHaveClickRateUsKey")
  421. UserDefaults.standard.synchronize()
  422. NotificationCenter.default.post(name: NSNotification.Name(rawValue: "kUserHaveClickRateUsNotification"), object: self)
  423. }
  424. iRate.sharedInstance().openRatingsPageInAppStore()
  425. }
  426. }
  427. extension KMTools {
  428. @objc class func getRawSystemInfo() -> String {
  429. let info = GBDeviceInfo.deviceInfo().rawSystemInfoString
  430. if (info == nil) {
  431. return ""
  432. }
  433. return info!
  434. }
  435. @objc class func pageRangeTypeString(pageRange: KMPageRange) -> String {
  436. switch pageRange {
  437. case .all:
  438. return NSLocalizedString("All Pages", comment: "")
  439. case .current:
  440. return NSLocalizedString("Current Page", comment: "")
  441. case .odd:
  442. return NSLocalizedString("Odd Pages", comment: "")
  443. case .even:
  444. return NSLocalizedString("Even Pages", comment: "")
  445. case .custom:
  446. return NSLocalizedString("Customize", comment: "")
  447. case .horizontal:
  448. return NSLocalizedString("Horizontal Pages", comment: "")
  449. case .vertical:
  450. return NSLocalizedString("Vertical Pages", comment: "")
  451. }
  452. }
  453. @objc class func pageRangePlaceholderString() -> String {
  454. return NSLocalizedString("e.g. 1,3-5,10", comment: "")
  455. }
  456. @objc class func saveWatermarkDocumentToTemp(document: CPDFDocument, secureOptions: [CPDFDocumentWriteOption : Any]? = nil, removePWD: Bool = false) -> URL? {
  457. // 将文档存入临时目录
  458. if let data = self.getTempFloderPath(), !FileManager.default.fileExists(atPath: data) {
  459. try?FileManager.default.createDirectory(atPath: data, withIntermediateDirectories: false)
  460. }
  461. guard let filePath = self.getTempFloderPath()?.stringByAppendingPathComponent("temp_saveDocumentFor_temp.pdf") else {
  462. return nil
  463. }
  464. // 清除临时数据
  465. if (FileManager.default.fileExists(atPath: filePath)) {
  466. try?FileManager.default.removeItem(atPath: filePath)
  467. }
  468. return self.saveWatermarkDocument(document: document, to: URL(fileURLWithPath: filePath), secureOptions: secureOptions, removePWD: removePWD)
  469. }
  470. @objc class func saveWatermarkDocument(document: CPDFDocument, to url: URL, secureOptions: [CPDFDocumentWriteOption : Any]? = nil, documentAttribute: [CPDFDocumentAttribute : Any]? = nil, removePWD: Bool = false) -> URL? {
  471. guard let _document = self._saveDocumentForWatermark(document: document) else {
  472. return nil
  473. }
  474. // 保存文档
  475. if let data = secureOptions, !data.isEmpty {
  476. _document.setDocumentAttributes(documentAttribute)
  477. _document.write(to: url, withOptions: data)
  478. } else if (removePWD) {
  479. _document.writeDecrypt(to: url)
  480. } else {
  481. _document.write(to: url)
  482. }
  483. // 清除临时数据
  484. if let _fileUrl = _document.documentURL, FileManager.default.fileExists(atPath: _fileUrl.path) {
  485. try?FileManager.default.removeItem(atPath: _fileUrl.path)
  486. }
  487. return url
  488. }
  489. @objc class func saveWatermarkDocumentForCompress(document: CPDFDocument, to url: URL, imageQuality: Int) -> URL? {
  490. guard let _document = self._saveDocumentForWatermark(document: document) else {
  491. return nil
  492. }
  493. // _document.write(to: _document.documentURL)
  494. // 保存文档
  495. let result = _document.writeOptimize(to: url, withOptions: [.imageQualityOption : imageQuality])
  496. // 清除临时数据
  497. if let _fileUrl = _document.documentURL, FileManager.default.fileExists(atPath: _fileUrl.path) {
  498. try?FileManager.default.removeItem(atPath: _fileUrl.path)
  499. }
  500. if (result) {
  501. return url
  502. }
  503. return nil
  504. }
  505. @objc class func saveWatermarkDocumentForFlatten(document: CPDFDocument, to url: URL) -> URL? {
  506. guard let _document = self._saveDocumentForWatermark(document: document) else {
  507. return nil
  508. }
  509. // 保存文档
  510. let result = _document.writeFlatten(to: url)
  511. // 清除临时数据
  512. if let _fileUrl = _document.documentURL, FileManager.default.fileExists(atPath: _fileUrl.path) {
  513. try?FileManager.default.removeItem(atPath: _fileUrl.path)
  514. }
  515. if (result) {
  516. return url
  517. }
  518. return nil
  519. }
  520. @objc class func saveDocumentToTemp(document: CPDFDocument, fileID: String, needUnlock: Bool = true) -> URL? {
  521. // 将文档存入临时目录
  522. if let data = self.getTempFloderPath(), !FileManager.default.fileExists(atPath: data) {
  523. if let rootPath = self.getTempRootPath(), !FileManager.default.fileExists(atPath: rootPath) {
  524. try?FileManager.default.createDirectory(atPath: rootPath, withIntermediateDirectories: false)
  525. }
  526. try?FileManager.default.createDirectory(atPath: data, withIntermediateDirectories: false)
  527. }
  528. guard let filePath = self.getTempFloderPath()?.stringByAppendingPathComponent("temp_saveDocumentFor\(fileID).pdf") else {
  529. return nil
  530. }
  531. // 清除临时数据
  532. if (FileManager.default.fileExists(atPath: filePath)) {
  533. try?FileManager.default.removeItem(atPath: filePath)
  534. }
  535. document.write(toFile: filePath)
  536. if (!FileManager.default.fileExists(atPath: filePath)) {
  537. return nil
  538. }
  539. guard let _document = CPDFDocument(url: URL(fileURLWithPath: filePath)) else {
  540. return nil
  541. }
  542. if (!needUnlock) {
  543. return _document.documentURL
  544. }
  545. // 如果加锁,则去解锁
  546. if let pwd = document.password, !pwd.isEmpty, _document.isLocked {
  547. _document.unlock(withPassword: document.password)
  548. }
  549. if (_document.isLocked) {
  550. return nil
  551. }
  552. return _document.documentURL
  553. }
  554. @objc class func trackEvent(type: KMSubscribeWaterMarkType) -> Void {
  555. }
  556. private static var dateFormatter_: DateFormatter?
  557. @objc class func timeString(timeDate date: Date) -> String {
  558. if dateFormatter_ == nil {
  559. dateFormatter_ = DateFormatter()
  560. }
  561. let calendar = Calendar.current
  562. let nowCmps = calendar.dateComponents([.day, .month, .year], from: Date())
  563. let currentCmps = calendar.dateComponents([.day, .month, .year], from: date)
  564. if (currentCmps.year == nowCmps.year) {
  565. if (currentCmps.month == nowCmps.month && currentCmps.day == nowCmps.day) {
  566. dateFormatter_?.dateFormat = "HH:mm"
  567. } else {
  568. dateFormatter_?.dateFormat = "MM-dd, HH:mm"
  569. }
  570. } else {
  571. dateFormatter_?.dateFormat = "yyyy-MM-dd, HH:mm"
  572. }
  573. return dateFormatter_?.string(from: date) ?? ""
  574. }
  575. @objc class func timeString(timeDate date: Date, formatString: String) -> String {
  576. if dateFormatter_ == nil {
  577. dateFormatter_ = DateFormatter()
  578. }
  579. let calendar = Calendar.current
  580. let nowCmps = calendar.dateComponents([.day, .month, .year], from: Date())
  581. let currentCmps = calendar.dateComponents([.day, .month, .year], from: date)
  582. dateFormatter_?.dateFormat = formatString
  583. return dateFormatter_?.string(from: date) ?? ""
  584. }
  585. @objc class func isFileGreaterThan10MB(atPath filePath: String) -> Bool {
  586. let fileManager = FileManager.default
  587. do {
  588. let fileAttributes = try fileManager.attributesOfItem(atPath: filePath)
  589. if let fileSize = fileAttributes[.size] as? UInt64 {
  590. let megabyteSize = fileSize / (1024 * 1024)
  591. return megabyteSize >= 10
  592. }
  593. } catch {
  594. KMPrint("Error: \(error)")
  595. }
  596. return false
  597. }
  598. // MARK: - Private Methods
  599. @objc fileprivate class func _documentAddWatermark(document: CPDFDocument) -> CPDFDocument? {
  600. // 添加水印
  601. let watermark = CPDFWatermark(document: document, type: .image)
  602. watermark?.image = NSImage(named: "KMImageNameWatermark")
  603. watermark?.horizontalPosition = .left
  604. watermark?.verticalPosition = .top
  605. watermark?.scale = 0.5
  606. document.addWatermark(watermark)
  607. // 添加 link注释
  608. var watermarkAnnoBounds = NSMakeRect(0, 0, 120, 32)
  609. for i in 0 ..< document.pageCount {
  610. guard let page = document.page(at: i) else {
  611. continue
  612. }
  613. // 水印注释 frame
  614. watermarkAnnoBounds.origin.y = page.bounds.size.height-watermarkAnnoBounds.size.height
  615. // 找到需要删除的水印注释(之前添加)
  616. var flagAnnos: [CPDFAnnotation] = []
  617. for anno in page.annotations {
  618. if let anno_link = anno as? CPDFLinkAnnotation, anno_link.url() == Store_Link, anno_link.bounds.equalTo(watermarkAnnoBounds) {
  619. flagAnnos.append(anno_link)
  620. }
  621. }
  622. // 删除之前的水印注释
  623. for anno in flagAnnos {
  624. page.removeAnnotation(anno)
  625. }
  626. // 新增新的水印注释
  627. let anno = CPDFLinkAnnotation(document: document)
  628. anno?.bounds = watermarkAnnoBounds
  629. anno?.setURL(Store_Link)
  630. page.addAnnotation(anno)
  631. }
  632. return document
  633. }
  634. @objc fileprivate class func _saveDocumentForWatermark(document: CPDFDocument) -> CPDFDocument? {
  635. // 将文档存入临时目录
  636. guard let _fileUrl = self.saveDocumentToTemp(document: document, fileID: "Watermark") else {
  637. return nil
  638. }
  639. guard let _document = CPDFDocument(url: _fileUrl) else {
  640. return nil
  641. }
  642. // 如果加锁,则去解锁
  643. if let pwd = document.password, !pwd.isEmpty, _document.isLocked {
  644. _document.unlock(withPassword: pwd)
  645. }
  646. // 添加水印
  647. return self._documentAddWatermark(document: _document)
  648. }
  649. }