summaryrefslogtreecommitdiffstats
path: root/pkg/gui/presentation/commit_files.go
blob: 59e026f16d8c90642cc0c33d56caf7450f83b86c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
package presentation

import (
	"github.com/fatih/color"
	"github.com/jesseduffield/lazygit/pkg/commands"
	"github.com/jesseduffield/lazygit/pkg/commands/patch"
	"github.com/jesseduffield/lazygit/pkg/theme"
)

func GetCommitFileListDisplayStrings(commitFiles []*commands.CommitFile, diffName string) [][]string {
	lines := make([][]string, len(commitFiles))

	for i := range commitFiles {
		diffed := commitFiles[i].Name == diffName
		lines[i] = getCommitFileDisplayStrings(commitFiles[i], diffed)
	}

	return lines
}

// getCommitFileDisplayStrings returns the display string of branch
func getCommitFileDisplayStrings(f *commands.CommitFile, diffed bool) []string {
	yellow := color.New(color.FgYellow)
	green := color.New(color.FgGreen)
	defaultColor := color.New(theme.DefaultTextColor)
	diffTerminalColor := color.New(theme.DiffTerminalColor)

	var colour *color.Color
	switch f.Status {
	case patch.UNSELECTED:
		colour = defaultColor
	case patch.WHOLE:
		colour = green
	case patch.PART:
		colour = yellow
	}
	if diffed {
		colour = diffTerminalColor
	}
	return []string{colour.Sprint(f.DisplayString)}
}