159 lines
2.7 KiB
Go
159 lines
2.7 KiB
Go
package tx
|
|
|
|
import "golang.org/x/term"
|
|
import "surdeus.su/core/cli/aes"
|
|
import "bufio"
|
|
import "fmt"
|
|
import "os"
|
|
import "os/signal"
|
|
import "syscall"
|
|
|
|
type Signal = os.Signal
|
|
const (
|
|
SignalWinChange = syscall.SIGWINCH
|
|
)
|
|
|
|
type Terminal struct {
|
|
x *term.Terminal
|
|
file *os.File
|
|
input *bufio.Reader
|
|
fd int
|
|
prevState *term.State
|
|
|
|
signals chan Signal
|
|
keys chan Key
|
|
}
|
|
|
|
func NewTerminal(input *os.File) (*Terminal, error) {
|
|
if input == nil {
|
|
return nil, ErrNoInputProvided
|
|
}
|
|
|
|
fd := int(input.Fd())
|
|
if !term.IsTerminal(fd) {
|
|
return nil, ErrNotTerminal
|
|
}
|
|
|
|
ret := Terminal{}
|
|
ret.file = input
|
|
ret.input = bufio.NewReader(input)
|
|
ret.fd = fd
|
|
ret.x = term.NewTerminal(input, "")
|
|
|
|
return &ret, nil
|
|
}
|
|
|
|
func (t *Terminal) MakeRaw() (chan Key, chan Signal, error) {
|
|
state, err := term.MakeRaw(t.fd)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
t.prevState = state
|
|
t.signals = make(chan Signal, 1)
|
|
signal.Notify(
|
|
t.signals,
|
|
syscall.SIGWINCH,
|
|
)
|
|
t.keys = make(chan Key, 1)
|
|
go func() {
|
|
for {
|
|
r, _, _ := t.input.ReadRune()
|
|
t.keys <- Key(r)
|
|
}
|
|
} ()
|
|
|
|
return t.keys, t.signals, nil
|
|
}
|
|
|
|
func (t *Terminal) Restore() error {
|
|
t.DisableAltBuffer()
|
|
if t == nil || t.prevState == nil {
|
|
return nil
|
|
}
|
|
|
|
err := term.Restore(t.fd, t.prevState)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (t *Terminal) Size() (int, int, error) {
|
|
return term.GetSize(t.fd)
|
|
}
|
|
|
|
func (t *Terminal) ReadLine() (string, error) {
|
|
return t.x.ReadLine()
|
|
}
|
|
|
|
/*
|
|
func (t *Terminal) ReadKey() (Key, int, error) {
|
|
r, size, err := t.input.ReadRune()
|
|
return Key(r), size, err
|
|
}
|
|
*/
|
|
|
|
func (t *Terminal) Write(bts []byte) (int, error) {
|
|
return t.x.Write(bts)
|
|
}
|
|
|
|
func (t *Terminal) Printf(
|
|
format string, v ...any,
|
|
) (int, error) {
|
|
str := fmt.Sprintf(format, v...)
|
|
return t.Write([]byte(str))
|
|
}
|
|
|
|
func (t *Terminal) Print(v ...any) (int, error) {
|
|
str := fmt.Sprint(v...)
|
|
return t.Write([]byte(str))
|
|
}
|
|
|
|
func (t *Terminal) MoveCursorRight(n int) {
|
|
t.Printf(aes.MoveCursorRightFormat, n)
|
|
}
|
|
|
|
func (t *Terminal) MoveCursorLeft(n int) {
|
|
t.Printf(aes.MoveCursorLeftFormat, n)
|
|
}
|
|
|
|
func (t *Terminal) MoveCursorUp(n int) {
|
|
t.Printf(aes.MoveCursorUpFormat, n)
|
|
}
|
|
|
|
func (t *Terminal) MoveCursorDown(n int) {
|
|
t.Printf(aes.MoveCursorDownFormat, n)
|
|
}
|
|
|
|
func (t *Terminal) MoveCursorHome() {
|
|
t.Printf("%s", aes.MoveCursorHome)
|
|
}
|
|
|
|
func (t *Terminal) SaveCursor() {
|
|
t.Printf("%s", aes.SaveCursorPositionDEC)
|
|
}
|
|
|
|
func (t *Terminal) UndoCursor() {
|
|
t.Printf("%s", aes.UndoCursorPositionDEC)
|
|
}
|
|
|
|
func (t *Terminal) NextLine(n int) {
|
|
t.Printf(
|
|
aes.MoveCursorToBeginOfNextLineDownFormat,
|
|
n,
|
|
)
|
|
}
|
|
|
|
func (t *Terminal) EnableAltBuffer() {
|
|
t.Printf("%s", aes.EnableAltBuffer)
|
|
}
|
|
|
|
func (t *Terminal) DisableAltBuffer() {
|
|
t.Printf("%s", aes.DisableAltBuffer)
|
|
}
|
|
|
|
func (t *Terminal) ClearScreen() {
|
|
t.Write([]byte(aes.EraseEntireScreen))
|
|
}
|