twitch-bot/internal/actors/quotedb/http.go

209 lines
5.7 KiB
Go
Raw Normal View History

package quotedb
import (
"encoding/json"
"net/http"
"strconv"
"strings"
"github.com/Luzifer/twitch-bot/plugins"
"github.com/gorilla/mux"
"github.com/pkg/errors"
)
func registerAPI(register plugins.HTTPRouteRegistrationFunc) {
register(plugins.HTTPRouteRegistrationArgs{
Description: "Add quotes for the given {channel}",
HandlerFunc: handleAddQuotes,
Method: http.MethodPost,
Module: "quotedb",
Name: "Add Quotes",
Path: "/{channel}",
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]+}",
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{
Description: "Lists all quotes for the given {channel}",
HandlerFunc: handleListQuotes,
Method: http.MethodGet,
Module: "quotedb",
Name: "List Quotes",
Path: "/{channel}",
ResponseType: plugins.HTTPRouteResponseTypeJSON,
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}",
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]+}",
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(&quotes); 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) {
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(&quotes); 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 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(&quotes); 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)
}