KittyPatch/kitty
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

DRYer

Browse Source
This commit is contained in:
Kovid Goyal 2023-03-21 17:06:03 +05:30
parent c2e549b79c
commit e2edacb629
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tools/cmd/diff/highlight.go
View File

@ -82,6 +82,13 @@ const SGR_SUFFIX = "m"
func ansi_formatter(w io.Writer, style *chroma.Style, it chroma.Iterator) error {
style = clear_background(style)
before, after := make([]byte, 0, 64), make([]byte, 0, 64)
write_sgr := func(which []byte) {
if len(which) > 1 {
w.Write(utils.UnsafeStringToBytes(SGR_PREFIX))
w.Write(which[:len(which)-1])
w.Write(utils.UnsafeStringToBytes(SGR_SUFFIX))
}
}
for token := it(); token != chroma.EOF; token = it() {
entry := style.Get(token.Type)
before, after = before[:0], after[:0]
@ -103,17 +110,9 @@ func ansi_formatter(w io.Writer, style *chroma.Style, it chroma.Iterator) error
after = append(after, '3', '9', ';')
}
}
if len(before) > 1 {
w.Write(utils.UnsafeStringToBytes(SGR_PREFIX))
w.Write(before[:len(before)-1])
w.Write(utils.UnsafeStringToBytes(SGR_SUFFIX))
}
write_sgr(before)
w.Write(utils.UnsafeStringToBytes(token.Value))
if len(after) > 1 {
w.Write(utils.UnsafeStringToBytes(SGR_PREFIX))
w.Write(after[:len(after)-1])
w.Write(utils.UnsafeStringToBytes(SGR_SUFFIX))
}
write_sgr(after)
}
return nil
}