norman/generator/funcs.go

package generator

import (
	"net/http"
	"strings"
	"text/template"

	"github.com/rancher/norman/types"
	"github.com/rancher/norman/types/convert"
)

func funcs() template.FuncMap {
	return template.FuncMap{
		"toLowerCamelCase": convert.LowerTitle,
		"capitalize":       convert.Capitalize,
		"upper":            strings.ToUpper,
		"toLower":          strings.ToLower,
		"hasGet":           hasGet,
	}
}

func addUnderscore(input string) string {
	return strings.ToLower(underscoreRegexp.ReplaceAllString(input, `${1}_${2}`))
}

func hasGet(schema *types.Schema) bool {
	return contains(schema.CollectionMethods, http.MethodGet)
}

func contains(list []string, needle string) bool {
	for _, i := range list {
		if i == needle {
			return true
		}
	}
	return false
}