Fix ANSI processor to handle multi-line regions

1 files changed, 5 insertions, 2 deletions

diff --git a/src/core.go b/src/core.go
index e38908a8..7a0f1199 100644
--- a/src/core.go
+++ b/src/core.go
@@ -69,14 +69,17 @@ func Run(opts *Options) {
 	}
 	if opts.Ansi {
 		if opts.Theme != nil {
+			var state *ansiState
 			ansiProcessor = func(data *string) (*string, []ansiOffset) {
-				return extractColor(data)
+				trimmed, offsets, newState := extractColor(data, state)
+				state = newState
+				return trimmed, offsets
 			}
 		} else {
 			// When color is disabled but ansi option is given,
 			// we simply strip out ANSI codes from the input
 			ansiProcessor = func(data *string) (*string, []ansiOffset) {
-				trimmed, _ := extractColor(data)
+				trimmed, _, _ := extractColor(data, nil)
 				return trimmed, nil
 			}
 		}