2022-08-12 01:19:39 +02:00
|
|
|
package components
|
2022-08-09 12:27:44 +02:00
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"strings"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/jesseduffield/lazygit/pkg/config"
|
|
|
|
"github.com/jesseduffield/lazygit/pkg/gui/types"
|
|
|
|
integrationTypes "github.com/jesseduffield/lazygit/pkg/integration/types"
|
|
|
|
)
|
|
|
|
|
2022-08-11 13:04:10 +02:00
|
|
|
type Input struct {
|
2022-08-13 04:56:04 +02:00
|
|
|
gui integrationTypes.GuiDriver
|
2022-08-09 12:27:44 +02:00
|
|
|
keys config.KeybindingConfig
|
2022-08-09 13:27:12 +02:00
|
|
|
assert *Assert
|
2022-08-09 12:27:44 +02:00
|
|
|
pushKeyDelay int
|
|
|
|
}
|
|
|
|
|
2022-08-13 04:56:04 +02:00
|
|
|
func NewInput(gui integrationTypes.GuiDriver, keys config.KeybindingConfig, assert *Assert, pushKeyDelay int) *Input {
|
2022-08-11 13:04:10 +02:00
|
|
|
return &Input{
|
2022-08-09 12:27:44 +02:00
|
|
|
gui: gui,
|
|
|
|
keys: keys,
|
|
|
|
assert: assert,
|
|
|
|
pushKeyDelay: pushKeyDelay,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-08-09 13:27:12 +02:00
|
|
|
// key is something like 'w' or '<space>'. It's best not to pass a direct value,
|
|
|
|
// but instead to go through the default user config to get a more meaningful key name
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) PressKeys(keyStrs ...string) {
|
2022-08-09 12:27:44 +02:00
|
|
|
for _, keyStr := range keyStrs {
|
|
|
|
self.pressKey(keyStr)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) pressKey(keyStr string) {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.Wait(self.pushKeyDelay)
|
|
|
|
|
|
|
|
self.gui.PressKey(keyStr)
|
|
|
|
}
|
|
|
|
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) SwitchToStatusWindow() {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.pressKey(self.keys.Universal.JumpToBlock[0])
|
|
|
|
}
|
|
|
|
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) SwitchToFilesWindow() {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.pressKey(self.keys.Universal.JumpToBlock[1])
|
|
|
|
}
|
|
|
|
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) SwitchToBranchesWindow() {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.pressKey(self.keys.Universal.JumpToBlock[2])
|
|
|
|
}
|
|
|
|
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) SwitchToCommitsWindow() {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.pressKey(self.keys.Universal.JumpToBlock[3])
|
|
|
|
}
|
|
|
|
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) SwitchToStashWindow() {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.pressKey(self.keys.Universal.JumpToBlock[4])
|
|
|
|
}
|
|
|
|
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) Type(content string) {
|
2022-08-09 12:27:44 +02:00
|
|
|
for _, char := range content {
|
|
|
|
self.pressKey(string(char))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-08-09 13:27:12 +02:00
|
|
|
// i.e. pressing enter
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) Confirm() {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.pressKey(self.keys.Universal.Confirm)
|
|
|
|
}
|
|
|
|
|
2022-08-09 13:27:12 +02:00
|
|
|
// i.e. pressing escape
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) Cancel() {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.pressKey(self.keys.Universal.Return)
|
|
|
|
}
|
|
|
|
|
2022-08-09 13:27:12 +02:00
|
|
|
// i.e. pressing space
|
2022-08-14 12:47:09 +02:00
|
|
|
func (self *Input) PrimaryAction() {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.pressKey(self.keys.Universal.Select)
|
|
|
|
}
|
|
|
|
|
2022-08-09 13:27:12 +02:00
|
|
|
// i.e. pressing down arrow
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) NextItem() {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.pressKey(self.keys.Universal.NextItem)
|
|
|
|
}
|
|
|
|
|
2022-08-09 13:27:12 +02:00
|
|
|
// i.e. pressing up arrow
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) PreviousItem() {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.pressKey(self.keys.Universal.PrevItem)
|
|
|
|
}
|
|
|
|
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) ContinueMerge() {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.PressKeys(self.keys.Universal.CreateRebaseOptionsMenu)
|
2022-08-14 12:13:39 +02:00
|
|
|
self.assert.MatchSelectedLine(Contains("continue"))
|
2022-08-09 12:27:44 +02:00
|
|
|
self.Confirm()
|
|
|
|
}
|
|
|
|
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) ContinueRebase() {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.ContinueMerge()
|
|
|
|
}
|
|
|
|
|
2022-08-09 13:27:12 +02:00
|
|
|
// for when you want to allow lazygit to process something before continuing
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) Wait(milliseconds int) {
|
2022-08-09 12:27:44 +02:00
|
|
|
time.Sleep(time.Duration(milliseconds) * time.Millisecond)
|
|
|
|
}
|
|
|
|
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) LogUI(message string) {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.gui.LogUI(message)
|
|
|
|
}
|
|
|
|
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) Log(message string) {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.gui.LogUI(message)
|
|
|
|
}
|
|
|
|
|
2022-08-09 13:27:12 +02:00
|
|
|
// this will look for a list item in the current panel and if it finds it, it will
|
|
|
|
// enter the keypresses required to navigate to it.
|
|
|
|
// The test will fail if:
|
2022-08-12 01:24:39 +02:00
|
|
|
// - the user is not in a list item
|
|
|
|
// - no list item is found containing the given text
|
|
|
|
// - multiple list items are found containing the given text in the initial page of items
|
2022-08-09 13:27:12 +02:00
|
|
|
//
|
2022-08-09 12:27:44 +02:00
|
|
|
// NOTE: this currently assumes that ViewBufferLines returns all the lines that can be accessed.
|
|
|
|
// If this changes in future, we'll need to update this code to first attempt to find the item
|
|
|
|
// in the current page and failing that, jump to the top of the view and iterate through all of it,
|
|
|
|
// looking for the item.
|
2022-08-11 13:04:10 +02:00
|
|
|
func (self *Input) NavigateToListItemContainingText(text string) {
|
2022-08-09 12:27:44 +02:00
|
|
|
self.assert.InListContext()
|
|
|
|
|
|
|
|
currentContext := self.gui.CurrentContext().(types.IListContext)
|
|
|
|
|
|
|
|
view := currentContext.GetView()
|
|
|
|
|
|
|
|
// first we look for a duplicate on the current screen. We won't bother looking beyond that though.
|
|
|
|
matchCount := 0
|
|
|
|
matchIndex := -1
|
|
|
|
for i, line := range view.ViewBufferLines() {
|
|
|
|
if strings.Contains(line, text) {
|
|
|
|
matchCount++
|
|
|
|
matchIndex = i
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if matchCount > 1 {
|
|
|
|
self.assert.Fail(fmt.Sprintf("Found %d matches for %s, expected only a single match", matchCount, text))
|
|
|
|
}
|
|
|
|
if matchCount == 1 {
|
|
|
|
selectedLineIdx := view.SelectedLineIdx()
|
|
|
|
if selectedLineIdx == matchIndex {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if selectedLineIdx < matchIndex {
|
|
|
|
for i := selectedLineIdx; i < matchIndex; i++ {
|
|
|
|
self.NextItem()
|
|
|
|
}
|
|
|
|
return
|
|
|
|
} else {
|
|
|
|
for i := selectedLineIdx; i > matchIndex; i-- {
|
|
|
|
self.PreviousItem()
|
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
self.assert.Fail(fmt.Sprintf("Could not find item containing text: %s", text))
|
|
|
|
}
|