// InviteCommand is a command for accepting an invite to a channel. func InviteCommand(bot *bruxism.Bot, service bruxism.Service, message bruxism.Message, command string, parts []string) { if service.Name() == bruxism.DiscordServiceName { discord := service.(*bruxism.Discord) if discord.ApplicationClientID != "" { service.SendMessage(message.Channel(), fmt.Sprintf("Please visit https://discordapp.com/oauth2/authorize?client_id=%s&scope=bot to add %s to your server.", discord.ApplicationClientID, service.UserName())) return } } if len(parts) == 1 { join := parts[0] if service.Name() == bruxism.DiscordServiceName { join = discordInviteID(join) } if err := service.Join(join); err != nil { if service.Name() == bruxism.DiscordServiceName && err == bruxism.ErrAlreadyJoined { service.PrivateMessage(message.UserID(), "I have already joined that server.") return } log.Println("Error joining %s %v", service.Name(), err) } else if service.Name() == bruxism.DiscordServiceName { service.PrivateMessage(message.UserID(), "I have joined that server.") } } }
// MTGCommand is a command for getting information about MTG cards.. func MTGCommand(bot *bruxism.Bot, service bruxism.Service, message bruxism.Message, command string, parts []string) { cardNames := fuzzy.RankFindFold(command, MTGCardNames) if len(cardNames) == 0 { service.SendMessage(message.Channel(), "Could not find a card with that name, sorry.") return } sort.Sort(cardNames) card := MTGCardMap[cardNames[0].Target] rest := "" if card.Text != "" { rest += "\n" } if card.Power != nil { rest += MTGRestReplacer.Replace(fmt.Sprintf("%s/%s", *card.Power, *card.Toughness)) } if card.Loyalty != nil { rest += MTGRestReplacer.Replace(fmt.Sprintf("%d", *card.Loyalty)) } if card.ID != nil { if rest != "" && rest != "\n" { rest += "\n" } rest += fmt.Sprintf("(http://gatherer.wizards.com/Handlers/Image.ashx?multiverseid=%d&type=card)", *card.ID) } if service.Name() == bruxism.DiscordServiceName { service.SendMessage(message.Channel(), fmt.Sprintf("**%s** %s\n*%s*\n%s%s", card.Name, card.ManaCost, card.Type, MTGTextReplacer.Replace(card.Text), rest)) } else { service.SendMessage(message.Channel(), strings.Replace(fmt.Sprintf("%s. %s. %s. %s%s", card.Name, card.Type, card.ManaCost, card.Text, rest), "\n", " ", -1)) } }
func (p *carbonitexPlugin) carbonitexPluginLoadFunc(bot *bruxism.Bot, service bruxism.Service, data []byte) error { if service.Name() != bruxism.DiscordServiceName { panic("Carbonitex Plugin only supports Discord.") } go p.Run(bot, service) return nil }
func (p *ReminderPlugin) randomReminder(service bruxism.Service) string { ticks := "" if service.Name() == bruxism.DiscordServiceName { ticks = "`" } return fmt.Sprintf("%s%sreminder %s %s%s", ticks, service.CommandPrefix(), p.random(randomTimes), p.random(randomMessages), ticks) }
func (p *chartPlugin) randomChart(service bruxism.Service) string { ticks := "" if service.Name() == bruxism.DiscordServiceName { ticks = "`" } return fmt.Sprintf("%s%schart %s %s, %s%s", ticks, service.CommandPrefix(), p.random(randomDirection), p.random(randomY), p.random(randomX), ticks) }
// Load will load plugin state from a byte array. func (p *playedPlugin) Load(bot *bruxism.Bot, service bruxism.Service, data []byte) error { if service.Name() != bruxism.DiscordServiceName { panic("Played Plugin only supports Discord.") } if data != nil { if err := json.Unmarshal(data, p); err != nil { log.Println("Error loading data", err) } } go p.Run(bot, service) return nil }
// InviteHelp will return the help text for the invite command. func InviteHelp(bot *bruxism.Bot, service bruxism.Service, message bruxism.Message) (string, string) { switch service.Name() { case bruxism.DiscordServiceName: discord := service.(*bruxism.Discord) if discord.ApplicationClientID != "" { return "", fmt.Sprintf("Returns a URL to add %s to your server.", service.UserName()) } return "<discordinvite>", "Joins the provided Discord server." case bruxism.YouTubeServiceName: return "<videoid>", "Joins the provided YouTube live stream." } return "<channel>", "Joins the provided channel." }
// Load will load plugin state from a byte array. func (p *MusicPlugin) Load(bot *bruxism.Bot, service bruxism.Service, data []byte) (err error) { if service.Name() != bruxism.DiscordServiceName { panic("Music Plugin only supports Discord.") } if data != nil { if err = json.Unmarshal(data, p); err != nil { log.Println("musicplugin: loading data err:", err) } } go p.init() return nil }
func messageFunc(bot *bruxism.Bot, service bruxism.Service, message bruxism.Message) { if service.IsMe(message) || !bruxism.MatchesCommand(service, "myson", message) || service.Name() != bruxism.DiscordServiceName { return } discord := service.(*bruxism.Discord) discord.Session.ChannelMessageSendEmbed(message.Channel(), &discordgo.MessageEmbed{ Color: discord.UserColor(service.UserID(), message.Channel()), Description: "Don't ever talk to me or my son ever again.", Author: &discordgo.MessageEmbedAuthor{ Name: discord.NicknameForID(service.UserID(), service.UserName(), message.Channel()), IconURL: discordgo.EndpointUserAvatar(service.UserID(), discord.Session.State.User.Avatar), }, }) }
func (p *ReminderPlugin) Message(bot *bruxism.Bot, service bruxism.Service, message bruxism.Message) { if !service.IsMe(message) { if bruxism.MatchesCommand(service, "remind", message) || bruxism.MatchesCommand(service, "reminder", message) { _, parts := bruxism.ParseCommand(service, message) if len(parts) < 2 { service.SendMessage(message.Channel(), fmt.Sprintf("Invalid reminder, no time or message. eg: %s", p.randomReminder(service))) return } t, r, err := p.parseReminder(parts) now := time.Now() if err != nil || t.Before(now) || t.After(now.Add(time.Hour*24*365+time.Hour)) { service.SendMessage(message.Channel(), fmt.Sprintf("Invalid time. eg: %s", strings.Join(randomTimes, ", "))) return } if r == "" { service.SendMessage(message.Channel(), fmt.Sprintf("Invalid reminder, no message. eg: %s", p.randomReminder(service))) return } requester := message.UserName() if service.Name() == bruxism.DiscordServiceName { requester = fmt.Sprintf("<@%s>", message.UserID()) } err = p.AddReminder(&Reminder{ StartTime: now, Time: t, Requester: requester, Target: message.Channel(), Message: r, IsPrivate: service.IsPrivate(message), }) if err != nil { service.SendMessage(message.Channel(), err.Error()) return } service.SendMessage(message.Channel(), fmt.Sprintf("Reminder set for %s.", humanize.Time(t))) } } }
// Load will load plugin state from a byte array. func (p *playingPlugin) Load(bot *bruxism.Bot, service bruxism.Service, data []byte) error { if service.Name() != bruxism.DiscordServiceName { panic("Playing Plugin only supports Discord.") } if data != nil { if err := json.Unmarshal(data, p); err != nil { log.Println("Error loading data", err) } } if p.Game != "" { service.(*bruxism.Discord).Session.UpdateStreamingStatus(0, p.Game, p.URL) } else { service.(*bruxism.Discord).Session.UpdateStatus(0, p.Game) } return nil }
// Load will load plugin state from a byte array. func (p *YouTubeJoinPlugin) Load(bot *bruxism.Bot, service bruxism.Service, data []byte) error { if service.Name() != bruxism.YouTubeServiceName { panic("YouTubeJoin plugin only supports YouTube.") } if data != nil { if err := json.Unmarshal(data, p); err != nil { log.Println("Error loading data", err) } } p.youtube = service.(*bruxism.YouTube) for channel, _ := range p.Channels { p.monitor(channel, false) } go p.Run(bot, service) return nil }
func emojiMessageFunc(bot *bruxism.Bot, service bruxism.Service, message bruxism.Message) { if service.Name() == bruxism.DiscordServiceName && !service.IsMe(message) { if bruxism.MatchesCommand(service, "emoji", message) || bruxism.MatchesCommand(service, "hugemoji", message) { base := "emoji/twitter" if bruxism.MatchesCommand(service, "hugemoji", message) { base = "emoji/twitterhuge" } _, parts := bruxism.ParseCommand(service, message) if len(parts) == 1 { submatches := discordRegex.FindStringSubmatch(parts[0]) if len(submatches) != 0 { h, err := http.Get("https://cdn.discordapp.com/emojis/" + submatches[1] + ".png") if err != nil { return } service.SendFile(message.Channel(), "emoji.png", h.Body) h.Body.Close() return } s := strings.TrimSpace(parts[0]) for i := range s { filename := emojiFile(base, s[i:]) if filename != "" { if f, err := os.Open(fmt.Sprintf("%s/%s.png", base, filename)); err == nil { defer f.Close() service.SendFile(message.Channel(), "emoji.png", f) return } } } } } } }
// Help returns a list of help strings that are printed when the user requests them. func (p *comicPlugin) Help(bot *bruxism.Bot, service bruxism.Service, message bruxism.Message, detailed bool) []string { help := bruxism.CommandHelp(service, "comic", "[1-10]", "Creates a comic from recent messages, or a number of messages if provided.") ticks := "" if service.Name() == bruxism.DiscordServiceName { ticks = "`" } if detailed { help = append(help, []string{ bruxism.CommandHelp(service, "customcomic", "[id|name:] <text> | [id|name:] <text>", fmt.Sprintf("Creates a custom comic. Available names: %s%s%s", ticks, strings.Join(comicgen.CharacterNames, ", "), ticks))[0], bruxism.CommandHelp(service, "customcomicsimple", "[id:] <text> | [id:] <text>", "Creates a simple custom comic.")[0], "Examples:", bruxism.CommandHelp(service, "comic", "5", "Creates a comic from the last 5 messages")[0], bruxism.CommandHelp(service, "customcomic", "A | B | C", "Creates a comic with 3 lines.")[0], bruxism.CommandHelp(service, "customcomic", "0: Hi! | 1: Hello! | 0: Goodbye.", "Creates a comic with 3 lines, the second line spoken by a different character")[0], bruxism.CommandHelp(service, "customcomic", "tiki: Hi! | jordy: Hello! | tiki: Goodbye.", "Creates a comic with 3 lines, containing tiki and jordy.")[0], bruxism.CommandHelp(service, "customcomicsimple", "0: Foo | 1: Bar", "Creates a comic with 2 lines, both spoken by different characters.")[0], }...) } return help }
func (p *streamerPlugin) messageFunc(bot *bruxism.Bot, service bruxism.Service, message bruxism.Message) { if !service.IsMe(message) { if bruxism.MatchesCommand(service, "streamer", message) { query, parts := bruxism.ParseCommand(service, message) if len(parts) == 0 { return } r, _ := p.requests[query] if r == nil { r = &streamerPluginRequest{} p.requests[query] = r } n := time.Now() if !n.After(r.lastUpdate.Add(60 * time.Minute)) { if r.lastMessage != "" { service.SendMessage(message.Channel(), fmt.Sprintf("%s *Last updated %s.*", r.lastMessage, humanize.Time(r.lastUpdate))) } return } service.Typing(message.Channel()) r.lastUpdate = n m, err := p.streamer(query, service.Name() == bruxism.DiscordServiceName) if err != nil { service.SendMessage(message.Channel(), "There was an error while requesting the streamer, please try again later.") return } service.SendMessage(message.Channel(), m) r.lastMessage = m } } }
func (p *comicPlugin) makeComic(bot *bruxism.Bot, service bruxism.Service, message bruxism.Message, script *comicgen.Script) { p.Comics++ comic := comicgen.NewComicGen("arial", service.Name() != bruxism.DiscordServiceName) image, err := comic.MakeComic(script) if err != nil { service.SendMessage(message.Channel(), fmt.Sprintf("Sorry %s, there was an error creating the comic. %s", message.UserName(), err)) } else { go func() { if service.Name() == bruxism.DiscordServiceName { discord := service.(*bruxism.Discord) p, err := discord.UserChannelPermissions(message.UserID(), message.Channel()) if err == nil && p&discordgo.PermissionAttachFiles != 0 { b := &bytes.Buffer{} err = png.Encode(b, image) if err != nil { service.SendMessage(message.Channel(), fmt.Sprintf("Sorry %s, there was a problem creating your comic.", message.UserName())) return } if err := service.SendFile(message.Channel(), "comic.png", b); err == nil { return } } } b := &bytes.Buffer{} err = png.Encode(b, image) if err != nil { service.SendMessage(message.Channel(), fmt.Sprintf("Sorry %s, there was a problem creating your comic.", message.UserName())) return } url, err := bot.UploadToImgur(b, "comic.png") if err != nil { service.SendMessage(message.Channel(), fmt.Sprintf("Sorry %s, there was a problem uploading the comic to imgur.", message.UserName())) log.Println("Error uploading comic: ", err) return } if service.Name() == bruxism.DiscordServiceName { service.SendMessage(message.Channel(), fmt.Sprintf("Here's your comic <@%s>: %s", message.UserID(), url)) } else { service.SendMessage(message.Channel(), fmt.Sprintf("Here's your comic %s: %s", message.UserName(), url)) } runtime.GC() }() } }
func (p *playedPlugin) Message(bot *bruxism.Bot, service bruxism.Service, message bruxism.Message) { defer bruxism.MessageRecover() if service.Name() == bruxism.DiscordServiceName && !service.IsMe(message) { if bruxism.MatchesCommand(service, "played", message) { query := strings.Join(strings.Split(message.RawMessage(), " ")[1:], " ") id := message.UserID() match := userIDRegex.FindStringSubmatch(query) if match != nil { id = match[1] } p.Lock() defer p.Unlock() u := p.Users[id] if u == nil { service.SendMessage(message.Channel(), "I haven't seen that user.") return } if len(u.Entries) == 0 { service.SendMessage(message.Channel(), "I haven't seen anything played by that user.") return } lc := humanize.Time(u.LastChanged) u.Update(u.Current, time.Now()) pes := make(byDuration, len(u.Entries)) i := 0 for _, pe := range u.Entries { pes[i] = pe i++ } sort.Sort(pes) messageText := fmt.Sprintf("*First seen %s, last update %s*\n", humanize.Time(u.FirstSeen), lc) for i = 0; i < len(pes) && i < 5; i++ { pe := pes[i] du := pe.Duration ds := "" hours := int(du / time.Hour) if hours > 0 { ds += fmt.Sprintf("%dh ", hours) du -= time.Duration(hours) * time.Hour } minutes := int(du / time.Minute) if minutes > 0 || len(ds) > 0 { ds += fmt.Sprintf("%dm ", minutes) du -= time.Duration(minutes) * time.Minute } seconds := int(du / time.Second) ds += fmt.Sprintf("%ds", seconds) messageText += fmt.Sprintf("**%s**: %s\n", pe.Name, ds) } service.SendMessage(message.Channel(), messageText) } } }
// Message handler. func (p *livePlugin) Message(bot *bruxism.Bot, service bruxism.Service, message bruxism.Message) { defer bruxism.MessageRecover() if !service.IsMe(message) { messageChannel := message.Channel() if bruxism.MatchesCommand(service, "live", message) { ticks := "" if service.Name() == bruxism.DiscordServiceName { ticks = "`" } _, parts := bruxism.ParseCommand(service, message) if len(parts) == 0 { service.SendMessage(messageChannel, fmt.Sprintf("Incorrect command. eg: %s%slive [add|remove|list] <%s>%s", ticks, service.CommandPrefix(), "UCGmC0A8mEAPdlELQdP9xJbw", ticks)) } switch parts[0] { case "list": if !service.IsModerator(message) { service.SendMessage(messageChannel, "I'm sorry, you must be the channel owner to list live announcements.") return } list := []string{} p.Lock() for ytChannel := range p.ChannelToYouTubeChannels[messageChannel] { list = append(list, fmt.Sprintf("%s (%s)", p.ytLiveChannel.ChannelName(ytChannel), ytChannel)) } p.Unlock() if len(list) == 0 { service.SendMessage(messageChannel, "No Channels are being announced.") } else { service.SendMessage(messageChannel, fmt.Sprintf("Currently announcing: %s", strings.Join(list, ","))) } case "add": if !service.IsModerator(message) { service.SendMessage(messageChannel, "I'm sorry, you must be the channel owner to add live announcements.") return } if len(parts) != 2 || len(parts[1]) != 24 { service.SendMessage(messageChannel, fmt.Sprintf("Incorrect Channel ID. eg: %s%slive %s %s%s", ticks, service.CommandPrefix(), parts[0], "UCGmC0A8mEAPdlELQdP9xJbw", ticks)) return } err := p.monitor(messageChannel, parts[1]) if err != nil { service.SendMessage(messageChannel, fmt.Sprintf("Could not add Channel ID. %s", err)) return } service.SendMessage(messageChannel, fmt.Sprintf("Messages will be sent here when %s goes live.", p.ytLiveChannel.ChannelName(parts[1]))) case "remove": if !service.IsModerator(message) { service.SendMessage(messageChannel, "I'm sorry, you must be the channel owner to remove live announcements.") return } if len(parts) != 2 || len(parts[1]) != 24 { service.SendMessage(messageChannel, fmt.Sprintf("Incorrect Channel ID. eg: %s%slive %s %s%s", ticks, service.CommandPrefix(), parts[0], "UCGmC0A8mEAPdlELQdP9xJbw", ticks)) return } p.Lock() delete(p.ChannelToYouTubeChannels[messageChannel], parts[1]) delete(p.youTubeChannelToChannels[parts[1]], messageChannel) p.Unlock() service.SendMessage(messageChannel, fmt.Sprintf("Messages will no longer be sent here when %s goes live.", p.ytLiveChannel.ChannelName(parts[1]))) } } } }
// StatsCommand returns bot statistics. func StatsCommand(bot *bruxism.Bot, service bruxism.Service, message bruxism.Message, command string, parts []string) { stats := runtime.MemStats{} runtime.ReadMemStats(&stats) w := &tabwriter.Writer{} buf := &bytes.Buffer{} w.Init(buf, 0, 4, 0, ' ', 0) if service.Name() == bruxism.DiscordServiceName { fmt.Fprintf(w, "```\n") } fmt.Fprintf(w, "Bruxism: \t%s\n", bruxism.VersionString) if service.Name() == bruxism.DiscordServiceName { fmt.Fprintf(w, "Discordgo: \t%s\n", discordgo.VERSION) } fmt.Fprintf(w, "Go: \t%s\n", runtime.Version()) fmt.Fprintf(w, "Uptime: \t%s\n", getDurationString(time.Now().Sub(statsStartTime))) fmt.Fprintf(w, "Memory used: \t%s / %s (%s garbage collected)\n", humanize.Bytes(stats.Alloc), humanize.Bytes(stats.Sys), humanize.Bytes(stats.TotalAlloc)) fmt.Fprintf(w, "Concurrent tasks: \t%d\n", runtime.NumGoroutine()) if service.Name() == bruxism.DiscordServiceName { discord := service.(*bruxism.Discord) fmt.Fprintf(w, "Connected servers: \t%d\n", service.ChannelCount()) if len(discord.Sessions) > 1 { shards := 0 for _, s := range discord.Sessions { if s.DataReady { shards++ } } if shards == len(discord.Sessions) { fmt.Fprintf(w, "Shards: \t%d\n", shards) } else { fmt.Fprintf(w, "Shards: \t%d (%d connected)\n", len(discord.Sessions), shards) } guild, err := discord.Channel(message.Channel()) if err == nil { id, err := strconv.Atoi(guild.ID) if err == nil { fmt.Fprintf(w, "Current shard: \t%d\n", ((id>>22)%len(discord.Sessions) + 1)) } } } } else { fmt.Fprintf(w, "Connected channels: \t%d\n", service.ChannelCount()) } plugins := bot.Services[service.Name()].Plugins names := []string{} for _, plugin := range plugins { names = append(names, plugin.Name()) sort.Strings(names) } for _, name := range names { stats := plugins[name].Stats(bot, service, message) for _, stat := range stats { fmt.Fprint(w, stat) } } if service.Name() == bruxism.DiscordServiceName { fmt.Fprintf(w, "\n```") } w.Flush() out := buf.String() end := "" if IsSeptapus { end += "Septapus community: <https://discord.gg/HWN9pwj>\nPatreon: <https://www.patreon.com/iopred>\nBuilt with love by iopred." } if service.SupportsMultiline() { if end != "" { out += "\n" + end } service.SendMessage(message.Channel(), out) } else { service.SendMessage(message.Channel(), strings.Join(strings.Split(out, "\n"), " ")) service.SendMessage(message.Channel(), strings.Join(strings.Split(end, "\n"), " ")) } }
func (p *chartPlugin) messageFunc(bot *bruxism.Bot, service bruxism.Service, message bruxism.Message) { if service.IsMe(message) { return } if bruxism.MatchesCommand(service, "chart", message) { query, parts := bruxism.ParseCommand(service, message) if len(parts) == 0 { service.SendMessage(message.Channel(), fmt.Sprintf("Invalid chart eg: %s", p.randomChart(service))) return } start, end := 0.5, 0.5 switch parts[0] { case "up": start, end = 0, 1 case "down": start, end = 1, 0 case "flat": case "straight": default: service.SendMessage(message.Channel(), fmt.Sprintf("Invalid chart direction. eg: %s", p.randomChart(service))) return } axes := strings.Split(query[len(parts[0]):], ",") if len(axes) != 2 { service.SendMessage(message.Channel(), fmt.Sprintf("Invalid chart axis labels eg: %s", p.randomChart(service))) return } pl, err := plot.New() if err != nil { service.SendMessage(message.Channel(), fmt.Sprintf("Error making chart, sorry! eg: %s", p.randomChart(service))) return } service.Typing(message.Channel()) pl.Y.Label.Text = axes[0] pl.X.Label.Text = axes[1] num := 5 + rand.Intn(15) start *= float64(num) end *= float64(num) pts := make(plotter.XYs, num) for i := range pts { pts[i].X = float64(i) + rand.Float64()*0.5 - 0.2 pts[i].Y = start + float64(end-start)/float64(num-1)*float64(i) + rand.Float64()*0.5 - 0.25 } pl.X.Tick.Label.Color = color.Transparent pl.Y.Tick.Label.Color = color.Transparent pl.X.Min = -0.5 pl.X.Max = float64(num) + 0.5 pl.Y.Min = -0.5 pl.Y.Max = float64(num) + 0.5 lpLine, lpPoints, err := plotter.NewLinePoints(pts) if err != nil { service.SendMessage(message.Channel(), fmt.Sprintf("Sorry %s, there was a problem creating your chart.", message.UserName())) } lpLine.Color = plotutil.Color(rand.Int()) lpLine.Width = vg.Points(1 + 0.5*rand.Float64()) lpLine.Dashes = plotutil.Dashes(rand.Int()) lpPoints.Shape = plotutil.Shape(rand.Int()) lpPoints.Color = lpLine.Color pl.Add(lpLine, lpPoints) w, err := pl.WriterTo(320, 240, "png") if err != nil { service.SendMessage(message.Channel(), fmt.Sprintf("Sorry %s, there was a problem creating your chart.", message.UserName())) return } b := &bytes.Buffer{} w.WriteTo(b) go func() { if service.Name() == bruxism.DiscordServiceName { discord := service.(*bruxism.Discord) p, err := discord.UserChannelPermissions(message.UserID(), message.Channel()) if err == nil && p&discordgo.PermissionAttachFiles != 0 { service.SendFile(message.Channel(), "chart.png", b) return } } url, err := bot.UploadToImgur(b, "chart.png") if err != nil { service.SendMessage(message.Channel(), fmt.Sprintf("Sorry %s, there was a problem uploading the chart to imgur.", message.UserName())) log.Println("Error uploading chart: ", err) return } if service.Name() == bruxism.DiscordServiceName { service.SendMessage(message.Channel(), fmt.Sprintf("Here's your chart <@%s>: %s", message.UserID(), url)) } else { service.SendMessage(message.Channel(), fmt.Sprintf("Here's your chart %s: %s", message.UserName(), url)) } }() } }
func avatarLoadFunc(bot *bruxism.Bot, service bruxism.Service, data []byte) error { if service.Name() != bruxism.DiscordServiceName { panic("DiscordAvatar plugin only supports Discord.") } return nil }