1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
package cobol
import (
"os"
"strings"
"unicode"
"go.wit.com/log"
"golang.org/x/term"
)
var WIDTH int = 120
var TERMSIZE int = 80
// getTerminalWidth returns the width of the active terminal.
// If the output is not an interactive terminal (e.g., it's being piped to a file
// or another command), it returns a default width and false.
func getTerminalWidth() (int, bool) {
// term.IsTerminal checks if the given file descriptor is connected to a terminal.
// We use os.Stdout.Fd() to check the standard output.
if term.IsTerminal(int(os.Stdout.Fd())) {
var err error
// term.GetSize returns the dimensions of the given terminal.
WIDTH, _, err = term.GetSize(int(os.Stdout.Fd()))
if err != nil {
// If we can't get the size for some reason, fall back to the default.
log.Printf("could not get terminal size: %v", err)
return WIDTH, false
}
return WIDTH, true
}
// If it's not a terminal, return the default width.
return WIDTH, false
}
// like the perl Chomp but with the terminal width
func TerminalChomp(cut string) string {
i, _ := getTerminalWidth()
// log.Info("cobol.TerminalCut() at ", i)
// TrimRightFunc removes all trailing runes r from the string s that satisfy f(r).
// unicode.IsSpace reports whether the rune is a space character.
cut = strings.TrimRightFunc(cut, unicode.IsSpace)
if i >= len(cut) {
return cut
} else {
return log.Sprintf("%s", cut[0:i])
}
}
|
