refactor for protobuf

1 files changed, 15 insertions, 0 deletions

diff --git a/treeWidget.go b/treeWidget.go
index 4d92abd..1559f68 100644
--- a/treeWidget.go
+++ b/treeWidget.go
@@ -152,3 +152,18 @@ func (tk *guiWidget) findWidgetByView(v *gocui.View) *guiWidget {
 	}
 	return nil
 }
+
+func (tk *guiWidget) WidgetType() widget.WidgetType {
+	if tk.node == nil {
+		return widget.Label
+	}
+	return tk.node.WidgetType
+}
+
+func (tk *guiWidget) WidgetId() int {
+	return tk.node.WidgetId
+}
+
+func (tk *guiWidget) GetLabel() string {
+	return tk.node.GetLabel()
+}