1 files changed, 37 insertions, 0 deletions
diff --git a/lib/parse/ansi.go b/lib/parse/ansi.go
new file mode 100644
index 00000000..265ef1f4
--- /dev/null
+++ b/lib/parse/ansi.go
@@ -0,0 +1,37 @@
+package parse
+
+import (
+	"bufio"
+	"bytes"
+	"fmt"
+	"io"
+	"os"
+	"regexp"
+
+	"git.sr.ht/~rjarry/aerc/logging"
+)
+
+var ansi = regexp.MustCompile("\x1B\\[[0-?]*[ -/]*[@-~]")
+
+// StripAnsi strips ansi escape codes from the reader
+func StripAnsi(r io.Reader) io.Reader {
+	buf := bytes.NewBuffer(nil)
+	scanner := bufio.NewScanner(r)
+	scanner.Buffer(nil, 1024*1024*1024)
+	for scanner.Scan() {
+		line := scanner.Bytes()
+		line = ansi.ReplaceAll(line, []byte(""))
+		_, err := buf.Write(line)
+		if err != nil {
+			logging.Warnf("failed write ", err)
+		}
+		_, err = buf.Write([]byte("\n"))
+		if err != nil {
+			logging.Warnf("failed write ", err)
+		}
+	}
+	if err := scanner.Err(); err != nil {
+		fmt.Fprintf(os.Stderr, "failed to read line: %v\n", err)
+	}
+	return buf
+}