Don’t ask for instance, but rather just its protocol

onekiloparsec · Oct 21, 2018 · c4f7f64 · c4f7f64
1 parent c388928
commit c4f7f64
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/KPCJumpBarControl/JumpBarControl.swift b/KPCJumpBarControl/JumpBarControl.swift
@@ -140,7 +140,7 @@ open class JumpBarControl: NSControl, JumpBarSegmentControlDelegate {
         self.useItemsTree(self.binding!.treeController!.arrangedRootObjects())
     }
 
-    open func useItemsTree(_ itemsTree: [JumpBarItem]) {
+    open func useItemsTree(_ itemsTree: [JumpBarItemProtocol]) {
         self.segmentControls().forEach { $0.removeFromSuperview() }
 
         // At that stage, we do items all in one shot. Might be optimized later.

diff --git a/KPCJumpBarControl/NSMenu+Utils.swift b/KPCJumpBarControl/NSMenu+Utils.swift
@@ -54,7 +54,7 @@ extension NSImage {
 
 extension NSMenu {
 
-    static func menuWithSegmentsTree(_ segmentsTree: [JumpBarItem],
+    static func menuWithSegmentsTree(_ segmentsTree: [JumpBarItemProtocol],
                                      target: AnyObject,
                                      action: Selector) -> NSMenu {
         let menu = NSMenu()

diff --git a/KPCJumpBarControl/NSTreeController+Utils.swift b/KPCJumpBarControl/NSTreeController+Utils.swift
@@ -9,7 +9,7 @@
 import AppKit
 
 extension NSTreeController {
-    func arrangedRootObjects() -> [JumpBarItem] {
+    func arrangedRootObjects() -> [JumpBarItemProtocol] {
         guard let rootTreeNode = self.arrangedObjects as AnyObject? else {
             return []
         }
@@ -18,7 +18,7 @@ extension NSTreeController {
             return []
         }
 
-        return proxyChildren.map({ $0.representedObject as! JumpBarItem })
+        return proxyChildren.map({ $0.representedObject as! JumpBarItemProtocol })
     }    
 }