refactor controllers

1 files changed, 16 insertions, 2 deletions

diff --git a/pkg/gui/context/local_commits_context.go b/pkg/gui/context/local_commits_context.go
index 533d97cb2..9da4721e3 100644
--- a/pkg/gui/context/local_commits_context.go
+++ b/pkg/gui/context/local_commits_context.go
@@ -23,7 +23,7 @@ func NewLocalCommitsContext(
 	onRenderToMain func(...types.OnFocusOpts) error,
 	onFocusLost func() error,
 
-	c *types.ControllerCommon,
+	c *types.HelperCommon,
 ) *LocalCommitsContext {
 	viewModel := NewLocalCommitsViewModel(getModel)
 
@@ -61,8 +61,14 @@ func (self *LocalCommitsContext) GetSelectedItemId() string {
 
 type LocalCommitsViewModel struct {
 	*traits.ListCursor
+	getModel func() []*models.Commit
+
+	// If this is true we limit the amount of commits we load, for the sake of keeping things fast.
+	// If the user attempts to scroll past the end of the list, we will load more commits.
 	limitCommits bool
-	getModel     func() []*models.Commit
+
+	// If this is true we'll use git log --all when fetching the commits.
+	showWholeGitGraph bool
 }
 
 func NewLocalCommitsViewModel(getModel func() []*models.Commit) *LocalCommitsViewModel {
@@ -95,3 +101,11 @@ func (self *LocalCommitsViewModel) SetLimitCommits(value bool) {
 func (self *LocalCommitsViewModel) GetLimitCommits() bool {
 	return self.limitCommits
 }
+
+func (self *LocalCommitsViewModel) SetShowWholeGitGraph(value bool) {
+	self.showWholeGitGraph = value
+}
+
+func (self *LocalCommitsViewModel) GetShowWholeGitGraph() bool {
+	return self.showWholeGitGraph
+}