updates

page.go

@@ -1,57 +1,23 @@
 package filemanager
 
 import (
-	"bytes"
+	"encoding/json"
 	"html/template"
 	"log"
+	"net/http"
 	"strings"
 )
 
-type Page struct {
-	Name   string
-	Path   string
-	Config *Config
-	Data   interface{}
-}
-
-func (f FileManager) formatAsHTML(page *Page, templates ...string) (*bytes.Buffer, error) {
-	buf := new(bytes.Buffer)
-
-	templates = append(templates, "base")
-	var tpl *template.Template
-
-	// For each template, add it to the the tpl variable
-	for i, t := range templates {
-		// Get the template from the assets
-		page, err := Asset("templates/" + t + ".tmpl")
-
-		// Check if there is some error. If so, the template doesn't exist
-		if err != nil {
-			log.Print(err)
-			return new(bytes.Buffer), err
-		}
-
-		// If it's the first iteration, creates a new template and add the
-		// functions map
-		if i == 0 {
-			tpl, err = template.New(t).Parse(string(page))
-		} else {
-			tpl, err = tpl.Parse(string(page))
-		}
-
-		if err != nil {
-			log.Print(err)
-			return new(bytes.Buffer), err
-		}
-	}
-
-	err := tpl.Execute(buf, page)
-	return buf, err
+// PageInfo contains the information of a page
+type PageInfo struct {
+	Name string
+	Path string
+	Data interface{}
 }
 
 // BreadcrumbMap returns p.Path where every element is a map
 // of URLs and path segment names.
-func (p Page) BreadcrumbMap() map[string]string {
+func (p PageInfo) BreadcrumbMap() map[string]string {
 	result := map[string]string{}
 
 	if len(p.Path) == 0 {
@@ -76,3 +42,59 @@ func (p Page) BreadcrumbMap() map[string]string {
 
 	return result
 }
+
+// Page contains the informations and functions needed to show the page
+type Page struct {
+	Info *PageInfo
+}
+
+// PrintAsHTML formats the page in HTML and executes the template
+func (p Page) PrintAsHTML(w http.ResponseWriter, templates ...string) (int, error) {
+	templates = append(templates, "base")
+	var tpl *template.Template
+
+	// For each template, add it to the the tpl variable
+	for i, t := range templates {
+		// Get the template from the assets
+		page, err := Asset("templates/" + t + ".tmpl")
+
+		// Check if there is some error. If so, the template doesn't exist
+		if err != nil {
+			log.Print(err)
+			return http.StatusInternalServerError, err
+		}
+
+		// If it's the first iteration, creates a new template and add the
+		// functions map
+		if i == 0 {
+			tpl, err = template.New(t).Parse(string(page))
+		} else {
+			tpl, err = tpl.Parse(string(page))
+		}
+
+		if err != nil {
+			log.Print(err)
+			return http.StatusInternalServerError, err
+		}
+	}
+
+	w.Header().Set("Content-Type", "text/html; charset=utf-8")
+	err := tpl.Execute(w, p.Info)
+
+	if err != nil {
+		return http.StatusInternalServerError, err
+	}
+
+	return http.StatusOK, nil
+}
+
+// PrintAsJSON prints the current page infromation in JSON
+func (p Page) PrintAsJSON(w http.ResponseWriter) (int, error) {
+	marsh, err := json.Marshal(p.Info.Data)
+	if err != nil {
+		return http.StatusInternalServerError, err
+	}
+
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	return w.Write(marsh)
+}