Migrate gitlab to multisource widget

wtfutil · May 24, 2019 · a283dd8 · a283dd8
1 parent 9a877b5
commit a283dd8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 27 deletions.
diff --git a/modules/gitlab/keyboard.go b/modules/gitlab/keyboard.go
@@ -5,9 +5,9 @@ import "github.com/gdamore/tcell"
 func (widget *Widget) initializeKeyboardControls() {
 	widget.SetKeyboardChar("/", widget.ShowHelp, "Show/hide this help prompt")
 	widget.SetKeyboardChar("r", widget.Refresh, "Refresh widget")
-	widget.SetKeyboardChar("h", widget.Prev, "Select previous item")
-	widget.SetKeyboardChar("l", widget.Next, "Select next item")
+	widget.SetKeyboardChar("h", widget.PrevSource, "Select previous project")
+	widget.SetKeyboardChar("l", widget.NextSource, "Select next project")
 
-	widget.SetKeyboardKey(tcell.KeyLeft, widget.Prev, "Select previous item")
-	widget.SetKeyboardKey(tcell.KeyRight, widget.Next, "Select next item")
+	widget.SetKeyboardKey(tcell.KeyLeft, widget.PrevSource, "Select previous project")
+	widget.SetKeyboardKey(tcell.KeyRight, widget.NextSource, "Select next project")
 }
diff --git a/modules/gitlab/widget.go b/modules/gitlab/widget.go
@@ -8,10 +8,10 @@ import (
 
 type Widget struct {
 	wtf.KeyboardWidget
+	wtf.MultiSourceWidget
 	wtf.TextWidget
 
 	GitlabProjects []*GitlabProject
-	Idx            int
 
 	gitlab   *glb.Client
 	settings *Settings
@@ -26,10 +26,9 @@ func NewWidget(app *tview.Application, pages *tview.Pages, settings *Settings) *
 	}
 
 	widget := Widget{
-		KeyboardWidget: wtf.NewKeyboardWidget(app, pages, settings.common),
-		TextWidget:     wtf.NewTextWidget(app, settings.common, true),
-
-		Idx: 0,
+		KeyboardWidget:    wtf.NewKeyboardWidget(app, pages, settings.common),
+		MultiSourceWidget: wtf.NewMultiSourceWidget(settings.common, "repository", "repositories"),
+		TextWidget:        wtf.NewTextWidget(app, settings.common, true),
 
 		gitlab:   gitlab,
 		settings: settings,
@@ -39,6 +38,7 @@ func NewWidget(app *tview.Application, pages *tview.Pages, settings *Settings) *
 
 	widget.initializeKeyboardControls()
 	widget.View.SetInputCapture(widget.InputCapture)
+	widget.SetDisplayFunction(widget.display)
 
 	widget.KeyboardWidget.SetView(widget.View)
 
@@ -55,24 +55,6 @@ func (widget *Widget) Refresh() {
 	widget.display()
 }
 
-func (widget *Widget) Next() {
-	widget.Idx++
-	if widget.Idx == len(widget.GitlabProjects) {
-		widget.Idx = 0
-	}
-
-	widget.display()
-}
-
-func (widget *Widget) Prev() {
-	widget.Idx--
-	if widget.Idx < 0 {
-		widget.Idx = len(widget.GitlabProjects) - 1
-	}
-
-	widget.display()
-}
-
 func (widget *Widget) HelpText() string {
 	return widget.KeyboardWidget.HelpText()
 }