Read all lines from task when starting to search

Author: stefanhaller

pkg/gui/controllers/main_view_controller.go

@@ -1,23 +1,24 @@
 package controllers
 
 import (
+	"github.com/jesseduffield/lazygit/pkg/gui/context"
 	"github.com/jesseduffield/lazygit/pkg/gui/types"
 )
 
 type MainViewController struct {
 	baseController
 	c *ControllerCommon
 
-	context      types.Context
-	otherContext types.Context
+	context      *context.MainContext
+	otherContext *context.MainContext
 }
 
 var _ types.IController = &MainViewController{}
 
 func NewMainViewController(
 	c *ControllerCommon,
-	context types.Context,
-	otherContext types.Context,
+	context *context.MainContext,
+	otherContext *context.MainContext,
 ) *MainViewController {
 	return &MainViewController{
 		baseController: baseController{},
@@ -41,6 +42,13 @@ func (self *MainViewController) GetKeybindings(opts types.KeybindingsOpts) []*ty
 			Handler:     self.escape,
 			Description: self.c.Tr.ExitFocusedMainView,
 		},
+		{
+			// overriding this because we want to read all of the task's output before we start searching
+			Key:         opts.GetKey(opts.Config.Universal.StartSearch),
+			Handler:     self.openSearch,
+			Description: self.c.Tr.StartSearch,
+			Tag:         "navigation",
+		},
 	}
 }
 
@@ -61,3 +69,15 @@ func (self *MainViewController) escape() error {
 	self.c.Context().Pop()
 	return nil
 }
+
+func (self *MainViewController) openSearch() error {
+	if manager := self.c.GetViewBufferManagerForView(self.context.GetView()); manager != nil {
+		manager.ReadToEnd(func() {
+			self.c.OnUIThread(func() error {
+				return self.c.Helpers().Search.OpenSearchPrompt(self.context)
+			})
+		})
+	}
+
+	return nil
+}