KMBatchProcessingView.swift 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325
  1. //
  2. // KMBatchProcessingView.swift
  3. // PDF Master
  4. //
  5. // Created by lizhe on 2022/11/16.
  6. //
  7. import Cocoa
  8. import KMComponentLibrary
  9. class KMBatchProcessingView: BaseXibView {
  10. @IBOutlet weak var tableView: KMBatchProcessingTableView!
  11. @IBOutlet weak var selectedFilesView: KMBatchSelectedFilesView!
  12. // @IBOutlet weak var addFilesButton: ComponentDropdownTool!
  13. @IBOutlet weak var deleteButton: ComponentButton!
  14. @IBOutlet weak var addFilesButton: ComponentButton!
  15. @IBOutlet weak var addFilesMoreButton: ComponentButton!
  16. @IBOutlet var addFilesSepline: ComponentDivider!
  17. @IBOutlet weak var lineView: NSView!
  18. var groupView: ComponentGroup?
  19. weak var delegate: KMBatchProcessingViewDelegate?
  20. var inputType: KMBatchCollectionViewType = .convertPDF {
  21. didSet {
  22. self.selectedFilesView.inputType = inputType
  23. self.tableView.inputType = inputType
  24. }
  25. }
  26. /**
  27. @abstract 外部传入数据
  28. @param inputData 文件路劲
  29. */
  30. var inputData: [URL]! {
  31. didSet {
  32. self.tableView.inputData = inputData
  33. self.reloadData()
  34. }
  35. }
  36. override func draw(_ dirtyRect: NSRect) {
  37. super.draw(dirtyRect)
  38. // Drawing code here.
  39. }
  40. deinit {
  41. NotificationCenter.default.removeObserver(self)
  42. }
  43. override func setup() {
  44. self.backgroundColor(ComponentLibrary.shared.getComponentColorFromKey("colorBg/layout-low"))
  45. self.border(ComponentLibrary.shared.getComponentColorFromKey("colorBorder/divider"), 0.5, 0)
  46. self.lineView.backgroundColor(ComponentLibrary.shared.getComponentColorFromKey("colorBorder/divider"))
  47. addFilesSepline.properties = ComponentDividerProperty(type: .vertical)
  48. let properties = ComponentButtonProperty()
  49. properties.type = .gray
  50. properties.size = .s
  51. properties.buttonText = KMLocalizedString("Add Files...")
  52. addFilesButton.properties = properties
  53. properties.propertyInfo.cornerRadius_topRight = 0
  54. properties.propertyInfo.cornerRadius_bottomRight = 0
  55. addFilesButton.reloadData()
  56. addFilesButton.setTarget(self, action: #selector(addFileAction))
  57. let properties2 = ComponentButtonProperty()
  58. properties2.type = .gray
  59. properties2.size = .s
  60. properties2.onlyIcon = true
  61. properties2.icon = NSImage(named: "KMImageNameImageToPDFDropdown")
  62. addFilesMoreButton.properties = properties2
  63. properties2.propertyInfo.cornerRadius_topLeft = 0
  64. properties2.propertyInfo.cornerRadius_bottomLeft = 0
  65. addFilesMoreButton.reloadData()
  66. addFilesMoreButton.setTarget(self, action: #selector(addFileAction))
  67. self.deleteButton.properties = ComponentButtonProperty(type: .default_tertiary,
  68. size: .s,
  69. state: .normal,
  70. onlyIcon: true,
  71. icon:NSImage(named: "icon_btn_clear_norm_false"),
  72. keepPressState: false)
  73. self.deleteButton.setTarget(self, action: #selector(cleanAll(_:)))
  74. self.deleteButton.reloadData()
  75. self.selectedFilesView.dragEnd = {[unowned self] (data) in
  76. self.inputData = data
  77. }
  78. self.selectedFilesView.addAction = {[unowned self] (sender) in
  79. self.addFiles()
  80. }
  81. self.tableView.delegate = self
  82. self.reloadData()
  83. NotificationCenter.default.addObserver(self, selector: #selector(bacthFilesProcessingNotification), name: NSNotification.Name(kBacthFilesProcessNotification), object: nil)
  84. }
  85. func reloadData() {
  86. if (self.tableView.data != nil) && self.tableView.data!.count != 0 {
  87. self.selectedFilesView.isHidden = true
  88. self.tableView.isHidden = false
  89. self.lineView.isHidden = false
  90. } else {
  91. self.selectedFilesView.isHidden = false
  92. self.tableView.isHidden = true
  93. self.lineView.isHidden = true
  94. }
  95. self.tableView.reloadData()
  96. }
  97. func showGroupView() {
  98. var viewHeight: CGFloat = 8
  99. var menuItemArr: [ComponentMenuitemProperty] = []
  100. for i in ["Add Files", "Add Folder"] {
  101. let properties_Menuitem: ComponentMenuitemProperty = ComponentMenuitemProperty(multipleSelect: false,
  102. itemSelected: false,
  103. isDisabled: false,
  104. keyEquivalent: nil,
  105. text: KMLocalizedString(i))
  106. menuItemArr.append(properties_Menuitem)
  107. viewHeight += 36
  108. }
  109. if groupView == nil {
  110. groupView = ComponentGroup.createFromNib(in: ComponentLibrary.shared.componentBundle())
  111. }
  112. groupView?.groupDelegate = self
  113. groupView?.frame = CGRectMake(310, 0, 200, viewHeight)
  114. groupView?.updateGroupInfo(menuItemArr)
  115. var point = addFilesButton.convert(addFilesButton.frame.origin, to: nil)
  116. point.y -= viewHeight
  117. groupView?.showWithPoint(point, relativeTo: addFilesButton)
  118. // addFileButton_?.properties.state = .pressed
  119. // addFileButton_?.reloadData()
  120. }
  121. }
  122. protocol KMBatchProcessingViewAction{}
  123. extension KMBatchProcessingView: KMBatchProcessingViewAction {
  124. @IBAction func cleanAll(_ sender: Any) {
  125. if self.tableView.selectModels.count != 0 {
  126. self.tableView.presenter.deleteData(model: (self.tableView.selectModels.first)!)
  127. self.tableView.selectModels.removeObject(self.tableView.selectModels.first!)
  128. } else {
  129. self.tableView.presenter.cleanAll()
  130. self.tableView.selectModels.removeAll()
  131. }
  132. self.reloadData()
  133. }
  134. func addFileAction() {
  135. self.showGroupView()
  136. }
  137. func addFiles() {
  138. KMBatchProcessingView.openfiles(window: window!) { [unowned self] openPanel in
  139. if self.inputType == .OCR {
  140. openPanel.allowedFileTypes = KMOCRModel.supportedTypes()
  141. } else if self.inputType == .imageToPDF {
  142. openPanel.allowedFileTypes = KMBatchProcessingTableViewModel.supportedImageTypes()
  143. } else {
  144. openPanel.allowedFileTypes = ["pdf","PDF"]
  145. }
  146. openPanel.prompt = "Open"
  147. openPanel.canChooseDirectories = false //是否允许选择目录
  148. openPanel.canChooseFiles = true //是否可以选择文件
  149. openPanel.allowsMultipleSelection = true //是否允许多选
  150. if openPanel.allowsMultipleSelection == true {
  151. openPanel.message = NSLocalizedString("Press the keyboard ⌘/⇧ to select multiple files", comment: "")
  152. }
  153. } completion: { [unowned self] (panel ,data) in
  154. if data.count != 0 {
  155. self.tableView.presenter.addData(data: data)
  156. self.reloadData()
  157. }
  158. }
  159. }
  160. func addFolder() {
  161. KMPrint("addFolder")
  162. KMBatchProcessingView.openfiles(window: window!) { [unowned self] openPanel in
  163. if self.inputType == .OCR {
  164. openPanel.allowedFileTypes = KMOCRModel.supportedTypes()
  165. } else if self.inputType == .imageToPDF {
  166. openPanel.allowedFileTypes = KMBatchProcessingTableViewModel.supportedImageTypes()
  167. } else {
  168. openPanel.allowedFileTypes = ["pdf","PDF"]
  169. }
  170. openPanel.prompt = "Open"
  171. openPanel.canChooseDirectories = true //是否允许选择目录
  172. openPanel.canChooseFiles = false //是否可以选择文件
  173. openPanel.allowsMultipleSelection = false //是否允许多选
  174. } completion: { [unowned self] (panel ,data) in
  175. if data.count != 0 {
  176. var filePaths: [URL] = []
  177. var types: [String] = []
  178. if self.inputType == .OCR {
  179. types = KMOCRModel.supportedTypes()
  180. } else if self.inputType == .imageToPDF {
  181. types = KMBatchProcessingTableViewModel.supportedImageTypes()
  182. } else {
  183. types = ["pdf","PDF"]
  184. }
  185. KMMergeTools.findAllFiles(folder: data.first!, result: &filePaths, types: types)
  186. self.tableView.presenter.addData(data: filePaths)
  187. self.reloadData()
  188. }
  189. }
  190. }
  191. func addOpenFiles() {
  192. // let array = ImageToPDFTools.fetchOpenDocumentFiles()
  193. // self.tableView.presenter.addData(data: array)
  194. // self.reloadData()
  195. }
  196. func canAddOpenFile() -> Bool {
  197. var result: Bool = false
  198. // let array = ImageToPDFTools.fetchOpenDocumentFiles()
  199. // if array.count != 0 {
  200. // result = true
  201. // }
  202. return result
  203. }
  204. }
  205. protocol KMBatchProcessingViewOpenPanel {}
  206. extension KMBatchProcessingView: KMBatchProcessingViewOpenPanel {
  207. /**
  208. @abstract 打开文件
  209. @param window 显示window
  210. @param title 标题
  211. */
  212. static func openfiles(window:NSWindow, openPanel:((_ openPanel: NSOpenPanel) -> Void)?, completion:@escaping/**逃逸闭包*/ (_ panel: NSOpenPanel ,_ data:[URL]) -> Void) {
  213. let panel = NSOpenPanel()
  214. if openPanel != nil {
  215. openPanel!(panel)
  216. } else {
  217. panel.prompt = ""
  218. panel.canChooseDirectories = false //是否允许选择目录
  219. panel.canChooseFiles = true //是否可以选择文件
  220. panel.allowsMultipleSelection = true //是否允许多选
  221. panel.allowedFileTypes = KMBatchProcessingTableViewModel.supportedImageTypes() //支持类型
  222. // let button = NSButton.init(checkboxWithTitle: "保存后打开文档", target: nil, action: nil)
  223. // button.state = .off
  224. // panel.accessoryView = button
  225. }
  226. panel.beginSheetModal(for: window, completionHandler: { result in
  227. if result != .OK {
  228. KMPrint("取消")
  229. completion(panel, [])
  230. } else {
  231. completion(panel, panel.urls)
  232. }
  233. })
  234. }
  235. }
  236. extension KMBatchProcessingView: KMBatchProcessingTableViewDelegate {
  237. func reloadData(data: [KMBatchProcessingTableViewModel]) {
  238. self.reloadData()
  239. if (self.delegate != nil) {
  240. self.delegate?.reloadData(data: data)
  241. }
  242. }
  243. }
  244. protocol KMBatchProcessingViewDelegate: NSObject {
  245. /**
  246. 数据刷新
  247. */
  248. func reloadData(data: [KMBatchProcessingTableViewModel])
  249. }
  250. extension KMBatchProcessingView: ComponentGroupDelegate {
  251. func componentGroupDidSelect(group: ComponentGroup?, menuItemProperty: ComponentMenuitemProperty?) {
  252. let familyString = menuItemProperty?.text
  253. if familyString == KMLocalizedString("Add Files", comment: "") {
  254. self.addFiles()
  255. } else if familyString == KMLocalizedString("Add Folder", comment: "") {
  256. self.addFolder()
  257. } else if familyString == KMLocalizedString("Add Open Files", comment: "") {
  258. self.addOpenFiles()
  259. }
  260. }
  261. func componentGroupDidDismiss(group: ComponentGroup?) {
  262. self.addFilesButton.properties.state = .normal
  263. self.addFilesMoreButton.properties.state = .normal
  264. self.addFilesButton.reloadData()
  265. self.addFilesMoreButton.reloadData()
  266. }
  267. }
  268. extension KMBatchProcessingView {
  269. @objc func bacthFilesProcessingNotification() {
  270. self.reloadData()
  271. }
  272. }