New class Term

term.go

@@ -0,0 +1,62 @@
+package main
+
+import (
+	"os"
+
+	"golang.org/x/term"
+
+	"github.com/aymanbagabas/go-pty"
+
+	"github.com/hymkor/go-windows1x-virtualterminal"
+)
+
+type Term struct {
+	pty.Pty
+	closers []func()
+}
+
+func (T *Term) Close() {
+	for i := len(T.closers) - 1; i >= 0; i-- {
+		T.closers[i]()
+	}
+	T.closers = nil
+}
+
+func NewTerm() (*Term, error) {
+	T := &Term{}
+
+	disableStdout, err := virtualterminal.EnableStdout()
+	if err != nil {
+		return nil, err
+	}
+	T.closers = append(T.closers, disableStdout)
+
+	disableStdin, err := virtualterminal.EnableStdin()
+	if err != nil {
+		T.Close()
+		return nil, err
+	}
+	T.closers = append(T.closers, disableStdin)
+
+	T.Pty, err = pty.New()
+	if err != nil {
+		T.Close()
+		return nil, err
+	}
+	T.closers = append(T.closers, func() { T.Pty.Close() })
+
+	width, height, err := term.GetSize(int(os.Stdout.Fd()))
+	if err != nil {
+		T.Close()
+		return nil, err
+	}
+	T.Pty.Resize(width, height)
+
+	oldState, err := term.MakeRaw(int(os.Stdin.Fd()))
+	if err != nil {
+		T.Close()
+		return nil, err
+	}
+	T.closers = append(T.closers, func() { term.Restore(int(os.Stdin.Fd()), oldState) })
+	return T, nil
+}