package quotedb import ( _ "embed" "encoding/json" "net/http" "strconv" "strings" "github.com/Luzifer/twitch-bot/plugins" "github.com/gorilla/mux" "github.com/pkg/errors" ) var ( //go:embed list.html listFrontend []byte //go:embed list.js listScript []byte ) func registerAPI(register plugins.HTTPRouteRegistrationFunc) { register(plugins.HTTPRouteRegistrationArgs{ HandlerFunc: handleScript, Method: http.MethodGet, Module: "quotedb", Path: "/app.js", SkipDocumentation: true, }) register(plugins.HTTPRouteRegistrationArgs{ Description: "Add quotes for the given {channel}", HandlerFunc: handleAddQuotes, Method: http.MethodPost, Module: "quotedb", Name: "Add Quotes", Path: "/{channel}", RequiresWriteAuth: true, ResponseType: plugins.HTTPRouteResponseTypeTextPlain, RouteParams: []plugins.HTTPRouteParamDocumentation{ { Description: "Channel to delete the quote in", Name: "channel", }, }, }) register(plugins.HTTPRouteRegistrationArgs{ Description: "Deletes quote with given {idx} from {channel}", HandlerFunc: handleDeleteQuote, Method: http.MethodDelete, Module: "quotedb", Name: "Delete Quote", Path: "/{channel}/{idx:[0-9]+}", RequiresWriteAuth: true, ResponseType: plugins.HTTPRouteResponseTypeTextPlain, RouteParams: []plugins.HTTPRouteParamDocumentation{ { Description: "Channel to delete the quote in", Name: "channel", }, { Description: "Index of the quote to delete", Name: "idx", }, }, }) register(plugins.HTTPRouteRegistrationArgs{ Accept: []string{"application/json", "text/html"}, Description: "Lists all quotes for the given {channel}", HandlerFunc: handleListQuotes, Method: http.MethodGet, Module: "quotedb", Name: "List Quotes", Path: "/{channel}", ResponseType: plugins.HTTPRouteResponseTypeMultiple, RouteParams: []plugins.HTTPRouteParamDocumentation{ { Description: "Channel to delete the quote in", Name: "channel", }, }, }) register(plugins.HTTPRouteRegistrationArgs{ Description: "Set quotes for the given {channel} (will overwrite ALL quotes!)", HandlerFunc: handleReplaceQuotes, Method: http.MethodPut, Module: "quotedb", Name: "Set Quotes", Path: "/{channel}", RequiresWriteAuth: true, ResponseType: plugins.HTTPRouteResponseTypeTextPlain, RouteParams: []plugins.HTTPRouteParamDocumentation{ { Description: "Channel to delete the quote in", Name: "channel", }, }, }) register(plugins.HTTPRouteRegistrationArgs{ Description: "Updates quote with given {idx} from {channel}", HandlerFunc: handleUpdateQuote, Method: http.MethodPut, Module: "quotedb", Name: "Update Quote", Path: "/{channel}/{idx:[0-9]+}", RequiresWriteAuth: true, ResponseType: plugins.HTTPRouteResponseTypeTextPlain, RouteParams: []plugins.HTTPRouteParamDocumentation{ { Description: "Channel to delete the quote in", Name: "channel", }, { Description: "Index of the quote to delete", Name: "idx", }, }, }) } func handleAddQuotes(w http.ResponseWriter, r *http.Request) { channel := "#" + strings.TrimLeft(mux.Vars(r)["channel"], "#") var quotes []string if err := json.NewDecoder(r.Body).Decode("es); err != nil { http.Error(w, errors.Wrap(err, "parsing input").Error(), http.StatusBadRequest) return } for _, q := range quotes { storedObject.AddQuote(channel, q) } if err := store.SetModuleStore(moduleUUID, storedObject); err != nil { http.Error(w, errors.Wrap(err, "storing quote database").Error(), http.StatusInternalServerError) return } w.WriteHeader(http.StatusCreated) } func handleDeleteQuote(w http.ResponseWriter, r *http.Request) { var ( channel = "#" + strings.TrimLeft(mux.Vars(r)["channel"], "#") idxStr = mux.Vars(r)["idx"] ) idx, err := strconv.Atoi(idxStr) if err != nil { http.Error(w, errors.Wrap(err, "parsing index").Error(), http.StatusBadRequest) return } storedObject.DelQuote(channel, idx) if err := store.SetModuleStore(moduleUUID, storedObject); err != nil { http.Error(w, errors.Wrap(err, "storing quote database").Error(), http.StatusInternalServerError) return } w.WriteHeader(http.StatusNoContent) } func handleListQuotes(w http.ResponseWriter, r *http.Request) { if strings.HasPrefix(r.Header.Get("Accept"), "text/html") { w.Header().Set("Content-Type", "text/html") w.Write(listFrontend) return } channel := "#" + strings.TrimLeft(mux.Vars(r)["channel"], "#") quotes := storedObject.GetChannelQuotes(channel) if err := json.NewEncoder(w).Encode(quotes); err != nil { http.Error(w, errors.Wrap(err, "enocding quote list").Error(), http.StatusInternalServerError) return } } func handleReplaceQuotes(w http.ResponseWriter, r *http.Request) { channel := "#" + strings.TrimLeft(mux.Vars(r)["channel"], "#") var quotes []string if err := json.NewDecoder(r.Body).Decode("es); err != nil { http.Error(w, errors.Wrap(err, "parsing input").Error(), http.StatusBadRequest) return } storedObject.SetQuotes(channel, quotes) if err := store.SetModuleStore(moduleUUID, storedObject); err != nil { http.Error(w, errors.Wrap(err, "storing quote database").Error(), http.StatusInternalServerError) return } w.WriteHeader(http.StatusNoContent) } func handleScript(w http.ResponseWriter, r *http.Request) { w.Header().Set("Content-Type", "text/javascript") w.Write(listScript) } func handleUpdateQuote(w http.ResponseWriter, r *http.Request) { var ( channel = "#" + strings.TrimLeft(mux.Vars(r)["channel"], "#") idxStr = mux.Vars(r)["idx"] ) idx, err := strconv.Atoi(idxStr) if err != nil { http.Error(w, errors.Wrap(err, "parsing index").Error(), http.StatusBadRequest) return } var quotes []string if err := json.NewDecoder(r.Body).Decode("es); err != nil { http.Error(w, errors.Wrap(err, "parsing input").Error(), http.StatusBadRequest) return } if len(quotes) != 1 { http.Error(w, "input must be list with one quote", http.StatusBadRequest) return } storedObject.UpdateQuote(channel, idx, quotes[0]) if err := store.SetModuleStore(moduleUUID, storedObject); err != nil { http.Error(w, errors.Wrap(err, "storing quote database").Error(), http.StatusInternalServerError) return } w.WriteHeader(http.StatusNoContent) }