65 lines
1.1 KiB
Go
65 lines
1.1 KiB
Go
package tprint
|
|
|
|
import (
|
|
"strings"
|
|
"unicode"
|
|
|
|
"github.com/mattn/go-runewidth"
|
|
)
|
|
|
|
func maxLengthStr(a, b string) string {
|
|
if strLen(a) > strLen(b) {
|
|
return a
|
|
}
|
|
return b
|
|
}
|
|
|
|
func maxLengthSlice[T any](a, b []T) []T {
|
|
if len(a) > len(b) {
|
|
return a
|
|
}
|
|
return b
|
|
}
|
|
|
|
func strLen(str string) int {
|
|
l := 0
|
|
|
|
runes := []rune(str)
|
|
|
|
for i := 0; i < len(runes); i++ {
|
|
rn := runes[i]
|
|
|
|
// skip control sequences
|
|
if unicode.IsControl(rn) {
|
|
for j := i; j < len(runes)-1 && runes[j] != 'm'; j++ {
|
|
i = j
|
|
}
|
|
i++
|
|
continue
|
|
}
|
|
|
|
if rn <= 0xFF || rn >= '─' && rn <= '╿' {
|
|
l++
|
|
} else {
|
|
l += runewidth.RuneWidth(rn)
|
|
}
|
|
}
|
|
|
|
return l
|
|
}
|
|
|
|
func padStringRight(str string, pad rune, length int) string {
|
|
padding := length - strLen(str)
|
|
return str + strings.Repeat(string(pad), padding)
|
|
}
|
|
|
|
func padStringLeft(str string, pad rune, length int) string {
|
|
padding := length - strLen(str)
|
|
return strings.Repeat(string(pad), padding) + str
|
|
}
|
|
|
|
func padStringCenter(str string, pad rune, length int) string {
|
|
l := strLen(str)
|
|
return padStringLeft(padStringRight(str, pad, (length-l)/2+l), pad, length)
|
|
}
|