forked from diamondburned/arikawa
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
61 lines (47 loc) · 1.34 KB
/
main.go
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
// Package main demonstrates a bare simple bot without a state cache. It logs
// all messages it sees into stderr.
package main
import (
"context"
"log"
"os"
"github.com/hi117/arikawa/v3/gateway"
"github.com/hi117/arikawa/v3/session/shard"
"github.com/hi117/arikawa/v3/state"
)
// To run, do `BOT_TOKEN="TOKEN HERE" go run .`
func main() {
var token = os.Getenv("BOT_TOKEN")
if token == "" {
log.Fatalln("No $BOT_TOKEN given.")
}
newShard := state.NewShardFunc(func(m *shard.Manager, s *state.State) {
// Add the needed Gateway intents.
s.AddIntents(gateway.IntentGuildMessages)
s.AddIntents(gateway.IntentDirectMessages)
s.AddHandler(func(c *gateway.MessageCreateEvent) {
_, shardIx := m.FromGuildID(c.GuildID)
log.Println(c.Author.Tag(), "sent", c.Content, "on shard", shardIx)
})
})
m, err := shard.NewManager("Bot "+token, newShard)
if err != nil {
log.Fatalln("failed to create shard manager:", err)
}
if err := m.Open(context.Background()); err != nil {
log.Fatalln("failed to connect shards:", err)
}
defer m.Close()
var shardNum int
m.ForEach(func(s shard.Shard) {
state := s.(*state.State)
u, err := state.Me()
if err != nil {
log.Fatalln("failed to get myself:", err)
}
log.Printf("Shard %d/%d started as %s", shardNum, m.NumShards()-1, u.Tag())
shardNum++
})
// Block forever.
select {}
}