1
0
mirror of https://github.com/jesseduffield/lazygit.git synced 2025-07-30 03:23:08 +03:00

support recording sessions for testing purposes

This commit is contained in:
Jesse Duffield
2020-10-04 17:41:21 +11:00
parent 37bb89dac3
commit ece93e5eef
4 changed files with 145 additions and 6 deletions

View File

@ -31,6 +31,7 @@ import (
"github.com/jesseduffield/lazygit/pkg/theme"
"github.com/jesseduffield/lazygit/pkg/updates"
"github.com/jesseduffield/lazygit/pkg/utils"
"github.com/jesseduffield/termbox-go"
"github.com/mattn/go-runewidth"
"github.com/sirupsen/logrus"
)
@ -107,6 +108,16 @@ type Gui struct {
showRecentRepos bool
Contexts ContextTree
ViewTabContextMap map[string][]tabContext
// this array either includes the events that we're recording in this session
// or the events we've recorded in a prior session
RecordedEvents []RecordedEvent
StartTime time.Time
}
type RecordedEvent struct {
Timestamp int64
Event *termbox.Event
}
type listPanelState struct {
@ -399,6 +410,7 @@ func NewGui(log *logrus.Entry, gitCommand *commands.GitCommand, oSCommand *oscom
statusManager: &statusManager{},
viewBufferManagerMap: map[string]*tasks.ViewBufferManager{},
showRecentRepos: showRecentRepos,
RecordedEvents: []RecordedEvent{},
}
gui.resetState()
@ -417,12 +429,18 @@ func NewGui(log *logrus.Entry, gitCommand *commands.GitCommand, oSCommand *oscom
func (gui *Gui) Run() error {
gui.resetState()
g, err := gocui.NewGui(gocui.Output256, OverlappingEdges)
recordEvents := recordingEvents()
g, err := gocui.NewGui(gocui.Output256, OverlappingEdges, recordEvents)
if err != nil {
return err
}
defer g.Close()
if recordEvents {
go gui.recordEvents()
}
if gui.State.Modes.Filtering.Active() {
gui.State.ScreenMode = SCREEN_HALF
} else {
@ -475,6 +493,9 @@ func (gui *Gui) Run() error {
// if the error returned from a run is a ErrSubProcess, it runs the subprocess
// otherwise it handles the error, possibly by quitting the application
func (gui *Gui) RunWithSubprocesses() error {
gui.StartTime = time.Now()
go gui.replayRecordedEvents()
for {
gui.stopChan = make(chan struct{})
if err := gui.Run(); err != nil {
@ -497,6 +518,10 @@ func (gui *Gui) RunWithSubprocesses() error {
}
}
if err := gui.saveRecordedEvents(); err != nil {
return err
}
return nil
case gui.Errors.ErrSwitchRepo, gui.Errors.ErrRestart:
continue