KMBatchProcessingView.swift 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366
  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. var inputDataModel: [KMBatchProcessingTableViewModel]! {
  37. didSet {
  38. self.tableView.presenter.addDataModel(datas: inputDataModel)
  39. self.reloadData()
  40. }
  41. }
  42. override func draw(_ dirtyRect: NSRect) {
  43. super.draw(dirtyRect)
  44. // Drawing code here.
  45. }
  46. deinit {
  47. NotificationCenter.default.removeObserver(self)
  48. }
  49. override func setup() {
  50. self.backgroundColor(ComponentLibrary.shared.getComponentColorFromKey("colorBg/layout-low"))
  51. self.border(ComponentLibrary.shared.getComponentColorFromKey("colorBorder/divider"), 0.5, 0)
  52. self.lineView.backgroundColor(ComponentLibrary.shared.getComponentColorFromKey("colorBorder/divider"))
  53. addFilesSepline.properties = ComponentDividerProperty(type: .vertical)
  54. let properties = ComponentButtonProperty()
  55. properties.type = .gray
  56. properties.size = .s
  57. properties.buttonText = KMLocalizedString("Add Files...")
  58. addFilesButton.properties = properties
  59. properties.propertyInfo.cornerRadius_topRight = 0
  60. properties.propertyInfo.cornerRadius_bottomRight = 0
  61. addFilesButton.reloadData()
  62. addFilesButton.setTarget(self, action: #selector(addFileAction))
  63. let properties2 = ComponentButtonProperty()
  64. properties2.type = .gray
  65. properties2.size = .s
  66. properties2.onlyIcon = true
  67. properties2.icon = NSImage(named: "KMImageNameImageToPDFDropdown")
  68. addFilesMoreButton.properties = properties2
  69. properties2.propertyInfo.cornerRadius_topLeft = 0
  70. properties2.propertyInfo.cornerRadius_bottomLeft = 0
  71. addFilesMoreButton.reloadData()
  72. addFilesMoreButton.setTarget(self, action: #selector(addFileAction))
  73. self.deleteButton.properties = ComponentButtonProperty(type: .default_tertiary,
  74. size: .s,
  75. state: .normal,
  76. onlyIcon: true,
  77. icon:NSImage(named: "icon_btn_clear_norm_false"),
  78. keepPressState: false)
  79. self.deleteButton.setTarget(self, action: #selector(cleanAll(_:)))
  80. self.deleteButton.reloadData()
  81. self.selectedFilesView.dragEnd = {[unowned self] (data) in
  82. self.inputData = data
  83. }
  84. self.selectedFilesView.addAction = {[unowned self] (sender) in
  85. self.addFiles()
  86. }
  87. self.tableView.delegate = self
  88. self.reloadData()
  89. NotificationCenter.default.addObserver(self, selector: #selector(bacthFilesProcessingNotification), name: NSNotification.Name(kBacthFilesProcessNotification), object: nil)
  90. }
  91. func reloadData() {
  92. if (self.tableView.data != nil) && self.tableView.data!.count != 0 {
  93. self.deleteButton.properties.isDisabled = false
  94. self.selectedFilesView.isHidden = true
  95. self.tableView.isHidden = false
  96. self.lineView.isHidden = false
  97. } else {
  98. self.selectedFilesView.isHidden = false
  99. self.tableView.isHidden = true
  100. self.lineView.isHidden = true
  101. self.deleteButton.properties.isDisabled = true
  102. }
  103. self.deleteButton.reloadData()
  104. self.tableView.reloadData()
  105. }
  106. func showGroupView() {
  107. var viewHeight: CGFloat = 8
  108. var menuItemArr: [ComponentMenuitemProperty] = []
  109. for i in ["Add Files", "Add Folder"] {
  110. let properties_Menuitem: ComponentMenuitemProperty = ComponentMenuitemProperty(multipleSelect: false,
  111. itemSelected: false,
  112. isDisabled: false,
  113. keyEquivalent: nil,
  114. text: KMLocalizedString(i))
  115. menuItemArr.append(properties_Menuitem)
  116. viewHeight += 36
  117. }
  118. if groupView == nil {
  119. groupView = ComponentGroup.createFromNib(in: ComponentLibrary.shared.componentBundle())
  120. }
  121. groupView?.groupDelegate = self
  122. groupView?.frame = CGRectMake(310, 0, 200, viewHeight)
  123. groupView?.updateGroupInfo(menuItemArr)
  124. var point = addFilesButton.convert(addFilesButton.frame.origin, to: nil)
  125. point.y -= viewHeight
  126. groupView?.showWithPoint(point, relativeTo: addFilesButton)
  127. // addFileButton_?.properties.state = .pressed
  128. // addFileButton_?.reloadData()
  129. }
  130. }
  131. protocol KMBatchProcessingViewAction{}
  132. extension KMBatchProcessingView: KMBatchProcessingViewAction {
  133. @IBAction func cleanAll(_ sender: Any) {
  134. self.fileDeleteAction()
  135. }
  136. func _cleanAll() {
  137. if self.tableView.selectModels.count != 0 {
  138. self.tableView.presenter.deleteData(models: self.tableView.selectModels)
  139. self.tableView.selectModels.removeAll()
  140. } else {
  141. self.tableView.presenter.cleanAll()
  142. self.tableView.selectModels.removeAll()
  143. }
  144. self.reloadData()
  145. }
  146. func fileDeleteAction() {
  147. if UserDefaults.standard.object(forKey: "kKMBatchProcessingView") != nil {
  148. self._cleanAll()
  149. return
  150. }
  151. let alert = NSAlert()
  152. alert.alertStyle = .informational
  153. if HistoryFilesManager.manager.selectFiles.count > 0 {
  154. alert.informativeText = KMLocalizedString("Remove Selected Files from your Recent Files?")
  155. } else {
  156. alert.informativeText = KMLocalizedString("Remove Files from your Recent Files?")
  157. }
  158. alert.messageText = KMLocalizedString("The file will disappear from the recent list.")
  159. alert.addButton(withTitle: KMLocalizedString("Delete"))
  160. alert.addButton(withTitle: KMLocalizedString("Cancel"))
  161. alert.showsSuppressionButton = true
  162. let response = alert.runModal()
  163. if response.rawValue == 1000 {
  164. self._cleanAll()
  165. if alert.suppressionButton?.state == .on {
  166. UserDefaults.standard.set("YES", forKey: "kKMBatchProcessingView")
  167. UserDefaults.standard.synchronize()
  168. }
  169. }
  170. }
  171. func addFileAction() {
  172. self.showGroupView()
  173. }
  174. func addFiles() {
  175. KMBatchProcessingView.openfiles(window: window!) { [unowned self] openPanel in
  176. if self.inputType == .OCR {
  177. openPanel.allowedFileTypes = KMOCRModel.supportedTypes()
  178. } else if self.inputType == .imageToPDF {
  179. openPanel.allowedFileTypes = KMBatchProcessingTableViewModel.supportedImageTypes()
  180. } else {
  181. openPanel.allowedFileTypes = ["pdf","PDF"]
  182. }
  183. openPanel.prompt = "Open"
  184. openPanel.canChooseDirectories = false //是否允许选择目录
  185. openPanel.canChooseFiles = true //是否可以选择文件
  186. openPanel.allowsMultipleSelection = true //是否允许多选
  187. if openPanel.allowsMultipleSelection == true {
  188. openPanel.message = NSLocalizedString("Press the keyboard ⌘/⇧ to select multiple files", comment: "")
  189. }
  190. } completion: { [unowned self] (panel ,data) in
  191. if data.count != 0 {
  192. self.tableView.presenter.addData(data: data)
  193. self.reloadData()
  194. }
  195. }
  196. }
  197. func addFolder() {
  198. KMPrint("addFolder")
  199. KMBatchProcessingView.openfiles(window: window!) { [unowned self] openPanel in
  200. if self.inputType == .OCR {
  201. openPanel.allowedFileTypes = KMOCRModel.supportedTypes()
  202. } else if self.inputType == .imageToPDF {
  203. openPanel.allowedFileTypes = KMBatchProcessingTableViewModel.supportedImageTypes()
  204. } else {
  205. openPanel.allowedFileTypes = ["pdf","PDF"]
  206. }
  207. openPanel.prompt = "Open"
  208. openPanel.canChooseDirectories = true //是否允许选择目录
  209. openPanel.canChooseFiles = false //是否可以选择文件
  210. openPanel.allowsMultipleSelection = false //是否允许多选
  211. } completion: { [unowned self] (panel ,data) in
  212. if data.count != 0 {
  213. var filePaths: [URL] = []
  214. var types: [String] = []
  215. if self.inputType == .OCR {
  216. types = KMOCRModel.supportedTypes()
  217. } else if self.inputType == .imageToPDF {
  218. types = KMBatchProcessingTableViewModel.supportedImageTypes()
  219. } else {
  220. types = ["pdf","PDF"]
  221. }
  222. KMMergeTools.findAllFiles(folder: data.first!, result: &filePaths, types: types)
  223. self.tableView.presenter.addData(data: filePaths)
  224. self.reloadData()
  225. }
  226. }
  227. }
  228. func addOpenFiles() {
  229. // let array = ImageToPDFTools.fetchOpenDocumentFiles()
  230. // self.tableView.presenter.addData(data: array)
  231. // self.reloadData()
  232. }
  233. func canAddOpenFile() -> Bool {
  234. var result: Bool = false
  235. // let array = ImageToPDFTools.fetchOpenDocumentFiles()
  236. // if array.count != 0 {
  237. // result = true
  238. // }
  239. return result
  240. }
  241. }
  242. protocol KMBatchProcessingViewOpenPanel {}
  243. extension KMBatchProcessingView: KMBatchProcessingViewOpenPanel {
  244. /**
  245. @abstract 打开文件
  246. @param window 显示window
  247. @param title 标题
  248. */
  249. static func openfiles(window:NSWindow, openPanel:((_ openPanel: NSOpenPanel) -> Void)?, completion:@escaping/**逃逸闭包*/ (_ panel: NSOpenPanel ,_ data:[URL]) -> Void) {
  250. let panel = NSOpenPanel()
  251. if openPanel != nil {
  252. openPanel!(panel)
  253. } else {
  254. panel.prompt = ""
  255. panel.canChooseDirectories = false //是否允许选择目录
  256. panel.canChooseFiles = true //是否可以选择文件
  257. panel.allowsMultipleSelection = true //是否允许多选
  258. panel.allowedFileTypes = KMBatchProcessingTableViewModel.supportedImageTypes() //支持类型
  259. // let button = NSButton.init(checkboxWithTitle: "保存后打开文档", target: nil, action: nil)
  260. // button.state = .off
  261. // panel.accessoryView = button
  262. }
  263. panel.beginSheetModal(for: window, completionHandler: { result in
  264. if result != .OK {
  265. KMPrint("取消")
  266. completion(panel, [])
  267. } else {
  268. completion(panel, panel.urls)
  269. }
  270. })
  271. }
  272. }
  273. extension KMBatchProcessingView: KMBatchProcessingTableViewDelegate {
  274. func reloadData(data: [KMBatchProcessingTableViewModel]) {
  275. self.reloadData()
  276. if (self.delegate != nil) {
  277. self.delegate?.reloadData(data: data)
  278. }
  279. }
  280. }
  281. protocol KMBatchProcessingViewDelegate: NSObject {
  282. /**
  283. 数据刷新
  284. */
  285. func reloadData(data: [KMBatchProcessingTableViewModel])
  286. }
  287. extension KMBatchProcessingView: ComponentGroupDelegate {
  288. func componentGroupDidSelect(group: ComponentGroup?, menuItemProperty: ComponentMenuitemProperty?) {
  289. let familyString = menuItemProperty?.text
  290. if familyString == KMLocalizedString("Add Files", comment: "") {
  291. self.addFiles()
  292. } else if familyString == KMLocalizedString("Add Folder", comment: "") {
  293. self.addFolder()
  294. } else if familyString == KMLocalizedString("Add Open Files", comment: "") {
  295. self.addOpenFiles()
  296. }
  297. }
  298. func componentGroupDidDismiss(group: ComponentGroup?) {
  299. self.addFilesButton.properties.state = .normal
  300. self.addFilesMoreButton.properties.state = .normal
  301. self.addFilesButton.reloadData()
  302. self.addFilesMoreButton.reloadData()
  303. }
  304. }
  305. extension KMBatchProcessingView {
  306. @objc func bacthFilesProcessingNotification() {
  307. self.tableView.reloadData()
  308. }
  309. }