filebrowser/setup.go

109 lines
2.0 KiB
Go

package filemanager
import (
"fmt"
"io/ioutil"
"net/http"
"text/template"
"github.com/mholt/caddy"
"github.com/mholt/caddy/caddyhttp/httpserver"
)
func init() {
caddy.RegisterPlugin("filemanager", caddy.Plugin{
ServerType: "http",
Action: setup,
})
}
// setup configures a new Browse middleware instance.
func setup(c *caddy.Controller) error {
// Second argument would be the template file to use
tplBytes, err := Asset("templates/template.tmpl")
if err != nil {
return err
}
tplText := string(tplBytes)
// Build the template
tpl, err := template.New("listing").Parse(tplText)
if err != nil {
return err
}
Template = tpl
configs, err := fileManagerParse(c)
if err != nil {
return err
}
f := FileManager{
Configs: configs,
IgnoreIndexes: false,
}
httpserver.GetConfig(c.Key).AddMiddleware(func(next httpserver.Handler) httpserver.Handler {
f.Next = next
return f
})
return nil
}
func fileManagerParse(c *caddy.Controller) ([]Config, error) {
var configs []Config
appendCfg := func(fmc Config) error {
for _, c := range configs {
if c.PathScope == fmc.PathScope {
return fmt.Errorf("duplicate file managing config for %s", c.PathScope)
}
}
configs = append(configs, fmc)
return nil
}
for c.Next() {
var fmc = Config{
PathScope: ".",
BaseURL: "/",
StyleSheet: "",
}
for c.NextBlock() {
switch c.Val() {
case "show":
if !c.NextArg() {
return configs, c.ArgErr()
}
fmc.PathScope = c.Val()
case "on":
if !c.NextArg() {
return configs, c.ArgErr()
}
fmc.BaseURL = c.Val()
case "styles":
if !c.NextArg() {
return configs, c.ArgErr()
}
tplBytes, err := ioutil.ReadFile(c.Val())
if err != nil {
return configs, err
}
fmc.StyleSheet = string(tplBytes)
}
}
fmc.Root = http.Dir(fmc.PathScope)
// Save configuration
err := appendCfg(fmc)
if err != nil {
return configs, err
}
}
return configs, nil
}