adding cases and templates for checkbox, file, richtext, select, tags, textarea

1 files changed, 16 insertions, 9 deletions

diff --git a/cmd/ponzu/generate.go b/cmd/ponzu/generate.go
index 62655f0..3536212 100644
--- a/cmd/ponzu/generate.go
+++ b/cmd/ponzu/generate.go
@@ -154,21 +154,28 @@ func setFieldView(field *generateField, viewType string) error {
 		return template.ParseFiles(filepath.Join(tmplDir, filename))
 	}
 
-	switch strings.ToLower(viewType) {
+	viewType = strings.ToLower(viewType)
+	switch viewType {
+	case "checkbox":
+		tmpl, err = tmplFrom("gen-checkbox.tmpl")
+	case "custom":
+		tmpl, err = tmplFrom("gen-custom.tmpl")
+	case "file":
+		tmpl, err = tmplFrom("gen-file.tmpl")
 	case "hidden":
 		tmpl, err = tmplFrom("gen-hidden.tmpl")
-	case "textarea":
+	case "input", "text":
+		tmpl, err = tmplFrom("gen-input.tmpl")
 	case "richtext":
+		tmpl, err = tmplFrom("gen-richtext.tmpl")
 	case "select":
-	case "input":
-		tmpl, err = tmplFrom("gen-input.tmpl")
-	case "checkbox":
-	case "file":
+		tmpl, err = tmplFrom("gen-select.tmpl")
+	case "textarea":
+		tmpl, err = tmplFrom("gen-textarea.tmpl")
 	case "tags":
-	case "custom":
-		tmpl, err = tmplFrom("gen-custom.tmpl")
+		tmpl, err = tmplFrom("gen-tags.tmpl")
 	default:
-		msg := fmt.Sprintf("'%s' is not a recognized view type. Using 'input' instead.")
+		msg := fmt.Sprintf("'%s' is not a recognized view type. Using 'input' instead.", viewType)
 		fmt.Println(msg)
 		tmpl, err = tmplFrom("gen-input.tmpl")
 	}