139 lines
2.5 KiB
Go
139 lines
2.5 KiB
Go
package util
|
|
|
|
import (
|
|
"math"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/mattn/go-isatty"
|
|
"github.com/mattn/go-runewidth"
|
|
"github.com/rivo/uniseg"
|
|
)
|
|
|
|
// RunesWidth returns runes width
|
|
func RunesWidth(runes []rune, prefixWidth int, tabstop int, limit int) (int, int) {
|
|
width := 0
|
|
gr := uniseg.NewGraphemes(string(runes))
|
|
idx := 0
|
|
for gr.Next() {
|
|
rs := gr.Runes()
|
|
var w int
|
|
if len(rs) == 1 && rs[0] == '\t' {
|
|
w = tabstop - (prefixWidth+width)%tabstop
|
|
} else {
|
|
s := string(rs)
|
|
w = runewidth.StringWidth(s) + strings.Count(s, "\n")
|
|
}
|
|
width += w
|
|
if limit > 0 && width > limit {
|
|
return width, idx
|
|
}
|
|
idx += len(rs)
|
|
}
|
|
return width, -1
|
|
}
|
|
|
|
// Max returns the largest integer
|
|
func Max(first int, second int) int {
|
|
if first >= second {
|
|
return first
|
|
}
|
|
return second
|
|
}
|
|
|
|
// Max16 returns the largest integer
|
|
func Max16(first int16, second int16) int16 {
|
|
if first >= second {
|
|
return first
|
|
}
|
|
return second
|
|
}
|
|
|
|
// Max32 returns the largest 32-bit integer
|
|
func Max32(first int32, second int32) int32 {
|
|
if first > second {
|
|
return first
|
|
}
|
|
return second
|
|
}
|
|
|
|
// Min returns the smallest integer
|
|
func Min(first int, second int) int {
|
|
if first <= second {
|
|
return first
|
|
}
|
|
return second
|
|
}
|
|
|
|
// Min32 returns the smallest 32-bit integer
|
|
func Min32(first int32, second int32) int32 {
|
|
if first <= second {
|
|
return first
|
|
}
|
|
return second
|
|
}
|
|
|
|
// Constrain32 limits the given 32-bit integer with the upper and lower bounds
|
|
func Constrain32(val int32, min int32, max int32) int32 {
|
|
if val < min {
|
|
return min
|
|
}
|
|
if val > max {
|
|
return max
|
|
}
|
|
return val
|
|
}
|
|
|
|
// Constrain limits the given integer with the upper and lower bounds
|
|
func Constrain(val int, min int, max int) int {
|
|
if val < min {
|
|
return min
|
|
}
|
|
if val > max {
|
|
return max
|
|
}
|
|
return val
|
|
}
|
|
|
|
func AsUint16(val int) uint16 {
|
|
if val > math.MaxUint16 {
|
|
return math.MaxUint16
|
|
} else if val < 0 {
|
|
return 0
|
|
}
|
|
return uint16(val)
|
|
}
|
|
|
|
// DurWithin limits the given time.Duration with the upper and lower bounds
|
|
func DurWithin(
|
|
val time.Duration, min time.Duration, max time.Duration) time.Duration {
|
|
if val < min {
|
|
return min
|
|
}
|
|
if val > max {
|
|
return max
|
|
}
|
|
return val
|
|
}
|
|
|
|
// IsTty returns true if stdin is a terminal
|
|
func IsTty() bool {
|
|
return isatty.IsTerminal(os.Stdin.Fd())
|
|
}
|
|
|
|
// ToTty returns true if stdout is a terminal
|
|
func ToTty() bool {
|
|
return isatty.IsTerminal(os.Stdout.Fd())
|
|
}
|
|
|
|
// Once returns a function that returns the specified boolean value only once
|
|
func Once(nextResponse bool) func() bool {
|
|
state := nextResponse
|
|
return func() bool {
|
|
prevState := state
|
|
state = false
|
|
return prevState
|
|
}
|
|
}
|