1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
package main
import (
"flag"
"fmt"
"log"
"os"
"regexp"
"time"
"github.com/bwmarrin/discordgo"
_ "github.com/joho/godotenv/autoload"
"within.website/x/internal/pvfm/bot"
"within.website/x/internal/pvfm/commands/source"
)
type aerial struct {
cs *bot.CommandSet
s *discordgo.Session
}
func (a *aerial) Handle(s *discordgo.Session, m *discordgo.MessageCreate) {
err := a.cs.Run(s, m.Message)
if err != nil {
log.Println(err)
}
}
var (
token = os.Getenv("TOKEN")
youtubeSpamRoomID = os.Getenv("DISCORD_YOUTUBESPAM_ROOMID")
musicLinkRegex = regexp.MustCompile(`(.*)((http(s?):\/\/(www\.)?soundcloud.com\/.*)|(http(s?):\/\/(www\.)?youtube.com\/.*)|(http(s?):\/\/(www\.)?youtu.be\/.*))(.*)|(.*)http(s?):\/\/(www\.)?mixcloud.com\/.*`)
)
func main() {
flag.Parse()
dg, err := discordgo.New("Bot " + token)
if err != nil {
log.Fatal(err)
}
a := &aerial{
cs: bot.NewCommandSet(),
s: dg,
}
a.cs.Prefix = ";"
a.cs.AddCmd("np", "shows radio station statistics for Ponyville FM", bot.NoPermissions, stats)
a.cs.AddCmd("stats", "shows radio station statistics for Ponyville FM", bot.NoPermissions, stats)
a.cs.AddCmd("dj", "shows which DJ is up next on Ponyville FM", bot.NoPermissions, stats)
a.cs.AddCmd("schedule", "shows the future radio schedule for Ponyville FM", bot.NoPermissions, schedule)
a.cs.AddCmd("source", "source code information", bot.NoPermissions, source.Source)
a.cs.AddCmd("time", "shows the current bot time", bot.NoPermissions, curTime)
a.cs.AddCmd("streams", "shows the different Ponyville FM stream links", bot.NoPermissions, streams)
a.cs.AddCmd("servers", "shows the different Ponyville FM stream links", bot.NoPermissions, streams)
dg.AddHandler(a.Handle)
dg.AddHandler(pesterLink)
dg.AddHandler(messageCreate)
err = dg.Open()
if err != nil {
log.Fatal(err)
}
log.Println("ready")
<-make(chan struct{})
}
// This function will be called (due to AddHandler above) every time a new
// message is created on any channel that the autenticated bot has access to.
func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
// Print message to stdout.
fmt.Printf("%20s %20s %20s %20s > %s\n", m.Author.ID, m.ChannelID, time.Now().Format(time.Stamp), m.Author.Username, m.Content)
}
|
