89 lines
2.4 KiB
Go
89 lines
2.4 KiB
Go
package bot
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
|
|
twitch "github.com/gempir/go-twitch-irc/v2"
|
|
"github.com/gempir/justlog/humanize"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func (b *Bot) handlePrivateMessage(message twitch.PrivateMessage) {
|
|
if contains(b.cfg.Admins, message.User.Name) {
|
|
if strings.HasPrefix(message.Message, "!justlog status") || strings.HasPrefix(message.Message, "!status") {
|
|
uptime := humanize.TimeSince(b.startTime)
|
|
b.twitchClient.Say(message.Channel, message.User.DisplayName+", uptime: "+uptime)
|
|
}
|
|
if strings.HasPrefix(message.Message, "!justlog join ") {
|
|
b.handleJoin(message)
|
|
}
|
|
if strings.HasPrefix(message.Message, "!justlog messageType ") {
|
|
b.handleMessageType(message)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (b *Bot) handleJoin(message twitch.PrivateMessage) {
|
|
input := strings.TrimPrefix(message.Message, "!justlog join ")
|
|
|
|
users, err := b.helixClient.GetUsersByUsernames(strings.Split(input, ","))
|
|
if err != nil {
|
|
log.Error(err)
|
|
b.twitchClient.Say(message.Channel, message.User.DisplayName+", something went wrong requesting the userids")
|
|
}
|
|
|
|
ids := []string{}
|
|
for _, user := range users {
|
|
ids = append(ids, user.ID)
|
|
log.Infof("[bot] joining %s", user.Login)
|
|
b.twitchClient.Join(user.Login)
|
|
}
|
|
b.cfg.AddChannels(ids...)
|
|
b.twitchClient.Say(message.Channel, fmt.Sprintf("%s, added channels: %v", message.User.DisplayName, ids))
|
|
}
|
|
|
|
func (b *Bot) handleMessageType(message twitch.PrivateMessage) {
|
|
input := strings.TrimPrefix(message.Message, "!justlog messageType ")
|
|
|
|
parts := strings.Split(input, " ")
|
|
if len(parts) < 2 {
|
|
return
|
|
}
|
|
|
|
users, err := b.helixClient.GetUsersByUsernames([]string{parts[0]})
|
|
if err != nil {
|
|
log.Error(err)
|
|
return
|
|
}
|
|
|
|
if parts[1] == "reset" {
|
|
b.cfg.ResetMessageTypes(users[parts[0]].ID)
|
|
log.Infof("[bot] setting %s config messageTypes to default", parts[0])
|
|
} else {
|
|
var messageTypes []twitch.MessageType
|
|
for _, msgType := range strings.Split(parts[1], ",") {
|
|
messageType, err := strconv.Atoi(msgType)
|
|
if err != nil {
|
|
log.Error(err)
|
|
return
|
|
}
|
|
|
|
messageTypes = append(messageTypes, twitch.MessageType(messageType))
|
|
}
|
|
|
|
b.cfg.SetMessageTypes(users[parts[0]].ID, messageTypes)
|
|
b.UpdateMessageTypesToLog()
|
|
log.Infof("[bot] setting %s config messageTypes to %v", parts[0], messageTypes)
|
|
}
|
|
}
|
|
|
|
func contains(arr []string, str string) bool {
|
|
for _, x := range arr {
|
|
if x == str {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|