KMBookMarkViewController.swift 35 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927
  1. //
  2. // KMBookMarkViewController.swift
  3. // PDF Reader Pro
  4. //
  5. // Created by lxy on 2022/10/10.
  6. //
  7. import Cocoa
  8. import KMComponentLibrary
  9. typealias KMBookMarkViewControllerBookMarkDidChange = (_ controller: KMBookMarkViewController, _ bookMarks: [KMBookmarkItem]) -> Void
  10. @objc protocol KMBookMarkViewControllerDelegate: NSObjectProtocol {
  11. @objc optional func bkControllerAddAction(controller: KMBookMarkViewController, bookmark: CPDFBookmark?, info: [String : Any]?)
  12. }
  13. extension KMNSearchKey.wholeWords {
  14. static let bookmark = "BookmarkSearchWholeWordsKey"
  15. }
  16. extension KMNSearchKey.caseSensitive {
  17. static let bookmark = "BookmarkSearchCaseSensitiveKey"
  18. }
  19. class KMBookMarkViewController: KMNBotaBaseViewController {
  20. @IBOutlet weak var topView: NSView!
  21. @IBOutlet weak var addBookButton: NSButton!
  22. @IBOutlet weak var titleTextField: NSTextField!
  23. @IBOutlet weak var sortButton: NSButton!
  24. @IBOutlet var topSeplineView: NSView!
  25. @IBOutlet weak var bookTableView: KMBotaTableView!
  26. @IBOutlet weak var emptyView: NSView!
  27. @IBOutlet weak var bigTipLabel: NSTextField!
  28. @IBOutlet weak var tipLabel: NSTextField!
  29. private lazy var addButton_: ComponentButton = {
  30. let view = ComponentButton()
  31. view.properties = ComponentButtonProperty(type: .text_gray, size: .xxs, onlyIcon: true, icon: NSImage(named: "KMBookmarkAdd"))
  32. return view
  33. }()
  34. private var emptyView_: ComponentEmpty = {
  35. let view = ComponentEmpty()
  36. view.properties = ComponentEmptyProperty(emptyType: .noBookmark, state: .normal, image: NSImage(named: "KMBookmarkEmpty"), text: KMLocalizedString("No Bookmark"), subText: KMLocalizedString("Here is the description."))
  37. return view
  38. }()
  39. private var groupView: ComponentGroup? = ComponentGroup.createFromNib(in: ComponentLibrary.shared.componentBundle())
  40. private var menuGroupView_: ComponentGroup?
  41. var dataSource: [KMBookmarkItem] = []
  42. var renameTextField: NSTextField?
  43. var renamePDFBook: KMBookmarkItem?
  44. var renameCellView: KMBookCellView?
  45. weak var document: CPDFDocument?
  46. var isLocalEvent: Bool = false //区分外部点击还是内部点击
  47. var selectItems: [KMBookmarkItem] = []
  48. var bookMarkDidChange: KMBookMarkViewControllerBookMarkDidChange?
  49. weak var delegate: KMBookMarkViewControllerDelegate?
  50. private lazy var handdler_ = KMNBookmarkHanddler()
  51. var handdler: KMNBookmarkHanddler {
  52. get {
  53. return handdler_
  54. }
  55. }
  56. convenience init() {
  57. self.init(nibName: "KMBookMarkViewController", bundle: nil)
  58. }
  59. override func viewDidLoad() {
  60. super.viewDidLoad()
  61. handdler.delegate = self
  62. titleTextField.font = ComponentLibrary.shared.getFontFromKey("mac/body-m-bold")
  63. addBookButton.addSubview(addButton_)
  64. addButton_.frame = addBookButton.bounds
  65. addButton_.autoresizingMask = [.width, .height]
  66. addButton_.setTarget(self, action: #selector(addBookmarkAction))
  67. bookTableView.style = NSTableView.Style.plain
  68. bookTableView.allowsMultipleSelection = true
  69. bookTableView.doubleAction = #selector(renameBookAction)
  70. bookTableView.hasImageToolTips = true
  71. bookTableView.botaDelegate = self
  72. bookTableView.menuClickedAction = { [unowned self] point in
  73. let idxs = self.bookTableView.selectedRowIndexes.count
  74. let convertP = self.bookTableView.convert(point, from: nil)
  75. let row = self.bookTableView.row(at: convertP)
  76. if row == -1 {
  77. var viewHeight: CGFloat = 0
  78. let items: [String] = ["Add Bookmark", "All Select", "Delete"]
  79. var menuItemArr: [ComponentMenuitemProperty] = []
  80. for value in items {
  81. let properties_Menuitem: ComponentMenuitemProperty = ComponentMenuitemProperty(multipleSelect: false,
  82. itemSelected: false,
  83. isDisabled: false,
  84. keyEquivalent: nil,
  85. text: KMLocalizedString(value),
  86. identifier: value)
  87. if value == "Delete" {
  88. properties_Menuitem.keyEquivalent = "⌘ " + String(Unicode.Scalar(NSBackspaceCharacter)!)
  89. properties_Menuitem.isDisabled = idxs == 0
  90. }
  91. menuItemArr.append(properties_Menuitem)
  92. viewHeight += 36
  93. }
  94. self.menuGroupView_ = ComponentGroup.createFromNib(in: ComponentLibrary.shared.componentBundle())
  95. self.menuGroupView_?.groupDelegate = self
  96. self.menuGroupView_?.frame = CGRectMake(0, 0, 180, viewHeight)
  97. self.menuGroupView_?.updateGroupInfo(menuItemArr)
  98. self.menuGroupView_?.showWithPoint(CGPoint(x: point.x, y: point.y - viewHeight), relativeTo: self.bookTableView)
  99. return NSMenu()
  100. }
  101. return NSMenu()
  102. }
  103. topView.addSubview(searchButton)
  104. searchButton.km_add_size_constraint(size: NSMakeSize(24, 24))
  105. searchButton.km_add_centerY_constraint(constant: 1)
  106. searchButton.km_add_trailing_constraint(equalTo: addBookButton, attribute: .leading, constant: -4)
  107. searchButton.setTarget(self, action: #selector(_searchAction))
  108. emptyView.addSubview(emptyView_)
  109. emptyView_.km_add_top_constraint(constant: 232)
  110. emptyView_.km_add_bottom_constraint()
  111. emptyView_.km_add_leading_constraint()
  112. emptyView_.km_add_trailing_constraint()
  113. if let data = headerSearchView {
  114. topView.addSubview(data)
  115. headerSearchView?.frame = topView.bounds
  116. headerSearchView?.autoresizingMask = [.width, .height]
  117. }
  118. hideHeaderSearch()
  119. headerSearchView?.itemClick = { [weak self] idx, params in
  120. if idx == 1 { // 显示搜索限制条件
  121. guard let button = params.first as? ComponentButton else {
  122. return
  123. }
  124. self?.showSearchGroupView(sender: button)
  125. } else if idx == 2 { // 关闭搜索
  126. self?.hideHeaderSearch()
  127. self?.reloadData()
  128. }
  129. }
  130. headerSearchView?.valueDidChange = { [weak self] sender, info in
  131. let value = info?[.newKey] as? String ?? ""
  132. self?.handdler.searchKey = value
  133. self?.reloadData()
  134. }
  135. self.reloadData()
  136. }
  137. override func updateUILanguage() {
  138. super.updateUILanguage()
  139. KMMainThreadExecute {
  140. self.titleTextField.stringValue = KMLocalizedString("Bookmarks")
  141. }
  142. }
  143. override func updateUIThemeColor() {
  144. super.updateUIThemeColor()
  145. KMMainThreadExecute {
  146. self.view.wantsLayer = true
  147. let color = ComponentLibrary.shared.getComponentColorFromKey("colorBg/layout-middle")
  148. self.view.layer?.backgroundColor = color.cgColor
  149. self.topSeplineView.wantsLayer = true
  150. self.topSeplineView.layer?.backgroundColor = ComponentLibrary.shared.getComponentColorFromKey("colorBorder/divider").cgColor
  151. self.titleTextField.textColor = ComponentLibrary.shared.getComponentColorFromKey("colorText/2")
  152. self.addButton_.properties.icon = NSImage(named: "KMBookmarkAdd")
  153. self.addButton_.reloadData()
  154. self.searchButton.properties.icon = NSImage(named: "KMImageNameOutlineSearch")
  155. self.searchButton.reloadData()
  156. }
  157. }
  158. func reloadData() {
  159. var bookMarks: [KMBookmarkItem] = []
  160. for bookmark in handdler.bookmarks() {
  161. let item = KMBookmarkItem()
  162. item.bookMark = bookmark
  163. item.index = UInt(bookmark.pageIndex)
  164. item.label = bookmark.label
  165. bookMarks.append(item)
  166. }
  167. self.dataSource = bookMarks
  168. self.dataSource.sort(){
  169. guard let index = $0.bookMark?.pageIndex else {
  170. return false
  171. }
  172. guard let otherIndex = $1.bookMark?.pageIndex else {
  173. return false
  174. }
  175. return index < otherIndex
  176. }
  177. if handdler.isValidSearchMode() {
  178. var items: [KMBookmarkItem] = []
  179. for item in dataSource {
  180. guard let bookmark = item.bookMark else {
  181. continue
  182. }
  183. if hasContainString(handdler.searchKey, bookmark: bookmark) {
  184. items.append(item)
  185. }
  186. }
  187. self.dataSource = items
  188. }
  189. self.bookTableView.reloadData()
  190. self.updateAddBookMarkState()
  191. }
  192. func addBookMarkAndEdit(newBookMark: KMBookmarkItem) {
  193. _ = self.dataSource.contains { item in
  194. if handdler.bookmarkIsEqual(bookmark: item.bookMark, otherBookmark: newBookMark.bookMark) {
  195. let index = KMOCToolClass.arrayIndexOf(array: self.dataSource, item: item) ?? 0
  196. self.didSelectItem(row: index, event: NSEvent())
  197. self.renameBookWithRow(row: index)
  198. return true
  199. }
  200. return false
  201. }
  202. }
  203. func showSearchGroupView(sender: ComponentButton) {
  204. var viewHeight: CGFloat = 8
  205. var menuItemArr: [ComponentMenuitemProperty] = []
  206. let titles = ["Whole Words","Case Sensitive"]
  207. for i in titles {
  208. let menuI = ComponentMenuitemProperty(text: KMLocalizedString(i))
  209. menuItemArr.append(menuI)
  210. viewHeight += 36
  211. }
  212. if let info = menuItemArr.first {
  213. if KMDataManager.ud_bool(forKey: KMNSearchKey.wholeWords.bookmark) {
  214. info.righticon = NSImage(named: "KMNImageNameMenuSelect")
  215. }
  216. }
  217. if let info = menuItemArr.last {
  218. if KMDataManager.ud_bool(forKey: KMNSearchKey.caseSensitive.bookmark) {
  219. info.righticon = NSImage(named: "KMNImageNameMenuSelect")
  220. }
  221. }
  222. let groupView = ComponentGroup.createFromNib(in: ComponentLibrary.shared.componentBundle())
  223. searchGroupView = groupView
  224. groupView?.groupDelegate = self
  225. groupView?.frame = CGRectMake(310, 0, 200, viewHeight)
  226. groupView?.updateGroupInfo(menuItemArr)
  227. var point = sender.convert(sender.frame.origin, to: nil)
  228. point.y -= viewHeight
  229. groupView?.showWithPoint(point, relativeTo: sender)
  230. searchGroupTarget = sender
  231. }
  232. override func showHeaderSearch() {
  233. super.showHeaderSearch()
  234. handdler.isSearchMode = true
  235. }
  236. override func hideHeaderSearch() {
  237. super.hideHeaderSearch()
  238. handdler.isSearchMode = false
  239. }
  240. func hasContainString(_ searchString: String, bookmark: CPDFBookmark) -> Bool {
  241. var label = bookmark.label ?? ""
  242. var searchLabel = searchString
  243. if handdler.caseSensitive == false {
  244. label = label.lowercased()
  245. searchLabel = searchLabel.lowercased()
  246. }
  247. if label.contains(searchLabel) {
  248. if handdler.wholeWords {
  249. let words = label.words()
  250. return words.contains(searchLabel)
  251. }
  252. return true
  253. }
  254. return false
  255. }
  256. func didSelectItem(row: Int, event: NSEvent?, needJump: Bool = true) {
  257. //当选中一个时
  258. if self.bookTableView.selectedRowIndexes.count == 1 || (event != nil &&
  259. (!event!.modifierFlags.contains(NSEvent.ModifierFlags.command) &&
  260. !event!.modifierFlags.contains(NSEvent.ModifierFlags.shift))) {
  261. self.bookTableView.selectRowIndexes(IndexSet(integer: IndexSet.Element(row)), byExtendingSelection: false)
  262. }
  263. //原始数据置空
  264. for model in self.selectItems {
  265. for (idx, item) in self.dataSource.enumerated() {
  266. if item.bookMark != model.bookMark {
  267. continue
  268. }
  269. item.select = false
  270. let rowView = self.bookTableView.rowView(atRow: idx, makeIfNecessary: false) as? KMBookMarkTableRowView
  271. rowView?.reloadData()
  272. }
  273. }
  274. //获取最新数据
  275. var items: [KMBookmarkItem] = []
  276. for index in self.bookTableView.selectedRowIndexes {
  277. if index < self.dataSource.count {
  278. let model = self.dataSource[index]
  279. model.select = true
  280. let rowView = self.bookTableView.rowView(atRow: index, makeIfNecessary: false) as? KMBookMarkTableRowView
  281. rowView?.reloadData()
  282. items.append(model)
  283. }
  284. }
  285. self.selectItems = items
  286. //刷新数据
  287. if needJump {
  288. self.updateListViewData()
  289. }
  290. }
  291. func updateListViewData() {
  292. let selectedRowIndexes = self.bookTableView.selectedRowIndexes
  293. if selectedRowIndexes.count != 1 {
  294. return
  295. }
  296. let idx = selectedRowIndexes.first ?? 0
  297. guard let item = self.dataSource.safe_element(for: idx) as? KMBookmarkItem else {
  298. return
  299. }
  300. handdler.goBookmark(item.bookMark, animated: true)
  301. }
  302. func cancelSelect() {
  303. self.bookTableView.deselectAll(nil)
  304. for model in self.selectItems {
  305. model.select = false
  306. if let index = self.dataSource.firstIndex(of: model) {
  307. let rowView = self.bookTableView.rowView(atRow: index, makeIfNecessary: false) as? KMBookMarkTableRowView
  308. rowView?.reloadData()
  309. }
  310. }
  311. }
  312. func updateAddBookMarkState() {
  313. // addButton_.properties.isDisabled = !canAddBorkMark()
  314. // addButton_.properties.state = .normal
  315. // addButton_.reloadData()
  316. }
  317. // MARK: - Public Methods
  318. public func selectIndex(index: Int) {
  319. if index < 0 || index >= self.dataSource.count {
  320. return
  321. }
  322. self.bookTableView.selectRowIndexes(IndexSet(integer: IndexSet.Element(index)), byExtendingSelection: false)
  323. self.didSelectItem(row: index, event: NSEvent(), needJump: false)
  324. }
  325. public func currentPageDidChangedAction(document: CPDFDocument?) {
  326. guard let theDocument = document, theDocument.isEqual(to: handdler.document) else {
  327. return
  328. }
  329. if !isLocalEvent {
  330. var containSelIndex:Bool = false
  331. for (index, value) in self.dataSource.enumerated() {
  332. let currentBookmark = handdler.bookmark(for: handdler.currentPageIndex)
  333. if handdler.bookmarkIsEqual(bookmark: value.bookMark, otherBookmark: currentBookmark) {
  334. containSelIndex = true
  335. self.didSelectItem(row: index, event: NSEvent())
  336. break
  337. }
  338. }
  339. if !containSelIndex {
  340. self.cancelSelect()
  341. }
  342. }
  343. isLocalEvent = false
  344. self.updateAddBookMarkState()
  345. }
  346. public func canAddBorkMark() -> Bool {
  347. return handdler.canAddBorkmark()
  348. }
  349. public func addCurrentBookmark(callback: ((CPDFBookmark?)->Void)?) {
  350. handdler.addCurrentBookmark(callback: callback)
  351. }
  352. public func removeBookmark(for index: Int) -> Bool {
  353. return handdler.removeBookmark(for: index)
  354. }
  355. public func removeBookmarks(for indexs: IndexSet) -> Bool {
  356. return handdler.removeBookmarks(for: indexs)
  357. }
  358. public func removeAllBookmarks() -> Bool {
  359. return handdler.removeAllBookmarks()
  360. }
  361. public func rename(bookmark: CPDFBookmark, label: String) {
  362. handdler.rename(bookmark: bookmark, label: label)
  363. }
  364. //MARK: - Menu Action
  365. @objc func renameBookAction() {
  366. if self.bookTableView.selectedRowIndexes.count == 1 {
  367. self.renameBookWithRow(row: self.bookTableView.selectedRowIndexes.first!)
  368. } else {
  369. __NSBeep()
  370. }
  371. }
  372. @IBAction func escButtonAction(_ sender: Any) {
  373. self.bookTableView.deselectAll(nil)
  374. }
  375. @IBAction func addBookmarkAction(_ sender: Any) {
  376. addButton_.properties.state = .normal
  377. addButton_.reloadData()
  378. let currentPageIndex = handdler.currentPageIndex
  379. if let data = handdler.bookmark(for: currentPageIndex) {
  380. delegate?.bkControllerAddAction?(controller: self, bookmark: data, info: ["result" : false])
  381. return
  382. }
  383. handdler.addCurrentBookmark(callback: { [unowned self] bookmark in
  384. self.delegate?.bkControllerAddAction?(controller: self, bookmark: bookmark, info: nil)
  385. self.reloadData()
  386. DispatchQueue.main.async {
  387. if let bookmark = self.handdler.bookmark(for: Int(currentPageIndex)) {
  388. let item = KMBookmarkItem()
  389. item.bookMark = bookmark
  390. item.label = bookmark.label
  391. item.index = UInt(bookmark.pageIndex)
  392. self.addBookMarkAndEdit(newBookMark: item)
  393. }
  394. }
  395. })
  396. }
  397. @objc func changeLocationAction() {
  398. if self.bookTableView.selectedRowIndexes.count == 1 {
  399. let item = self.dataSource[self.bookTableView.selectedRowIndexes.first!]
  400. let alter = NSAlert()
  401. alter.alertStyle = NSAlert.Style.informational
  402. alter.messageText = KMLocalizedString("Are you sure you want to set the selected page as the bookmark location?", comment: "")
  403. alter.addButton(withTitle: KMLocalizedString("Yes", comment:""))
  404. alter.addButton(withTitle: KMLocalizedString("No", comment:""))
  405. let modlres = alter.runModal()
  406. if modlres == .alertFirstButtonReturn {
  407. let bookMark = KMBookmarkItem()
  408. bookMark.bookMark = item.bookMark
  409. bookMark.label = item.label
  410. bookMark.index = UInt(handdler.currentPageIndex)
  411. self.changeLocation(oldBookMark: item,
  412. newBookMark: bookMark)
  413. }
  414. } else {
  415. __NSBeep()
  416. }
  417. }
  418. @objc func deleteBookAction() {
  419. if self.bookTableView.selectedRowIndexes.count != 0 {
  420. var bookMarks:[KMBookmarkItem] = []
  421. for index in self.bookTableView.selectedRowIndexes {
  422. let item = self.dataSource[index]
  423. bookMarks.append(item)
  424. }
  425. self.deleteBookMark(bookMarks: bookMarks)
  426. } else {
  427. __NSBeep()
  428. }
  429. }
  430. private func renameBookWithRow(row: Int) {
  431. self.renamePDFBook = self.dataSource[row]
  432. self.renameCellView = self.bookTableView.view(atColumn: 0, row: row, makeIfNecessary: true) as? KMBookCellView
  433. self.renameTextField = self.renameCellView?.inputTF
  434. self.renameTextField?.delegate = self
  435. self.renameTextField?.isEditable = true
  436. self.renameTextField?.becomeFirstResponder()
  437. }
  438. @objc private func _searchAction() {
  439. searchButton.properties.state = .normal
  440. searchButton.reloadData()
  441. showHeaderSearch()
  442. }
  443. }
  444. // MARK: - NSTextFieldDelegate
  445. extension KMBookMarkViewController: NSTextFieldDelegate {
  446. func controlTextDidEndEditing(_ obj: Notification) {
  447. guard let bookmark = renamePDFBook?.bookMark else {
  448. return
  449. }
  450. if (self.renameTextField!.isEqual(obj.object)) {
  451. let textField : NSTextField = obj.object as! NSTextField
  452. handdler.rename(bookmark: bookmark, label: textField.stringValue)
  453. self.renameTextField?.isEditable = false
  454. renameCellView?.inputBox.borderWidth = 0
  455. renameCellView?.inputBox.fillColor = .clear
  456. }
  457. }
  458. }
  459. // MARK: - NSTableViewDelegate,NSTableViewDataSource
  460. extension KMBookMarkViewController : NSTableViewDelegate,NSTableViewDataSource {
  461. func numberOfRows(in tableView: NSTableView) -> Int {
  462. let count = self.dataSource.count
  463. if count == 0 {
  464. self.emptyView.isHidden = false
  465. } else {
  466. self.emptyView.isHidden = true
  467. }
  468. return count
  469. }
  470. func tableView(_ tableView: NSTableView, viewFor tableColumn: NSTableColumn?, row: Int) -> NSView? {
  471. if row < self.dataSource.count {
  472. let item: KMBookmarkItem = self.dataSource[row]
  473. let cell : KMBookCellView = tableView.makeView(withIdentifier: NSUserInterfaceItemIdentifier(rawValue: "KMBookCellView"), owner: self) as! KMBookCellView
  474. if handdler.isValidSearchMode() {
  475. var label = item.bookMark?.label ?? ""
  476. let attri = NSMutableAttributedString(string: label, attributes: [
  477. .font : NSFont.SFProTextRegularFont(13),
  478. .foregroundColor : KMNColorTools.colorText_1()])
  479. var _searchKey = handdler.searchKey
  480. if handdler.caseSensitive == false {
  481. label = label.lowercased()
  482. _searchKey = _searchKey.lowercased()
  483. }
  484. let ranges = label.ranges(of: _searchKey)
  485. for range in ranges.nsRnage {
  486. attri.addAttributes([.font : NSFont.SFProTextBoldFont(13), .foregroundColor: KMNColorTools.colorPrimary_textLight()], range: range)
  487. }
  488. cell.inputTF.attributedStringValue = attri
  489. } else {
  490. cell.inputTF.font = ComponentLibrary.shared.getFontFromKey("mac/body-s-regular")
  491. cell.inputTF.textColor = ComponentLibrary.shared.getComponentColorFromKey("colorText/1")
  492. cell.inputTF.stringValue = item.bookMark?.label ?? ""
  493. cell.bookTitle.stringValue = "\(item.bookMark?.pageIndex ?? 0 + 1)"
  494. cell.inputBox.borderWidth = 0
  495. if let data = item.bookMark?.date {
  496. cell.dateLabel.stringValue = KMTools.timeString(timeDate: data)
  497. } else {
  498. cell.dateLabel.stringValue = ""
  499. }
  500. }
  501. cell.textFieldDidEndEditingCallback = { [weak self] textF in
  502. self?.renameCellView = cell
  503. if let bookmark = item.bookMark {
  504. self?.handdler.rename(bookmark: bookmark, label: textF.stringValue)
  505. }
  506. }
  507. return cell
  508. }
  509. return nil
  510. }
  511. func tableView(_ tableView: NSTableView, rowViewForRow row: Int) -> NSTableRowView? {
  512. let rowView = KMBookMarkTableRowView()
  513. rowView.selectionHighlightStyle = .none
  514. if row < self.dataSource.count {
  515. rowView.model = self.dataSource[row]
  516. rowView.menuClickedAction = { [weak self] point in
  517. let idxs = self?.bookTableView.selectedRowIndexes.count ?? 0
  518. let tempView = self?.bookTableView.rowView(atRow: row, makeIfNecessary: false)
  519. var viewHeight: CGFloat = 0
  520. var items: [String] = ["Rename", "Setting Destination", "Delete"]
  521. if idxs > 1 {
  522. items = ["Delete"]
  523. }
  524. var menuItemArr: [ComponentMenuitemProperty] = []
  525. for value in items {
  526. let properties_Menuitem: ComponentMenuitemProperty = ComponentMenuitemProperty(multipleSelect: false,
  527. itemSelected: false,
  528. isDisabled: false,
  529. keyEquivalent: nil,
  530. text: KMLocalizedString(value),
  531. identifier: value)
  532. if value == "Delete" {
  533. properties_Menuitem.keyEquivalent = "⌘ " + String(Unicode.Scalar(NSBackspaceCharacter)!)
  534. }
  535. menuItemArr.append(properties_Menuitem)
  536. viewHeight += 36
  537. }
  538. if self?.groupView != nil {
  539. self?.groupView?.groupDelegate = self
  540. self?.groupView?.frame = CGRectMake(0, 0, 180, viewHeight)
  541. self?.groupView?.updateGroupInfo(menuItemArr)
  542. self?.groupView?.showWithPoint(CGPoint(x: point.x, y: point.y - viewHeight), relativeTo: tempView)
  543. }
  544. return NSMenu()
  545. }
  546. rowView.mouseDownAction = { [unowned self] (view, event) in
  547. self.didSelectItem(row: row, event: event)
  548. }
  549. rowView.rightMouseDownAction = { [unowned self] (view, event) in
  550. if !KMOCToolClass.arrayContains(array: self.selectItems, annotation: rowView.model) ||
  551. self.selectItems.count == 1 {
  552. self.selectIndex(index: row)
  553. }
  554. if row < 0 || row >= self.dataSource.count {
  555. return
  556. }
  557. }
  558. rowView.hoverCallback = { [unowned self] (mouseEntered, mouseBox) in
  559. if let value = ComponentLibrary.shared.getComponentValueFromKey("radius/xs") {
  560. let currentValue = value as? CGFloat ?? 0
  561. rowView.box?.cornerRadius = currentValue
  562. }
  563. self.bookTableView.enumerateAvailableRowViews { view, row in
  564. if let data = view as? KMBookMarkTableRowView {
  565. data.model.hover = false
  566. data.reloadData()
  567. }
  568. }
  569. if mouseEntered {
  570. rowView.model.hover = true
  571. rowView.reloadData()
  572. } else {
  573. rowView.model.hover = false
  574. rowView.reloadData()
  575. }
  576. }
  577. }
  578. return rowView
  579. }
  580. func tableView(_ tableView: NSTableView, heightOfRow row: Int) -> CGFloat {
  581. return 36
  582. }
  583. func tableView(_ tableView: NSTableView, shouldSelect tableColumn: NSTableColumn?) -> Bool {
  584. self.isLocalEvent = true
  585. return true
  586. }
  587. func tableView(_ tableView: NSTableView, shouldSelectRow row: Int) -> Bool {
  588. self.isLocalEvent = true
  589. return true
  590. }
  591. func tableViewSelectionDidChange(_ notification: Notification) {
  592. if self.bookTableView.selectedRow == -1 {
  593. self.cancelSelect()
  594. }
  595. }
  596. }
  597. // MARK: - KMBotaTableViewDelegate
  598. extension KMBookMarkViewController: KMBotaTableViewDelegate {
  599. func tableView(_ aTableView: NSTableView, imageContextForRow rowIndex: Int) -> AnyObject? {
  600. if aTableView.isEqual(to: self.bookTableView) {
  601. let cnt = self.dataSource.count
  602. if rowIndex >= cnt {
  603. return nil
  604. }
  605. let model = self.dataSource[rowIndex]
  606. return model.bookMark
  607. }
  608. return nil
  609. }
  610. }
  611. //MARK: - undoRedo
  612. extension KMBookMarkViewController {
  613. func changeLocation(oldBookMark: KMBookmarkItem, newBookMark: KMBookmarkItem) {
  614. document?.removeBookmark(forPageIndex: oldBookMark.index)
  615. document?.addBookmark(newBookMark.label, forPageIndex: newBookMark.index)
  616. reloadData()
  617. }
  618. func renamePDFBook(bookmark : KMBookmarkItem! , label:String) {
  619. if bookmark.bookMark?.label == label {
  620. return
  621. }
  622. let temp = bookmark.bookMark?.label
  623. bookmark.bookMark?.label = label
  624. self.reloadData()
  625. var indexSet = IndexSet()
  626. indexSet.insert(self.bookTableView.row(for: self.renameCellView!))
  627. self.bookTableView.selectRowIndexes(indexSet, byExtendingSelection: false)
  628. }
  629. func deleteBookMark(bookMarks: [KMBookmarkItem]) {
  630. for bookMark in bookMarks {
  631. if ((document?.removeBookmark(forPageIndex: bookMark.index)) != nil) {
  632. KMPrint("删除标签成功")
  633. }
  634. }
  635. self.reloadData()
  636. guard let callBack = bookMarkDidChange else { return }
  637. callBack(self, bookMarks)
  638. }
  639. func addBookMark(bookMarks: [KMBookmarkItem]) {
  640. for bookmark in bookMarks {
  641. document?.addBookmark(bookmark.label, forPageIndex: UInt(bookmark.index))
  642. }
  643. self.reloadData()
  644. if bookMarks.count == 1 {
  645. DispatchQueue.main.async {
  646. if let bookmark = self.handdler.bookmark(for: Int(bookMarks.first!.index)) {
  647. let item = KMBookmarkItem()
  648. item.bookMark = bookmark
  649. item.label = bookmark.label
  650. item.index = UInt(bookmark.pageIndex)
  651. self.addBookMarkAndEdit(newBookMark: item)
  652. }
  653. }
  654. }
  655. guard let callBack = bookMarkDidChange else { return }
  656. callBack(self, bookMarks)
  657. }
  658. @IBAction func undo(_ sender: Any) {
  659. handdler.undo()
  660. }
  661. @IBAction func redo(_ sender: Any) {
  662. handdler.redo()
  663. }
  664. }
  665. // MARK: - NSMenuDelegate, NSMenuItemValidation
  666. extension KMBookMarkViewController: NSMenuDelegate, NSMenuItemValidation {
  667. func validateMenuItem(_ menuItem: NSMenuItem) -> Bool {
  668. let action = menuItem.action
  669. if (action == #selector(undo)) {
  670. return handdler.canUndo()
  671. }
  672. if (action == #selector(redo)) {
  673. return handdler.canRedo()
  674. }
  675. if action == #selector(renameBookAction) ||
  676. action == #selector(changeLocationAction) ||
  677. action == #selector(deleteBookAction) {
  678. if self.bookTableView.selectedRowIndexes.count > 1 {
  679. if action == #selector(changeLocationAction) {
  680. return false
  681. } else if action == #selector(renameBookAction) {
  682. return false
  683. }
  684. } else if self.bookTableView.selectedRowIndexes.count == 1 {
  685. return true
  686. } else {
  687. if self.bookTableView.selectedRowIndexes.count == 0 {
  688. if action == #selector(changeLocationAction) {}
  689. } else {
  690. return false
  691. }
  692. }
  693. }
  694. return true
  695. }
  696. }
  697. // MARK: - KMNBookmarkHanddlerDelegate
  698. extension KMBookMarkViewController: KMNBookmarkHanddlerDelegate {
  699. func handdler(_ handdler: KMNBookmarkHanddler, didAdd bookmark: CPDFBookmark?, info: [String : Any]?) {
  700. KMMainThreadExecute {
  701. self.reloadData()
  702. }
  703. }
  704. func handdler(_ handdler: KMNBookmarkHanddler, didRemove bookmark: CPDFBookmark?, info: [String : Any]?) {
  705. KMMainThreadExecute {
  706. self.reloadData()
  707. }
  708. }
  709. func handdler(_ handdler: KMNBookmarkHanddler, didRemoveAll info: [String : Any]?) {
  710. KMMainThreadExecute {
  711. self.reloadData()
  712. }
  713. }
  714. func handdler(_ handdler: KMNBookmarkHanddler, didRename bookmark: CPDFBookmark?, info: [String : Any]?) {
  715. KMMainThreadExecute {
  716. self.reloadData()
  717. }
  718. }
  719. }
  720. // MARK: - ComponentGroupDelegate
  721. extension KMBookMarkViewController: ComponentGroupDelegate {
  722. func componentGroupDidSelect(group: ComponentGroup?, menuItemProperty: ComponentMenuitemProperty?) {
  723. if group == groupView {
  724. if let selItem = menuItemProperty {
  725. if selItem.text == KMLocalizedString("Delete") {
  726. var pageIndexs = IndexSet()
  727. for i in bookTableView.selectedRowIndexes {
  728. guard let item = dataSource.safe_element(for: i) as? KMBookmarkItem else {
  729. continue
  730. }
  731. if let bookmark = item.bookMark {
  732. pageIndexs.insert(bookmark.pageIndex)
  733. }
  734. }
  735. _ = handdler.removeBookmarks(for: pageIndexs)
  736. } else if selItem.text == KMLocalizedString("Rename") {
  737. renameBookAction()
  738. } else if selItem.text == KMLocalizedString("Setting Destination") {
  739. changeLocationAction()
  740. }
  741. }
  742. } else if group == searchGroupView {
  743. guard let menuI = menuItemProperty else {
  744. return
  745. }
  746. let idx = group?.menuItemArr.firstIndex(of: menuI)
  747. if idx == 0 {
  748. let key = KMNSearchKey.wholeWords.bookmark
  749. let value = KMDataManager.ud_bool(forKey: key)
  750. KMDataManager.ud_set(!value, forKey: key)
  751. handdler.wholeWords = !value
  752. } else if idx == 1 {
  753. let key = KMNSearchKey.caseSensitive.bookmark
  754. let value = KMDataManager.ud_bool(forKey: key)
  755. KMDataManager.ud_set(!value, forKey: key)
  756. handdler.caseSensitive = !value
  757. }
  758. } else if group == menuGroupView_ {
  759. guard let menuI = menuItemProperty else {
  760. return
  761. }
  762. let idx = group?.menuItemArr.firstIndex(of: menuI)
  763. if idx == 0 {
  764. addBookmarkAction(addButton_)
  765. } else if idx == 1 {
  766. var indexs = IndexSet()
  767. for (i, _) in dataSource.enumerated() {
  768. indexs.insert(i)
  769. }
  770. bookTableView.km_safe_selectRowIndexes(indexs, byExtendingSelection: false)
  771. for i in indexs {
  772. didSelectItem(row: i, event: nil)
  773. }
  774. } else if idx == 2 {
  775. deleteBookAction()
  776. }
  777. }
  778. }
  779. func componentGroupDidDismiss(group: ComponentGroup?) {
  780. if group == searchGroupView {
  781. searchGroupTarget?.properties.state = .normal
  782. searchGroupTarget?.reloadData()
  783. searchGroupTarget = nil
  784. }
  785. }
  786. }