term.go

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{}

	isTerm := term.IsTerminal(int(os.Stdout.Fd()))
	if isTerm {
		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() })

	if isTerm {
		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
}