1package diffview
 2
 3import (
 4	"slices"
 5
 6	"github.com/aymanbagabas/go-udiff"
 7	"github.com/charmbracelet/x/exp/slice"
 8)
 9
10type splitHunk struct {
11	fromLine int
12	toLine   int
13	lines    []*splitLine
14}
15
16type splitLine struct {
17	before *udiff.Line
18	after  *udiff.Line
19}
20
21func hunkToSplit(h *udiff.Hunk) (sh splitHunk) {
22	lines := slices.Clone(h.Lines)
23	sh = splitHunk{
24		fromLine: h.FromLine,
25		toLine:   h.ToLine,
26		lines:    make([]*splitLine, 0, len(lines)),
27	}
28
29	for {
30		var ul udiff.Line
31		var ok bool
32		ul, lines, ok = slice.Shift(lines)
33		if !ok {
34			break
35		}
36
37		var sl splitLine
38
39		switch ul.Kind {
40		// For equal lines, add as is
41		case udiff.Equal:
42			sl.before = &ul
43			sl.after = &ul
44
45		// For inserted lines, set after and keep before as nil
46		case udiff.Insert:
47			sl.before = nil
48			sl.after = &ul
49
50		// For deleted lines, set before and loop over the next lines
51		// searching for the equivalent after line.
52		case udiff.Delete:
53			sl.before = &ul
54
55		inner:
56			for i, l := range lines {
57				switch l.Kind {
58				case udiff.Insert:
59					var ll udiff.Line
60					ll, lines, _ = slice.DeleteAt(lines, i)
61					sl.after = &ll
62					break inner
63				case udiff.Equal:
64					break inner
65				}
66			}
67		}
68
69		sh.lines = append(sh.lines, &sl)
70	}
71
72	return sh
73}