add reference.SelectRepeater and necessary struct and function exports

1 files changed, 101 insertions, 5 deletions

diff --git a/addons/github.com/ponzu-cms/addons/reference/reference.go b/addons/github.com/ponzu-cms/addons/reference/reference.go
index f90964c..9918f36 100644
--- a/addons/github.com/ponzu-cms/addons/reference/reference.go
+++ b/addons/github.com/ponzu-cms/addons/reference/reference.go
@@ -9,6 +9,7 @@ import (
 	"fmt"
 	"html/template"
 	"log"
+	"strings"
 
 	"github.com/ponzu-cms/ponzu/management/editor"
 	"github.com/ponzu-cms/ponzu/system/addon"
@@ -19,7 +20,101 @@ import (
 // The `fieldName` argument will cause a panic if it is not exactly the string
 // form of the struct field that this editor input is representing
 func Select(fieldName string, p interface{}, attrs map[string]string, contentType, tmplString string) []byte {
-	// decode all content type from db into options map
+	options, err := encodeDataToOptions(contentType, tmplString)
+	if err != nil {
+		log.Println("Error encoding data to options for", contentType, err)
+		return nil
+	}
+
+	return editor.Select(fieldName, p, attrs, options)
+}
+
+// SelectRepeater returns the []byte of a <select> HTML element plus internal <options> with a label.
+// It also includes repeat controllers (+ / -) so the element can be
+// dynamically multiplied or reduced.
+// IMPORTANT:
+// The `fieldName` argument will cause a panic if it is not exactly the string
+// form of the struct field that this editor input is representing
+func SelectRepeater(fieldName string, p interface{}, attrs map[string]string, contentType, tmplString string) []byte {
+	scope := editor.TagNameFromStructField(fieldName, p)
+	html := bytes.Buffer{}
+	_, err := html.WriteString(`<span class="__ponzu-repeat ` + scope + `">`)
+	if err != nil {
+		log.Println("Error writing HTML string to SelectRepeater buffer")
+		return nil
+	}
+
+	if _, ok := attrs["class"]; ok {
+		attrs["class"] += " browser-default"
+	} else {
+		attrs["class"] = "browser-default"
+	}
+
+	// find the field values in p to determine if an option is pre-selected
+	fieldVals := editor.ValueFromStructField(fieldName, p)
+	vals := strings.Split(fieldVals, "__ponzu")
+
+	options, err := encodeDataToOptions(contentType, tmplString)
+	if err != nil {
+		log.Println("Error encoding data to options for", contentType, err)
+		return nil
+	}
+
+	for _, val := range vals {
+		sel := editor.NewElement("select", attrs["label"], fieldName, p, attrs)
+		var opts []*editor.Element
+
+		// provide a call to action for the select element
+		cta := &editor.Element{
+			TagName: "option",
+			Attrs:   map[string]string{"disabled": "true", "selected": "true"},
+			Data:    "Select an option...",
+			ViewBuf: &bytes.Buffer{},
+		}
+
+		// provide a selection reset (will store empty string in db)
+		reset := &editor.Element{
+			TagName: "option",
+			Attrs:   map[string]string{"value": ""},
+			Data:    "None",
+			ViewBuf: &bytes.Buffer{},
+		}
+
+		opts = append(opts, cta, reset)
+
+		for k, v := range options {
+			optAttrs := map[string]string{"value": k}
+			if k == val {
+				optAttrs["selected"] = "true"
+			}
+			opt := &editor.Element{
+				TagName: "option",
+				Attrs:   optAttrs,
+				Data:    v,
+				ViewBuf: &bytes.Buffer{},
+			}
+
+			opts = append(opts, opt)
+		}
+
+		_, err := html.Write(editor.DOMElementWithChildrenSelect(sel, opts))
+		if err != nil {
+			log.Println("Error writing DOMElementWithChildrenSelect to SelectRepeater buffer")
+			return nil
+		}
+	}
+
+	_, err = html.WriteString("</span>")
+	if err != nil {
+		log.Println("Error writing HTML string to SelectRepeater buffer")
+		return nil
+	}
+
+	return append(html.Bytes(), editor.RepeatController(fieldName, p, "select", ".input-field")...)
+}
+
+func encodeDataToOptions(contentType, tmplString string) (map[string]string, error) {
+	// encode all content type from db into options map
 	// options in form of map["?type=<contentType>&id=<id>"]t.String()
 	options := make(map[string]string)
 
@@ -28,7 +123,7 @@ func Select(fieldName string, p interface{}, attrs map[string]string, contentTyp
 
 	err := json.Unmarshal(j, &all)
 	if err != nil {
-		return nil
+		return nil, err
 	}
 
 	// make template for option html display
@@ -43,12 +138,13 @@ func Select(fieldName string, p interface{}, attrs map[string]string, contentTyp
 		v := &bytes.Buffer{}
 		err := tmpl.Execute(v, item)
 		if err != nil {
-			log.Println("Error executing template for reference of:", contentType)
-			return nil
+			return nil, fmt.Errorf(
+				"Error executing template for reference of %s: %s",
+				contentType, err.Error())
 		}
 
 		options[k] = v.String()
 	}
 
-	return editor.Select(fieldName, p, attrs, options)
+	return options, nil
 }