lazygit/pkg/gui/diffing.go

package gui

import (
	"fmt"
	"strings"

	"github.com/jesseduffield/gocui"
	"github.com/jesseduffield/lazygit/pkg/commands"
)

func (gui *Gui) inDiffMode() bool {
	return gui.State.Diff.Ref != ""
}

func (gui *Gui) exitDiffMode() error {
	gui.State.Diff = DiffState{}
	return gui.refreshSidePanels(refreshOptions{mode: ASYNC})
}

func (gui *Gui) renderDiff() error {
	gui.getMainView().Title = "Diff"
	gui.State.SplitMainPanel = false
	filterArg := ""
	if gui.inFilterMode() {
		filterArg = fmt.Sprintf(" -- %s", gui.State.FilterPath)
	}

	cmd := gui.OSCommand.ExecutableFromString(
		fmt.Sprintf("git diff --color %s %s", gui.diffStr(), filterArg),
	)
	if err := gui.newPtyTask("main", cmd); err != nil {
		gui.Log.Error(err)
	}
	return nil
}

// currentDiffTerminals returns the current diff terminals of the currently selected item.
// in the case of a branch it returns both the branch and it's upstream name,
// which becomes an option when you bring up the diff menu, but when you're just
// flicking through branches it will be using the local branch name.
func (gui *Gui) currentDiffTerminals() []string {
	currentView := gui.g.CurrentView()
	if currentView == nil {
		return nil
	}
	names := []string{}
	switch currentView.Name() {
	case "files":
	// not supporting files for now
	// file, err := gui.getSelectedFile()
	// if err == nil {
	// 	names = append(names, file.Name)
	// }
	case "commitFiles":
		// not supporting commit files for now
		// file := gui.getSelectedCommitFile()
		// if file != nil {
		// 	names = append(names, file.Name)
		// }
	case "commits":
		var commit *commands.Commit
		switch gui.getCommitsView().Context {
		case "reflog-commits":
			commit = gui.getSelectedReflogCommit()
		case "branch-commits":
			commit = gui.getSelectedCommit()
		}
		if commit != nil {
			names = append(names, commit.Sha)
		}
	case "stash":
		entry := gui.getSelectedStashEntry()
		if entry != nil {
			names = append(names, entry.RefName())
		}
	case "branches":
		switch gui.getBranchesView().Context {
		case "local-branches":
			branch := gui.getSelectedBranch()
			if branch != nil {
				names = append(names, branch.Name)
				if branch.UpstreamName != "" {
					names = append(names, branch.UpstreamName)
				}
			}
		case "remotes":
			remote := gui.getSelectedRemote()
			if remote != nil {
				names = append(names, remote.Name)
			}
		case "remote-branches":
			remoteBranch := gui.getSelectedRemoteBranch()
			if remoteBranch != nil {
				names = append(names, remoteBranch.FullName())
			}
		case "tags":
			tag := gui.getSelectedTag()
			if tag != nil {
				names = append(names, tag.Name)
			}
		}
	}
	return names
}

func (gui *Gui) currentDiffTerminal() string {
	names := gui.currentDiffTerminals()
	if len(names) == 0 {
		return ""
	}
	return names[0]
}

func (gui *Gui) diffStr() string {
	output := gui.State.Diff.Ref

	right := gui.currentDiffTerminal()
	if right != "" {
		output += " " + right
	}
	if gui.State.Diff.Reverse {
		output += " -R"
	}
	return output
}

func (gui *Gui) handleCreateDiffingMenuPanel(g *gocui.Gui, v *gocui.View) error {
	if gui.popupPanelFocused() {
		return nil
	}

	names := gui.currentDiffTerminals()

	menuItems := []*menuItem{}
	for _, name := range names {
		name := name
		menuItems = append(menuItems, []*menuItem{
			{
				displayString: fmt.Sprintf("%s %s", gui.Tr.SLocalize("diff"), name),
				onPress: func() error {
					gui.State.Diff.Ref = name
					// can scope this down based on current view but too lazy right now
					return gui.refreshSidePanels(refreshOptions{mode: ASYNC})
				},
			},
		}...)
	}

	menuItems = append(menuItems, []*menuItem{
		{
			displayString: gui.Tr.SLocalize("enterRefToDiff"),
			onPress: func() error {
				return gui.prompt(v, gui.Tr.SLocalize("enteRefName"), "", func(response string) error {
					gui.State.Diff.Ref = strings.TrimSpace(response)
					return gui.refreshSidePanels(refreshOptions{mode: ASYNC})
				})
			},
		},
	}...)

	if gui.inDiffMode() {
		menuItems = append(menuItems, []*menuItem{
			{
				displayString: gui.Tr.SLocalize("swapDiff"),
				onPress: func() error {
					gui.State.Diff.Reverse = !gui.State.Diff.Reverse
					return gui.refreshSidePanels(refreshOptions{mode: ASYNC})
				},
			},
			{
				displayString: gui.Tr.SLocalize("exitDiffMode"),
				onPress: func() error {
					gui.State.Diff = DiffState{}
					return gui.refreshSidePanels(refreshOptions{mode: ASYNC})
				},
			},
		}...)
	}

	return gui.createMenu(gui.Tr.SLocalize("DiffingMenuTitle"), menuItems, createMenuOptions{showCancel: true})
}