active/core/ansi/non_escape_writer.go

// ------------------------------------------------------------------------
//   Project active2
//   Active Thing (activething.com) git.activething.com/go
//
//   File n  non_colorable_writer.go
//   Created by DEV
//   Modified   06/04/2024
//
//   Copyright 2024 activething.com 
// ------------------------------------------------------------------------
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// ------------------------------------------------------------------------

package ansi

import (
	"bytes"
	"io"
)

type (

	NonEscapeWriter struct {
		writer		io.Writer
	}

)


func NewNonEscapeWriter(w io.Writer) *NonEscapeWriter {
	return &NonEscapeWriter{ writer: w } }



func (w *NonEscapeWriter) Write(data []byte) (n int, e error) {
	var pt bytes.Buffer
	var ef bool

	for _, b := range data {
		if b == 0x1b {
			ef = true
			continue
		}

		if ef && b == 0x5b {
			ef = false
			continue
		}

		if ef {
			ef = g.charsEscs[b] == 0
			continue
		}

		pt.WriteByte(b)
	}

	_, e = pt.WriteTo(w.writer)
	if e != nil {
		return n, e
	}

	return len(data), nil
}