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. //
  268. // func addItemForBookmark(_ bookmark: KMBookmark, toMenu menu: NSMenu, isFolder: Bool, isAlternate: Bool) {
  269. // var item: NSMenuItem?
  270. // if isFolder {
  271. // item = menu.addItem(withSubmenuAndTitle: bookmark.label)
  272. // item?.submenu?.delegate = self
  273. // } else {
  274. // item = menu.addItem(withTitle: bookmark.label, action: #selector(openBookmark(_:)), target: self)
  275. // }
  276. // item?.representedObject = bookmark
  277. // if isAlternate {
  278. // item?.keyEquivalentModifierMask = .alternate
  279. // item?.isAlternate = true
  280. // item?.setImageAndSize(bookmark.alternateIcon)
  281. // } else {
  282. // item?.setImageAndSize(bookmark.icon)
  283. // }
  284. // }
  285. //
  286. // func menuNeedsUpdate(_ menu: NSMenu) {
  287. // if menu == outlineView.menu {
  288. // let row = outlineView.clickedRow
  289. // menu.removeAllItems()
  290. // if row != -1 {
  291. // menu.addItem(withTitle: NSLocalizedString("Remove", comment: "Menu item title"), action: #selector(deleteBookmarks(_:)), target: self)
  292. // menu.addItem(withTitle: NSLocalizedString("Open", comment: "Menu item title"), action: #selector(openBookmarks(_:)), target: self)
  293. // menu.addItem(withTitle: NSLocalizedString("Quick Look", comment: "Menu item title"), action: #selector(previewBookmarks(_:)), target: self)
  294. // menu.addItem(.separator())
  295. // }
  296. // menu.addItem(withTitle: NSLocalizedString("New Folder", comment: "Menu item title"), action: #selector(insertBookmarkFolder(_:)), target: self)
  297. // menu.addItem(withTitle: NSLocalizedString("New Separator", comment: "Menu item title"), action: #selector(insertBookmarkSeparator(_:)), target: self)
  298. // } else {
  299. // 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 }
  300. //
  301. // let bookmarks = bm.children
  302. // var i = menu.numberOfItems
  303. //
  304. // while i > 0 {
  305. // if let menuItem = menu.item(at: i - 1), menuItem.isSeparatorItem || menuItem.representedObject != nil {
  306. // menu.removeItem(menuItem)
  307. // }
  308. // i -= 1
  309. // }
  310. //
  311. // if supermenu == NSApp.mainMenu, let previousSession = previousSession {
  312. // menu.addItem(.separator())
  313. // addItemForBookmark(previousSession, toMenu: menu, isFolder: false, isAlternate: false)
  314. // addItemForBookmark(previousSession, toMenu: menu, isFolder: true, isAlternate: true)
  315. // }
  316. //
  317. // if menu.numberOfItems > 0, bookmarks.count > 0 {
  318. // menu.addItem(.separator())
  319. // }
  320. //
  321. // for bm in bookmarks {
  322. // switch bm.bookmarkType {
  323. // case .folder:
  324. // addItemForBookmark(bm, toMenu: menu, isFolder: true, isAlternate: false)
  325. // addItemForBookmark(bm, toMenu: menu, isFolder: false, isAlternate: true)
  326. // case .session:
  327. // addItemForBookmark(bm, toMenu: menu, isFolder: false, isAlternate: false)
  328. // addItemForBookmark(bm, toMenu: menu, isFolder: true, isAlternate: true)
  329. // case .separator:
  330. // menu.addItem(.separator())
  331. // default:
  332. // addItemForBookmark(bm, toMenu: menu, isFolder: false, isAlternate: false)
  333. // }
  334. // }
  335. // }
  336. // }
  337. //
  338. // // avoid rebuilding the bookmarks menu on every key event
  339. // func menuHasKeyEquivalent(_ menu: NSMenu, for event: NSEvent, target: AutoreleasingUnsafeMutablePointer<AnyObject?>?, action: UnsafeMutablePointer<Selector?>?) -> Bool { false }
  340. //
  341. // // MARK: - Toolbar
  342. //
  343. func setupToolbar() {
  344. // Create a new toolbar instance, and attach it to our document window
  345. let toolbar = NSToolbar(identifier: kBookmarksToolbarIdentifier)
  346. var dict = [String: NSToolbarItem]()
  347. // Set up toolbar properties: Allow customization, give a default display mode, and remember state in user defaults
  348. toolbar.allowsUserCustomization = true
  349. toolbar.autosavesConfiguration = true
  350. toolbar.displayMode = .default
  351. // We are the delegate
  352. toolbar.delegate = self
  353. // Add template toolbar items
  354. var item = NSToolbarItem(itemIdentifier: NSToolbarItem.Identifier(kBookmarksNewFolderToolbarItemIdentifier))
  355. item.label = NSLocalizedString("New Folder", comment: "Toolbar item label")
  356. item.paletteLabel = NSLocalizedString("New Folder", comment: "Toolbar item label")
  357. item.toolTip = NSLocalizedString("Add a New Folder", comment: "Tool tip message")
  358. // item.image = NSImage(named: "NewFolder")
  359. item.image = NSImage(named: NSImage.folderName)!
  360. item.target = self
  361. item.action = #selector(insertBookmarkFolder(_:))
  362. dict[kBookmarksNewFolderToolbarItemIdentifier] = item
  363. item = NSToolbarItem(itemIdentifier: NSToolbarItem.Identifier(kBookmarksNewSeparatorToolbarItemIdentifier))
  364. item.label = NSLocalizedString("New Separator", comment: "Toolbar item label")
  365. item.paletteLabel = NSLocalizedString("New Separator", comment: "Toolbar item label")
  366. item.toolTip = NSLocalizedString("Add a New Separator", comment: "Tool tip message")
  367. // item.image = NSImage(named: "NewSeparator")
  368. item.image = NSImage(named: NSImage.shareTemplateName)!
  369. item.target = self
  370. item.action = #selector(insertBookmarkSeparator(_:))
  371. dict[kBookmarksNewSeparatorToolbarItemIdentifier] = item
  372. item = NSToolbarItem(itemIdentifier: NSToolbarItem.Identifier(kBookmarksDeleteToolbarItemIdentifier))
  373. item.label = NSLocalizedString("Delete", comment: "Toolbar item label")
  374. item.paletteLabel = NSLocalizedString("Delete", comment: "Toolbar item label")
  375. item.toolTip = NSLocalizedString("Delete Selected Items", comment: "Tool tip message")
  376. item.image = NSWorkspace.shared.icon(forFileType: NSFileTypeForHFSTypeCode(OSType(kToolbarDeleteIcon)))
  377. item.target = self
  378. item.action = #selector(deleteBookmark(_:))
  379. dict[kBookmarksDeleteToolbarItemIdentifier] = item
  380. toolbarItems = dict
  381. // Attach the toolbar to the window
  382. self.window?.toolbar = toolbar
  383. }
  384. //
  385. // // MARK: - Quick Look Panel Support
  386. //
  387. // func acceptsPreviewPanelControl(_ panel: QLPreviewPanel) -> Bool {
  388. // return true
  389. // }
  390. //
  391. // func beginPreviewPanelControl(_ panel: QLPreviewPanel) {
  392. // panel.delegate = self
  393. // panel.dataSource = self
  394. // }
  395. //
  396. // func endPreviewPanelControl(_ panel: QLPreviewPanel) {
  397. // }
  398. //
  399. // func previewItems() -> [KMBookmark] {
  400. // var items = [KMBookmark]()
  401. //
  402. // outlineView.selectedRowIndexes.enumerated().forEach { (idx, _) in
  403. // if let item = outlineView.item(atRow: idx) as? KMBookmark {
  404. // if item.bookmarkType == .bookmark {
  405. // items.append(item)
  406. // } else if item.bookmarkType == .session {
  407. // items.append(contentsOf: item.children)
  408. // }
  409. // }
  410. // }
  411. // return items
  412. // }
  413. //
  414. // func numberOfPreviewItems(in panel: QLPreviewPanel) -> Int {
  415. // return previewItems().count
  416. // }
  417. //
  418. // func previewPanel(_ panel: QLPreviewPanel, previewItemAt anIndex: Int) -> QLPreviewItem {
  419. // return previewItems()[anIndex]
  420. // }
  421. //
  422. // func previewPanel(_ panel: QLPreviewPanel, sourceFrameOnScreenForPreviewItem item: QLPreviewItem) -> NSRect {
  423. // var item = item
  424. // if let parent = (item as? KMBookmark)?.parent, parent.bookmarkType == .session {
  425. // item = parent
  426. // }
  427. // let row = outlineView.row(forItem: item)
  428. // var iconRect = NSZeroRect
  429. // if let item = item as? KMBookmark, row != -1 {
  430. // let cell = outlineView.preparedCell(atColumn: 0, row: row) as? SKTextWithIconCell
  431. // iconRect = cell?.iconRect(forBounds: outlineView.frameOfCell(atColumn: 0, row: row)) ?? NSZeroRect
  432. // if outlineView.visibleRect.intersects(iconRect) {
  433. // iconRect = outlineView.convert(iconRect, to: nil)
  434. // } else {
  435. // iconRect = NSZeroRect
  436. // }
  437. // }
  438. // return iconRect
  439. // }
  440. //
  441. // func previewPanel(_ panel: QLPreviewPanel, transitionImageForPreviewItem item: QLPreviewItem, contentRect: UnsafeMutablePointer<NSRect>) -> NSImage? {
  442. // var item = item
  443. // if let parent = (item as? KMBookmark)?.parent, parent.bookmarkType == .session {
  444. // item = parent
  445. // }
  446. // return (item as? KMBookmark)?.icon
  447. // }
  448. //
  449. // func previewPanel(_ panel: QLPreviewPanel, handle event: NSEvent) -> Bool {
  450. // if event.type == .keyDown {
  451. // outlineView.keyDown(with: event)
  452. // return true
  453. // }
  454. // return false
  455. // }
  456. //
  457. }
  458. extension KMBookmarkController: NSOutlineViewDelegate, NSOutlineViewDataSource {
  459. //MARK: NSOutlineViewDataSource
  460. func minimumCoverForBookmarks(_ items: [KMBookmark]) -> [KMBookmark] {
  461. var lastBm: KMBookmark?
  462. var minimalCover = [KMBookmark]()
  463. for bm in items {
  464. if !(bm.isDescendant(of: lastBm)) {
  465. minimalCover.append(bm)
  466. lastBm = bm
  467. }
  468. }
  469. return minimalCover
  470. }
  471. func outlineView(_ ov: NSOutlineView, numberOfChildrenOfItem item: Any?) -> Int {
  472. let bookmark = item as? KMBookmark ?? bookmarkRoot
  473. return bookmark.bookmarkType == .folder ? bookmark.children.count : 0
  474. }
  475. func outlineView(_ ov: NSOutlineView, isItemExpandable item: Any) -> Bool {
  476. let bookmark = item as! KMBookmark
  477. return bookmark.bookmarkType == .folder
  478. }
  479. func outlineView(_ ov: NSOutlineView, child index: Int, ofItem item: Any?) -> Any {
  480. let bookmark = (item as? KMBookmark) ?? bookmarkRoot
  481. return bookmark.objectOfChidren(index: index)
  482. }
  483. func outlineView(_ ov: NSOutlineView, objectValueFor tableColumn: NSTableColumn?, byItem item: Any?) -> Any? {
  484. guard let column = tableColumn else { return nil }
  485. guard let bm = item as? KMBookmark else { return nil }
  486. let tcID = column.identifier
  487. switch tcID {
  488. case kLabelIdentifier:
  489. return [kTextWithIconStringKey: bm.label, kTextWithIconImageKey: bm.icon]
  490. case kFileIdentifier:
  491. if bm.bookmarkType == .folder || bm.bookmarkType == .session {
  492. let count = bm.children.count
  493. return count == 1 ? NSLocalizedString("1 item", comment: "Bookmark folder description") : String.localizedStringWithFormat(NSLocalizedString("%ld items", comment: "Bookmark folder description"), count)
  494. } else {
  495. return bm.fileURL?.path ?? ""
  496. }
  497. case kPageIdentifier:
  498. return bm.pageNumber
  499. default:
  500. return nil
  501. }
  502. }
  503. func outlineView(_ ov: NSOutlineView, setObjectValue object: Any?, for tableColumn: NSTableColumn?, byItem item: Any?) {
  504. guard let column = tableColumn else { return }
  505. guard let bm = item as? KMBookmark else { return }
  506. let tcID = column.identifier
  507. switch tcID {
  508. case kLabelIdentifier:
  509. if let newLabel = (object as? [String: Any])?[kTextWithIconStringKey] as? String, newLabel != bm.label {
  510. bm.label = newLabel
  511. }
  512. case kPageIdentifier:
  513. if let newPageNumber = object as? Int, newPageNumber != bm.pageNumber.intValue {
  514. bm.pageNumber = newPageNumber as NSNumber
  515. }
  516. default:
  517. break
  518. }
  519. }
  520. func outlineView(_ ov: NSOutlineView, writeItems items: [Any], to pboard: NSPasteboard) -> Bool {
  521. draggedBookmarks = minimumCoverForBookmarks(items as! [KMBookmark])
  522. pboard.clearContents()
  523. pboard.setData(Data(), forType: kPasteboardTypeBookmarkRows)
  524. return true
  525. }
  526. func outlineView(_ ov: NSOutlineView, validateDrop info: NSDraggingInfo, proposedItem item: Any?, proposedChildIndex index: Int) -> NSDragOperation {
  527. guard index != NSOutlineViewDropOnItemIndex else { return [] }
  528. let pboard = info.draggingPasteboard
  529. if pboard.canReadItem(withDataConformingToTypes: [kPasteboardTypeBookmarkRows.rawValue]) && info.draggingSource as? NSOutlineView == ov {
  530. return .move
  531. } else if NSURL.canReadFileURL(from: pboard) {
  532. return .every
  533. }
  534. return []
  535. }
  536. func outlineView(_ ov: NSOutlineView, acceptDrop info: NSDraggingInfo, item: Any?, childIndex index: Int) -> Bool {
  537. let pboard = info.draggingPasteboard
  538. if pboard.canReadItem(withDataConformingToTypes: [kPasteboardTypeBookmarkRows.rawValue]) && info.draggingSource as? NSOutlineView == ov {
  539. var movedBookmarks = [KMBookmark]()
  540. var indexes = IndexSet()
  541. var insertionIndex = index
  542. let targetItem = item as? KMBookmark ?? bookmarkRoot
  543. for bookmark in draggedBookmarks {
  544. guard let parent = bookmark.parent else { continue }
  545. guard let bookmarkIndex = parent.children.firstIndex(of: bookmark) else { continue }
  546. if targetItem == parent {
  547. if insertionIndex > bookmarkIndex {
  548. insertionIndex -= 1
  549. }
  550. if insertionIndex == bookmarkIndex {
  551. continue
  552. }
  553. }
  554. parent.removeObjectFromChildren(index: bookmarkIndex)
  555. targetItem.insert(child: bookmark, atIndex: insertionIndex)
  556. movedBookmarks.append(bookmark)
  557. insertionIndex += 1
  558. }
  559. for bookmark in movedBookmarks {
  560. let row = ov.row(forItem: bookmark)
  561. if row != -1 {
  562. indexes.insert(row)
  563. }
  564. }
  565. if !indexes.isEmpty {
  566. ov.selectRowIndexes(indexes, byExtendingSelection: false)
  567. }
  568. return true
  569. } else {
  570. let urls = NSURL.readFileURLs(from: pboard)
  571. let newBookmarks = KMBookmark.bookmarks(urls: urls)
  572. if !newBookmarks.isEmpty {
  573. var indexes = IndexSet(integersIn: index..<(index + newBookmarks.count))
  574. (item as? KMBookmark ?? bookmarkRoot).mutableArrayValue(forKey: "children").insert(newBookmarks, at: indexes)
  575. if (item as? KMBookmark ?? bookmarkRoot) === bookmarkRoot || ov.isItemExpanded(item) {
  576. if (item as? KMBookmark ?? bookmarkRoot) !== bookmarkRoot {
  577. indexes.shift(startingAt: 0, by: ov.row(forItem: item) + 1)
  578. }
  579. ov.selectRowIndexes(indexes, byExtendingSelection: false)
  580. }
  581. return true
  582. }
  583. return false
  584. }
  585. }
  586. func outlineView(_ ov: NSOutlineView, dragEndedWith operation: NSDragOperation) {
  587. draggedBookmarks.removeAll()
  588. }
  589. // MARK: NSOutlineViewDelegate
  590. func outlineView(_ ov: NSOutlineView, dataCellFor tableColumn: NSTableColumn?, item: Any) -> Any? {
  591. if tableColumn == nil {
  592. return (item as? KMBookmark)?.bookmarkType == .separator ? KMSeparatorCell() : nil
  593. }
  594. return tableColumn?.dataCell(forRow: ov.row(forItem: item))
  595. }
  596. func outlineView(_ ov: NSOutlineView, willDisplayCell cell: Any, for tableColumn: NSTableColumn?, item: Any) {
  597. guard let column = tableColumn else { return }
  598. guard let cell = cell as? NSCell else { return }
  599. if column.identifier == kFileIdentifier {
  600. if let bm = item as? KMBookmark {
  601. // if bm.bookmarkType == .folder || bm.bookmarkType == .session {
  602. // cell.textColor = .disabledControlTextColor
  603. // } else {
  604. // cell.textColor = .controlTextColor
  605. // }
  606. }
  607. }
  608. }
  609. func outlineView(_ ov: NSOutlineView, shouldEdit tableColumn: NSTableColumn?, item: Any) -> Bool {
  610. guard let column = tableColumn else { return false }
  611. guard let bm = item as? KMBookmark else { return false }
  612. let tcID = column.identifier
  613. switch tcID {
  614. case kLabelIdentifier:
  615. return bm.bookmarkType != .separator
  616. case kPageIdentifier:
  617. return bm.pageIndex != NSNotFound
  618. default:
  619. return false
  620. }
  621. }
  622. func outlineView(_ ov: NSOutlineView, toolTipFor cell: NSCell, rect: UnsafeMutablePointer<NSRect>, tableColumn tc: NSTableColumn?, item: Any, mouseLocation: NSPoint) -> String {
  623. guard let column = tc else { return "" }
  624. guard let bm = item as? KMBookmark else { return "" }
  625. let tcID = column.identifier
  626. switch tcID {
  627. case kLabelIdentifier:
  628. return bm.label
  629. case kFileIdentifier:
  630. if bm.bookmarkType == .session {
  631. return ""
  632. // return bm.children.map { $0.path ?? "" }.joined(separator: "\n")
  633. } else if bm.bookmarkType == .folder {
  634. let count = bm.children.count
  635. return count == 1 ? NSLocalizedString("1 item", comment: "Bookmark folder description") : String.localizedStringWithFormat(NSLocalizedString("%ld items", comment: "Bookmark folder description"), count)
  636. } else {
  637. return bm.fileURL?.path ?? ""
  638. }
  639. case kPageIdentifier:
  640. return bm.pageNumber.stringValue
  641. default:
  642. return ""
  643. }
  644. }
  645. func outlineViewSelectionDidChange(_ notification: Notification) {
  646. updateStatus()
  647. if QLPreviewPanel.sharedPreviewPanelExists(), let previewPanel = QLPreviewPanel.shared(), previewPanel.isVisible, previewPanel.dataSource === self {
  648. previewPanel.reloadData()
  649. }
  650. }
  651. func outlineView(_ ov: NSOutlineView, deleteItems items: [Any]) {
  652. for item in minimumCoverForBookmarks(items as! [KMBookmark]).reversed() {
  653. guard let parent = item.parent, let itemIndex = parent.children.firstIndex(of: item) else { continue }
  654. parent.removeObjectFromChildren(index: itemIndex)
  655. }
  656. }
  657. func outlineView(_ ov: NSOutlineView, canDeleteItems items: [Any]) -> Bool {
  658. return !items.isEmpty
  659. }
  660. func outlineView(_ ov: NSOutlineView, copyItems items: [Any]) {
  661. var urls = [URL]()
  662. addBookmarkURLsToArray(minimumCoverForBookmarks(items as! [KMBookmark]), &urls)
  663. if !urls.isEmpty {
  664. let pboard = NSPasteboard.general
  665. pboard.clearContents()
  666. pboard.writeObjects(urls as [NSPasteboardWriting])
  667. }
  668. }
  669. func outlineView(_ ov: NSOutlineView, canCopyItems items: [Any]) -> Bool {
  670. return !items.isEmpty
  671. }
  672. func outlineView(_ ov: NSOutlineView, pasteFromPasteboard pboard: NSPasteboard) {
  673. let urls = NSURL.readFileURLs(from: pboard)
  674. let newBookmarks = KMBookmark.bookmarks(urls: urls)
  675. if !newBookmarks.isEmpty {
  676. var item: KMBookmark?
  677. var anIndex = 0
  678. getInsertionFolder(&item, childIndex: &anIndex)
  679. var indexes = IndexSet(integersIn: anIndex..<(anIndex + newBookmarks.count))
  680. (item ?? bookmarkRoot).mutableArrayValue(forKey: "children").insert(newBookmarks, at: indexes)
  681. if item === bookmarkRoot || ov.isItemExpanded(item) {
  682. if item !== bookmarkRoot {
  683. indexes.shift(startingAt: 0, by: ov.row(forItem: item) + 1)
  684. }
  685. ov.selectRowIndexes(indexes, byExtendingSelection: false)
  686. }
  687. }
  688. }
  689. func outlineView(_ ov: NSOutlineView, canPasteFromPasteboard pboard: NSPasteboard) -> Bool {
  690. return NSURL.canReadFileURL(from: pboard)
  691. }
  692. func outlineView(_ ov: NSOutlineView, typeSelectHelperSelectionStrings typeSelectHelper: SKTypeSelectHelper) -> [String] {
  693. let count = ov.numberOfRows
  694. var labels = [String]()
  695. for i in 0..<count {
  696. if let label = ov.item(atRow: i) {
  697. labels.append(label as! String)
  698. }
  699. }
  700. return labels
  701. }
  702. // func outlineView(_ ov: NSOutlineView, typeSelectHelper typeSelectHelper: SKTypeSelectHelper, didFailToFindMatchForSearchString searchString: String) {
  703. // statusBar.setLeftStringValue(String.localizedStringWithFormat(NSLocalizedString("No match: \"%@\"", comment: "Status message"), searchString))
  704. // }
  705. //
  706. // func outlineView(_ ov: NSOutlineView, typeSelectHelper typeSelectHelper: SKTypeSelectHelper, updateSearchString searchString: String?) {
  707. // if let searchString = searchString {
  708. // statusBar.setLeftStringValue(String.localizedStringWithFormat(NSLocalizedString("Finding: \"%@\"", comment: "Status message"), searchString))
  709. // } else {
  710. // updateStatus()
  711. // }
  712. // }
  713. func addBookmarkURLsToArray(_ items: [KMBookmark], _ array: inout [URL]) {
  714. for bm in items {
  715. if bm.bookmarkType == .bookmark {
  716. if let url = bm.fileURL {
  717. array.append(url)
  718. }
  719. } else if bm.bookmarkType != .separator {
  720. addBookmarkURLsToArray(bm.children, &array)
  721. }
  722. }
  723. }
  724. }
  725. extension KMBookmarkController: NSToolbarDelegate, NSToolbarItemValidation {
  726. func toolbarDefaultItemIdentifiers(_ toolbar: NSToolbar) -> [NSToolbarItem.Identifier] {
  727. // return [.flexibleSpace, .yourItem1, .yourItem2, .yourItem3]
  728. return [
  729. NSToolbarItem.Identifier(kBookmarksNewFolderToolbarItemIdentifier),
  730. NSToolbarItem.Identifier(kBookmarksNewSeparatorToolbarItemIdentifier),
  731. NSToolbarItem.Identifier(kBookmarksDeleteToolbarItemIdentifier)
  732. ]
  733. }
  734. func toolbarAllowedItemIdentifiers(_ toolbar: NSToolbar) -> [NSToolbarItem.Identifier] {
  735. // return [.yourItem1, .yourItem2, .yourItem3, .flexibleSpace, .space]
  736. return [
  737. NSToolbarItem.Identifier(kBookmarksNewFolderToolbarItemIdentifier),
  738. NSToolbarItem.Identifier(kBookmarksNewSeparatorToolbarItemIdentifier),
  739. NSToolbarItem.Identifier(kBookmarksDeleteToolbarItemIdentifier),
  740. .flexibleSpace,
  741. .space
  742. ]
  743. }
  744. func toolbar(_ toolbar: NSToolbar, itemForItemIdentifier itemIdentifier: NSToolbarItem.Identifier, willBeInsertedIntoToolbar flag: Bool) -> NSToolbarItem? {
  745. return toolbarItems[itemIdentifier.rawValue]
  746. }
  747. func validateToolbarItem(_ item: NSToolbarItem) -> Bool {
  748. // guard let toolbar = self.window?.toolbar else { return false }
  749. //
  750. // if toolbar.customizationPaletteIsRunning {
  751. // return false
  752. // } else if toolbarItem.itemIdentifier == kBookmarksDeleteToolbarItemIdentifier {
  753. // return outlineView.canDelete
  754. // }
  755. return true
  756. }
  757. }
  758. extension KMBookmarkController: NSMenuDelegate, NSMenuItemValidation {
  759. func validateMenuItem(_ menuItem: NSMenuItem) -> Bool {
  760. // if menuItem.action == #selector(toggleStatusBar(_:)) {
  761. // if statusBar.isVisible {
  762. // menuItem.title = NSLocalizedString("Hide Status Bar", comment: "Menu item title")
  763. // } else {
  764. // menuItem.title = NSLocalizedString("Show Status Bar", comment: "Menu item title")
  765. // }
  766. // return true
  767. // } else if menuItem.action == #selector(addBookmark(_:)) {
  768. // return menuItem.tag == 0
  769. // }
  770. return true
  771. }
  772. }
  773. extension NSURL {
  774. static func canReadFileURL(from pboard: NSPasteboard) -> Bool {
  775. let canReadFileURLsOnly = [NSPasteboard.ReadingOptionKey.urlReadingFileURLsOnly: true]
  776. let canReadClasses = [NSURL.self]
  777. return pboard.canReadObject(forClasses: canReadClasses, options: canReadFileURLsOnly) ||
  778. pboard.canReadItem(withDataConformingToTypes: [NSPasteboard.PasteboardType.fileURL.rawValue])
  779. }
  780. static func readFileURLs(from pboard: NSPasteboard) -> [URL] {
  781. if let fileURLs = pboard.readObjects(forClasses: [NSURL.self], options: [.urlReadingFileURLsOnly: true]) as? [URL], !fileURLs.isEmpty {
  782. return fileURLs
  783. } else if ((pboard.types?.contains(.fileURL)) != nil) {
  784. if let filenames = pboard.propertyList(forType: .fileURL) as? [String] {
  785. return filenames.compactMap { URL(fileURLWithPath: $0) }
  786. }
  787. }
  788. return []
  789. }
  790. }