Codebase list golang-github-alecthomas-kong / 9c19b2d3-dc8b-41a0-9c9a-78506f97c374/upstream/sid guesswidth_unix.go
9c19b2d3-dc8b-41a0-9c9a-78506f97c374/upstream/sid

Tree @9c19b2d3-dc8b-41a0-9c9a-78506f97c374/upstream/sid (Download .tar.gz)

guesswidth_unix.go @9c19b2d3-dc8b-41a0-9c9a-78506f97c374/upstream/sidraw · history · blame

//go:build (!appengine && linux) || freebsd || darwin || dragonfly || netbsd || openbsd
// +build !appengine,linux freebsd darwin dragonfly netbsd openbsd

package kong

import (
	"io"
	"os"
	"strconv"
	"syscall"
	"unsafe"
)

func guessWidth(w io.Writer) int {
	// check if COLUMNS env is set to comply with
	// http://pubs.opengroup.org/onlinepubs/009604499/basedefs/xbd_chap08.html
	colsStr := os.Getenv("COLUMNS")
	if colsStr != "" {
		if cols, err := strconv.Atoi(colsStr); err == nil {
			return cols
		}
	}

	if t, ok := w.(*os.File); ok {
		fd := t.Fd()
		var dimensions [4]uint16

		if _, _, err := syscall.Syscall6(
			syscall.SYS_IOCTL,
			uintptr(fd), // nolint: unconvert
			uintptr(syscall.TIOCGWINSZ),
			uintptr(unsafe.Pointer(&dimensions)), // nolint: gas
			0, 0, 0,
		); err == 0 {
			if dimensions[1] == 0 {
				return 80
			}
			return int(dimensions[1])
		}
	}
	return 80
}