package context import ( "github.com/jesseduffield/gocui" "github.com/jesseduffield/lazygit/pkg/commands/models" "github.com/jesseduffield/lazygit/pkg/gui/types" ) type BranchesContext struct { *BasicViewModel[*models.Branch] *ListContextTrait } var ( _ types.IListContext = (*BranchesContext)(nil) _ types.DiffableContext = (*BranchesContext)(nil) ) func NewBranchesContext( getModel func() []*models.Branch, view *gocui.View, getDisplayStrings func(startIdx int, length int) [][]string, c *types.HelperCommon, ) *BranchesContext { viewModel := NewBasicViewModel(getModel) self := &BranchesContext{ BasicViewModel: viewModel, ListContextTrait: &ListContextTrait{ Context: NewSimpleContext(NewBaseContext(NewBaseContextOpts{ View: view, WindowName: "branches", Key: LOCAL_BRANCHES_CONTEXT_KEY, Kind: types.SIDE_CONTEXT, Focusable: true, })), list: viewModel, getDisplayStrings: getDisplayStrings, c: c, }, } return self } func (self *BranchesContext) GetSelectedItemId() string { item := self.GetSelected() if item == nil { return "" } return item.ID() } func (self *BranchesContext) GetSelectedRef() types.Ref { branch := self.GetSelected() if branch == nil { return nil } return branch } func (self *BranchesContext) GetDiffTerminals() []string { // for our local branches we want to include both the branch and its upstream branch := self.GetSelected() if branch != nil { names := []string{branch.ID()} if branch.IsTrackingRemote() { names = append(names, branch.ID()+"@{u}") } return names } return nil }