58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
|
package fs
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TerminalColorMode describes how ANSI codes should be handled
|
||
|
|
type TerminalColorMode byte
|
||
|
|
|
||
|
|
// TerminalColorMode constants
|
||
|
|
const (
|
||
|
|
TerminalColorModeAuto TerminalColorMode = iota
|
||
|
|
TerminalColorModeNever
|
||
|
|
TerminalColorModeAlways
|
||
|
|
)
|
||
|
|
|
||
|
|
var terminalColorModeToString = []string{
|
||
|
|
TerminalColorModeAuto: "AUTO",
|
||
|
|
TerminalColorModeNever: "NEVER",
|
||
|
|
TerminalColorModeAlways: "ALWAYS",
|
||
|
|
}
|
||
|
|
|
||
|
|
// String converts a TerminalColorMode to a string
|
||
|
|
func (m TerminalColorMode) String() string {
|
||
|
|
if m >= TerminalColorMode(len(terminalColorModeToString)) {
|
||
|
|
return fmt.Sprintf("TerminalColorMode(%d)", m)
|
||
|
|
}
|
||
|
|
return terminalColorModeToString[m]
|
||
|
|
}
|
||
|
|
|
||
|
|
// Set a TerminalColorMode
|
||
|
|
func (m *TerminalColorMode) Set(s string) error {
|
||
|
|
for n, name := range terminalColorModeToString {
|
||
|
|
if s != "" && name == strings.ToUpper(s) {
|
||
|
|
*m = TerminalColorMode(n)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return fmt.Errorf("unknown terminal color mode %q", s)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Type of TerminalColorMode
|
||
|
|
func (m TerminalColorMode) Type() string {
|
||
|
|
return "string"
|
||
|
|
}
|
||
|
|
|
||
|
|
// UnmarshalJSON converts a string/integer in JSON to a TerminalColorMode
|
||
|
|
func (m *TerminalColorMode) UnmarshalJSON(in []byte) error {
|
||
|
|
return UnmarshalJSONFlag(in, m, func(i int64) error {
|
||
|
|
if i < 0 || i >= int64(len(terminalColorModeToString)) {
|
||
|
|
return fmt.Errorf("out of range terminal color mode %d", i)
|
||
|
|
}
|
||
|
|
*m = (TerminalColorMode)(i)
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
}
|