39 lines
681 B
Go
39 lines
681 B
Go
package ansi
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"github.com/mattn/go-runewidth"
|
|
)
|
|
|
|
// Buffer is a buffer aware of ANSI escape sequences.
|
|
type Buffer struct {
|
|
bytes.Buffer
|
|
}
|
|
|
|
// PrintableRuneWidth returns the width of all printable runes in the buffer.
|
|
func (w Buffer) PrintableRuneWidth() int {
|
|
return PrintableRuneWidth(w.String())
|
|
}
|
|
|
|
func PrintableRuneWidth(s string) int {
|
|
var n int
|
|
var ansi bool
|
|
|
|
for _, c := range s {
|
|
if c == '\x1B' {
|
|
// ANSI escape sequence
|
|
ansi = true
|
|
} else if ansi {
|
|
if (c >= 0x40 && c <= 0x5a) || (c >= 0x61 && c <= 0x7a) {
|
|
// ANSI sequence terminated
|
|
ansi = false
|
|
}
|
|
} else {
|
|
n += runewidth.StringWidth(string(c))
|
|
}
|
|
}
|
|
|
|
return n
|
|
}
|