mirror of
https://gitee.com/dromara/mayfly-go
synced 2026-05-18 17:05:21 +08:00
44 lines
744 B
Go
44 lines
744 B
Go
package ansiterm
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
type ByteStream struct {
|
|
*Stream
|
|
utf8Decoder func(data []byte) (string, error)
|
|
}
|
|
|
|
func (b *ByteStream) Feed(data []byte) {
|
|
var err error
|
|
var dataStr string
|
|
if b.UseUTF8 {
|
|
dataStr, err = b.utf8Decoder(data)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
} else {
|
|
dataStr = BytesToString(data)
|
|
}
|
|
b.Stream.Feed(dataStr)
|
|
|
|
}
|
|
|
|
func (b *ByteStream) selectOtherCharset(code string) {
|
|
if code == "@" {
|
|
b.UseUTF8 = false
|
|
} else if strings.Contains("G8", code) {
|
|
b.UseUTF8 = true
|
|
}
|
|
}
|
|
|
|
func InitByteStream(screen *Screen, strict bool) *ByteStream {
|
|
stream := initializeStream(screen, strict)
|
|
bs := &ByteStream{
|
|
Stream: stream,
|
|
utf8Decoder: DecodeUTF8WithReplacement,
|
|
}
|
|
return bs
|
|
}
|