Merge pull request #5 from moul/dev/moul/helpers

feat: add template helpers
moul · Sep 8, 2020 · 53c9e24 · 53c9e24
2 parents a80a645 + b6ad7e0
commit 53c9e24
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) }