lazygit/pkg/gui/context/main_context.go

package context

import (
	"github.com/jesseduffield/lazygit/pkg/gocui"
	"github.com/jesseduffield/lazygit/pkg/gui/types"
)

type MainContext struct {
	*SimpleContext
	*SearchTrait

	diffSelect types.DiffSelectState
	// dragAnchorViewLine is the view line a mouse-down landed on, remembered so that a
	// drag that follows can anchor its range there. The click may have selected a whole
	// hunk, whose range anchor is the block's far end, so the clicked line can't be
	// read back from the view.
	dragAnchorViewLine int
	// selectableContentRenderKey names the render whose content HasSelectableContent
	// was worked out from. What there is to select is a property of the content, so an
	// answer about the content of another render says nothing about this one.
	selectableContentRenderKey string
	// contentIsDiff says the render this pane was last given is the diff the panel
	// beneath offers here, rather than a message or an explanation (see
	// types.ContentIsDiff). It is known as the render is asked for, well before there
	// is any content to look at.
	contentIsDiff bool
}

var (
	_ types.ISearchableContext = (*MainContext)(nil)
	_ types.DiffPaneContext    = (*MainContext)(nil)
)

// DiffSelectState returns the focused main view's selection mode state, for the
// controllers to read and mutate directly.
func (self *MainContext) DiffSelectState() *types.DiffSelectState {
	return &self.diffSelect
}

// ResetDiffSelectMode returns the pane's selection to the default mode — a single
// line, no range — for whenever it is established from scratch rather than moved. The
// view's range anchor is cleared too, so the next render highlights the cursor line
// only.
func (self *MainContext) ResetDiffSelectMode() {
	self.diffSelect.Mode = types.DiffSelectModeLine
	self.diffSelect.RangeIsSticky = false
	self.diffSelect.UserEnabledHunkMode = false
	self.GetView().CancelRangeSelect()
}

// SetDragAnchorViewLine records the view line a mouse-down landed on, so that a drag
// that follows can anchor its range there (see dragAnchorViewLine).
func (self *MainContext) SetDragAnchorViewLine(viewLine int) {
	self.dragAnchorViewLine = viewLine
}

// DragAnchorViewLine returns the view line the last mouse-down landed on.
func (self *MainContext) DragAnchorViewLine() int {
	return self.dragAnchorViewLine
}

// SelectableContentRenderKey returns the render HasSelectableContent describes (see
// selectableContentRenderKey).
func (self *MainContext) SelectableContentRenderKey() string {
	return self.selectableContentRenderKey
}

// SetSelectableContentRenderKey records which render HasSelectableContent describes.
func (self *MainContext) SetSelectableContentRenderKey(key string) {
	self.selectableContentRenderKey = key
}

// ContentIsDiff reports whether this pane is showing the panel's diff (see
// contentIsDiff).
func (self *MainContext) ContentIsDiff() bool {
	return self.contentIsDiff
}

// SetContentIsDiff records what kind of render the pane is being given.
func (self *MainContext) SetContentIsDiff(value bool) {
	self.contentIsDiff = value
}

func NewMainContext(
	view *gocui.View,
	windowName string,
	key types.ContextKey,
	c *ContextCommon,
) *MainContext {
	ctx := &MainContext{
		SimpleContext: NewSimpleContext(
			NewBaseContext(NewBaseContextOpts{
				Kind:       types.MAIN_CONTEXT,
				View:       view,
				WindowName: windowName,
				Key:        key,
				Focusable:  true,
			})),
		SearchTrait: NewSearchTrait(c),
	}

	return ctx
}

func (self *MainContext) ModelSearchResults(searchStr string, caseSensitive bool) []gocui.SearchPosition {
	return nil
}

// When selecting a search result, collapse a range selection (whether sticky or not)
// or a hunk selection to just the matching line.
func (self *MainContext) OnSearchSelect(int) {
	self.ResetDiffSelectMode()
}