mirror of
https://github.com/jesseduffield/lazygit.git
synced 2024-12-12 11:15:00 +02:00
498 lines
14 KiB
Go
498 lines
14 KiB
Go
package gui
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/jesseduffield/gocui"
|
|
"github.com/jesseduffield/lazygit/pkg/gui/context"
|
|
"github.com/jesseduffield/lazygit/pkg/gui/types"
|
|
)
|
|
|
|
func (gui *Gui) popupViewNames() []string {
|
|
result := []string{}
|
|
for _, context := range gui.State.Contexts.Flatten() {
|
|
if context.GetKind() == types.PERSISTENT_POPUP || context.GetKind() == types.TEMPORARY_POPUP {
|
|
result = append(result, context.GetViewName())
|
|
}
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
func (gui *Gui) currentContextKeyIgnoringPopups() types.ContextKey {
|
|
gui.State.ContextManager.RLock()
|
|
defer gui.State.ContextManager.RUnlock()
|
|
|
|
stack := gui.State.ContextManager.ContextStack
|
|
|
|
for i := range stack {
|
|
reversedIndex := len(stack) - 1 - i
|
|
context := stack[reversedIndex]
|
|
kind := stack[reversedIndex].GetKind()
|
|
if kind != types.TEMPORARY_POPUP && kind != types.PERSISTENT_POPUP {
|
|
return context.GetKey()
|
|
}
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
// use replaceContext when you don't want to return to the original context upon
|
|
// hitting escape: you want to go that context's parent instead.
|
|
func (gui *Gui) replaceContext(c types.Context) error {
|
|
if !c.IsFocusable() {
|
|
return nil
|
|
}
|
|
|
|
gui.State.ContextManager.Lock()
|
|
|
|
if len(gui.State.ContextManager.ContextStack) == 0 {
|
|
gui.State.ContextManager.ContextStack = []types.Context{c}
|
|
} else {
|
|
// replace the last item with the given item
|
|
gui.State.ContextManager.ContextStack = append(gui.State.ContextManager.ContextStack[0:len(gui.State.ContextManager.ContextStack)-1], c)
|
|
}
|
|
|
|
defer gui.State.ContextManager.Unlock()
|
|
|
|
return gui.activateContext(c)
|
|
}
|
|
|
|
func (gui *Gui) pushContext(c types.Context, opts ...types.OnFocusOpts) error {
|
|
// using triple dot but you should only ever pass one of these opt structs
|
|
if len(opts) > 1 {
|
|
return errors.New("cannot pass multiple opts to pushContext")
|
|
}
|
|
|
|
if !c.IsFocusable() {
|
|
return nil
|
|
}
|
|
|
|
gui.State.ContextManager.Lock()
|
|
|
|
// push onto stack
|
|
// if we are switching to a side context, remove all other contexts in the stack
|
|
if c.GetKind() == types.SIDE_CONTEXT {
|
|
for _, stackContext := range gui.State.ContextManager.ContextStack {
|
|
if stackContext.GetKey() != c.GetKey() {
|
|
if err := gui.deactivateContext(stackContext); err != nil {
|
|
gui.State.ContextManager.Unlock()
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
gui.State.ContextManager.ContextStack = []types.Context{c}
|
|
} else if len(gui.State.ContextManager.ContextStack) == 0 || gui.currentContextWithoutLock().GetKey() != c.GetKey() {
|
|
// Do not append if the one at the end is the same context (e.g. opening a menu from a menu)
|
|
// In that case we'll just close the menu entirely when the user hits escape.
|
|
|
|
// TODO: think about other exceptional cases
|
|
gui.State.ContextManager.ContextStack = append(gui.State.ContextManager.ContextStack, c)
|
|
}
|
|
|
|
gui.State.ContextManager.Unlock()
|
|
|
|
return gui.activateContext(c, opts...)
|
|
}
|
|
|
|
// asynchronous code idea: functions return an error via a channel, when done
|
|
|
|
// pushContextWithView is to be used when you don't know which context you
|
|
// want to switch to: you only know the view that you want to switch to. It will
|
|
// look up the context currently active for that view and switch to that context
|
|
func (gui *Gui) pushContextWithView(viewName string) error {
|
|
return gui.c.PushContext(gui.State.ViewContextMap.Get(viewName))
|
|
}
|
|
|
|
func (gui *Gui) returnFromContext() error {
|
|
gui.State.ContextManager.Lock()
|
|
|
|
if len(gui.State.ContextManager.ContextStack) == 1 {
|
|
// cannot escape from bottommost context
|
|
gui.State.ContextManager.Unlock()
|
|
return nil
|
|
}
|
|
|
|
n := len(gui.State.ContextManager.ContextStack) - 1
|
|
|
|
currentContext := gui.State.ContextManager.ContextStack[n]
|
|
newContext := gui.State.ContextManager.ContextStack[n-1]
|
|
|
|
gui.State.ContextManager.ContextStack = gui.State.ContextManager.ContextStack[:n]
|
|
|
|
gui.g.SetCurrentContext(string(newContext.GetKey()))
|
|
|
|
gui.State.ContextManager.Unlock()
|
|
|
|
if err := gui.deactivateContext(currentContext); err != nil {
|
|
return err
|
|
}
|
|
|
|
return gui.activateContext(newContext)
|
|
}
|
|
|
|
func (gui *Gui) deactivateContext(c types.Context) error {
|
|
view, _ := gui.g.View(c.GetViewName())
|
|
|
|
if view != nil && view.IsSearching() {
|
|
if err := gui.onSearchEscape(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// if we are the kind of context that is sent to back upon deactivation, we should do that
|
|
if view != nil && (c.GetKind() == types.TEMPORARY_POPUP || c.GetKind() == types.PERSISTENT_POPUP || c.GetKey() == context.COMMIT_FILES_CONTEXT_KEY) {
|
|
view.Visible = false
|
|
}
|
|
|
|
if err := c.HandleFocusLost(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// postRefreshUpdate is to be called on a context after the state that it depends on has been refreshed
|
|
// if the context's view is set to another context we do nothing.
|
|
// if the context's view is the current view we trigger a focus; re-selecting the current item.
|
|
func (gui *Gui) postRefreshUpdate(c types.Context) error {
|
|
if gui.State.ViewContextMap.Get(c.GetViewName()).GetKey() != c.GetKey() {
|
|
return nil
|
|
}
|
|
|
|
if err := c.HandleRender(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if gui.currentViewName() == c.GetViewName() {
|
|
if err := c.HandleFocus(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (gui *Gui) activateContext(c types.Context, opts ...types.OnFocusOpts) error {
|
|
viewName := c.GetViewName()
|
|
v, err := gui.g.View(viewName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
originalViewContext := gui.State.ViewContextMap.Get(viewName)
|
|
var originalViewContextKey types.ContextKey = ""
|
|
if originalViewContext != nil {
|
|
originalViewContextKey = originalViewContext.GetKey()
|
|
}
|
|
|
|
gui.setWindowContext(c)
|
|
gui.setViewTabForContext(c)
|
|
|
|
if viewName == "main" {
|
|
gui.changeMainViewsContext(c)
|
|
} else {
|
|
gui.changeMainViewsContext(gui.State.Contexts.Normal)
|
|
}
|
|
|
|
gui.g.SetCurrentContext(string(c.GetKey()))
|
|
if _, err := gui.g.SetCurrentView(viewName); err != nil {
|
|
return err
|
|
}
|
|
|
|
v.Visible = true
|
|
|
|
// if the new context's view was previously displaying another context, render the new context
|
|
if originalViewContextKey != c.GetKey() {
|
|
if err := c.HandleRender(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
gui.ViewContextMapSet(viewName, c)
|
|
|
|
gui.g.Cursor = v.Editable
|
|
|
|
// render the options available for the current context at the bottom of the screen
|
|
optionsMap := c.GetOptionsMap()
|
|
if optionsMap == nil {
|
|
optionsMap = gui.globalOptionsMap()
|
|
}
|
|
gui.renderOptionsMap(optionsMap)
|
|
|
|
if err := c.HandleFocus(opts...); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (gui *Gui) optionsMapToString(optionsMap map[string]string) string {
|
|
optionsArray := make([]string, 0)
|
|
for key, description := range optionsMap {
|
|
optionsArray = append(optionsArray, key+": "+description)
|
|
}
|
|
sort.Strings(optionsArray)
|
|
return strings.Join(optionsArray, ", ")
|
|
}
|
|
|
|
func (gui *Gui) renderOptionsMap(optionsMap map[string]string) {
|
|
_ = gui.renderString(gui.Views.Options, gui.optionsMapToString(optionsMap))
|
|
}
|
|
|
|
// also setting context on view for now. We'll need to pick one of these two approaches to stick with.
|
|
func (gui *Gui) ViewContextMapSet(viewName string, c types.Context) {
|
|
gui.State.ViewContextMap.Set(viewName, c)
|
|
view, err := gui.g.View(viewName)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
view.Context = string(c.GetKey())
|
|
}
|
|
|
|
// // currently unused
|
|
// func (gui *Gui) renderContextStack() string {
|
|
// result := ""
|
|
// for _, context := range gui.State.ContextManager.ContextStack {
|
|
// result += string(context.GetKey()) + "\n"
|
|
// }
|
|
// return result
|
|
// }
|
|
|
|
func (gui *Gui) currentContext() types.Context {
|
|
gui.State.ContextManager.RLock()
|
|
defer gui.State.ContextManager.RUnlock()
|
|
|
|
return gui.currentContextWithoutLock()
|
|
}
|
|
|
|
func (gui *Gui) currentContextWithoutLock() types.Context {
|
|
if len(gui.State.ContextManager.ContextStack) == 0 {
|
|
return gui.defaultSideContext()
|
|
}
|
|
|
|
return gui.State.ContextManager.ContextStack[len(gui.State.ContextManager.ContextStack)-1]
|
|
}
|
|
|
|
// the status panel is not yet a list context (and may never be), so this method is not
|
|
// quite the same as currentSideContext()
|
|
func (gui *Gui) currentSideListContext() types.IListContext {
|
|
context := gui.currentSideContext()
|
|
listContext, ok := context.(types.IListContext)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
return listContext
|
|
}
|
|
|
|
func (gui *Gui) currentSideContext() types.Context {
|
|
gui.State.ContextManager.RLock()
|
|
defer gui.State.ContextManager.RUnlock()
|
|
|
|
stack := gui.State.ContextManager.ContextStack
|
|
|
|
// on startup the stack can be empty so we'll return an empty string in that case
|
|
if len(stack) == 0 {
|
|
return gui.defaultSideContext()
|
|
}
|
|
|
|
// find the first context in the stack with the type of types.SIDE_CONTEXT
|
|
for i := range stack {
|
|
context := stack[len(stack)-1-i]
|
|
|
|
if context.GetKind() == types.SIDE_CONTEXT {
|
|
return context
|
|
}
|
|
}
|
|
|
|
return gui.defaultSideContext()
|
|
}
|
|
|
|
// static as opposed to popup
|
|
func (gui *Gui) currentStaticContext() types.Context {
|
|
gui.State.ContextManager.RLock()
|
|
defer gui.State.ContextManager.RUnlock()
|
|
|
|
stack := gui.State.ContextManager.ContextStack
|
|
|
|
if len(stack) == 0 {
|
|
return gui.defaultSideContext()
|
|
}
|
|
|
|
// find the first context in the stack without a popup type
|
|
for i := range stack {
|
|
context := stack[len(stack)-1-i]
|
|
|
|
if context.GetKind() != types.TEMPORARY_POPUP && context.GetKind() != types.PERSISTENT_POPUP {
|
|
return context
|
|
}
|
|
}
|
|
|
|
return gui.defaultSideContext()
|
|
}
|
|
|
|
func (gui *Gui) defaultSideContext() types.Context {
|
|
if gui.State.Modes.Filtering.Active() {
|
|
return gui.State.Contexts.LocalCommits
|
|
} else {
|
|
return gui.State.Contexts.Files
|
|
}
|
|
}
|
|
|
|
// getFocusLayout returns a manager function for when view gain and lose focus
|
|
func (gui *Gui) getFocusLayout() func(g *gocui.Gui) error {
|
|
var previousView *gocui.View
|
|
return func(g *gocui.Gui) error {
|
|
newView := gui.g.CurrentView()
|
|
if err := gui.onViewFocusChange(); err != nil {
|
|
return err
|
|
}
|
|
// for now we don't consider losing focus to a popup panel as actually losing focus
|
|
if newView != previousView && !gui.isPopupPanel(newView.Name()) {
|
|
if err := gui.onViewFocusLost(previousView, newView); err != nil {
|
|
return err
|
|
}
|
|
|
|
previousView = newView
|
|
}
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func (gui *Gui) onViewFocusChange() error {
|
|
gui.g.Mutexes.ViewsMutex.Lock()
|
|
defer gui.g.Mutexes.ViewsMutex.Unlock()
|
|
|
|
currentView := gui.g.CurrentView()
|
|
for _, view := range gui.g.Views() {
|
|
view.Highlight = view.Name() != "main" && view.Name() != "extras" && view == currentView
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (gui *Gui) onViewFocusLost(oldView *gocui.View, newView *gocui.View) error {
|
|
if oldView == nil {
|
|
return nil
|
|
}
|
|
|
|
_ = oldView.SetOriginX(0)
|
|
|
|
if oldView == gui.Views.CommitFiles && newView != gui.Views.Main && newView != gui.Views.Secondary && newView != gui.Views.Search {
|
|
gui.resetWindowContext(gui.State.Contexts.CommitFiles)
|
|
if err := gui.deactivateContext(gui.State.Contexts.CommitFiles); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// changeContext is a helper function for when we want to change a 'main' context
|
|
// which currently just means a context that affects both the main and secondary views
|
|
// other views can have their context changed directly but this function helps
|
|
// keep the main and secondary views in sync
|
|
func (gui *Gui) changeMainViewsContext(c types.Context) {
|
|
if gui.State.MainContext == c.GetKey() {
|
|
return
|
|
}
|
|
|
|
switch c.GetKey() {
|
|
case context.MAIN_NORMAL_CONTEXT_KEY, context.MAIN_PATCH_BUILDING_CONTEXT_KEY, context.MAIN_STAGING_CONTEXT_KEY, context.MAIN_MERGING_CONTEXT_KEY:
|
|
gui.ViewContextMapSet(gui.Views.Main.Name(), c)
|
|
gui.ViewContextMapSet(gui.Views.Secondary.Name(), c)
|
|
default:
|
|
panic(fmt.Sprintf("unknown context for main: %s", c.GetKey()))
|
|
}
|
|
|
|
gui.State.MainContext = c.GetKey()
|
|
}
|
|
|
|
func (gui *Gui) viewTabNames(viewName string) []string {
|
|
tabContexts := gui.State.ViewTabContextMap[viewName]
|
|
|
|
if len(tabContexts) == 0 {
|
|
return nil
|
|
}
|
|
|
|
result := make([]string, len(tabContexts))
|
|
for i, tabContext := range tabContexts {
|
|
result[i] = tabContext.Tab
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
func (gui *Gui) setViewTabForContext(c types.Context) {
|
|
// search for the context in our map and if we find it, set the tab for the corresponding view
|
|
tabContexts, ok := gui.State.ViewTabContextMap[c.GetViewName()]
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
for tabIndex, tabContext := range tabContexts {
|
|
for _, context := range tabContext.Contexts {
|
|
if context.GetKey() == c.GetKey() {
|
|
// get the view, set the tab
|
|
v, err := gui.g.View(c.GetViewName())
|
|
if err != nil {
|
|
gui.c.Log.Error(err)
|
|
return
|
|
}
|
|
v.TabIndex = tabIndex
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (gui *Gui) contextForContextKey(contextKey types.ContextKey) (types.Context, bool) {
|
|
for _, context := range gui.State.Contexts.Flatten() {
|
|
if context.GetKey() == contextKey {
|
|
return context, true
|
|
}
|
|
}
|
|
|
|
return nil, false
|
|
}
|
|
|
|
func (gui *Gui) rerenderView(view *gocui.View) error {
|
|
return gui.State.ViewContextMap.Get(view.Name()).HandleRender()
|
|
}
|
|
|
|
func (gui *Gui) getSideContextSelectedItemId() string {
|
|
currentSideContext := gui.currentSideListContext()
|
|
if currentSideContext == nil {
|
|
return ""
|
|
}
|
|
|
|
return currentSideContext.GetSelectedItemId()
|
|
}
|
|
|
|
func (gui *Gui) isContextVisible(c types.Context) bool {
|
|
return gui.State.WindowViewNameMap[c.GetWindowName()] == c.GetViewName() && gui.State.ViewContextMap.Get(c.GetViewName()).GetKey() == c.GetKey()
|
|
}
|
|
|
|
// currently unused
|
|
// func (gui *Gui) getCurrentSideView() *gocui.View {
|
|
// currentSideContext := gui.currentSideContext()
|
|
// if currentSideContext == nil {
|
|
// return nil
|
|
// }
|
|
|
|
// view, _ := gui.g.View(currentSideContext.GetViewName())
|
|
|
|
// return view
|
|
// }
|
|
|
|
// currently unused
|
|
// func (gui *Gui) renderContextStack() string {
|
|
// result := ""
|
|
// for _, context := range gui.State.ContextManager.ContextStack {
|
|
// result += context.GetViewName() + "\n"
|
|
// }
|
|
// return result
|
|
// }
|