package colorable

import (

	"bytes"

	"io"

)

// NonColorable holds writer but removes escape sequence.

type NonColorable struct {

	out io.Writer

}

// NewNonColorable returns new instance of Writer which removes escape sequence from Writer.

func NewNonColorable(w io.Writer) io.Writer {

	return &NonColorable{out: w}

}

// Write writes data on console

func (w *NonColorable) Write(data []byte) (n int, err error) {

	er := bytes.NewReader(data)

	var bw [1]byte

loop:

	for {

		c1, err := er.ReadByte()

		if err != nil {

			break loop

		}

		if c1 != 0x1b {

			bw[0] = c1

			w.out.Write(bw[:])

			continue

		}

		c2, err := er.ReadByte()

		if err != nil {

			break loop

		}

		if c2 != 0x5b {

			continue

		}

		var buf bytes.Buffer

		for {

			c, err := er.ReadByte()

			if err != nil {

				break loop

			}

			if ('a' <= c && c <= 'z') || ('A' <= c && c <= 'Z') || c == '@' {

				break

			}

			buf.Write([]byte(string(c)))

		}

	}

	return len(data), nil

}