You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
131 lines
3.0 KiB
Go
131 lines
3.0 KiB
Go
package nstr
|
|
|
|
import "fmt"
|
|
|
|
// PosFlag type
|
|
type PosFlag uint8
|
|
|
|
// Position for padding/resize string
|
|
const (
|
|
PosLeft PosFlag = iota
|
|
PosRight
|
|
PosMiddle
|
|
)
|
|
|
|
/*************************************************************
|
|
* String padding operation
|
|
*************************************************************/
|
|
|
|
// Padding a string.
|
|
func Padding(s, pad string, length int, pos PosFlag) string {
|
|
diff := len(s) - length
|
|
if diff >= 0 { // do not need padding.
|
|
return s
|
|
}
|
|
|
|
if pad == "" || pad == " " {
|
|
mark := ""
|
|
if pos == PosRight { // to right
|
|
mark = "-"
|
|
}
|
|
|
|
// padding left: "%7s", padding right: "%-7s"
|
|
tpl := fmt.Sprintf("%s%d", mark, length)
|
|
return fmt.Sprintf(`%`+tpl+`s`, s)
|
|
}
|
|
|
|
if pos == PosRight { // to right
|
|
return s + Repeat(pad, -diff)
|
|
}
|
|
return Repeat(pad, -diff) + s
|
|
}
|
|
|
|
// PadLeft a string.
|
|
func PadLeft(s, pad string, length int) string {
|
|
return Padding(s, pad, length, PosLeft)
|
|
}
|
|
|
|
// PadRight a string.
|
|
func PadRight(s, pad string, length int) string {
|
|
return Padding(s, pad, length, PosRight)
|
|
}
|
|
|
|
// PadAround a string. Both left and right
|
|
func PadAround(s, pad string, length int) string {
|
|
return PadRight(PadLeft(s, pad, length-1), pad, length)
|
|
}
|
|
|
|
// Resize a string by given length and align settings. padding space.
|
|
func Resize(s string, length int, align PosFlag) string {
|
|
diff := len(s) - length
|
|
if diff >= 0 { // do not need padding.
|
|
return s
|
|
}
|
|
|
|
if align == PosMiddle {
|
|
strLn := len(s)
|
|
padLn := (length - strLn) / 2
|
|
padStr := string(make([]byte, padLn))
|
|
|
|
if diff := length - padLn*2; diff > 0 {
|
|
s += " "
|
|
}
|
|
return padStr + s + padStr
|
|
}
|
|
|
|
return Padding(s, " ", length, align)
|
|
}
|
|
|
|
// PadChars padding a rune/byte to want length and with position flag
|
|
func PadChars[T byte | rune](cs []T, pad T, length int, pos PosFlag) []T {
|
|
ln := len(cs)
|
|
if ln >= length {
|
|
ns := make([]T, length)
|
|
copy(ns, cs[:length])
|
|
return ns
|
|
}
|
|
|
|
idx := length - ln
|
|
ns := make([]T, length)
|
|
ps := RepeatChars(pad, idx)
|
|
if pos == PosRight {
|
|
copy(ns, cs)
|
|
copy(ns[idx:], ps)
|
|
} else { // to left
|
|
copy(ns[:idx], ps)
|
|
copy(ns[idx:], cs)
|
|
}
|
|
|
|
return ns
|
|
}
|
|
|
|
// PadBytes padding a byte to want length and with position flag
|
|
func PadBytes(bs []byte, pad byte, length int, pos PosFlag) []byte {
|
|
return PadChars(bs, pad, length, pos)
|
|
}
|
|
|
|
// PadBytesLeft a byte to want length
|
|
func PadBytesLeft(bs []byte, pad byte, length int) []byte {
|
|
return PadChars(bs, pad, length, PosLeft)
|
|
}
|
|
|
|
// PadBytesRight a byte to want length
|
|
func PadBytesRight(bs []byte, pad byte, length int) []byte {
|
|
return PadChars(bs, pad, length, PosRight)
|
|
}
|
|
|
|
// PadRunes padding a rune to want length and with position flag
|
|
func PadRunes(rs []rune, pad rune, length int, pos PosFlag) []rune {
|
|
return PadChars(rs, pad, length, pos)
|
|
}
|
|
|
|
// PadRunesLeft a rune to want length
|
|
func PadRunesLeft(rs []rune, pad rune, length int) []rune {
|
|
return PadChars(rs, pad, length, PosLeft)
|
|
}
|
|
|
|
// PadRunesRight a rune to want length
|
|
func PadRunesRight(rs []rune, pad rune, length int) []rune {
|
|
return PadChars(rs, pad, length, PosRight)
|
|
}
|