Add SelectedNode method

gizak · Jul 15, 2019 · e3d12be · e3d12be
1 parent 691e2f0
commit e3d12be
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/widgets/tree.go b/widgets/tree.go
@@ -173,6 +173,13 @@ func (self *Tree) ScrollAmount(amount int) {
 	}
 }
 
+func (self *Tree) SelectedNode() *TreeNode {
+	if len(self.rows) == 0 {
+		return nil
+	}
+	return self.rows[self.SelectedRow]
+}
+
 func (self *Tree) ScrollUp() {
 	self.ScrollAmount(-1)
 }