mirror of
https://github.com/rancher/norman.git
synced 2025-07-01 01:32:03 +00:00
37 lines
701 B
Go
37 lines
701 B
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{
|
|
"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
|
|
}
|