filebrowser/settings.go

package filemanager

import (
	"encoding/json"
	"errors"
	"net/http"

	"github.com/mitchellh/mapstructure"
)

func commandsHandler(c *RequestContext, w http.ResponseWriter, r *http.Request) (int, error) {
	switch r.Method {
	case http.MethodGet:
		return commandsGetHandler(c, w, r)
	case http.MethodPut:
		return commandsPutHandler(c, w, r)
	}

	return http.StatusMethodNotAllowed, nil
}

func commandsGetHandler(c *RequestContext, w http.ResponseWriter, r *http.Request) (int, error) {
	if !c.User.Admin {
		return http.StatusForbidden, nil
	}

	return renderJSON(w, c.FM.Commands)
}

func commandsPutHandler(c *RequestContext, w http.ResponseWriter, r *http.Request) (int, error) {
	if !c.User.Admin {
		return http.StatusForbidden, nil
	}

	if r.Body == nil {
		return http.StatusBadGateway, errors.New("Empty request body")
	}

	var commands map[string][]string

	// Parses the user and checks for error.
	err := json.NewDecoder(r.Body).Decode(&commands)
	if err != nil {
		return http.StatusBadRequest, errors.New("Invalid JSON")
	}

	if err := c.FM.db.Set("config", "commands", commands); err != nil {
		return http.StatusInternalServerError, err
	}

	c.FM.Commands = commands
	return http.StatusOK, nil
}

func pluginsHandler(c *RequestContext, w http.ResponseWriter, r *http.Request) (int, error) {
	switch r.Method {
	case http.MethodGet:
		return pluginsGetHandler(c, w, r)
	case http.MethodPut:
		return pluginsPutHandler(c, w, r)
	}

	return http.StatusMethodNotAllowed, nil
}

func pluginsGetHandler(c *RequestContext, w http.ResponseWriter, r *http.Request) (int, error) {
	if !c.User.Admin {
		return http.StatusForbidden, nil
	}

	return renderJSON(w, c.FM.Plugins)
}

func pluginsPutHandler(c *RequestContext, w http.ResponseWriter, r *http.Request) (int, error) {
	if !c.User.Admin {
		return http.StatusForbidden, nil
	}

	if r.Body == nil {
		return http.StatusBadGateway, errors.New("Empty request body")
	}

	var raw map[string]map[string]interface{}

	// Parses the user and checks for error.
	err := json.NewDecoder(r.Body).Decode(&raw)
	if err != nil {
		return http.StatusBadRequest, err
	}

	for name, plugin := range raw {
		err = mapstructure.Decode(plugin, c.FM.Plugins[name])
		if err != nil {
			return http.StatusInternalServerError, err
		}

		err = c.FM.db.Set("plugins", name, c.FM.Plugins[name])
		if err != nil {
			return http.StatusInternalServerError, err
		}
	}

	return http.StatusOK, nil
}