示例#1
0
文件: tty.go 项目: burke/zeus
func (s *StatusChart) lengthOfOutput() int {
	ts, err := ttyutils.Winsize(os.Stdout)
	if err != nil {
		// This can happen when the output is redirected to a device
		// that blows up on the ioctl Winsize uses. We don't care about fancy drawing in this case.
		return 0
	}
	width := int(ts.Columns)
	if width == 0 { // output has been redirected
		return 0
	}

	lines := strings.Split(s.extraOutput, "\n")

	numLines := 0
	for _, line := range lines {
		n := (len(line) + width - 1) / width
		if n == 0 {
			n = 1
		}
		numLines += n
	}

	return numLines - 1
}
// SetupRawTerminal will set terminal to RAW mode and return function to
// restore normal terminal.
func SetupRawTerminal(setSize func(cols, row uint16) error) func() {
	// Switch terminal to raw mode
	state, err := ttyutils.MakeTerminalRaw(os.Stdout.Fd())
	if err != nil {
		// If we can't do it just continue as we were
		return func() {}
	}

	// Get TTY size
	size, err := ttyutils.Winsize(os.Stdout)
	if err == nil && setSize != nil {
		setSize(size.Columns, size.Lines)
	}

	// Handle SIGWINCH signals (window resize signals from graphical terminals)
	sigWinch := make(chan os.Signal, 1)
	signal.Notify(sigWinch, syscall.SIGWINCH)
	go func() {
		for range sigWinch {
			size, err := ttyutils.Winsize(os.Stdout)
			if err == nil && setSize != nil {
				setSize(size.Columns, size.Lines)
			}
		}
	}()

	// Return a function to cleanup
	return func() {
		// Stop handling SIGWINCH, and close channel to clean up go routine
		signal.Stop(sigWinch)
		close(sigWinch)

		// Restore terminal state
		ttyutils.RestoreTerminalState(os.Stdout.Fd(), state)
		fmt.Println("")
	}
}
示例#3
0
func TestWinSize(t *testing.T) {
	mpty, mtty, err := pty.Open()
	if err != nil {
		t.Fatal(err)
	}
	defer mtty.Close()
	defer mpty.Close()

	size, err := ttyutils.Winsize(mtty)

	if err != nil {
		t.Fatal(err)
	}

	if !reflect.DeepEqual(size, ttyutils.Ttysize{Lines: 0, Columns: 0}) {
		t.Errorf("expected 0x0 terminal, got %#v", size)
	}
}
示例#4
0
func (s *StatusChart) lengthOfOutput() int {
	ts, err := ttyutils.Winsize(os.Stdout)
	if err != nil {
		slog.Error(err)
	}
	width := int(ts.Columns)

	lines := strings.Split(s.extraOutput, "\n")

	numLines := 0
	for _, line := range lines {
		n := (len(line) + width - 1) / width
		if n == 0 {
			n = 1
		}
		numLines += n
	}

	return numLines - 1
}