2022-02-05 08:04:10 +02:00
|
|
|
package context
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/jesseduffield/lazygit/pkg/commands/models"
|
2023-03-21 12:38:37 +02:00
|
|
|
"github.com/jesseduffield/lazygit/pkg/gui/presentation"
|
2022-02-05 08:04:10 +02:00
|
|
|
"github.com/jesseduffield/lazygit/pkg/gui/types"
|
|
|
|
)
|
|
|
|
|
|
|
|
type BranchesContext struct {
|
2023-05-27 06:14:43 +02:00
|
|
|
*FilteredListViewModel[*models.Branch]
|
2022-02-05 08:04:10 +02:00
|
|
|
*ListContextTrait
|
|
|
|
}
|
|
|
|
|
2022-12-30 14:24:24 +02:00
|
|
|
var (
|
|
|
|
_ types.IListContext = (*BranchesContext)(nil)
|
|
|
|
_ types.DiffableContext = (*BranchesContext)(nil)
|
|
|
|
)
|
2022-02-05 08:04:10 +02:00
|
|
|
|
2023-03-23 03:35:07 +02:00
|
|
|
func NewBranchesContext(c *ContextCommon) *BranchesContext {
|
2023-05-27 06:14:43 +02:00
|
|
|
viewModel := NewFilteredListViewModel(
|
|
|
|
func() []*models.Branch { return c.Model().Branches },
|
|
|
|
func(branch *models.Branch) []string {
|
|
|
|
return []string{branch.Name}
|
|
|
|
},
|
|
|
|
)
|
2022-02-05 08:04:10 +02:00
|
|
|
|
2023-03-21 12:38:37 +02:00
|
|
|
getDisplayStrings := func(startIdx int, length int) [][]string {
|
|
|
|
return presentation.GetBranchListDisplayStrings(
|
2023-05-27 06:14:43 +02:00
|
|
|
viewModel.GetItems(),
|
2023-03-21 12:38:37 +02:00
|
|
|
c.State().GetRepoState().GetScreenMode() != types.SCREEN_NORMAL,
|
|
|
|
c.Modes().Diffing.Ref,
|
|
|
|
c.Tr,
|
2022-11-07 07:35:36 +02:00
|
|
|
c.UserConfig,
|
2023-07-17 05:40:26 +02:00
|
|
|
c.Model().Worktrees,
|
2023-03-21 12:38:37 +02:00
|
|
|
)
|
|
|
|
}
|
|
|
|
|
2022-12-30 14:24:24 +02:00
|
|
|
self := &BranchesContext{
|
2023-05-27 06:14:43 +02:00
|
|
|
FilteredListViewModel: viewModel,
|
2022-02-05 08:04:10 +02:00
|
|
|
ListContextTrait: &ListContextTrait{
|
|
|
|
Context: NewSimpleContext(NewBaseContext(NewBaseContextOpts{
|
2023-03-21 12:06:39 +02:00
|
|
|
View: c.Views().Branches,
|
2022-02-05 08:04:10 +02:00
|
|
|
WindowName: "branches",
|
|
|
|
Key: LOCAL_BRANCHES_CONTEXT_KEY,
|
|
|
|
Kind: types.SIDE_CONTEXT,
|
|
|
|
Focusable: true,
|
2023-03-21 12:01:58 +02:00
|
|
|
})),
|
2022-02-05 08:04:10 +02:00
|
|
|
list: viewModel,
|
|
|
|
getDisplayStrings: getDisplayStrings,
|
|
|
|
c: c,
|
|
|
|
},
|
|
|
|
}
|
2022-12-30 14:24:24 +02:00
|
|
|
|
|
|
|
return self
|
2022-02-05 08:04:10 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func (self *BranchesContext) GetSelectedItemId() string {
|
|
|
|
item := self.GetSelected()
|
|
|
|
if item == nil {
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
|
|
|
|
return item.ID()
|
|
|
|
}
|
|
|
|
|
2022-03-26 15:18:08 +02:00
|
|
|
func (self *BranchesContext) GetSelectedRef() types.Ref {
|
2022-05-05 14:41:44 +02:00
|
|
|
branch := self.GetSelected()
|
|
|
|
if branch == nil {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
return branch
|
2022-03-26 05:44:30 +02:00
|
|
|
}
|
2022-12-30 14:24:24 +02:00
|
|
|
|
|
|
|
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
|
|
|
|
}
|