131 lines
3.0 KiB
Go
131 lines
3.0 KiB
Go
|
package page
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"encoding/json"
|
||
|
"html/template"
|
||
|
"log"
|
||
|
"net/http"
|
||
|
"strings"
|
||
|
|
||
|
"github.com/hacdias/caddy-filemanager/assets"
|
||
|
"github.com/hacdias/caddy-filemanager/config"
|
||
|
"github.com/hacdias/caddy-filemanager/utils/variables"
|
||
|
)
|
||
|
|
||
|
// Page contains the informations and functions needed to show the page
|
||
|
type Page struct {
|
||
|
*Info
|
||
|
}
|
||
|
|
||
|
// Info contains the information of a page
|
||
|
type Info struct {
|
||
|
Name string
|
||
|
Path string
|
||
|
IsDir bool
|
||
|
Config *config.Config
|
||
|
Data interface{}
|
||
|
}
|
||
|
|
||
|
// BreadcrumbMap returns p.Path where every element is a map
|
||
|
// of URLs and path segment names.
|
||
|
func (i Info) BreadcrumbMap() map[string]string {
|
||
|
result := map[string]string{}
|
||
|
|
||
|
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 == 0 && part == "" {
|
||
|
// Leading slash (root)
|
||
|
result["/"] = "/"
|
||
|
continue
|
||
|
}
|
||
|
result[strings.Join(parts[:i+1], "/")] = part
|
||
|
}
|
||
|
|
||
|
return result
|
||
|
}
|
||
|
|
||
|
// PreviousLink returns the path of the previous folder
|
||
|
func (i Info) PreviousLink() string {
|
||
|
path := strings.TrimSuffix(i.Path, "/")
|
||
|
path = strings.TrimPrefix(path, "/")
|
||
|
path = i.Config.BaseURL + "/" + path
|
||
|
path = path[0 : len(path)-len(i.Name)]
|
||
|
|
||
|
if len(path) < len(i.Config.BaseURL+"/") {
|
||
|
return ""
|
||
|
}
|
||
|
|
||
|
return path
|
||
|
}
|
||
|
|
||
|
// PrintAsHTML formats the page in HTML and executes the template
|
||
|
func (p Page) PrintAsHTML(w http.ResponseWriter, templates ...string) (int, error) {
|
||
|
// Create the functions map, then the template, check for erros and
|
||
|
// execute the template if there aren't errors
|
||
|
functions := template.FuncMap{
|
||
|
"SplitCapitalize": variables.SplitCapitalize,
|
||
|
"Defined": variables.Defined,
|
||
|
}
|
||
|
|
||
|
templates = append(templates, "actions", "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 := assets.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).Funcs(functions).Parse(string(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.Info)
|
||
|
|
||
|
if err != nil {
|
||
|
return http.StatusInternalServerError, err
|
||
|
}
|
||
|
|
||
|
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||
|
_, err = buf.WriteTo(w)
|
||
|
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)
|
||
|
}
|