mirror of
https://github.com/jesseduffield/lazygit.git
synced 2025-02-03 13:21:56 +02:00
7075b66bc6
Very similar to WithWaitingStatus, except that the status is shown in a view next to the affected item, rather than in the status bar. Not used by anything yet; again, committing separately to get smaller commits.
130 lines
3.0 KiB
Go
130 lines
3.0 KiB
Go
package helpers
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/jesseduffield/gocui"
|
|
"github.com/jesseduffield/lazygit/pkg/gui/types"
|
|
"github.com/jesseduffield/lazygit/pkg/utils"
|
|
"github.com/sasha-s/go-deadlock"
|
|
)
|
|
|
|
type InlineStatusHelper struct {
|
|
c *HelperCommon
|
|
|
|
contextsWithInlineStatus map[types.ContextKey]*inlineStatusInfo
|
|
mutex *deadlock.Mutex
|
|
}
|
|
|
|
func NewInlineStatusHelper(c *HelperCommon) *InlineStatusHelper {
|
|
return &InlineStatusHelper{
|
|
c: c,
|
|
contextsWithInlineStatus: make(map[types.ContextKey]*inlineStatusInfo),
|
|
mutex: &deadlock.Mutex{},
|
|
}
|
|
}
|
|
|
|
type InlineStatusOpts struct {
|
|
Item types.HasUrn
|
|
Operation types.ItemOperation
|
|
ContextKey types.ContextKey
|
|
}
|
|
|
|
type inlineStatusInfo struct {
|
|
refCount int
|
|
stop chan struct{}
|
|
}
|
|
|
|
// A custom task for WithInlineStatus calls; it wraps the original one and
|
|
// hides the status whenever the task is paused, and shows it again when
|
|
// continued.
|
|
type inlineStatusHelperTask struct {
|
|
gocui.Task
|
|
|
|
inlineStatusHelper *InlineStatusHelper
|
|
opts InlineStatusOpts
|
|
}
|
|
|
|
// poor man's version of explicitly saying that struct X implements interface Y
|
|
var _ gocui.Task = inlineStatusHelperTask{}
|
|
|
|
func (self inlineStatusHelperTask) Pause() {
|
|
self.inlineStatusHelper.stop(self.opts)
|
|
self.Task.Pause()
|
|
|
|
self.inlineStatusHelper.renderContext(self.opts.ContextKey)
|
|
}
|
|
|
|
func (self inlineStatusHelperTask) Continue() {
|
|
self.Task.Continue()
|
|
self.inlineStatusHelper.start(self.opts)
|
|
}
|
|
|
|
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)
|
|
})
|
|
}
|
|
|
|
self.stop(opts)
|
|
})
|
|
}
|
|
|
|
func (self *InlineStatusHelper) start(opts InlineStatusOpts) {
|
|
self.c.State().SetItemOperation(opts.Item, opts.Operation)
|
|
|
|
self.mutex.Lock()
|
|
defer self.mutex.Unlock()
|
|
|
|
info := self.contextsWithInlineStatus[opts.ContextKey]
|
|
if info == nil {
|
|
info = &inlineStatusInfo{refCount: 0, stop: make(chan struct{})}
|
|
self.contextsWithInlineStatus[opts.ContextKey] = info
|
|
|
|
go utils.Safe(func() {
|
|
ticker := time.NewTicker(time.Millisecond * utils.LoaderAnimationInterval)
|
|
defer ticker.Stop()
|
|
outer:
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
self.renderContext(opts.ContextKey)
|
|
case <-info.stop:
|
|
break outer
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
info.refCount++
|
|
}
|
|
|
|
func (self *InlineStatusHelper) stop(opts InlineStatusOpts) {
|
|
self.mutex.Lock()
|
|
|
|
if info := self.contextsWithInlineStatus[opts.ContextKey]; info != nil {
|
|
info.refCount--
|
|
if info.refCount <= 0 {
|
|
info.stop <- struct{}{}
|
|
delete(self.contextsWithInlineStatus, opts.ContextKey)
|
|
}
|
|
|
|
}
|
|
|
|
self.mutex.Unlock()
|
|
|
|
self.c.State().ClearItemOperation(opts.Item)
|
|
}
|
|
|
|
func (self *InlineStatusHelper) renderContext(contextKey types.ContextKey) {
|
|
self.c.OnUIThread(func() error {
|
|
_ = self.c.ContextForKey(contextKey).HandleRender()
|
|
return nil
|
|
})
|
|
}
|