moving stuff around

1 files changed, 32 insertions, 0 deletions

diff --git a/cmd/rr/utils/cprint.go b/cmd/rr/utils/cprint.go
new file mode 100644
index 00000000..f6f828f8
--- /dev/null
+++ b/cmd/rr/utils/cprint.go
@@ -0,0 +1,32 @@
+package utils
+
+import (
+	"fmt"
+	"gopkg.in/AlecAivazis/survey.v1/core"
+	"regexp"
+	"strings"
+)
+
+// Printf works identically to fmt.Print but adds `<white+hb>color formatting support for CLI</reset>`.
+func Printf(format string, args ...interface{}) {
+	fmt.Print(Sprintf(format, args...))
+}
+
+// Sprintf works identically to fmt.Sprintf but adds `<white+hb>color formatting support for CLI</reset>`.
+func Sprintf(format string, args ...interface{}) string {
+	r, err := regexp.Compile(`<([^>]+)>`)
+	if err != nil {
+		panic(err)
+	}
+
+	format = r.ReplaceAllStringFunc(format, func(s string) string {
+		return fmt.Sprintf(`{{color "%s"}}`, strings.Trim(s, "<>/"))
+	})
+
+	out, err := core.RunTemplate(fmt.Sprintf(format, args...), nil)
+	if err != nil {
+		panic(err)
+	}
+
+	return out
+}