feat: add template helpers

Signed-off-by: Manfred Touron <94029+moul@users.noreply.github.com>
moul · Sep 8, 2020 · b6ad7e0 · b6ad7e0
1 parent a80a645
commit b6ad7e0
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 9 deletions.
diff --git a/go.mod b/go.mod
diff --git a/go.sum b/go.sum
diff --git a/prefix.go b/prefix.go
@@ -2,17 +2,22 @@ package prefix
 
 import (
 	"bytes"
+	"fmt"
 	"text/template"
 
 	"github.com/Masterminds/sprig"
 )
 
-const DefaultFormat = `{{printf "%-3d" .LineNumber}} `
+/// Public API
+
+const DefaultFormat = `{{.LineNumber3}} `
 
 type LinePrefixer interface {
 	PrefixLine(string) string
 }
 
+/// Main implementation
+
 type linePrefixer struct {
 	Format     string
 	LineNumber int
@@ -40,3 +45,9 @@ func (p *linePrefixer) PrefixLine(line string) string {
 	_ = p.t.Execute(&prefix, p)
 	return prefix.String() + line
 }
+
+/// Template helpers
+
+func (p *linePrefixer) LineNumber3() string { return fmt.Sprintf("%-3d", p.LineNumber) }
+func (p *linePrefixer) LineNumber4() string { return fmt.Sprintf("%-4d", p.LineNumber) }
+func (p *linePrefixer) LineNumber5() string { return fmt.Sprintf("%-5d", p.LineNumber) }