summaryrefslogtreecommitdiffstats
path: root/pkg/commands/patch/patch_line.go
blob: 994d7c4e92c42e921bb426d7d71c616879e98008 (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
package patch

import "github.com/samber/lo"

type PatchLineKind int

const (
	PATCH_HEADER PatchLineKind = iota
	HUNK_HEADER
	ADDITION
	DELETION
	CONTEXT
	NEWLINE_MESSAGE
)

type PatchLine struct {
	Kind    PatchLineKind
	Content string // something like '+ hello' (note the first character is not removed)
}

func (self *PatchLine) isChange() bool {
	return self.Kind == ADDITION || self.Kind == DELETION
}

// Returns the number of lines in the given slice that have one of the given kinds
func nLinesWithKind(lines []*PatchLine, kinds []PatchLineKind) int {
	return lo.CountBy(lines, func(line *PatchLine) bool {
		return lo.Contains(kinds, line.Kind)
	})
}