cmd: moved RegisterStyle to main.go

i582 · Nov 14, 2020 · e0b85ac · e0b85ac
1 parent 2900978
commit e0b85ac
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 15 deletions.
diff --git a/cmd/register.go b/cmd/register.go
diff --git a/main.go b/main.go
@@ -6,8 +6,19 @@ import (
 	"os"
 
 	"github.com/i582/cfmt/internal/parser"
+	"github.com/i582/cfmt/internal/styles"
 )
 
+func RegisterStyle(name string, fn func(string) string) {
+	styles.Map[name] = func(f func(text string) string) func(text string) string {
+		return func(text string) string {
+			t := f(text)
+			t = fn(t)
+			return t
+		}
+	}
+}
+
 func Sprint(a ...interface{}) string {
 	text := fmt.Sprint(a...)
 	p := parser.Parser{}