mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-11-19 05:12:48 +00:00
54cadd2f45
This commit refactors the whole package gateway as well as utils/ws
(formerly utils/wsutil) and voice/voicegateway. The new refactor
utilizes a design pattern involving a concurrent loop and an arriving
event channel.
An additional change was made to the way gateway events are typed.
Before, pretty much any type will satisfy a gateway event type, since
the actual type was just interface{}. The new refactor defines a
concrete interface that events can implement:
type Event interface {
Op() OpCode
EventType() EventType
}
Using this interface, the user can easily add custom gateway events
independently of the library without relying on string maps. This adds a
lot of type safety into the library and makes type-switching on Event
types much more reasonable.
Gateway error callbacks are also almost entirely removed in favor of
custom gateway events. A catch-all can easily be added like this:
s.AddHandler(func(err error) {
log.Println("gateway error:, err")
})
62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
// 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/diamondburned/arikawa/v3/gateway"
|
|
"github.com/diamondburned/arikawa/v3/session/shard"
|
|
"github.com/diamondburned/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 {}
|
|
}
|