mirror of
https://github.com/jesseduffield/lazygit.git
synced 2024-12-14 11:23:09 +02:00
132 lines
2.5 KiB
Go
132 lines
2.5 KiB
Go
package gui
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/jesseduffield/generics/slices"
|
|
"github.com/jesseduffield/lazygit/pkg/utils"
|
|
"github.com/sasha-s/go-deadlock"
|
|
)
|
|
|
|
// statusManager's job is to handle rendering of loading states and toast notifications
|
|
// that you see at the bottom left of the screen.
|
|
type statusManager struct {
|
|
statuses []appStatus
|
|
nextId int
|
|
mutex deadlock.Mutex
|
|
}
|
|
|
|
type appStatus struct {
|
|
message string
|
|
statusType string
|
|
id int
|
|
}
|
|
|
|
func (m *statusManager) removeStatus(id int) {
|
|
m.mutex.Lock()
|
|
defer m.mutex.Unlock()
|
|
|
|
m.statuses = slices.Filter(m.statuses, func(status appStatus) bool {
|
|
return status.id != id
|
|
})
|
|
}
|
|
|
|
func (m *statusManager) addWaitingStatus(message string) int {
|
|
m.mutex.Lock()
|
|
defer m.mutex.Unlock()
|
|
|
|
m.nextId += 1
|
|
id := m.nextId
|
|
|
|
newStatus := appStatus{
|
|
message: message,
|
|
statusType: "waiting",
|
|
id: id,
|
|
}
|
|
m.statuses = append([]appStatus{newStatus}, m.statuses...)
|
|
|
|
return id
|
|
}
|
|
|
|
func (m *statusManager) addToastStatus(message string) int {
|
|
m.mutex.Lock()
|
|
defer m.mutex.Unlock()
|
|
|
|
m.nextId++
|
|
id := m.nextId
|
|
|
|
newStatus := appStatus{
|
|
message: message,
|
|
statusType: "toast",
|
|
id: id,
|
|
}
|
|
m.statuses = append([]appStatus{newStatus}, m.statuses...)
|
|
|
|
go func() {
|
|
time.Sleep(time.Second * 2)
|
|
|
|
m.removeStatus(id)
|
|
}()
|
|
|
|
return id
|
|
}
|
|
|
|
func (m *statusManager) getStatusString() string {
|
|
if len(m.statuses) == 0 {
|
|
return ""
|
|
}
|
|
topStatus := m.statuses[0]
|
|
if topStatus.statusType == "waiting" {
|
|
return topStatus.message + " " + utils.Loader()
|
|
}
|
|
return topStatus.message
|
|
}
|
|
|
|
func (m *statusManager) showStatus() bool {
|
|
return len(m.statuses) > 0
|
|
}
|
|
|
|
func (gui *Gui) toast(message string) {
|
|
gui.statusManager.addToastStatus(message)
|
|
|
|
gui.renderAppStatus()
|
|
}
|
|
|
|
func (gui *Gui) renderAppStatus() {
|
|
go utils.Safe(func() {
|
|
ticker := time.NewTicker(time.Millisecond * 50)
|
|
defer ticker.Stop()
|
|
for range ticker.C {
|
|
appStatus := gui.statusManager.getStatusString()
|
|
gui.c.OnUIThread(func() error {
|
|
return gui.renderString(gui.Views.AppStatus, appStatus)
|
|
})
|
|
|
|
if appStatus == "" {
|
|
return
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
// withWaitingStatus wraps a function and shows a waiting status while the function is still executing
|
|
func (gui *Gui) withWaitingStatus(message string, f func() error) error {
|
|
go utils.Safe(func() {
|
|
id := gui.statusManager.addWaitingStatus(message)
|
|
|
|
defer func() {
|
|
gui.statusManager.removeStatus(id)
|
|
}()
|
|
|
|
gui.renderAppStatus()
|
|
|
|
if err := f(); err != nil {
|
|
gui.c.OnUIThread(func() error {
|
|
return gui.c.Error(err)
|
|
})
|
|
}
|
|
})
|
|
|
|
return nil
|
|
}
|