use generics to DRY up context code

1 files changed, 3 insertions, 31 deletions

diff --git a/pkg/gui/context/reflog_commits_context.go b/pkg/gui/context/reflog_commits_context.go
index fa136a7d4..815805515 100644
--- a/pkg/gui/context/reflog_commits_context.go
+++ b/pkg/gui/context/reflog_commits_context.go
@@ -3,12 +3,11 @@ package context
 import (
 	"github.com/jesseduffield/gocui"
 	"github.com/jesseduffield/lazygit/pkg/commands/models"
-	"github.com/jesseduffield/lazygit/pkg/gui/context/traits"
 	"github.com/jesseduffield/lazygit/pkg/gui/types"
 )
 
 type ReflogCommitsContext struct {
-	*ReflogCommitsViewModel
+	*BasicViewModel[*models.Commit]
 	*ListContextTrait
 }
 
@@ -25,10 +24,10 @@ func NewReflogCommitsContext(
 
 	c *types.HelperCommon,
 ) *ReflogCommitsContext {
-	viewModel := NewReflogCommitsViewModel(getModel)
+	viewModel := NewBasicViewModel(getModel)
 
 	return &ReflogCommitsContext{
-		ReflogCommitsViewModel: viewModel,
+		BasicViewModel: viewModel,
 		ListContextTrait: &ListContextTrait{
 			Context: NewSimpleContext(NewBaseContext(NewBaseContextOpts{
 				ViewName:   "commits",
@@ -71,30 +70,3 @@ func (self *ReflogCommitsContext) GetSelectedRefName() string {
 
 	return item.RefName()
 }
-
-type ReflogCommitsViewModel struct {
-	*traits.ListCursor
-	getModel func() []*models.Commit
-}
-
-func NewReflogCommitsViewModel(getModel func() []*models.Commit) *ReflogCommitsViewModel {
-	self := &ReflogCommitsViewModel{
-		getModel: getModel,
-	}
-
-	self.ListCursor = traits.NewListCursor(self)
-
-	return self
-}
-
-func (self *ReflogCommitsViewModel) GetItemsLength() int {
-	return len(self.getModel())
-}
-
-func (self *ReflogCommitsViewModel) GetSelected() *models.Commit {
-	if self.GetItemsLength() == 0 {
-		return nil
-	}
-
-	return self.getModel()[self.GetSelectedLineIdx()]
-}