151 lines
2.7 KiB
Go
151 lines
2.7 KiB
Go
// Copyright 2015 Zack Guo <gizak@icloud.com>. All rights reserved.
|
|
// Use of this source code is governed by a MIT license that can
|
|
// be found in the LICENSE file.
|
|
|
|
package termui
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
|
|
tm "github.com/nsf/termbox-go"
|
|
)
|
|
import rw "github.com/mattn/go-runewidth"
|
|
|
|
/* ---------------Port from termbox-go --------------------- */
|
|
|
|
// Attribute is printable cell's color and style.
|
|
type Attribute uint16
|
|
|
|
const (
|
|
ColorDefault Attribute = iota
|
|
ColorBlack
|
|
ColorRed
|
|
ColorGreen
|
|
ColorYellow
|
|
ColorBlue
|
|
ColorMagenta
|
|
ColorCyan
|
|
ColorWhite
|
|
)
|
|
|
|
const (
|
|
AttrBold Attribute = 1 << (iota + 9)
|
|
AttrUnderline
|
|
AttrReverse
|
|
)
|
|
|
|
var (
|
|
dot = "…"
|
|
dotw = rw.StringWidth(dot)
|
|
)
|
|
|
|
/* ----------------------- End ----------------------------- */
|
|
|
|
func toTmAttr(x Attribute) tm.Attribute {
|
|
return tm.Attribute(x)
|
|
}
|
|
|
|
func str2runes(s string) []rune {
|
|
return []rune(s)
|
|
}
|
|
|
|
// Here for backwards-compatibility.
|
|
func trimStr2Runes(s string, w int) []rune {
|
|
return TrimStr2Runes(s, w)
|
|
}
|
|
|
|
// TrimStr2Runes trims string to w[-1 rune], appends …, and returns the runes
|
|
// of that string if string is grather then n. If string is small then w,
|
|
// return the runes.
|
|
func TrimStr2Runes(s string, w int) []rune {
|
|
if w <= 0 {
|
|
return []rune{}
|
|
}
|
|
|
|
sw := rw.StringWidth(s)
|
|
if sw > w {
|
|
return []rune(rw.Truncate(s, w, dot))
|
|
}
|
|
return str2runes(s)
|
|
}
|
|
|
|
// TrimStrIfAppropriate trim string to "s[:-1] + …"
|
|
// if string > width otherwise return string
|
|
func TrimStrIfAppropriate(s string, w int) string {
|
|
if w <= 0 {
|
|
return ""
|
|
}
|
|
|
|
sw := rw.StringWidth(s)
|
|
if sw > w {
|
|
return rw.Truncate(s, w, dot)
|
|
}
|
|
|
|
return s
|
|
}
|
|
|
|
func strWidth(s string) int {
|
|
return rw.StringWidth(s)
|
|
}
|
|
|
|
func charWidth(ch rune) int {
|
|
return rw.RuneWidth(ch)
|
|
}
|
|
|
|
var whiteSpaceRegex = regexp.MustCompile(`\s`)
|
|
|
|
// StringToAttribute converts text to a termui attribute. You may specifiy more
|
|
// then one attribute like that: "BLACK, BOLD, ...". All whitespaces
|
|
// are ignored.
|
|
func StringToAttribute(text string) Attribute {
|
|
text = whiteSpaceRegex.ReplaceAllString(strings.ToLower(text), "")
|
|
attributes := strings.Split(text, ",")
|
|
result := Attribute(0)
|
|
|
|
for _, theAttribute := range attributes {
|
|
var match Attribute
|
|
switch theAttribute {
|
|
case "reset", "default":
|
|
match = ColorDefault
|
|
|
|
case "black":
|
|
match = ColorBlack
|
|
|
|
case "red":
|
|
match = ColorRed
|
|
|
|
case "green":
|
|
match = ColorGreen
|
|
|
|
case "yellow":
|
|
match = ColorYellow
|
|
|
|
case "blue":
|
|
match = ColorBlue
|
|
|
|
case "magenta":
|
|
match = ColorMagenta
|
|
|
|
case "cyan":
|
|
match = ColorCyan
|
|
|
|
case "white":
|
|
match = ColorWhite
|
|
|
|
case "bold":
|
|
match = AttrBold
|
|
|
|
case "underline":
|
|
match = AttrUnderline
|
|
|
|
case "reverse":
|
|
match = AttrReverse
|
|
}
|
|
|
|
result |= match
|
|
}
|
|
|
|
return result
|
|
}
|