KMBatesTemplateController.swift 15 KB

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