147 lines
3.3 KiB
Go
147 lines
3.3 KiB
Go
package http
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"net/http"
|
|
"reflect"
|
|
|
|
fm "github.com/filebrowser/filebrowser"
|
|
"github.com/mitchellh/mapstructure"
|
|
)
|
|
|
|
type modifySettingsRequest struct {
|
|
modifyRequest
|
|
Data struct {
|
|
CSS string `json:"css"`
|
|
Commands map[string][]string `json:"commands"`
|
|
StaticGen map[string]interface{} `json:"staticGen"`
|
|
} `json:"data"`
|
|
}
|
|
|
|
type option struct {
|
|
Variable string `json:"variable"`
|
|
Name string `json:"name"`
|
|
Value interface{} `json:"value"`
|
|
}
|
|
|
|
func parsePutSettingsRequest(r *http.Request) (*modifySettingsRequest, error) {
|
|
// Checks if the request body is empty.
|
|
if r.Body == nil {
|
|
return nil, fm.ErrEmptyRequest
|
|
}
|
|
|
|
// Parses the request body and checks if it's well formed.
|
|
mod := &modifySettingsRequest{}
|
|
err := json.NewDecoder(r.Body).Decode(mod)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Checks if the request type is right.
|
|
if mod.What != "settings" {
|
|
return nil, fm.ErrWrongDataType
|
|
}
|
|
|
|
return mod, nil
|
|
}
|
|
|
|
func settingsHandler(c *fm.Context, w http.ResponseWriter, r *http.Request) (int, error) {
|
|
if r.URL.Path != "" && r.URL.Path != "/" {
|
|
return http.StatusNotFound, nil
|
|
}
|
|
|
|
switch r.Method {
|
|
case http.MethodGet:
|
|
return settingsGetHandler(c, w, r)
|
|
case http.MethodPut:
|
|
return settingsPutHandler(c, w, r)
|
|
}
|
|
|
|
return http.StatusMethodNotAllowed, nil
|
|
}
|
|
|
|
type settingsGetRequest struct {
|
|
CSS string `json:"css"`
|
|
Commands map[string][]string `json:"commands"`
|
|
StaticGen []option `json:"staticGen"`
|
|
}
|
|
|
|
func settingsGetHandler(c *fm.Context, w http.ResponseWriter, r *http.Request) (int, error) {
|
|
if !c.User.Admin {
|
|
return http.StatusForbidden, nil
|
|
}
|
|
|
|
result := &settingsGetRequest{
|
|
Commands: c.Commands,
|
|
StaticGen: []option{},
|
|
CSS: c.CSS,
|
|
}
|
|
|
|
if c.StaticGen != nil {
|
|
t := reflect.TypeOf(c.StaticGen).Elem()
|
|
|
|
for i := 0; i < t.NumField(); i++ {
|
|
if t.Field(i).Name[0] == bytes.ToLower([]byte{t.Field(i).Name[0]})[0] {
|
|
continue
|
|
}
|
|
|
|
result.StaticGen = append(result.StaticGen, option{
|
|
Variable: t.Field(i).Name,
|
|
Name: t.Field(i).Tag.Get("name"),
|
|
Value: reflect.ValueOf(c.StaticGen).Elem().FieldByName(t.Field(i).Name).Interface(),
|
|
})
|
|
}
|
|
}
|
|
|
|
return renderJSON(w, result)
|
|
}
|
|
|
|
func settingsPutHandler(c *fm.Context, w http.ResponseWriter, r *http.Request) (int, error) {
|
|
if !c.User.Admin {
|
|
return http.StatusForbidden, nil
|
|
}
|
|
|
|
mod, err := parsePutSettingsRequest(r)
|
|
if err != nil {
|
|
return http.StatusBadRequest, err
|
|
}
|
|
|
|
// Update the commands.
|
|
if mod.Which == "commands" {
|
|
if err := c.Store.Config.Save("commands", mod.Data.Commands); err != nil {
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
|
|
c.Commands = mod.Data.Commands
|
|
return http.StatusOK, nil
|
|
}
|
|
|
|
// Update the global CSS.
|
|
if mod.Which == "css" {
|
|
if err := c.Store.Config.Save("css", mod.Data.CSS); err != nil {
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
|
|
c.CSS = mod.Data.CSS
|
|
return http.StatusOK, nil
|
|
}
|
|
|
|
// Update the static generator options.
|
|
if mod.Which == "staticGen" {
|
|
err = mapstructure.Decode(mod.Data.StaticGen, c.StaticGen)
|
|
if err != nil {
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
|
|
err = c.Store.Config.Save("staticgen_"+c.StaticGen.Name(), c.StaticGen)
|
|
if err != nil {
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
|
|
return http.StatusOK, nil
|
|
}
|
|
|
|
return http.StatusMethodNotAllowed, nil
|
|
}
|