updates

page.go

@@ -0,0 +1,168 @@
+package filemanager
+
+import (
+	"bytes"
+	"encoding/base64"
+	"encoding/json"
+	"html/template"
+	"log"
+	"net/http"
+	"strings"
+
+	"github.com/hacdias/filemanager/variables"
+)
+
+// Page contains the informations and functions needed to show the Page
+type Page struct {
+	*PageInfo
+	Minimal bool
+}
+
+// PageInfo contains the information of a Page
+type PageInfo struct {
+	Name    string
+	Path    string
+	IsDir   bool
+	User    *User
+	Config  *FileManager
+	Data    interface{}
+	Editor  bool
+	Display string
+}
+
+// BreadcrumbMapItem ...
+type BreadcrumbMapItem struct {
+	Name string
+	URL  string
+}
+
+// BreadcrumbMap returns p.Path where every element is a map
+// of URLs and path segment names.
+func (i PageInfo) BreadcrumbMap() []BreadcrumbMapItem {
+	result := []BreadcrumbMapItem{}
+
+	if len(i.Path) == 0 {
+		return result
+	}
+
+	// skip trailing slash
+	lpath := i.Path
+	if lpath[len(lpath)-1] == '/' {
+		lpath = lpath[:len(lpath)-1]
+	}
+
+	parts := strings.Split(lpath, "/")
+	for i, part := range parts {
+		if i == len(parts)-1 {
+			continue
+		}
+
+		if i == 0 && part == "" {
+			result = append([]BreadcrumbMapItem{{
+				Name: "/",
+				URL:  "/",
+			}}, result...)
+			continue
+		}
+
+		result = append([]BreadcrumbMapItem{{
+			Name: part,
+			URL:  strings.Join(parts[:i+1], "/") + "/",
+		}}, result...)
+	}
+
+	return result
+}
+
+// PreviousLink returns the path of the previous folder
+func (i PageInfo) PreviousLink() string {
+	path := strings.TrimSuffix(i.Path, "/")
+	path = strings.TrimPrefix(path, "/")
+	path = i.Config.AbsoluteURL() + "/" + path
+	path = path[0 : len(path)-len(i.Name)]
+
+	if len(path) < len(i.Config.AbsoluteURL()+"/") {
+		return ""
+	}
+
+	return path
+}
+
+// Create the functions map, then the template, check for erros and
+// execute the template if there aren't errors
+var functions = template.FuncMap{
+	"Defined": variables.FieldInStruct,
+	"CSS": func(s string) template.CSS {
+		return template.CSS(s)
+	},
+	"Marshal": func(v interface{}) template.JS {
+		a, _ := json.Marshal(v)
+		return template.JS(a)
+	},
+	"EncodeBase64": func(s string) string {
+		return base64.StdEncoding.EncodeToString([]byte(s))
+	},
+}
+
+// PrintAsHTML formats the page in HTML and executes the template
+func (p Page) PrintAsHTML(w http.ResponseWriter, templates ...string) (int, error) {
+
+	if p.Minimal {
+		templates = append(templates, "minimal")
+	} else {
+		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 := p.Config.Assets.Templates.String(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).Funcs(functions).Parse(Page)
+		} else {
+			tpl, err = tpl.Parse(string(Page))
+		}
+
+		if err != nil {
+			log.Print(err)
+			return http.StatusInternalServerError, err
+		}
+	}
+
+	buf := &bytes.Buffer{}
+	err := tpl.Execute(buf, p.PageInfo)
+
+	if err != nil {
+		return http.StatusInternalServerError, err
+	}
+
+	w.Header().Set("Content-Type", "text/html; charset=utf-8")
+	_, err = buf.WriteTo(w)
+	return http.StatusOK, err
+}
+
+// PrintAsJSON prints the current Page information in JSON
+func (p Page) PrintAsJSON(w http.ResponseWriter) (int, error) {
+	marsh, err := json.MarshalIndent(p.PageInfo.Data, "", "    ")
+	if err != nil {
+		return http.StatusInternalServerError, err
+	}
+
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	if _, err := w.Write(marsh); err != nil {
+		return http.StatusInternalServerError, err
+	}
+
+	return http.StatusOK, nil
+}