KMBookmarkController.swift 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881
  1. //
  2. // KMBookmarkController.swift
  3. // PDF Reader Pro
  4. //
  5. // Created by lizhe on 2024/2/5.
  6. //
  7. import Cocoa
  8. private let kLabelIdentifier = NSUserInterfaceItemIdentifier("label")
  9. private let kFileIdentifier = NSUserInterfaceItemIdentifier("file")
  10. private let kPageIdentifier = NSUserInterfaceItemIdentifier("page")
  11. let kTextWithIconStringKey = "string";
  12. let kTextWithIconImageKey = "image";
  13. let kBookmarksToolbarIdentifier = "BookmarksToolbarIdentifier"
  14. let kBookmarksNewFolderToolbarItemIdentifier = "BookmarksNewFolderToolbarItemIdentifier"
  15. let kBookmarksNewSeparatorToolbarItemIdentifier = "BookmarksNewSeparatorToolbarItemIdentifier"
  16. let kBookmarksDeleteToolbarItemIdentifier = "BookmarksDeleteToolbarItemIdentifier"
  17. let kPasteboardTypeBookmarkRows = NSPasteboard.PasteboardType(rawValue: "pasteboard.bookmarkrows")
  18. class KMBookmarkController: NSWindowController {
  19. @IBOutlet weak var outlineView: KMCustomOutlineView!
  20. var draggedBookmarks: [KMBookmark] = []
  21. var recentDocuments: [[String: Any]] {
  22. get {
  23. return KMBookmarkManager.manager.recentDocuments
  24. }
  25. set {
  26. }
  27. }
  28. var bookmarkRoot: KMRootBookmark {
  29. get {
  30. return KMBookmarkManager.manager.rootBookmark
  31. }
  32. set {
  33. }
  34. }
  35. var toolbarItems: [String: NSToolbarItem] = [:]
  36. override func windowDidLoad() {
  37. super.windowDidLoad()
  38. setupToolbar()
  39. outlineView.delegate = self
  40. outlineView.dataSource = self
  41. outlineView.registerForDraggedTypes([kPasteboardTypeBookmarkRows, .fileURL, .string])
  42. outlineView.doubleAction = #selector(doubleClickBookmark(_:))
  43. }
  44. func updateStatus() {
  45. let row = outlineView.selectedRow
  46. var message = ""
  47. if row != -1 {
  48. if let bookmark = outlineView.item(atRow: row) as? KMBookmark {
  49. switch bookmark.bookmarkType {
  50. case .bookmark:
  51. message = bookmark.fileURL?.path ?? ""
  52. case .folder:
  53. let count = bookmark.children.count
  54. message = count == 1 ? NSLocalizedString("1 item", comment: "Bookmark folder description") : String(format: NSLocalizedString("%ld items", comment: "Bookmark folder description"), count)
  55. default:
  56. break
  57. }
  58. }
  59. }
  60. // statusBar.leftStringValue = message
  61. }
  62. //
  63. static func showBookmarkController() -> KMBookmarkController {
  64. let controller = KMBookmarkController.init(windowNibName: "KMBookmarkController")
  65. NSWindow.currentWindow().addChildWindow(controller.window!, ordered: NSWindow.OrderingMode.above)
  66. controller.window?.center()
  67. return controller
  68. }
  69. //
  70. //
  71. // //MARK: Recent Documents
  72. func recentDocumentInfo(at fileURL: URL) -> [String: Any]? {
  73. let path = fileURL.path
  74. for info in recentDocuments {
  75. // if let aliasData = info[ALIASDATA_KEY] as? Data,
  76. // let alias = SKAlias(aliasData),
  77. // alias.fileURLNoUI?.path.caseInsensitiveCompare(path) == .orderedSame {
  78. // return info
  79. // }
  80. }
  81. return nil
  82. }
  83. //
  84. func addRecentDocument(for fileURL: URL, pageIndex: UInt, scaleFactor factor: CGFloat, snapshots setups: [Any]?) {
  85. // if let info = recentDocumentInfo(at: fileURL) {
  86. // recentDocuments.removeObject(info)
  87. // }
  88. //
  89. // if let alias = SKAlias(url: fileURL) {
  90. // var bm: [String: Any] = [
  91. // PAGEINDEX_KEY: pageIndex,
  92. // SCALE_KEY: factor,
  93. // ALIASDATA_KEY: alias.data,
  94. // ALIAS_KEY: alias,
  95. // SNAPSHOTS_KEY: setups ?? []
  96. // ]
  97. // recentDocuments.insert(bm, at: 0)
  98. // if recentDocuments.count > maxRecentDocumentsCount {
  99. // recentDocuments.removeLastObject()
  100. // }
  101. // }
  102. }
  103. //
  104. // func pageIndex(forRecentDocumentAt fileURL: URL) -> UInt {
  105. // guard let fileURL = fileURL else { return UInt.max }
  106. // if let pageIndex = recentDocumentInfo(at: fileURL)?[PAGEINDEX_KEY] as? UInt {
  107. // return pageIndex
  108. // }
  109. // return UInt.max
  110. // }
  111. //
  112. // func scaleFactor(forRecentDocumentAt fileURL: URL) -> CGFloat {
  113. // guard let fileURL = fileURL else { return 0 }
  114. // if let scaleFactor = recentDocumentInfo(at: fileURL)?[SCALE_KEY] as? CGFloat {
  115. // return scaleFactor
  116. // }
  117. // return 0
  118. // }
  119. //
  120. // func snapshots(forRecentDocumentAt fileURL: URL) -> [Any]? {
  121. // guard let fileURL = fileURL else { return nil }
  122. // if let setups = recentDocumentInfo(at: fileURL)?[SNAPSHOTS_KEY] as? [Any], !setups.isEmpty {
  123. // return setups
  124. // }
  125. // return nil
  126. // }
  127. //
  128. // //MARK: Bookmarks support
  129. func getInsertionFolder(_ bookmarkPtr: inout KMBookmark?, childIndex indexPtr: inout Int) {
  130. let rowIndex = outlineView.clickedRow
  131. var indexes = outlineView.selectedRowIndexes
  132. if rowIndex != -1 && !indexes.contains(rowIndex) {
  133. indexes = IndexSet(integer: rowIndex)
  134. }
  135. let rowIdx = indexes.last ?? NSNotFound
  136. var item = KMBookmarkManager.manager.rootBookmark
  137. var idx = item.children.count
  138. if rowIdx != NSNotFound {
  139. if let selectedItem = outlineView.item(atRow: rowIdx) as? KMBookmark {
  140. if outlineView.isItemExpanded(selectedItem) {
  141. item = selectedItem as! KMRootBookmark
  142. idx = item.children.count
  143. } else if let parent = selectedItem.parent, let itemIdx = parent.children.firstIndex(of: selectedItem) {
  144. item = parent as! KMRootBookmark
  145. idx = itemIdx + 1
  146. }
  147. }
  148. }
  149. bookmarkPtr = item
  150. indexPtr = idx
  151. }
  152. @IBAction func openBookmark(_ sender: Any) {
  153. if let bookmark = (sender as AnyObject).representedObject as? KMBookmark {
  154. bookmark.open()
  155. }
  156. }
  157. @IBAction func doubleClickBookmark(_ sender: Any) {
  158. let row = outlineView.clickedRow
  159. if let bm = (row != -1 ? outlineView.item(atRow: row) : nil) as? KMBookmark,
  160. [KMBookmarkType.bookmark, .session].contains(bm.bookmarkType) {
  161. bm.open()
  162. }
  163. }
  164. func deleteBookmarks(bookmarks: [KMBookmark]) {
  165. for item in minimumCoverForBookmarks(bookmarks).reversed() {
  166. guard let parent = item.parent, let itemIndex = parent.children.firstIndex(of: item) else { continue }
  167. parent.removeObjectFromChildren(index: itemIndex)
  168. }
  169. outlineView.reloadData()
  170. }
  171. @IBAction func insertBookmarkFolder(_ sender: Any) {
  172. let folder = KMFolderBookmark.folderBookmark(label: NSLocalizedString("Folder", comment: "default folder name"))
  173. var item: KMBookmark?
  174. var idx: Int = 0
  175. getInsertionFolder(&item, childIndex: &idx)
  176. item?.insert(child: folder, atIndex: idx)
  177. let row = outlineView.row(forItem: folder)
  178. outlineView.selectRowIndexes(IndexSet(integer: row), byExtendingSelection: false)
  179. outlineView.editColumn(0, row: row, with: nil, select: true)
  180. }
  181. @IBAction func insertBookmarkSeparator(_ sender: Any) {
  182. let separator = KMSeparatorBookmark()
  183. var item: KMBookmark?
  184. var idx: Int = 0
  185. getInsertionFolder(&item, childIndex: &idx)
  186. item?.insert(child: separator, atIndex: idx)
  187. let row = outlineView.row(forItem: separator)
  188. outlineView.selectRowIndexes(IndexSet(integer: row), byExtendingSelection: false)
  189. }
  190. @IBAction func addBookmark(_ sender: Any) {
  191. let openPanel = NSOpenPanel()
  192. var types = [String]()
  193. for docClass in NSDocumentController.shared.documentClassNames {
  194. if let docClass = NSClassFromString(docClass) as? NSDocument.Type {
  195. types += docClass.readableTypes
  196. }
  197. }
  198. openPanel.allowsMultipleSelection = true
  199. openPanel.canChooseDirectories = true
  200. openPanel.allowedFileTypes = types
  201. openPanel.beginSheetModal(for: self.window!) { (result) in
  202. guard result == .OK else { return }
  203. let newBookmarks = KMBookmark.bookmarks(urls: openPanel.urls)
  204. if newBookmarks != nil {
  205. var item: KMBookmark?
  206. var index: Int = 0
  207. self.getInsertionFolder(&item, childIndex: &index)
  208. var indexes = IndexSet(integersIn: Int(index)..<Int(index + newBookmarks.count))
  209. item?.mutableArrayValue(forKey: "children").insert(newBookmarks, at: indexes)
  210. if item == self.bookmarkRoot || self.outlineView.isItemExpanded(item) {
  211. if item != self.bookmarkRoot {
  212. indexes.shift(startingAt: 0, by: self.outlineView.row(forItem: item) + 1)
  213. }
  214. self.outlineView.selectRowIndexes(indexes, byExtendingSelection: false)
  215. }
  216. }
  217. }
  218. }
  219. @IBAction func deleteBookmark(_ sender: Any) {
  220. print("deleteBookmark")
  221. guard let bookmark = outlineView.selectedItem() as? KMBookmark else { return }
  222. self.deleteBookmarks(bookmarks: [bookmark])
  223. }
  224. //
  225. // @IBAction func toggleStatusBar(_ sender: Any) {
  226. // UserDefaults.standard.set(!statusBar.isVisible, forKey: SKShowBookmarkStatusBarKey)
  227. // statusBar.toggle(below: outlineView.enclosingScrollView, animate: sender != nil)
  228. // }
  229. //
  230. // func clickedBookmarks() -> [Any]? {
  231. // let row = outlineView.clickedRow
  232. // guard row != -1 else { return nil }
  233. // var indexes = outlineView.selectedRowIndexes
  234. // if !indexes.contains(row) {
  235. // indexes = IndexSet(integer: row)
  236. // }
  237. // return indexes.compactMap { outlineView.item(atRow: $0) }
  238. // }
  239. //
  240. // @IBAction func deleteBookmarks(_ sender: Any) {
  241. // guard let items = clickedBookmarks() as? [KMBookmark] else { return }
  242. // for item in items.reversed() {
  243. // if let parent = item.parent, let itemIndex = parent.children.index(of: item) {
  244. // parent.removeObject(fromChildrenAtIndex: itemIndex)
  245. // }
  246. // }
  247. // }
  248. //
  249. // @IBAction func openBookmarks(_ sender: Any) {
  250. // guard let items = clickedBookmarks() as? [KMBookmark] else { return }
  251. // for item in items.reversed() {
  252. // item.open()
  253. // }
  254. // }
  255. //
  256. // @IBAction func previewBookmarks(_ sender: Any) {
  257. // if QLPreviewPanel.sharedPreviewPanelExists() && QLPreviewPanel.shared().isVisible {
  258. // QLPreviewPanel.shared().orderOut(nil)
  259. // } else if let row = outlineView.clickedRow {
  260. // outlineView.selectRowIndexes(IndexSet(integer: row), byExtendingSelection: false)
  261. // QLPreviewPanel.shared().makeKeyAndOrderFront(nil)
  262. // }
  263. // }
  264. //
  265. //
  266. // // MARK: - NSMenu delegate methods
  267. // func addItemForBookmark(_ bookmark: KMBookmark, toMenu menu: NSMenu, isFolder: Bool, isAlternate: Bool) {
  268. // var item: NSMenuItem?
  269. // if isFolder {
  270. // item = menu.addItem(withSubmenuAndTitle: bookmark.label)
  271. // item?.submenu?.delegate = self
  272. // } else {
  273. // item = menu.addItem(withTitle: bookmark.label, action: #selector(openBookmark(_:)), target: self)
  274. // }
  275. // item?.representedObject = bookmark
  276. // if isAlternate {
  277. // item?.keyEquivalentModifierMask = .alternate
  278. // item?.isAlternate = true
  279. // item?.setImageAndSize(bookmark.alternateIcon)
  280. // } else {
  281. // item?.setImageAndSize(bookmark.icon)
  282. // }
  283. // }
  284. //
  285. // func menuNeedsUpdate(_ menu: NSMenu) {
  286. // if menu == outlineView.menu {
  287. // let row = outlineView.clickedRow
  288. // menu.removeAllItems()
  289. // if row != -1 {
  290. // menu.addItem(withTitle: NSLocalizedString("Remove", comment: "Menu item title"), action: #selector(deleteBookmarks(_:)), target: self)
  291. // menu.addItem(withTitle: NSLocalizedString("Open", comment: "Menu item title"), action: #selector(openBookmarks(_:)), target: self)
  292. // menu.addItem(withTitle: NSLocalizedString("Quick Look", comment: "Menu item title"), action: #selector(previewBookmarks(_:)), target: self)
  293. // menu.addItem(.separator())
  294. // }
  295. // menu.addItem(withTitle: NSLocalizedString("New Folder", comment: "Menu item title"), action: #selector(insertBookmarkFolder(_:)), target: self)
  296. // menu.addItem(withTitle: NSLocalizedString("New Separator", comment: "Menu item title"), action: #selector(insertBookmarkSeparator(_:)), target: self)
  297. // } else {
  298. // guard let supermenu = menu.supermenu, let idx = supermenu.indexOfItem(withSubmenu: menu), let bm = (supermenu == NSApp.mainMenu) ? bookmarkRoot : supermenu.item(at: idx)?.representedObject as? KMBookmark else { return }
  299. //
  300. // let bookmarks = bm.children
  301. // var i = menu.numberOfItems
  302. //
  303. // while i > 0 {
  304. // if let menuItem = menu.item(at: i - 1), menuItem.isSeparatorItem || menuItem.representedObject != nil {
  305. // menu.removeItem(menuItem)
  306. // }
  307. // i -= 1
  308. // }
  309. //
  310. // if supermenu == NSApp.mainMenu, let previousSession = previousSession {
  311. // menu.addItem(.separator())
  312. // addItemForBookmark(previousSession, toMenu: menu, isFolder: false, isAlternate: false)
  313. // addItemForBookmark(previousSession, toMenu: menu, isFolder: true, isAlternate: true)
  314. // }
  315. //
  316. // if menu.numberOfItems > 0, bookmarks.count > 0 {
  317. // menu.addItem(.separator())
  318. // }
  319. //
  320. // for bm in bookmarks {
  321. // switch bm.bookmarkType {
  322. // case .folder:
  323. // addItemForBookmark(bm, toMenu: menu, isFolder: true, isAlternate: false)
  324. // addItemForBookmark(bm, toMenu: menu, isFolder: false, isAlternate: true)
  325. // case .session:
  326. // addItemForBookmark(bm, toMenu: menu, isFolder: false, isAlternate: false)
  327. // addItemForBookmark(bm, toMenu: menu, isFolder: true, isAlternate: true)
  328. // case .separator:
  329. // menu.addItem(.separator())
  330. // default:
  331. // addItemForBookmark(bm, toMenu: menu, isFolder: false, isAlternate: false)
  332. // }
  333. // }
  334. // }
  335. // }
  336. //
  337. // // avoid rebuilding the bookmarks menu on every key event
  338. // func menuHasKeyEquivalent(_ menu: NSMenu, for event: NSEvent, target: AutoreleasingUnsafeMutablePointer<AnyObject?>?, action: UnsafeMutablePointer<Selector?>?) -> Bool { false }
  339. //
  340. // // MARK: - Toolbar
  341. //
  342. func setupToolbar() {
  343. // Create a new toolbar instance, and attach it to our document window
  344. let toolbar = NSToolbar(identifier: kBookmarksToolbarIdentifier)
  345. var dict = [String: NSToolbarItem]()
  346. // Set up toolbar properties: Allow customization, give a default display mode, and remember state in user defaults
  347. toolbar.allowsUserCustomization = true
  348. toolbar.autosavesConfiguration = true
  349. toolbar.displayMode = .default
  350. // We are the delegate
  351. toolbar.delegate = self
  352. // Add template toolbar items
  353. var item = NSToolbarItem(itemIdentifier: NSToolbarItem.Identifier(kBookmarksNewFolderToolbarItemIdentifier))
  354. item.label = NSLocalizedString("New Folder", comment: "Toolbar item label")
  355. item.paletteLabel = NSLocalizedString("New Folder", comment: "Toolbar item label")
  356. item.toolTip = NSLocalizedString("Add a New Folder", comment: "Tool tip message")
  357. // item.image = NSImage(named: "NewFolder")
  358. item.image = NSImage(named: NSImage.folderName)!
  359. item.target = self
  360. item.action = #selector(insertBookmarkFolder(_:))
  361. dict[kBookmarksNewFolderToolbarItemIdentifier] = item
  362. item = NSToolbarItem(itemIdentifier: NSToolbarItem.Identifier(kBookmarksNewSeparatorToolbarItemIdentifier))
  363. item.label = NSLocalizedString("New Separator", comment: "Toolbar item label")
  364. item.paletteLabel = NSLocalizedString("New Separator", comment: "Toolbar item label")
  365. item.toolTip = NSLocalizedString("Add a New Separator", comment: "Tool tip message")
  366. // item.image = NSImage(named: "NewSeparator")
  367. item.image = NSImage(named: NSImage.shareTemplateName)!
  368. item.target = self
  369. item.action = #selector(insertBookmarkSeparator(_:))
  370. dict[kBookmarksNewSeparatorToolbarItemIdentifier] = item
  371. item = NSToolbarItem(itemIdentifier: NSToolbarItem.Identifier(kBookmarksDeleteToolbarItemIdentifier))
  372. item.label = NSLocalizedString("Delete", comment: "Toolbar item label")
  373. item.paletteLabel = NSLocalizedString("Delete", comment: "Toolbar item label")
  374. item.toolTip = NSLocalizedString("Delete Selected Items", comment: "Tool tip message")
  375. item.image = NSWorkspace.shared.icon(forFileType: NSFileTypeForHFSTypeCode(OSType(kToolbarDeleteIcon)))
  376. item.target = self
  377. item.action = #selector(deleteBookmark(_:))
  378. dict[kBookmarksDeleteToolbarItemIdentifier] = item
  379. toolbarItems = dict
  380. // Attach the toolbar to the window
  381. self.window?.toolbar = toolbar
  382. }
  383. //
  384. // // MARK: - Quick Look Panel Support
  385. //
  386. // func acceptsPreviewPanelControl(_ panel: QLPreviewPanel) -> Bool {
  387. // return true
  388. // }
  389. //
  390. // func beginPreviewPanelControl(_ panel: QLPreviewPanel) {
  391. // panel.delegate = self
  392. // panel.dataSource = self
  393. // }
  394. //
  395. // func endPreviewPanelControl(_ panel: QLPreviewPanel) {
  396. // }
  397. //
  398. // func previewItems() -> [KMBookmark] {
  399. // var items = [KMBookmark]()
  400. //
  401. // outlineView.selectedRowIndexes.enumerated().forEach { (idx, _) in
  402. // if let item = outlineView.item(atRow: idx) as? KMBookmark {
  403. // if item.bookmarkType == .bookmark {
  404. // items.append(item)
  405. // } else if item.bookmarkType == .session {
  406. // items.append(contentsOf: item.children)
  407. // }
  408. // }
  409. // }
  410. // return items
  411. // }
  412. //
  413. // func numberOfPreviewItems(in panel: QLPreviewPanel) -> Int {
  414. // return previewItems().count
  415. // }
  416. //
  417. // func previewPanel(_ panel: QLPreviewPanel, previewItemAt anIndex: Int) -> QLPreviewItem {
  418. // return previewItems()[anIndex]
  419. // }
  420. //
  421. // func previewPanel(_ panel: QLPreviewPanel, sourceFrameOnScreenForPreviewItem item: QLPreviewItem) -> NSRect {
  422. // var item = item
  423. // if let parent = (item as? KMBookmark)?.parent, parent.bookmarkType == .session {
  424. // item = parent
  425. // }
  426. // let row = outlineView.row(forItem: item)
  427. // var iconRect = NSZeroRect
  428. // if let item = item as? KMBookmark, row != -1 {
  429. // let cell = outlineView.preparedCell(atColumn: 0, row: row) as? SKTextWithIconCell
  430. // iconRect = cell?.iconRect(forBounds: outlineView.frameOfCell(atColumn: 0, row: row)) ?? NSZeroRect
  431. // if outlineView.visibleRect.intersects(iconRect) {
  432. // iconRect = outlineView.convert(iconRect, to: nil)
  433. // } else {
  434. // iconRect = NSZeroRect
  435. // }
  436. // }
  437. // return iconRect
  438. // }
  439. //
  440. // func previewPanel(_ panel: QLPreviewPanel, transitionImageForPreviewItem item: QLPreviewItem, contentRect: UnsafeMutablePointer<NSRect>) -> NSImage? {
  441. // var item = item
  442. // if let parent = (item as? KMBookmark)?.parent, parent.bookmarkType == .session {
  443. // item = parent
  444. // }
  445. // return (item as? KMBookmark)?.icon
  446. // }
  447. //
  448. // func previewPanel(_ panel: QLPreviewPanel, handle event: NSEvent) -> Bool {
  449. // if event.type == .keyDown {
  450. // outlineView.keyDown(with: event)
  451. // return true
  452. // }
  453. // return false
  454. // }
  455. //
  456. }
  457. extension KMBookmarkController: NSOutlineViewDelegate, NSOutlineViewDataSource {
  458. //MARK: NSOutlineViewDataSource
  459. func minimumCoverForBookmarks(_ items: [KMBookmark]) -> [KMBookmark] {
  460. var lastBm: KMBookmark?
  461. var minimalCover = [KMBookmark]()
  462. for bm in items {
  463. if !(bm.isDescendant(of: lastBm)) {
  464. minimalCover.append(bm)
  465. lastBm = bm
  466. }
  467. }
  468. return minimalCover
  469. }
  470. func outlineView(_ ov: NSOutlineView, numberOfChildrenOfItem item: Any?) -> Int {
  471. let bookmark = item as? KMBookmark ?? bookmarkRoot
  472. return bookmark.bookmarkType == .folder ? bookmark.children.count : 0
  473. }
  474. func outlineView(_ ov: NSOutlineView, isItemExpandable item: Any) -> Bool {
  475. let bookmark = item as! KMBookmark
  476. return bookmark.bookmarkType == .folder
  477. }
  478. func outlineView(_ ov: NSOutlineView, child index: Int, ofItem item: Any?) -> Any {
  479. let bookmark = (item as? KMBookmark) ?? bookmarkRoot
  480. return bookmark.objectOfChidren(index: index)
  481. }
  482. func outlineView(_ ov: NSOutlineView, objectValueFor tableColumn: NSTableColumn?, byItem item: Any?) -> Any? {
  483. guard let column = tableColumn else { return nil }
  484. guard let bm = item as? KMBookmark else { return nil }
  485. let tcID = column.identifier
  486. switch tcID {
  487. case kLabelIdentifier:
  488. return [kTextWithIconStringKey: bm.label, kTextWithIconImageKey: bm.icon]
  489. case kFileIdentifier:
  490. if bm.bookmarkType == .folder || bm.bookmarkType == .session {
  491. let count = bm.children.count
  492. return count == 1 ? NSLocalizedString("1 item", comment: "Bookmark folder description") : String.localizedStringWithFormat(NSLocalizedString("%ld items", comment: "Bookmark folder description"), count)
  493. } else {
  494. return bm.fileURL?.path ?? ""
  495. }
  496. case kPageIdentifier:
  497. return bm.pageNumber
  498. default:
  499. return nil
  500. }
  501. }
  502. func outlineView(_ ov: NSOutlineView, setObjectValue object: Any?, for tableColumn: NSTableColumn?, byItem item: Any?) {
  503. guard let column = tableColumn else { return }
  504. guard let bm = item as? KMBookmark else { return }
  505. let tcID = column.identifier
  506. switch tcID {
  507. case kLabelIdentifier:
  508. if let newLabel = (object as? [String: Any])?[kTextWithIconStringKey] as? String, newLabel != bm.label {
  509. bm.label = newLabel
  510. }
  511. case kPageIdentifier:
  512. if let newPageNumber = object as? Int, newPageNumber != bm.pageNumber.intValue {
  513. bm.pageNumber = newPageNumber as NSNumber
  514. }
  515. default:
  516. break
  517. }
  518. }
  519. func outlineView(_ ov: NSOutlineView, writeItems items: [Any], to pboard: NSPasteboard) -> Bool {
  520. draggedBookmarks = minimumCoverForBookmarks(items as! [KMBookmark])
  521. pboard.clearContents()
  522. pboard.setData(Data(), forType: kPasteboardTypeBookmarkRows)
  523. return true
  524. }
  525. func outlineView(_ ov: NSOutlineView, validateDrop info: NSDraggingInfo, proposedItem item: Any?, proposedChildIndex index: Int) -> NSDragOperation {
  526. guard index != NSOutlineViewDropOnItemIndex else { return [] }
  527. let pboard = info.draggingPasteboard
  528. if pboard.canReadItem(withDataConformingToTypes: [kPasteboardTypeBookmarkRows.rawValue]) && info.draggingSource as? NSOutlineView == ov {
  529. return .move
  530. } else if NSURL.canReadFileURL(from: pboard) {
  531. return .every
  532. }
  533. return []
  534. }
  535. func outlineView(_ ov: NSOutlineView, acceptDrop info: NSDraggingInfo, item: Any?, childIndex index: Int) -> Bool {
  536. let pboard = info.draggingPasteboard
  537. if pboard.canReadItem(withDataConformingToTypes: [kPasteboardTypeBookmarkRows.rawValue]) && info.draggingSource as? NSOutlineView == ov {
  538. var movedBookmarks = [KMBookmark]()
  539. var indexes = IndexSet()
  540. var insertionIndex = index
  541. let targetItem = item as? KMBookmark ?? bookmarkRoot
  542. for bookmark in draggedBookmarks {
  543. guard let parent = bookmark.parent else { continue }
  544. guard let bookmarkIndex = parent.children.firstIndex(of: bookmark) else { continue }
  545. if targetItem == parent {
  546. if insertionIndex > bookmarkIndex {
  547. insertionIndex -= 1
  548. }
  549. if insertionIndex == bookmarkIndex {
  550. continue
  551. }
  552. }
  553. parent.removeObjectFromChildren(index: bookmarkIndex)
  554. targetItem.insert(child: bookmark, atIndex: insertionIndex)
  555. movedBookmarks.append(bookmark)
  556. insertionIndex += 1
  557. }
  558. for bookmark in movedBookmarks {
  559. let row = ov.row(forItem: bookmark)
  560. if row != -1 {
  561. indexes.insert(row)
  562. }
  563. }
  564. if !indexes.isEmpty {
  565. ov.selectRowIndexes(indexes, byExtendingSelection: false)
  566. }
  567. return true
  568. } else {
  569. let urls = NSURL.readFileURLs(from: pboard)
  570. let newBookmarks = KMBookmark.bookmarks(urls: urls)
  571. if !newBookmarks.isEmpty {
  572. var indexes = IndexSet(integersIn: index..<(index + newBookmarks.count))
  573. (item as? KMBookmark ?? bookmarkRoot).mutableArrayValue(forKey: "children").insert(newBookmarks, at: indexes)
  574. if (item as? KMBookmark ?? bookmarkRoot) === bookmarkRoot || ov.isItemExpanded(item) {
  575. if (item as? KMBookmark ?? bookmarkRoot) !== bookmarkRoot {
  576. indexes.shift(startingAt: 0, by: ov.row(forItem: item) + 1)
  577. }
  578. ov.selectRowIndexes(indexes, byExtendingSelection: false)
  579. }
  580. return true
  581. }
  582. return false
  583. }
  584. }
  585. func outlineView(_ ov: NSOutlineView, dragEndedWith operation: NSDragOperation) {
  586. draggedBookmarks.removeAll()
  587. }
  588. // MARK: NSOutlineViewDelegate
  589. func outlineView(_ ov: NSOutlineView, dataCellFor tableColumn: NSTableColumn?, item: Any) -> Any? {
  590. if tableColumn == nil {
  591. return (item as? KMBookmark)?.bookmarkType == .separator ? KMSeparatorCell() : nil
  592. }
  593. return tableColumn?.dataCell(forRow: ov.row(forItem: item))
  594. }
  595. func outlineView(_ ov: NSOutlineView, willDisplayCell cell: Any, for tableColumn: NSTableColumn?, item: Any) {
  596. guard let column = tableColumn else { return }
  597. guard let cell = cell as? NSCell else { return }
  598. if column.identifier == kFileIdentifier {
  599. if let bm = item as? KMBookmark {
  600. // if bm.bookmarkType == .folder || bm.bookmarkType == .session {
  601. // cell.textColor = .disabledControlTextColor
  602. // } else {
  603. // cell.textColor = .controlTextColor
  604. // }
  605. }
  606. }
  607. }
  608. func outlineView(_ ov: NSOutlineView, shouldEdit tableColumn: NSTableColumn?, item: Any) -> Bool {
  609. guard let column = tableColumn else { return false }
  610. guard let bm = item as? KMBookmark else { return false }
  611. let tcID = column.identifier
  612. switch tcID {
  613. case kLabelIdentifier:
  614. return bm.bookmarkType != .separator
  615. case kPageIdentifier:
  616. return bm.pageIndex != NSNotFound
  617. default:
  618. return false
  619. }
  620. }
  621. func outlineView(_ ov: NSOutlineView, toolTipFor cell: NSCell, rect: UnsafeMutablePointer<NSRect>, tableColumn tc: NSTableColumn?, item: Any, mouseLocation: NSPoint) -> String {
  622. guard let column = tc else { return "" }
  623. guard let bm = item as? KMBookmark else { return "" }
  624. let tcID = column.identifier
  625. switch tcID {
  626. case kLabelIdentifier:
  627. return bm.label
  628. case kFileIdentifier:
  629. if bm.bookmarkType == .session {
  630. return ""
  631. // return bm.children.map { $0.path ?? "" }.joined(separator: "\n")
  632. } else if bm.bookmarkType == .folder {
  633. let count = bm.children.count
  634. return count == 1 ? NSLocalizedString("1 item", comment: "Bookmark folder description") : String.localizedStringWithFormat(NSLocalizedString("%ld items", comment: "Bookmark folder description"), count)
  635. } else {
  636. return bm.fileURL?.path ?? ""
  637. }
  638. case kPageIdentifier:
  639. return bm.pageNumber.stringValue
  640. default:
  641. return ""
  642. }
  643. }
  644. func outlineViewSelectionDidChange(_ notification: Notification) {
  645. updateStatus()
  646. if QLPreviewPanel.sharedPreviewPanelExists(), let previewPanel = QLPreviewPanel.shared(), previewPanel.isVisible, previewPanel.dataSource === self {
  647. previewPanel.reloadData()
  648. }
  649. }
  650. func outlineView(_ ov: NSOutlineView, deleteItems items: [Any]) {
  651. for item in minimumCoverForBookmarks(items as! [KMBookmark]).reversed() {
  652. guard let parent = item.parent, let itemIndex = parent.children.firstIndex(of: item) else { continue }
  653. parent.removeObjectFromChildren(index: itemIndex)
  654. }
  655. }
  656. func outlineView(_ ov: NSOutlineView, canDeleteItems items: [Any]) -> Bool {
  657. return !items.isEmpty
  658. }
  659. func outlineView(_ ov: NSOutlineView, copyItems items: [Any]) {
  660. var urls = [URL]()
  661. addBookmarkURLsToArray(minimumCoverForBookmarks(items as! [KMBookmark]), &urls)
  662. if !urls.isEmpty {
  663. let pboard = NSPasteboard.general
  664. pboard.clearContents()
  665. pboard.writeObjects(urls as [NSPasteboardWriting])
  666. }
  667. }
  668. func outlineView(_ ov: NSOutlineView, canCopyItems items: [Any]) -> Bool {
  669. return !items.isEmpty
  670. }
  671. func outlineView(_ ov: NSOutlineView, pasteFromPasteboard pboard: NSPasteboard) {
  672. let urls = NSURL.readFileURLs(from: pboard)
  673. let newBookmarks = KMBookmark.bookmarks(urls: urls)
  674. if !newBookmarks.isEmpty {
  675. var item: KMBookmark?
  676. var anIndex = 0
  677. getInsertionFolder(&item, childIndex: &anIndex)
  678. var indexes = IndexSet(integersIn: anIndex..<(anIndex + newBookmarks.count))
  679. (item ?? bookmarkRoot).mutableArrayValue(forKey: "children").insert(newBookmarks, at: indexes)
  680. if item === bookmarkRoot || ov.isItemExpanded(item) {
  681. if item !== bookmarkRoot {
  682. indexes.shift(startingAt: 0, by: ov.row(forItem: item) + 1)
  683. }
  684. ov.selectRowIndexes(indexes, byExtendingSelection: false)
  685. }
  686. }
  687. }
  688. func outlineView(_ ov: NSOutlineView, canPasteFromPasteboard pboard: NSPasteboard) -> Bool {
  689. return NSURL.canReadFileURL(from: pboard)
  690. }
  691. func outlineView(_ ov: NSOutlineView, typeSelectHelperSelectionStrings typeSelectHelper: SKTypeSelectHelper) -> [String] {
  692. let count = ov.numberOfRows
  693. var labels = [String]()
  694. for i in 0..<count {
  695. if let label = ov.item(atRow: i) {
  696. labels.append(label as! String)
  697. }
  698. }
  699. return labels
  700. }
  701. // func outlineView(_ ov: NSOutlineView, typeSelectHelper typeSelectHelper: SKTypeSelectHelper, didFailToFindMatchForSearchString searchString: String) {
  702. // statusBar.setLeftStringValue(String.localizedStringWithFormat(NSLocalizedString("No match: \"%@\"", comment: "Status message"), searchString))
  703. // }
  704. //
  705. // func outlineView(_ ov: NSOutlineView, typeSelectHelper typeSelectHelper: SKTypeSelectHelper, updateSearchString searchString: String?) {
  706. // if let searchString = searchString {
  707. // statusBar.setLeftStringValue(String.localizedStringWithFormat(NSLocalizedString("Finding: \"%@\"", comment: "Status message"), searchString))
  708. // } else {
  709. // updateStatus()
  710. // }
  711. // }
  712. func addBookmarkURLsToArray(_ items: [KMBookmark], _ array: inout [URL]) {
  713. for bm in items {
  714. if bm.bookmarkType == .bookmark {
  715. if let url = bm.fileURL {
  716. array.append(url)
  717. }
  718. } else if bm.bookmarkType != .separator {
  719. addBookmarkURLsToArray(bm.children, &array)
  720. }
  721. }
  722. }
  723. }
  724. extension KMBookmarkController: NSToolbarDelegate, NSToolbarItemValidation {
  725. func toolbarDefaultItemIdentifiers(_ toolbar: NSToolbar) -> [NSToolbarItem.Identifier] {
  726. // return [.flexibleSpace, .yourItem1, .yourItem2, .yourItem3]
  727. return [
  728. NSToolbarItem.Identifier(kBookmarksNewFolderToolbarItemIdentifier),
  729. NSToolbarItem.Identifier(kBookmarksNewSeparatorToolbarItemIdentifier),
  730. NSToolbarItem.Identifier(kBookmarksDeleteToolbarItemIdentifier)
  731. ]
  732. }
  733. func toolbarAllowedItemIdentifiers(_ toolbar: NSToolbar) -> [NSToolbarItem.Identifier] {
  734. // return [.yourItem1, .yourItem2, .yourItem3, .flexibleSpace, .space]
  735. return [
  736. NSToolbarItem.Identifier(kBookmarksNewFolderToolbarItemIdentifier),
  737. NSToolbarItem.Identifier(kBookmarksNewSeparatorToolbarItemIdentifier),
  738. NSToolbarItem.Identifier(kBookmarksDeleteToolbarItemIdentifier),
  739. .flexibleSpace,
  740. .space
  741. ]
  742. }
  743. func toolbar(_ toolbar: NSToolbar, itemForItemIdentifier itemIdentifier: NSToolbarItem.Identifier, willBeInsertedIntoToolbar flag: Bool) -> NSToolbarItem? {
  744. return toolbarItems[itemIdentifier.rawValue]
  745. }
  746. func validateToolbarItem(_ item: NSToolbarItem) -> Bool {
  747. // guard let toolbar = self.window?.toolbar else { return false }
  748. //
  749. // if toolbar.customizationPaletteIsRunning {
  750. // return false
  751. // } else if toolbarItem.itemIdentifier == kBookmarksDeleteToolbarItemIdentifier {
  752. // return outlineView.canDelete
  753. // }
  754. return true
  755. }
  756. }
  757. extension KMBookmarkController: NSMenuDelegate, NSMenuItemValidation {
  758. func validateMenuItem(_ menuItem: NSMenuItem) -> Bool {
  759. // if menuItem.action == #selector(toggleStatusBar(_:)) {
  760. // if statusBar.isVisible {
  761. // menuItem.title = NSLocalizedString("Hide Status Bar", comment: "Menu item title")
  762. // } else {
  763. // menuItem.title = NSLocalizedString("Show Status Bar", comment: "Menu item title")
  764. // }
  765. // return true
  766. // } else if menuItem.action == #selector(addBookmark(_:)) {
  767. // return menuItem.tag == 0
  768. // }
  769. return true
  770. }
  771. }
  772. extension NSURL {
  773. static func canReadFileURL(from pboard: NSPasteboard) -> Bool {
  774. let canReadFileURLsOnly = [NSPasteboard.ReadingOptionKey.urlReadingFileURLsOnly: true]
  775. let canReadClasses = [NSURL.self]
  776. return pboard.canReadObject(forClasses: canReadClasses, options: canReadFileURLsOnly) ||
  777. pboard.canReadItem(withDataConformingToTypes: [NSPasteboard.PasteboardType.fileURL.rawValue])
  778. }
  779. static func readFileURLs(from pboard: NSPasteboard) -> [URL] {
  780. if let fileURLs = pboard.readObjects(forClasses: [NSURL.self], options: [.urlReadingFileURLsOnly: true]) as? [URL], !fileURLs.isEmpty {
  781. return fileURLs
  782. } else if ((pboard.types?.contains(.fileURL)) != nil) {
  783. if let filenames = pboard.propertyList(forType: .fileURL) as? [String] {
  784. return filenames.compactMap { URL(fileURLWithPath: $0) }
  785. }
  786. }
  787. return []
  788. }
  789. }