mirror of
https://github.com/jesseduffield/lazygit.git
synced 2024-12-02 09:21:40 +02:00
a5f3515ad8
Something dumb that we're currently doing is expecting list items to define an ID method which returns a string. We use that when copying items to clipboard with ctrl+o and when getting a ref name for diffing. This commit gets us a little deeper into that hole by explicitly requiring list items to implement that method so that we can easily use the new helper functions in list_controller_trait.go. In future we need to just remove the whole ID thing entirely but I'm too lazy to do that right now.
86 lines
2.1 KiB
Go
86 lines
2.1 KiB
Go
package context
|
|
|
|
import (
|
|
"github.com/jesseduffield/lazygit/pkg/commands/models"
|
|
"github.com/jesseduffield/lazygit/pkg/gui/presentation"
|
|
"github.com/jesseduffield/lazygit/pkg/gui/types"
|
|
)
|
|
|
|
type BranchesContext struct {
|
|
*FilteredListViewModel[*models.Branch]
|
|
*ListContextTrait
|
|
}
|
|
|
|
var (
|
|
_ types.IListContext = (*BranchesContext)(nil)
|
|
_ types.DiffableContext = (*BranchesContext)(nil)
|
|
)
|
|
|
|
func NewBranchesContext(c *ContextCommon) *BranchesContext {
|
|
viewModel := NewFilteredListViewModel(
|
|
func() []*models.Branch { return c.Model().Branches },
|
|
func(branch *models.Branch) []string {
|
|
return []string{branch.Name}
|
|
},
|
|
)
|
|
|
|
getDisplayStrings := func(_ int, _ int) [][]string {
|
|
return presentation.GetBranchListDisplayStrings(
|
|
viewModel.GetItems(),
|
|
c.State().GetItemOperation,
|
|
c.State().GetRepoState().GetScreenMode() != types.SCREEN_NORMAL,
|
|
c.Modes().Diffing.Ref,
|
|
c.Views().Branches.Width(),
|
|
c.Tr,
|
|
c.UserConfig,
|
|
c.Model().Worktrees,
|
|
)
|
|
}
|
|
|
|
self := &BranchesContext{
|
|
FilteredListViewModel: viewModel,
|
|
ListContextTrait: &ListContextTrait{
|
|
Context: NewSimpleContext(NewBaseContext(NewBaseContextOpts{
|
|
View: c.Views().Branches,
|
|
WindowName: "branches",
|
|
Key: LOCAL_BRANCHES_CONTEXT_KEY,
|
|
Kind: types.SIDE_CONTEXT,
|
|
Focusable: true,
|
|
NeedsRerenderOnWidthChange: true,
|
|
})),
|
|
ListRenderer: ListRenderer{
|
|
list: viewModel,
|
|
getDisplayStrings: getDisplayStrings,
|
|
},
|
|
c: c,
|
|
},
|
|
}
|
|
|
|
return self
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
func (self *BranchesContext) ShowBranchHeadsInSubCommits() bool {
|
|
return true
|
|
}
|