Preserve the original color of each token when using --with-nth with --ansi

Close #1500

1 files changed, 9 insertions, 0 deletions

diff --git a/src/util/chars.go b/src/util/chars.go
index ec6fca0e..35b28297 100644
--- a/src/util/chars.go
+++ b/src/util/chars.go
@@ -171,3 +171,12 @@ func (chars *Chars) CopyRunes(dest []rune) {
 	}
 	return
 }
+
+func (chars *Chars) Wrap(prefix string, suffix string) {
+	if runes := chars.optionalRunes(); runes != nil {
+		runes = append(append([]rune(prefix), runes...), []rune(suffix)...)
+		chars.slice = *(*[]byte)(unsafe.Pointer(&runes))
+	} else {
+		chars.slice = append(append([]byte(prefix), chars.slice...), []byte(suffix)...)
+	}
+}