package filemanager import ( "encoding/json" "net/http" "reflect" "github.com/mitchellh/mapstructure" ) type modifySettingsRequest struct { *modifyRequest Data struct { Commands map[string][]string `json:"commands"` Plugins map[string]map[string]interface{} `json:"plugins"` } `json:"data"` } type pluginOption 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, 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, errWrongDataType } return mod, nil } func settingsHandler(c *RequestContext, 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 { Commands map[string][]string `json:"commands"` Plugins map[string][]pluginOption `json:"plugins"` } func settingsGetHandler(c *RequestContext, w http.ResponseWriter, r *http.Request) (int, error) { if !c.User.Admin { return http.StatusForbidden, nil } result := &settingsGetRequest{ Commands: c.Commands, Plugins: map[string][]pluginOption{}, } for name, p := range c.Plugins { result.Plugins[name] = []pluginOption{} t := reflect.TypeOf(p).Elem() for i := 0; i < t.NumField(); i++ { result.Plugins[name] = append(result.Plugins[name], pluginOption{ Variable: t.Field(i).Name, Name: t.Field(i).Tag.Get("name"), Value: reflect.ValueOf(p).Elem().FieldByName(t.Field(i).Name).Interface(), }) } } return renderJSON(w, result) } func settingsPutHandler(c *RequestContext, 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.db.Set("config", "commands", mod.Data.Commands); err != nil { return http.StatusInternalServerError, err } c.Commands = mod.Data.Commands return http.StatusOK, nil } // Update the plugins. if mod.Which == "plugins" { for name, plugin := range mod.Data.Plugins { err = mapstructure.Decode(plugin, c.Plugins[name]) if err != nil { return http.StatusInternalServerError, err } err = c.db.Set("plugins", name, c.Plugins[name]) if err != nil { return http.StatusInternalServerError, err } } return http.StatusOK, nil } return http.StatusMethodNotAllowed, nil }