KMFileAttribute.swift 8.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252
  1. //
  2. // KMFileAttribute.swift
  3. // PDF Master
  4. //
  5. // Created by tangchao on 2023/10/12.
  6. //
  7. import Cocoa
  8. @objc enum KMPDFSelectPageStringType: Int {
  9. case AllPages = 0
  10. case OnlyOdd
  11. case OnlyEven
  12. case PagesString
  13. }
  14. @objcMembers class KMFileAttribute: NSObject {
  15. var filePath: String = "" {
  16. didSet {
  17. self.reloadData()
  18. }
  19. }
  20. var oriFilePath: String?
  21. var myPDFDocument: PDFDocument?
  22. var pdfDocument: CPDFDocument?
  23. var bAllPage = true
  24. private var selectPages: [Int] = []
  25. var pagesString: String = ""
  26. var isLocked = false
  27. var password: String = "" {
  28. didSet {
  29. self.reloadData()
  30. }
  31. }
  32. var pagesType: KMPageRange = .all
  33. //新增参数
  34. var fileSize: CGFloat = 0
  35. var fileImage: NSImage = NSImage()
  36. /*
  37. @property (nonatomic, assign) BOOL pageRangeError;
  38. */
  39. func reloadData() {
  40. if filePath.count != 0 {
  41. let attribe = try?FileManager.default.attributesOfItem(atPath: filePath)
  42. fileSize = attribe?[FileAttributeKey.size] as! CGFloat
  43. let document = CPDFDocument.init(url: URL(fileURLWithPath: filePath))
  44. document?.unlock(withPassword: self.password)
  45. isLocked = document!.isLocked
  46. self.pdfDocument = document
  47. let page = document?.page(at: 0) ?? CPDFPage()
  48. let image = page.thumbnail(of: page.size)
  49. fileImage = image ?? NSImage()
  50. // fileImage = NSImage.previewForFile(path: URL(fileURLWithPath: filePath), ofSize: CGSizeMake(136, 214), asIcon: true) ?? NSImage()
  51. }
  52. }
  53. func fetchSelectPages() -> [Int] {
  54. if let doc = self.pdfDocument, doc.isLocked {
  55. doc.unlock(withPassword: self.password)
  56. }
  57. guard let pdfDocument = pdfDocument else { return [] }
  58. selectPages.removeAll()
  59. if pagesType == .all {
  60. for i in 0..<pdfDocument.pageCount {
  61. selectPages.append(Int(i) + 1)
  62. }
  63. pagesString = "1-\(pdfDocument.pageCount)"
  64. } else if pagesType == .odd {
  65. pagesString = ""
  66. for i in 0..<pdfDocument.pageCount where i % 2 == 0 {
  67. selectPages.append(Int(i) + 1)
  68. if pagesString == "" {
  69. pagesString = "\(i + 1)"
  70. } else {
  71. pagesString = "\(pagesString),\(i + 1)"
  72. }
  73. }
  74. } else if pagesType == .even {
  75. pagesString = ""
  76. for i in 0..<pdfDocument.pageCount where i % 2 != 0 {
  77. selectPages.append(Int(i) + 1)
  78. if pagesString == "" {
  79. pagesString = "\(i + 1)"
  80. } else {
  81. pagesString = "\(pagesString),\(i + 1)"
  82. }
  83. }
  84. } else {
  85. isInvalidString(pagesString)
  86. // let pages = KMPageRangeTools.findSelectPage(pageRangeString: pagesString, pageCount: Int(pdfDocument.pageCount ))
  87. // selectPages = pages
  88. if !bAllPage {
  89. self.QuickSort(&selectPages, startIndex: 0, endIndex: selectPages.count-1)
  90. }
  91. }
  92. return selectPages
  93. }
  94. func isInvalidString(_ text: String) -> Bool {
  95. var document: PDFDocument?
  96. if (self.myPDFDocument != nil) {
  97. document = self.myPDFDocument
  98. } else {
  99. document = PDFDocument(url: URL(fileURLWithPath: self.filePath ))
  100. }
  101. if let data = document?.isLocked, data {
  102. document?.unlock(withPassword: self.password )
  103. }
  104. let pageNumber = document?.pageCount ?? 1
  105. if (self.bAllPage) {
  106. self.selectPages = []
  107. for i in 1 ... pageNumber {
  108. self.selectPages.append(i)
  109. }
  110. return false
  111. }
  112. var pageNumbers: [Int] = []
  113. var isInvalid = false
  114. var c: unichar = 0
  115. for c in text {
  116. if (c != "0" && c != "1" && c != "2" && c != "3" && c != "4" && c != "5" && c != "6" && c != "7" && c != "8" && c != "9" && c != "," && c != "-") {
  117. isInvalid = true
  118. break
  119. }else{
  120. isInvalid = false
  121. }
  122. }
  123. if (!isInvalid) {
  124. let array = text.components(separatedBy: ",")
  125. for s in array {
  126. if s.isEmpty {
  127. isInvalid = true
  128. break
  129. }else{
  130. let pages = s.components(separatedBy: "-")
  131. if (pages.count>2) {
  132. isInvalid = true
  133. break
  134. }else if(pages.count==1){
  135. let p = pages.first!
  136. if p.isEmpty || Int(p)! > pageNumber || Int(p) == 0 {
  137. isInvalid = true
  138. break
  139. }else{
  140. var isEqual = false
  141. for pageNumber in pageNumbers {
  142. if pageNumber == Int(p) {
  143. isEqual = true
  144. isInvalid = true
  145. break
  146. }
  147. }
  148. if (!isEqual) {
  149. pageNumbers.append(Int(p)!)
  150. }
  151. }
  152. }else if(pages.count==2){
  153. let p1 = pages[0]
  154. let p2 = pages[1]
  155. if p1.isEmpty || p2.isEmpty || Int(p1)! >= Int(p2)! || Int(p2)! > pageNumber || Int(p1) == 0 {
  156. isInvalid = true
  157. break
  158. }else{
  159. var isEqual = false
  160. for i in Int(p1)! ... Int(p2)! {
  161. for pageNumber in pageNumbers {
  162. if pageNumber == i {
  163. isEqual = true
  164. isInvalid = true
  165. break
  166. }
  167. }
  168. }
  169. if (!isEqual) {
  170. for i in Int(p1)! ... Int(p2)! {
  171. pageNumbers.append(i)
  172. }
  173. }
  174. }
  175. }
  176. }
  177. }
  178. }
  179. if (text.isEmpty) {
  180. isInvalid = true
  181. }
  182. if (isInvalid) {
  183. self.selectPages = []
  184. }else{
  185. self.selectPages = pageNumbers
  186. }
  187. return isInvalid
  188. }
  189. func QuickSort(_ list: inout [Int],startIndex: Int, endIndex: Int) {
  190. if(startIndex >= endIndex) {
  191. return
  192. }
  193. let temp = list[startIndex]
  194. var tempIndex = startIndex
  195. for i in startIndex+1 ... endIndex {
  196. let t = list[i]
  197. if (temp > t) {
  198. tempIndex = tempIndex + 1
  199. let tmp = list[tempIndex]
  200. list[tempIndex] = list[i]
  201. list[i] = tmp
  202. }
  203. }
  204. let tmp = list[tempIndex]
  205. list[tempIndex] = list[startIndex]
  206. list[startIndex] = tmp
  207. self.QuickSort(&list, startIndex: startIndex, endIndex: tempIndex-1)
  208. self.QuickSort(&list, startIndex: tempIndex+1, endIndex: endIndex)
  209. }
  210. /*
  211. /* give our representation to the image browser */
  212. - (id)imageRepresentation
  213. {
  214. return self.filePath;
  215. }
  216. /* use the absolute filepath as identifier */
  217. - (NSString *)imageUID
  218. {
  219. return self.filePath;
  220. }
  221. - (NSString*)imageTitle
  222. {
  223. return [[self.filePath lastPathComponent] stringByDeletingPathExtension];
  224. }
  225. */
  226. }