63 lines
1.1 KiB
Go
63 lines
1.1 KiB
Go
package terminal
|
|
|
|
import (
|
|
"fmt"
|
|
"golang.org/x/term"
|
|
"os"
|
|
)
|
|
|
|
type Screen struct {
|
|
Rows int
|
|
Cols int
|
|
}
|
|
|
|
// Get the size of the terminal in rows and columns
|
|
func Size() *Screen {
|
|
cols := 80
|
|
rows := 24
|
|
|
|
// Try the syscall first
|
|
cols, rows, err := term.GetSize(int(os.Stdin.Fd()))
|
|
if err == nil {
|
|
return &Screen{rows, cols}
|
|
}
|
|
|
|
// Figure out the size the hard way
|
|
if term.IsTerminal(int(os.Stdin.Fd())) {
|
|
rows, cols = sizeTrick()
|
|
}
|
|
|
|
return &Screen{rows, cols}
|
|
}
|
|
|
|
func sizeTrick() (rows int, cols int) {
|
|
// Move cursor to location further than likely screen size
|
|
// The cursor will move to maximum available position
|
|
fmt.Print(Code("999C") + Code("999B"))
|
|
|
|
// Ask the terminal where the cursor is
|
|
fmt.Print(LocateCursor)
|
|
|
|
// Read stdin looking for the reported location
|
|
buffer := ""
|
|
for char, _ := ReadKey(); char != 'R'; char, _ = ReadKey() {
|
|
|
|
if char == '\x1b' || char == '[' {
|
|
continue
|
|
}
|
|
|
|
if char == 'R' || char == '\x00' {
|
|
break
|
|
}
|
|
|
|
buffer += string(char)
|
|
}
|
|
|
|
_, err := fmt.Sscanf(buffer, "%d;%d", &rows, &cols)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return rows, cols
|
|
}
|