package main import ( "bytes" "context" "encoding/json" "os" "os/exec" "github.com/Luzifer/twitch-bot/plugins" "github.com/Luzifer/twitch-bot/twitch" "github.com/go-irc/irc" "github.com/pkg/errors" ) func init() { registerAction(func() plugins.Actor { return &ActorScript{} }) } type ActorScript struct { Command []string `json:"command" yaml:"command"` } func (a ActorScript) Execute(c *irc.Client, m *irc.Message, r *plugins.Rule, eventData map[string]interface{}) (preventCooldown bool, err error) { if len(a.Command) == 0 { return false, nil } var command []string for _, arg := range a.Command { tmp, err := formatMessage(arg, m, r, eventData) if err != nil { return false, errors.Wrap(err, "execute command argument template") } command = append(command, tmp) } ctx, cancel := context.WithTimeout(context.Background(), cfg.CommandTimeout) defer cancel() var ( stdin = new(bytes.Buffer) stdout = new(bytes.Buffer) ) if err := json.NewEncoder(stdin).Encode(map[string]interface{}{ "badges": twitch.ParseBadgeLevels(m), "channel": m.Params[0], "message": m.Trailing(), "tags": m.Tags, "username": m.User, }); err != nil { return false, errors.Wrap(err, "encoding script input") } cmd := exec.CommandContext(ctx, command[0], command[1:]...) // #nosec G204 // This is expected to call a command with parameters cmd.Env = os.Environ() cmd.Stderr = os.Stderr cmd.Stdin = stdin cmd.Stdout = stdout if err := cmd.Run(); err != nil { return false, errors.Wrap(err, "running command") } if stdout.Len() == 0 { // Script was successful but did not yield actions return false, nil } var ( actions []*plugins.RuleAction decoder = json.NewDecoder(stdout) ) decoder.DisallowUnknownFields() if err := decoder.Decode(&actions); err != nil { return false, errors.Wrap(err, "decoding actions output") } for _, action := range actions { apc, err := triggerActions(c, m, r, action, eventData) if err != nil { return preventCooldown, errors.Wrap(err, "execute returned action") } preventCooldown = preventCooldown || apc } return preventCooldown, nil } func (a ActorScript) IsAsync() bool { return false } func (a ActorScript) Name() string { return "script" }