Reorganized directories and packages.

pkg/funcmap/funcmap.go

@@ -0,0 +1,66 @@
+package funcmap
+
+import (
+	"fmt"
+	"html/template"
+	"reflect"
+	"strings"
+
+	"github.com/mikestefanello/pagoda/config"
+
+	"github.com/Masterminds/sprig"
+	"github.com/labstack/gommon/random"
+)
+
+var (
+	// CacheBuster stores a random string used as a cache buster for static files.
+	CacheBuster = random.String(10)
+)
+
+// GetFuncMap provides a template function map
+func GetFuncMap() template.FuncMap {
+	// See http://masterminds.github.io/sprig/ for available funcs
+	funcMap := sprig.FuncMap()
+
+	// Provide a list of custom functions
+	// Expand this as you add more functions to this package
+	// Avoid using a name already in use by sprig
+	f := template.FuncMap{
+		"hasField": HasField,
+		"file":     File,
+		"link":     Link,
+	}
+
+	for k, v := range f {
+		funcMap[k] = v
+	}
+
+	return funcMap
+}
+
+// HasField checks if an interface contains a given field
+func HasField(v interface{}, name string) bool {
+	rv := reflect.ValueOf(v)
+	if rv.Kind() == reflect.Ptr {
+		rv = rv.Elem()
+	}
+	if rv.Kind() != reflect.Struct {
+		return false
+	}
+	return rv.FieldByName(name).IsValid()
+}
+
+// File appends a cache buster to a given filepath so it can remain cached until the app is restarted
+func File(filepath string) string {
+	return fmt.Sprintf("/%s/%s?v=%s", config.StaticPrefix, filepath, CacheBuster)
+}
+
+// Link outputs HTML for a link element, providing the ability to dynamically set the active class
+func Link(url, text, currentPath string, classes ...string) template.HTML {
+	if currentPath == url {
+		classes = append(classes, "is-active")
+	}
+
+	html := fmt.Sprintf(`<a class="%s" href="%s">%s</a>`, strings.Join(classes, " "), url, text)
+	return template.HTML(html)
+}

pkg/funcmap/funcmap_test.go

@@ -0,0 +1,39 @@
+package funcmap
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/mikestefanello/pagoda/config"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestHasField(t *testing.T) {
+	type example struct {
+		name string
+	}
+	var e example
+	assert.True(t, HasField(e, "name"))
+	assert.False(t, HasField(e, "abcd"))
+}
+
+func TestLink(t *testing.T) {
+	link := string(Link("/abc", "Text", "/abc"))
+	expected := `<a class="is-active" href="/abc">Text</a>`
+	assert.Equal(t, expected, link)
+
+	link = string(Link("/abc", "Text", "/abc", "first", "second"))
+	expected = `<a class="first second is-active" href="/abc">Text</a>`
+	assert.Equal(t, expected, link)
+
+	link = string(Link("/abc", "Text", "/def"))
+	expected = `<a class="" href="/abc">Text</a>`
+	assert.Equal(t, expected, link)
+}
+
+func TestGetFuncMap(t *testing.T) {
+	file := File("test.png")
+	expected := fmt.Sprintf("/%s/test.png?v=%s", config.StaticPrefix, CacheBuster)
+	assert.Equal(t, expected, file)
+}