mirror of
https://github.com/Luzifer/twitch-bot.git
synced 2024-11-09 16:50:01 +00:00
99 lines
2.8 KiB
Go
99 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/Luzifer/go_helpers/v2/str"
|
|
"github.com/Luzifer/twitch-bot/internal/actors/ban"
|
|
"github.com/Luzifer/twitch-bot/internal/actors/delay"
|
|
deleteactor "github.com/Luzifer/twitch-bot/internal/actors/delete"
|
|
"github.com/Luzifer/twitch-bot/internal/actors/modchannel"
|
|
"github.com/Luzifer/twitch-bot/internal/actors/nuke"
|
|
"github.com/Luzifer/twitch-bot/internal/actors/punish"
|
|
"github.com/Luzifer/twitch-bot/internal/actors/quotedb"
|
|
"github.com/Luzifer/twitch-bot/internal/actors/raw"
|
|
"github.com/Luzifer/twitch-bot/internal/actors/respond"
|
|
"github.com/Luzifer/twitch-bot/internal/actors/timeout"
|
|
"github.com/Luzifer/twitch-bot/internal/actors/whisper"
|
|
"github.com/Luzifer/twitch-bot/plugins"
|
|
"github.com/Luzifer/twitch-bot/twitch"
|
|
"github.com/pkg/errors"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
coreActorRegistations = []plugins.RegisterFunc{
|
|
ban.Register,
|
|
delay.Register,
|
|
deleteactor.Register,
|
|
modchannel.Register,
|
|
nuke.Register,
|
|
punish.Register,
|
|
quotedb.Register,
|
|
raw.Register,
|
|
respond.Register,
|
|
timeout.Register,
|
|
whisper.Register,
|
|
}
|
|
knownModules []string
|
|
)
|
|
|
|
func initCorePlugins() error {
|
|
args := getRegistrationArguments()
|
|
for _, rf := range coreActorRegistations {
|
|
if err := rf(args); err != nil {
|
|
return errors.Wrap(err, "registering core plugin")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func registerRoute(route plugins.HTTPRouteRegistrationArgs) error {
|
|
r := router.
|
|
PathPrefix(fmt.Sprintf("/%s/", route.Module)).
|
|
Subrouter()
|
|
|
|
if !str.StringInSlice(route.Module, knownModules) {
|
|
knownModules = append(knownModules, route.Module)
|
|
}
|
|
|
|
var hdl http.Handler = route.HandlerFunc
|
|
switch {
|
|
case route.RequiresEditorsAuth:
|
|
hdl = botEditorAuthMiddleware(hdl)
|
|
case route.RequiresWriteAuth:
|
|
hdl = writeAuthMiddleware(hdl, route.Module)
|
|
}
|
|
|
|
if route.IsPrefix {
|
|
r.PathPrefix(route.Path).
|
|
Handler(hdl).
|
|
Methods(route.Method)
|
|
} else {
|
|
r.Handle(route.Path, hdl).
|
|
Methods(route.Method)
|
|
}
|
|
|
|
if !route.SkipDocumentation {
|
|
return errors.Wrap(registerSwaggerRoute(route), "registering documentation")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func getRegistrationArguments() plugins.RegistrationArguments {
|
|
return plugins.RegistrationArguments{
|
|
FormatMessage: formatMessage,
|
|
GetLogger: func(moduleName string) *log.Entry { return log.WithField("module", moduleName) },
|
|
GetStorageManager: func() plugins.StorageManager { return store },
|
|
GetTwitchClient: func() *twitch.Client { return twitchClient },
|
|
RegisterActor: registerAction,
|
|
RegisterActorDocumentation: registerActorDocumentation,
|
|
RegisterAPIRoute: registerRoute,
|
|
RegisterCron: cronService.AddFunc,
|
|
RegisterRawMessageHandler: registerRawMessageHandler,
|
|
RegisterTemplateFunction: tplFuncs.Register,
|
|
SendMessage: sendMessage,
|
|
}
|
|
}
|