Add SelectedNode method

gizak · Jun 10, 2019 · 81435f7 · 81435f7
1 parent 7357ed4
commit 81435f7
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)
 }