KMFileAttribute.swift 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176
  1. //
  2. // KMFileAttribute.swift
  3. // PDF Master
  4. //
  5. // Created by tangchao on 2023/10/12.
  6. //
  7. import Cocoa
  8. @objcMembers class KMFileAttribute: NSObject {
  9. var filePath: String?
  10. var oriFilePath: String?
  11. var myPDFDocument: PDFDocument?
  12. var pdfDocument: CPDFDocument?
  13. var bAllPage = true
  14. private var _selectPages: [NSNumber] = []
  15. var pagesString: String = ""
  16. var isLocked = false
  17. var password: String?
  18. var pagesType: KMPageRange = .all
  19. /*
  20. @property (nonatomic, assign) BOOL pageRangeError;
  21. */
  22. func fetchSelectPages() -> [NSNumber] {
  23. let _ = self.isInvalidString(self.pagesString)
  24. if(!self.bAllPage) {
  25. self.QuickSort(&self._selectPages, startIndex: 0, endIndex: self._selectPages.count-1)
  26. }
  27. return self._selectPages
  28. }
  29. func isInvalidString(_ text: String) -> Bool {
  30. var document: PDFDocument?
  31. if (self.myPDFDocument != nil) {
  32. document = self.myPDFDocument
  33. } else {
  34. document = PDFDocument(url: URL(fileURLWithPath: self.filePath ?? ""))
  35. }
  36. let pageNumber = document?.pageCount ?? 1
  37. if (self.bAllPage) {
  38. self._selectPages = []
  39. for i in 1 ... pageNumber {
  40. self._selectPages.append(NSNumber(value: i))
  41. }
  42. return false
  43. }
  44. var pageNumbers: [NSNumber] = []
  45. var isInvalid = false
  46. var c: unichar = 0
  47. for c in text {
  48. if (c != "0" && c != "1" && c != "2" && c != "3" && c != "4" && c != "5" && c != "6" && c != "7" && c != "8" && c != "9" && c != "," && c != "-") {
  49. isInvalid = true
  50. break
  51. }else{
  52. isInvalid = false
  53. }
  54. }
  55. if (!isInvalid) {
  56. let array = text.components(separatedBy: ",")
  57. for s in array {
  58. if s.isEmpty {
  59. isInvalid = true
  60. break
  61. }else{
  62. let pages = s.components(separatedBy: "-")
  63. if (pages.count>2) {
  64. isInvalid = true
  65. break
  66. }else if(pages.count==1){
  67. let p = pages.first!
  68. if p.isEmpty || Int(p)! > pageNumber || Int(p) == 0 {
  69. isInvalid = true
  70. break
  71. }else{
  72. var isEqual = false
  73. for pageNumber in pageNumbers {
  74. if pageNumber.stringValue == p {
  75. isEqual = true
  76. isInvalid = true
  77. break
  78. }
  79. }
  80. if (!isEqual) {
  81. pageNumbers.append(NSNumber(value: Int(p)!))
  82. }
  83. }
  84. }else if(pages.count==2){
  85. let p1 = pages[0]
  86. let p2 = pages[1]
  87. if p1.isEmpty || p2.isEmpty || Int(p1)! >= Int(p2)! || Int(p2)! > pageNumber || Int(p1) == 0 {
  88. isInvalid = true
  89. break
  90. }else{
  91. var isEqual = false
  92. for i in Int(p1)! ... Int(p2)! {
  93. for pageNumber in pageNumbers {
  94. if pageNumber.intValue == i {
  95. isEqual = true
  96. isInvalid = true
  97. break
  98. }
  99. }
  100. }
  101. if (!isEqual) {
  102. for i in Int(p1)! ... Int(p2)! {
  103. pageNumbers.append(NSNumber(value: i))
  104. }
  105. }
  106. }
  107. }
  108. }
  109. }
  110. }
  111. if (text.isEmpty) {
  112. isInvalid = true
  113. }
  114. if (isInvalid) {
  115. self._selectPages = []
  116. }else{
  117. self._selectPages = pageNumbers
  118. }
  119. return isInvalid
  120. }
  121. func QuickSort(_ list: inout [NSNumber],startIndex: Int, endIndex: Int) {
  122. if(startIndex >= endIndex) {
  123. return
  124. }
  125. let temp = list[startIndex]
  126. var tempIndex = startIndex
  127. for i in startIndex+1 ... endIndex {
  128. let t = list[i]
  129. if (temp.intValue > t.intValue) {
  130. tempIndex = tempIndex + 1
  131. let tmp = list[tempIndex]
  132. list[tempIndex] = list[i]
  133. list[i] = tmp
  134. }
  135. }
  136. let tmp = list[tempIndex]
  137. list[tempIndex] = list[startIndex]
  138. list[startIndex] = tmp
  139. self.QuickSort(&list, startIndex: startIndex, endIndex: tempIndex-1)
  140. self.QuickSort(&list, startIndex: tempIndex+1, endIndex: endIndex)
  141. }
  142. /*
  143. /* give our representation to the image browser */
  144. - (id)imageRepresentation
  145. {
  146. return self.filePath;
  147. }
  148. /* use the absolute filepath as identifier */
  149. - (NSString *)imageUID
  150. {
  151. return self.filePath;
  152. }
  153. - (NSString*)imageTitle
  154. {
  155. return [[self.filePath lastPathComponent] stringByDeletingPathExtension];
  156. }
  157. */
  158. }