Skip to content

Commit

Permalink
Fall back to WithWaitingStatus if item is not visible (#3083)
Browse files Browse the repository at this point in the history
  • Loading branch information
stefanhaller authored Dec 10, 2023
2 parents cf82e69 + f99c59b commit d548f85
Show file tree
Hide file tree
Showing 12 changed files with 74 additions and 24 deletions.
13 changes: 13 additions & 0 deletions pkg/gui/context/list_context_trait.go
Original file line number Diff line number Diff line change
Expand Up @@ -97,3 +97,16 @@ func (self *ListContextTrait) OnSearchSelect(selectedLineIdx int) error {
self.GetList().SetSelectedLineIdx(selectedLineIdx)
return self.HandleFocus(types.OnFocusOpts{})
}

func (self *ListContextTrait) IsItemVisible(item types.HasUrn) bool {
startIdx, length := self.GetViewTrait().ViewPortYBounds()
selectionStart := self.ViewIndexToModelIndex(startIdx)
selectionEnd := self.ViewIndexToModelIndex(startIdx + length)
for i := selectionStart; i < selectionEnd; i++ {
iterItem := self.GetList().GetItem(i)
if iterItem != nil && iterItem.URN() == item.URN() {
return true
}
}
return false
}
10 changes: 9 additions & 1 deletion pkg/gui/context/list_view_model.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,9 @@
package context

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

type ListViewModel[T any] struct {
*traits.ListCursor
Expand Down Expand Up @@ -36,3 +39,8 @@ func (self *ListViewModel[T]) GetItems() []T {
func Zero[T any]() T {
return *new(T)
}

func (self *ListViewModel[T]) GetItem(index int) types.HasUrn {
item := self.getModel()[index]
return any(item).(types.HasUrn)
}
2 changes: 1 addition & 1 deletion pkg/gui/controllers.go
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ func (gui *Gui) resetHelpersAndControllers() {
Confirmation: helpers.NewConfirmationHelper(helperCommon),
Mode: modeHelper,
AppStatus: appStatusHelper,
InlineStatus: helpers.NewInlineStatusHelper(helperCommon),
InlineStatus: helpers.NewInlineStatusHelper(helperCommon, windowHelper),
WindowArrangement: helpers.NewWindowArrangementHelper(
gui.c,
windowHelper,
Expand Down
42 changes: 30 additions & 12 deletions pkg/gui/controllers/helpers/inline_status_helper.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ import (
"time"

"github.com/jesseduffield/gocui"
"github.com/jesseduffield/lazygit/pkg/gui/presentation"
"github.com/jesseduffield/lazygit/pkg/gui/types"
"github.com/jesseduffield/lazygit/pkg/utils"
"github.com/sasha-s/go-deadlock"
Expand All @@ -12,13 +13,15 @@ import (
type InlineStatusHelper struct {
c *HelperCommon

windowHelper *WindowHelper
contextsWithInlineStatus map[types.ContextKey]*inlineStatusInfo
mutex *deadlock.Mutex
}

func NewInlineStatusHelper(c *HelperCommon) *InlineStatusHelper {
func NewInlineStatusHelper(c *HelperCommon, windowHelper *WindowHelper) *InlineStatusHelper {
return &InlineStatusHelper{
c: c,
windowHelper: windowHelper,
contextsWithInlineStatus: make(map[types.ContextKey]*inlineStatusInfo),
mutex: &deadlock.Mutex{},
}
Expand Down Expand Up @@ -61,18 +64,33 @@ func (self inlineStatusHelperTask) Continue() {
}

func (self *InlineStatusHelper) WithInlineStatus(opts InlineStatusOpts, f func(gocui.Task) error) {
self.c.OnWorker(func(task gocui.Task) {
self.start(opts)

err := f(inlineStatusHelperTask{task, self, opts})
if err != nil {
self.c.OnUIThread(func() error {
return self.c.Error(err)
})
}
context := self.c.ContextForKey(opts.ContextKey).(types.IListContext)
view := context.GetView()
visible := view.Visible && self.windowHelper.TopViewInWindow(context.GetWindowName()) == view
if visible && context.IsItemVisible(opts.Item) {
self.c.OnWorker(func(task gocui.Task) {
self.start(opts)

err := f(inlineStatusHelperTask{task, self, opts})
if err != nil {
self.c.OnUIThread(func() error {
return self.c.Error(err)
})
}

self.stop(opts)
})
self.stop(opts)
})
} else {
message := presentation.ItemOperationToString(opts.Operation, self.c.Tr)
_ = self.c.WithWaitingStatus(message, func(t gocui.Task) error {
// We still need to set the item operation, because it might be used
// for other (non-presentation) purposes
self.c.State().SetItemOperation(opts.Item, opts.Operation)
defer self.c.State().ClearItemOperation(opts.Item)

return f(t)
})
}
}

func (self *InlineStatusHelper) start(opts InlineStatusOpts) {
Expand Down
4 changes: 2 additions & 2 deletions pkg/gui/controllers/helpers/window_helper.go
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ func (self *WindowHelper) MoveToTopOfWindow(context types.Context) {

topView := self.TopViewInWindow(window)

if view.Name() != topView.Name() {
if topView != nil && view.Name() != topView.Name() {
if err := self.c.GocuiGui().SetViewOnTopOf(view.Name(), topView.Name()); err != nil {
self.c.Log.Error(err)
}
Expand All @@ -106,7 +106,7 @@ func (self *WindowHelper) TopViewInWindow(windowName string) *gocui.View {
// The views list is ordered highest-last, so we're grabbing the last view of the window
var topView *gocui.View
for _, currentView := range self.c.GocuiGui().Views() {
if lo.Contains(viewNamesInWindow, currentView.Name()) {
if lo.Contains(viewNamesInWindow, currentView.Name()) && currentView.Visible {
topView = currentView
}
}
Expand Down
6 changes: 6 additions & 0 deletions pkg/gui/filetree/commit_file_tree.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ package filetree

import (
"github.com/jesseduffield/lazygit/pkg/commands/models"
"github.com/jesseduffield/lazygit/pkg/gui/types"
"github.com/samber/lo"
"github.com/sirupsen/logrus"
)
Expand Down Expand Up @@ -69,6 +70,11 @@ func (self *CommitFileTree) Len() int {
return self.tree.Size(self.collapsedPaths) - 1 // ignoring root
}

func (self *CommitFileTree) GetItem(index int) types.HasUrn {
// Unimplemented because we don't yet need to show inlines statuses in commit file views
return nil
}

func (self *CommitFileTree) GetAllFiles() []*models.CommitFile {
return self.getFiles()
}
Expand Down
7 changes: 7 additions & 0 deletions pkg/gui/filetree/file_tree.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ import (
"fmt"

"github.com/jesseduffield/lazygit/pkg/commands/models"
"github.com/jesseduffield/lazygit/pkg/gui/types"
"github.com/samber/lo"
"github.com/sirupsen/logrus"
)
Expand All @@ -24,6 +25,7 @@ type ITree[T any] interface {
ToggleShowTree()
GetIndexForPath(path string) (int, bool)
Len() int
GetItem(index int) types.HasUrn
SetTree()
IsCollapsed(path string) bool
ToggleCollapsed(path string)
Expand Down Expand Up @@ -139,6 +141,11 @@ func (self *FileTree) Len() int {
return self.tree.Size(self.collapsedPaths) - 1 // ignoring root
}

func (self *FileTree) GetItem(index int) types.HasUrn {
// Unimplemented because we don't yet need to show inlines statuses in commit file views
return nil
}

func (self *FileTree) GetAllFiles() []*models.File {
return self.getFiles()
}
Expand Down
6 changes: 1 addition & 5 deletions pkg/gui/main_panels.go
Original file line number Diff line number Diff line change
Expand Up @@ -39,12 +39,8 @@ func (gui *Gui) moveMainContextToTop(context types.Context) {
view := context.GetView()

topView := gui.helpers.Window.TopViewInWindow(context.GetWindowName())
if topView == nil {
gui.Log.Error("unexpected: topView is nil")
return
}

if topView != view {
if topView != nil && topView != view {
// We need to copy the content to avoid a flicker effect: If we're flicking
// through files in the files panel, we use a different view to render the
// files vs the directories, and if you select dir A, then file B, then dir
Expand Down
2 changes: 1 addition & 1 deletion pkg/gui/presentation/branches.go
Original file line number Diff line number Diff line change
Expand Up @@ -163,7 +163,7 @@ func ColoredBranchStatus(branch *models.Branch, itemOperation types.ItemOperatio
}

func BranchStatus(branch *models.Branch, itemOperation types.ItemOperation, tr *i18n.TranslationSet, now time.Time) string {
itemOperationStr := itemOperationToString(itemOperation, tr)
itemOperationStr := ItemOperationToString(itemOperation, tr)
if itemOperationStr != "" {
return itemOperationStr + " " + utils.Loader(now)
}
Expand Down
2 changes: 1 addition & 1 deletion pkg/gui/presentation/item_operations.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ import (
"github.com/jesseduffield/lazygit/pkg/i18n"
)

func itemOperationToString(itemOperation types.ItemOperation, tr *i18n.TranslationSet) string {
func ItemOperationToString(itemOperation types.ItemOperation, tr *i18n.TranslationSet) string {
switch itemOperation {
case types.ItemOperationNone:
return ""
Expand Down
2 changes: 1 addition & 1 deletion pkg/gui/presentation/tags.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ func getTagDisplayStrings(t *models.Tag, itemOperation types.ItemOperation, diff
}
descriptionColor := style.FgYellow
descriptionStr := descriptionColor.Sprint(t.Description())
itemOperationStr := itemOperationToString(itemOperation, tr)
itemOperationStr := ItemOperationToString(itemOperation, tr)
if itemOperationStr != "" {
descriptionStr = style.FgCyan.Sprint(itemOperationStr+" "+utils.Loader(time.Now())) + " " + descriptionStr
}
Expand Down
2 changes: 2 additions & 0 deletions pkg/gui/types/context.go
Original file line number Diff line number Diff line change
Expand Up @@ -136,6 +136,7 @@ type IListContext interface {
Context

GetSelectedItemId() string
IsItemVisible(item HasUrn) bool

GetList() IList
ViewIndexToModelIndex(int) int
Expand Down Expand Up @@ -215,6 +216,7 @@ type IController interface {
type IList interface {
IListCursor
Len() int
GetItem(index int) HasUrn
}

type IListCursor interface {
Expand Down

0 comments on commit d548f85

Please sign in to comment.