twitch-bot/internal/actors/commercial/actor.go

133 lines
3.5 KiB
Go
Raw Normal View History

package commercial
import (
"context"
"regexp"
"strconv"
"strings"
"github.com/go-irc/irc"
"github.com/pkg/errors"
"github.com/Luzifer/twitch-bot/v3/pkg/twitch"
"github.com/Luzifer/twitch-bot/v3/plugins"
)
const (
actorName = "commercial"
maxCommercialDuration = 180
)
var (
formatMessage plugins.MsgFormatter
permCheckFn plugins.ChannelPermissionCheckFunc
tcGetter func(string) (*twitch.Client, error)
commercialChatcommandRegex = regexp.MustCompile(`^/commercial ([0-9]+)$`)
)
func Register(args plugins.RegistrationArguments) error {
formatMessage = args.FormatMessage
permCheckFn = args.HasPermissionForChannel
tcGetter = args.GetTwitchClientForChannel
args.RegisterActor(actorName, func() plugins.Actor { return &actor{} })
args.RegisterActorDocumentation(plugins.ActionDocumentation{
Description: "Start Commercial",
Name: "Commercial",
Type: actorName,
Fields: []plugins.ActionDocumentationField{
{
Default: "",
Description: "Duration of the commercial (must not be longer than 180s and must yield an integer)",
Key: "duration",
Name: "Duration",
Optional: false,
SupportTemplate: true,
Type: plugins.ActionDocumentationFieldTypeString,
},
},
})
args.RegisterMessageModFunc("/commercial", handleChatCommand)
return nil
}
type actor struct{}
func (actor) Execute(_ *irc.Client, m *irc.Message, r *plugins.Rule, eventData *plugins.FieldCollection, attrs *plugins.FieldCollection) (preventCooldown bool, err error) {
ptrStringEmpty := func(v string) *string { return &v }("")
durationStr, err := formatMessage(attrs.MustString("duration", ptrStringEmpty), m, r, eventData)
if err != nil {
return false, errors.Wrap(err, "executing duration template")
}
return false, startCommercial(strings.TrimLeft(plugins.DeriveChannel(m, eventData), "#"), durationStr)
}
func (a actor) IsAsync() bool { return false }
func (a actor) Name() string { return actorName }
func (a actor) Validate(tplValidator plugins.TemplateValidatorFunc, attrs *plugins.FieldCollection) (err error) {
durationTemplate, err := attrs.String("duration")
if err != nil || durationTemplate == "" {
return errors.New("duration must be non-empty string")
}
if err = tplValidator(durationTemplate); err != nil {
return errors.Wrap(err, "validating duration template")
}
return nil
}
func handleChatCommand(m *irc.Message) error {
channel := strings.TrimLeft(plugins.DeriveChannel(m, nil), "#")
matches := commercialChatcommandRegex.FindStringSubmatch(m.Trailing())
if matches == nil {
return errors.New("ban message does not match required format")
}
if err := startCommercial(channel, matches[1]); err != nil {
return err
}
return plugins.ErrSkipSendingMessage
}
func startCommercial(channel, durationStr string) error {
duration, err := strconv.ParseInt(durationStr, 10, 64)
if err != nil {
return errors.Wrap(err, "parsing duration to integer")
}
if duration > maxCommercialDuration {
return errors.New("duration too long")
}
ok, err := permCheckFn(channel, twitch.ScopeChannelEditCommercial)
if err != nil {
return errors.Wrap(err, "checking for channel permissions")
}
if !ok {
return errors.Errorf("channel %q is missing permission %s", channel, twitch.ScopeChannelEditCommercial)
}
tc, err := tcGetter(channel)
if err != nil {
return errors.Wrap(err, "getting channel twitch-client")
}
return errors.Wrap(
tc.RunCommercial(context.Background(), channel, duration),
"running commercial",
)
}