142 lines · 4.3 KB · Go
RawDownload
package http import (	"bytes"	"embed"	"fmt"	"html/template"	"os"	"strings"	"time" 	"github.com/spf13/pflag" 	"github.com/rclone/rclone/fs"	"github.com/rclone/rclone/fs/config/flags") // TemplateHelp returns a string that describes how to use a custom templatefunc TemplateHelp(prefix string) string {	help := `#### Template ` + "`--{{ .Prefix }}template`" + ` allows a user to specify a custom markup template for HTTPand WebDAV serve functions.  The server exports the following markupto be used within the template to server pages: | Parameter   | Description || :---------- | :---------- || .Name       | The full path of a file/directory. || .Title      | Directory listing of .Name || .Sort       | The current sort used.  This is changeable via ?sort= parameter ||             | Sort Options: namedirfirst,name,size,time (default namedirfirst) || .Order      | The current ordering used.  This is changeable via ?order= parameter ||             | Order Options: asc,desc (default asc) || .Query      | Currently unused. || .Breadcrumb | Allows for creating a relative navigation ||-- .Link     | The relative to the root link of the Text. ||-- .Text     | The Name of the directory. || .Entries    | Information about a specific file/directory. ||-- .URL      | The 'url' of an entry.  ||-- .Leaf     | Currently same as 'URL' but intended to be 'just' the name. ||-- .IsDir    | Boolean for if an entry is a directory or not. ||-- .Size     | Size in Bytes of the entry. ||-- .ModTime  | The UTC timestamp of an entry. | The server also makes the following functions available so that they can be used within thetemplate. These functions help extend the options for dynamic rendering of HTML. They canbe used to render HTML based on specific conditions. | Function   | Description || :---------- | :---------- || afterEpoch  | Returns the time since the epoch for the given time. || contains    | Checks whether a given substring is present or not in a given string. || hasPrefix   | Checks whether the given string begins with the specified prefix. || hasSuffix   | Checks whether the given string end with the specified suffix. | ` 	tmpl, err := template.New("template help").Parse(help)	if err != nil {		fs.Fatal(nil, fmt.Sprint("Fatal error parsing template", err))	} 	data := struct {		Prefix string	}{		Prefix: prefix,	}	buf := &bytes.Buffer{}	err = tmpl.Execute(buf, data)	if err != nil {		fs.Fatal(nil, fmt.Sprint("Fatal error executing template", err))	}	return buf.String()} // TemplateConfigInfo descripts the Options in usevar TemplateConfigInfo = fs.Options{{	Name:    "template",	Default: "",	Help:    "User-specified template",}} // TemplateConfig for the templating functionalitytype TemplateConfig struct {	Path string `config:"template"`} // AddFlagsPrefix for the templating functionalityfunc (cfg *TemplateConfig) AddFlagsPrefix(flagSet *pflag.FlagSet, prefix string) {	flags.StringVarP(flagSet, &cfg.Path, prefix+"template", "", cfg.Path, "User-specified template", prefix)} // AddTemplateFlagsPrefix for the templating functionalityfunc AddTemplateFlagsPrefix(flagSet *pflag.FlagSet, prefix string, cfg *TemplateConfig) {	cfg.AddFlagsPrefix(flagSet, prefix)} // DefaultTemplateCfg returns a new config which can be customized by command line flags//// Note that this needs to be kept in sync with TemplateConfigInfo above and// can be removed when all callers have been converted.func DefaultTemplateCfg() TemplateConfig {	return TemplateConfig{}} // AfterEpoch returns the time since the epoch for the given timefunc AfterEpoch(t time.Time) bool {	return t.After(time.Time{})} // Assets holds the embedded filesystem for the default template////go:embed templatesvar Assets embed.FS // GetTemplate returns the HTML template for serving directories via HTTP/WebDAVfunc GetTemplate(tmpl string) (*template.Template, error) {	var readFile = os.ReadFile	if tmpl == "" {		tmpl = "templates/index.html"		readFile = Assets.ReadFile	} 	data, err := readFile(tmpl)	if err != nil {		return nil, err	} 	funcMap := template.FuncMap{		"afterEpoch": AfterEpoch,		"contains":   strings.Contains,		"hasPrefix":  strings.HasPrefix,		"hasSuffix":  strings.HasSuffix,	} 	tpl, err := template.New("index").Funcs(funcMap).Parse(string(data))	if err != nil {		return nil, err	} 	return tpl, nil}