KMBGTemplateController.swift 13 KB

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