KMBatesTemplateController.swift 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415
  1. //
  2. // KMBatesTemplateController.swift
  3. // PDF Reader Pro
  4. //
  5. // Created by Niehaoyu on 2024/11/8.
  6. //
  7. import Cocoa
  8. import KMComponentLibrary
  9. //删除数据时调用
  10. let kBatesListDeleteAllNotiName = NSNotification.Name("kBatesListDeleteAllNotiName")
  11. let kBatesListDeleteNotiName = NSNotification.Name("kBatesListDeleteNotiName")
  12. @objc protocol KMBatesTemplateControllerDelegate: AnyObject {
  13. @objc optional func templateControllerDidAddData(_ controller: KMBatesTemplateController)
  14. @objc optional func templateControllerDidEditData(_ controller: KMBatesTemplateController, _ data: KMBatesModel)
  15. //选中内容切换
  16. @objc optional func templateControllerDidSelectedChanged(_ controller: KMBatesTemplateController)
  17. }
  18. class KMBatesTemplateController: KMNBaseViewController {
  19. @IBOutlet var contendView: NSView!
  20. @IBOutlet var titleLabel: NSTextField!
  21. @IBOutlet var addButton: ComponentButton!
  22. @IBOutlet var emptyView: ComponentEmpty!
  23. @IBOutlet var scrollView: NSScrollView!
  24. @IBOutlet var collectionView: NSCollectionView!
  25. weak open var delegate: KMBatesTemplateControllerDelegate?
  26. var selectedDataModel: KMBatesModel?
  27. var groupView: ComponentGroup?
  28. var editCell: KMBatesTemplateItem?
  29. deinit {
  30. NotificationCenter.default.removeObserver(self)
  31. }
  32. override func viewDidLoad() {
  33. super.viewDidLoad()
  34. // Do view setup here.
  35. configUI()
  36. setupProperty()
  37. reloadData()
  38. NotificationCenter.default.addObserver(self, selector: #selector(deleteItemNoti), name: kBatesListDeleteNotiName, object: nil)
  39. NotificationCenter.default.addObserver(self, selector: #selector(deleteAllItemsNoti), name: kBatesListDeleteAllNotiName, object: nil)
  40. NotificationCenter.default.addObserver(self, selector: #selector(dataUpdated), name: KMBatesDatasUpdateNotiName , object: nil)
  41. }
  42. override func updateUILanguage() {
  43. super.updateUILanguage()
  44. configUI()
  45. }
  46. override func updateUIThemeColor() {
  47. super.updateUIThemeColor()
  48. configUI()
  49. }
  50. func setupProperty() {
  51. addButton.properties = ComponentButtonProperty(type: .text_gray, size: .xxs, onlyIcon: true, icon: NSImage(named: "watermark_template_add"), keepPressState: false)
  52. addButton.setTarget(self, action: #selector(buttonClicked(_:)))
  53. emptyView.properties = ComponentEmptyProperty(emptyType: .noWatermark, text: KMLocalizedString("No Templates"), subText: KMLocalizedString(""))
  54. }
  55. func configUI() {
  56. view.wantsLayer = true
  57. view.layer?.backgroundColor = ComponentLibrary.shared.getComponentColorFromKey("colorBg/layout-middle").cgColor
  58. titleLabel.stringValue = KMLocalizedString("Templates")
  59. titleLabel.textColor = ComponentLibrary.shared.getComponentColorFromKey("colorText/2")
  60. titleLabel.font = ComponentLibrary.shared.getFontFromKey("mac/body-m-bold")
  61. collectionView.backgroundColors = [NSColor.clear]
  62. collectionView.wantsLayer = true
  63. collectionView.layer?.backgroundColor = NSColor.clear.cgColor
  64. collectionView.delegate = self
  65. collectionView.dataSource = self
  66. collectionView.allowsEmptySelection = false
  67. collectionView.allowsMultipleSelection = false
  68. collectionView.register(KMBatesTemplateItem.self, forItemWithIdentifier: NSUserInterfaceItemIdentifier(rawValue: "KMBatesTemplateItem"))
  69. }
  70. //刷新所有数据
  71. public func reloadData() {
  72. if KMBatesManager.defaultManager.datas.count == 0 {
  73. collectionView.isHidden = true
  74. emptyView.isHidden = false
  75. } else {
  76. emptyView.isHidden = true
  77. collectionView.isHidden = false
  78. collectionView.reloadData()
  79. updateSelectedState()
  80. }
  81. }
  82. public func updateSelectedState() {
  83. if let selData = selectedDataModel {
  84. if let index = KMBatesManager.defaultManager.datas.firstIndex(of: selData) {
  85. let indexpath = IndexPath(item: index, section: 0)
  86. var set = Set<IndexPath>()
  87. set.insert(indexpath)
  88. collectionView.selectItems(at: set, scrollPosition: .top)
  89. }
  90. } else {
  91. collectionView.deselectAll(nil)
  92. }
  93. }
  94. private func collectionViewSelectedChanged() {
  95. let indexs = collectionView.selectionIndexPaths
  96. if indexs.count > 0 {
  97. for index in indexs {
  98. let data = KMBatesManager.defaultManager.datas[index.item]
  99. selectedDataModel = data
  100. }
  101. } else {
  102. selectedDataModel = nil
  103. }
  104. DispatchQueue.main.async {
  105. self.delegate?.templateControllerDidSelectedChanged?(self)
  106. }
  107. }
  108. //MARK: - action
  109. @objc func buttonClicked(_ sender: ComponentButton) {
  110. if sender == addButton {
  111. if selectedDataModel != nil {
  112. selectedDataModel = nil
  113. delegate?.templateControllerDidSelectedChanged?(self)
  114. }
  115. reloadData()
  116. delegate?.templateControllerDidAddData?(self)
  117. }
  118. }
  119. override func rightMouseUp(with event: NSEvent) {
  120. super.rightMouseUp(with: event)
  121. var items: [String] = []
  122. var point = self.view.convert(event.locationInWindow, from: nil)
  123. let collectionPoint = self.view.convert(point, to: collectionView)
  124. self.editCell = nil
  125. if let indexPath = self.collectionView.indexPathForItem(at: collectionPoint) {
  126. let cellView = collectionView.item(at: indexPath) as? KMBatesTemplateItem
  127. self.editCell = cellView
  128. items.append(KMLocalizedString("Rename"))
  129. items.append(KMLocalizedString("Edit"))
  130. items.append(KMLocalizedString("Delete"))
  131. items.append(KMLocalizedString("Remove All"))
  132. self.showGroupView(items, event.locationInWindow)
  133. }
  134. }
  135. func showRenameWindow() {
  136. let saveWindow: KMWatermarkSaveWindow = KMWatermarkSaveWindow(windowNibName: "KMWatermarkSaveWindow")
  137. if let item = self.editCell, let name = item.dataModel?.name {
  138. saveWindow.nameValue = name
  139. } else {
  140. saveWindow.nameValue = ""
  141. }
  142. saveWindow.saveHandler = {[weak self] string in
  143. guard let weakSelf = self else { return }
  144. DispatchQueue.main.async {
  145. weakSelf.editCell?.dataModel?.name = string ?? ""
  146. if let model = weakSelf.editCell?.dataModel, KMBatesManager.defaultManager.updateTemplate(model) {
  147. weakSelf.reloadData()
  148. }
  149. }
  150. }
  151. saveWindow.own_beginSheetModal(for: view.window) { string in
  152. }
  153. saveWindow.titleLabel.stringValue = KMLocalizedString("Rename")
  154. }
  155. func showGroupView(_ items: [String], _ point: CGPoint) {
  156. var viewHeight: CGFloat = 8
  157. var menuItemArr: [ComponentMenuitemProperty] = []
  158. for i in items {
  159. let properties_Menuitem: ComponentMenuitemProperty = ComponentMenuitemProperty(multipleSelect: false,
  160. itemSelected: false,
  161. isDisabled: false,
  162. keyEquivalent: nil,
  163. text: KMLocalizedString(i),
  164. identifier: i)
  165. menuItemArr.append(properties_Menuitem)
  166. viewHeight += 36
  167. }
  168. if groupView == nil {
  169. groupView = ComponentGroup.createFromNib(in: ComponentLibrary.shared.componentBundle())
  170. }
  171. groupView?.groupDelegate = self
  172. var viewPoint = point
  173. viewPoint.y -= viewHeight
  174. groupView?.frame = CGRectMake(310, 0, 200, viewHeight)
  175. groupView?.updateGroupInfo(menuItemArr)
  176. groupView?.showWithPoint(viewPoint, relativeTo: nil)
  177. }
  178. //MARK: - Noti
  179. @objc func deleteItemNoti(_ noti: Notification) {
  180. guard let notiTag = noti.object as? String else {
  181. return
  182. }
  183. if notiTag == self.selectedDataModel?.tag {
  184. self.selectedDataModel = nil
  185. self.delegate?.templateControllerDidSelectedChanged?(self)
  186. }
  187. self.reloadData()
  188. }
  189. @objc func deleteAllItemsNoti() {
  190. if let _ = self.selectedDataModel {
  191. self.selectedDataModel = nil
  192. self.delegate?.templateControllerDidSelectedChanged?(self)
  193. }
  194. self.reloadData()
  195. }
  196. @objc func dataUpdated() {
  197. self.reloadData()
  198. }
  199. }
  200. //MARK: - NSCollectionViewDelegate, NSCollectionViewDataSource
  201. extension KMBatesTemplateController: NSCollectionViewDelegate, NSCollectionViewDataSource, NSCollectionViewDelegateFlowLayout {
  202. public func collectionView(_ collectionView: NSCollectionView, numberOfItemsInSection section: Int) -> Int {
  203. return KMBatesManager.defaultManager.datas.count
  204. }
  205. public func collectionView(_ collectionView: NSCollectionView, itemForRepresentedObjectAt indexPath: IndexPath) -> NSCollectionViewItem {
  206. if indexPath.item >= KMBatesManager.defaultManager.datas.count {
  207. return NSCollectionViewItem()
  208. }
  209. let item: KMBatesTemplateItem = collectionView.makeItem(withIdentifier: NSUserInterfaceItemIdentifier(rawValue: "KMBatesTemplateItem"), for: indexPath) as! KMBatesTemplateItem
  210. item.delegate = self
  211. item.dataModel = KMBatesManager.defaultManager.datas[indexPath.item]
  212. item.reloadData()
  213. return item
  214. }
  215. public func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> NSSize {
  216. var itemHeight = 56
  217. let dataModel = KMBatesManager.defaultManager.datas[indexPath.item]
  218. if dataModel.topLeftString.count > 0 {
  219. itemHeight += 24
  220. itemHeight += 8
  221. }
  222. if dataModel.topCenterString.count > 0 {
  223. itemHeight += 24
  224. itemHeight += 8
  225. }
  226. if dataModel.topRightString.count > 0 {
  227. itemHeight += 24
  228. itemHeight += 8
  229. }
  230. if dataModel.bottomLeftString.count > 0 {
  231. itemHeight += 24
  232. itemHeight += 8
  233. }
  234. if dataModel.bottomCenterString.count > 0 {
  235. itemHeight += 24
  236. itemHeight += 8
  237. }
  238. if dataModel.bottomRightString.count > 0 {
  239. itemHeight += 24
  240. itemHeight += 8
  241. }
  242. itemHeight -= 8
  243. return CGSize(width: 232, height: itemHeight)
  244. }
  245. func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, minimumInteritemSpacingForSectionAt section: Int) -> CGFloat {
  246. return 8
  247. }
  248. func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, minimumLineSpacingForSectionAt section: Int) -> CGFloat {
  249. return 8
  250. }
  251. func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, insetForSectionAt section: Int) -> NSEdgeInsets {
  252. return NSEdgeInsets(top: 8, left: 0, bottom: 8, right: 0)
  253. }
  254. public func collectionView(_ collectionView: NSCollectionView, didSelectItemsAt indexPaths: Set<IndexPath>) {
  255. collectionViewSelectedChanged()
  256. }
  257. public func collectionView(_ collectionView: NSCollectionView, didDeselectItemsAt indexPaths: Set<IndexPath>) {
  258. collectionViewSelectedChanged()
  259. }
  260. }
  261. //MARK: - kmNWatermarkTemplateItemDelegate
  262. extension KMBatesTemplateController: KMBatesTemplateItemDelegate {
  263. func kmBatesTemplateItemDidEdit(_ view: KMBatesTemplateItem) {
  264. if let data = view.dataModel {
  265. delegate?.templateControllerDidEditData?(self, data)
  266. }
  267. }
  268. func kmBatesTemplateItemDidDelete(_ view: KMBatesTemplateItem) {
  269. if let data = view.dataModel {
  270. let alert = NSAlert()
  271. alert.alertStyle = .critical
  272. alert.messageText = String(format: KMLocalizedString("Are you sure you want to remove the Bates?"))
  273. alert.addButton(withTitle: KMLocalizedString("Delete"))
  274. alert.addButton(withTitle: KMLocalizedString("Cancel"))
  275. alert.beginSheetModal(for: NSApp.mainWindow!) { (response) in
  276. if response == .alertFirstButtonReturn {
  277. let _ = KMBatesManager.defaultManager.deleteTemplate(data)
  278. if data == self.selectedDataModel {
  279. self.selectedDataModel = nil
  280. self.delegate?.templateControllerDidSelectedChanged?(self)
  281. }
  282. self.reloadData()
  283. NotificationCenter.default.post(name: kWatermarkListDeleteNotiName, object: data.tag)
  284. }
  285. }
  286. }
  287. }
  288. func kmBatesTemplateItemDidDeleteAllItems() {
  289. let alert = NSAlert()
  290. alert.alertStyle = .critical
  291. alert.messageText = String(format: KMLocalizedString("Are you sure to delete all templates?"))
  292. alert.addButton(withTitle: KMLocalizedString("Delete"))
  293. alert.addButton(withTitle: KMLocalizedString("Cancel"))
  294. alert.beginSheetModal(for: NSApp.mainWindow!) { (response) in
  295. if response == .alertFirstButtonReturn {
  296. let _ = KMBatesManager.defaultManager.deleteAllTemplate()
  297. if let _ = self.selectedDataModel {
  298. self.selectedDataModel = nil
  299. self.delegate?.templateControllerDidSelectedChanged?(self)
  300. }
  301. self.reloadData()
  302. NotificationCenter.default.post(name: kWatermarkListDeleteAllNotiName, object: nil)
  303. }
  304. }
  305. }
  306. }
  307. //MARK: - ComponentGroupDelegate
  308. extension KMBatesTemplateController: ComponentGroupDelegate {
  309. func componentGroupDidSelect(group: ComponentGroup?, menuItemProperty: ComponentMenuitemProperty?) {
  310. if menuItemProperty?.identifier == KMLocalizedString("Rename") {
  311. self.showRenameWindow()
  312. } else if menuItemProperty?.identifier == KMLocalizedString("Edit") {
  313. if let cell = self.editCell {
  314. self.kmBatesTemplateItemDidEdit(cell)
  315. }
  316. } else if menuItemProperty?.identifier == KMLocalizedString("Delete") {
  317. if let cell = self.editCell {
  318. self.kmBatesTemplateItemDidDelete(cell)
  319. }
  320. } else if menuItemProperty?.identifier == KMLocalizedString("Remove All") {
  321. self.kmBatesTemplateItemDidDeleteAllItems()
  322. }
  323. }
  324. }