package modchannel import ( "context" "strings" "github.com/go-irc/irc" "github.com/pkg/errors" "github.com/Luzifer/twitch-bot/plugins" "github.com/Luzifer/twitch-bot/twitch" ) const actorName = "modchannel" var ( formatMessage plugins.MsgFormatter tcGetter func(string) (*twitch.Client, error) ) func Register(args plugins.RegistrationArguments) error { formatMessage = args.FormatMessage tcGetter = args.GetTwitchClientForChannel args.RegisterActor(actorName, func() plugins.Actor { return &actor{} }) args.RegisterActorDocumentation(plugins.ActionDocumentation{ Description: "Update stream information", Name: "Modify Stream", Type: "modchannel", Fields: []plugins.ActionDocumentationField{ { Default: "", Description: "Channel to update", Key: "channel", Name: "Channel", Optional: false, SupportTemplate: true, Type: plugins.ActionDocumentationFieldTypeString, }, { Default: "", Description: "Category / Game to set", Key: "game", Name: "Game", Optional: true, SupportTemplate: true, Type: plugins.ActionDocumentationFieldTypeString, }, { Default: "", Description: "Stream title to set", Key: "title", Name: "Title", Optional: true, SupportTemplate: true, Type: plugins.ActionDocumentationFieldTypeString, }, }, }) return nil } type actor struct{} func (a actor) Execute(c *irc.Client, m *irc.Message, r *plugins.Rule, eventData *plugins.FieldCollection, attrs *plugins.FieldCollection) (preventCooldown bool, err error) { var ( ptrStringEmpty = func(v string) *string { return &v }("") game = attrs.MustString("game", ptrStringEmpty) title = attrs.MustString("title", ptrStringEmpty) ) if game == "" && title == "" { return false, nil } var updGame, updTitle *string channel, err := formatMessage(attrs.MustString("channel", nil), m, r, eventData) if err != nil { return false, errors.Wrap(err, "parsing channel") } if game != "" { parsedGame, err := formatMessage(game, m, r, eventData) if err != nil { return false, errors.Wrap(err, "parsing game") } updGame = &parsedGame } if title != "" { parsedTitle, err := formatMessage(title, m, r, eventData) if err != nil { return false, errors.Wrap(err, "parsing title") } updTitle = &parsedTitle } twitchClient, err := tcGetter(strings.TrimLeft(channel, "#")) if err != nil { return false, errors.Wrap(err, "getting Twitch client") } return false, errors.Wrap( twitchClient.ModifyChannelInformation(context.Background(), strings.TrimLeft(channel, "#"), updGame, updTitle), "updating channel info", ) } func (a actor) IsAsync() bool { return false } func (a actor) Name() string { return actorName } func (a actor) Validate(attrs *plugins.FieldCollection) (err error) { if v, err := attrs.String("channel"); err != nil || v == "" { return errors.New("channel must be non-empty string") } return nil }