48 lines
1.4 KiB
Go
48 lines
1.4 KiB
Go
|
package remote
|
||
|
|
|
||
|
|
import (
|
||
|
|
"regexp"
|
||
|
|
|
||
|
|
"github.com/mitchellh/colorstring"
|
||
|
|
)
|
||
|
|
|
||
|
|
// colorsRe is used to find ANSI escaped color codes.
|
||
|
|
var colorsRe = regexp.MustCompile("\033\\[\\d{1,3}m")
|
||
|
|
|
||
|
|
// Colorer is the interface that must be implemented to colorize strings.
|
||
|
|
type Colorer interface {
|
||
|
|
Color(v string) string
|
||
|
|
}
|
||
|
|
|
||
|
|
// Colorize is used to print output when the -no-color flag is used. It will
|
||
|
|
// strip all ANSI escaped color codes which are set while the operation was
|
||
|
|
// executed in Terraform Enterprise.
|
||
|
|
//
|
||
|
|
// When Terraform Enterprise supports run specific variables, this code can be
|
||
|
|
// removed as we can then pass the CLI flag to the backend and prevent the color
|
||
|
|
// codes from being written to the output.
|
||
|
|
type Colorize struct {
|
||
|
|
cliColor *colorstring.Colorize
|
||
|
|
}
|
||
|
|
|
||
|
|
// Color will strip all ANSI escaped color codes and return a uncolored string.
|
||
|
|
func (c *Colorize) Color(v string) string {
|
||
|
|
return colorsRe.ReplaceAllString(c.cliColor.Color(v), "")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Colorize returns the Colorize structure that can be used for colorizing
|
||
|
|
// output. This is guaranteed to always return a non-nil value and so is useful
|
||
|
|
// as a helper to wrap any potentially colored strings.
|
||
|
|
func (b *Remote) Colorize() Colorer {
|
||
|
|
if b.CLIColor != nil && !b.CLIColor.Disable {
|
||
|
|
return b.CLIColor
|
||
|
|
}
|
||
|
|
if b.CLIColor != nil {
|
||
|
|
return &Colorize{cliColor: b.CLIColor}
|
||
|
|
}
|
||
|
|
return &Colorize{cliColor: &colorstring.Colorize{
|
||
|
|
Colors: colorstring.DefaultColors,
|
||
|
|
Disable: true,
|
||
|
|
}}
|
||
|
|
}
|