1 files changed, 142 insertions, 0 deletions
diff --git a/system/admin/export.go b/system/admin/export.go
new file mode 100644
index 0000000..2453eb8
--- /dev/null
+++ b/system/admin/export.go
@@ -0,0 +1,142 @@
+package admin
+
+import (
+	"encoding/csv"
+	"fmt"
+	"io/ioutil"
+	"log"
+	"net/http"
+	"os"
+	"strings"
+	"time"
+
+	"github.com/ponzu-cms/ponzu/management/format"
+	"github.com/ponzu-cms/ponzu/system/db"
+	"github.com/ponzu-cms/ponzu/system/item"
+
+	"github.com/tidwall/gjson"
+)
+
+func exportHandler(res http.ResponseWriter, req *http.Request) {
+	// /admin/contents/export?type=Blogpost&format=csv
+	q := req.URL.Query()
+	t := q.Get("type")
+	f := strings.ToLower(q.Get("format"))
+
+	if t == "" || f == "" {
+		v, err := Error400()
+		if err != nil {
+			res.WriteHeader(http.StatusInternalServerError)
+			return
+		}
+
+		res.WriteHeader(http.StatusBadRequest)
+		_, err = res.Write(v)
+		if err != nil {
+			res.WriteHeader(http.StatusInternalServerError)
+			return
+		}
+
+	}
+
+	pt, ok := item.Types[t]
+	if !ok {
+		res.WriteHeader(http.StatusBadRequest)
+		return
+	}
+
+	switch f {
+	case "csv":
+		csv, ok := pt().(format.CSVFormattable)
+		if !ok {
+			res.WriteHeader(http.StatusBadRequest)
+			return
+		}
+
+		fields := csv.FormatCSV()
+		exportCSV(res, req, pt, fields)
+
+	default:
+		res.WriteHeader(http.StatusBadRequest)
+		return
+	}
+}
+
+func exportCSV(res http.ResponseWriter, req *http.Request, pt func() interface{}, fields []string) {
+	tmpFile, err := ioutil.TempFile(os.TempDir(), "exportcsv-")
+	if err != nil {
+		log.Println("Failed to create tmp file for CSV export:", err)
+		res.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
+	err = os.Chmod(tmpFile.Name(), 0666)
+	if err != nil {
+		log.Println("chmod err:", err)
+	}
+
+	csvBuf := csv.NewWriter(tmpFile)
+
+	t := req.URL.Query().Get("type")
+
+	// get content data and loop through creating a csv row per result
+	bb := db.ContentAll(t)
+
+	// add field names to first row
+	err = csvBuf.Write(fields)
+	if err != nil {
+		res.WriteHeader(http.StatusInternalServerError)
+		log.Println("Failed to write column headers:", fields)
+		return
+	}
+
+	for row := range bb {
+		// unmarshal data and loop over fields
+		rowBuf := []string{}
+
+		for _, col := range fields {
+			// pull out each field as the column value
+			result := gjson.GetBytes(bb[row], col)
+
+			// append it to the buffer
+			rowBuf = append(rowBuf, result.String())
+		}
+
+		// write row to csv
+		err := csvBuf.Write(rowBuf)
+		if err != nil {
+			res.WriteHeader(http.StatusInternalServerError)
+			log.Println("Failed to write column headers:", fields)
+			return
+		}
+	}
+
+	csvBuf.Flush()
+
+	// write the buffer to a content-disposition response
+	fi, err := tmpFile.Stat()
+	if err != nil {
+		log.Println("Failed to read tmp file info for CSV export:", err)
+		res.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
+	err = tmpFile.Close()
+	if err != nil {
+		log.Println("Failed to close tmp file for CSV export:", err)
+	}
+
+	ts := time.Now().Unix()
+	disposition := `attachment; filename="export-%s-%d.csv"`
+
+	res.Header().Set("Content-Type", "text/csv")
+	res.Header().Set("Content-Disposition", fmt.Sprintf(disposition, t, ts))
+	res.Header().Set("Content-Length", fmt.Sprintf("%d", int(fi.Size())))
+
+	http.ServeFile(res, req, tmpFile.Name())
+
+	err = os.Remove(tmpFile.Name())
+	if err != nil {
+		log.Println("Failed to remove tmp file for CSV export:", err)
+	}
+}