gavalink/example/main.go
2018-11-05 17:24:52 -05:00

172 lines
3.3 KiB
Go

package main
import (
"flag"
"log"
"os"
"os/signal"
"strconv"
"strings"
"syscall"
"github.com/bwmarrin/discordgo"
"github.com/foxbot/gavalink"
)
var token string
var lavalink *gavalink.Lavalink
var player *gavalink.Player
func init() {
flag.StringVar(&token, "token", "", "token=unprefixed token")
}
func main() {
flag.Parse()
if token == "" {
panic("no token specified!")
}
token = "Bot " + token
dg, err := discordgo.New(token)
if err != nil {
panic(err)
}
dg.SyncEvents = false
dg.AddHandler(ready)
dg.AddHandler(messageCreate)
dg.AddHandler(voiceServerUpdate)
err = dg.Open()
if err != nil {
panic(err)
}
sc := make(chan os.Signal, 1)
signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill)
<-sc
dg.Close()
}
func ready(s *discordgo.Session, event *discordgo.Ready) {
log.Println("discordgo ready!")
s.UpdateStatus(0, "gavalink")
lavalink = gavalink.NewLavalink("1", event.User.ID)
err := lavalink.AddNodes(gavalink.NodeConfig{
REST: "http://localhost:2333",
WebSocket: "ws://localhost:2334",
Password: "youshallnotpass",
})
if err != nil {
log.Println(err)
}
}
func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
if m.Author.ID == s.State.User.ID {
return
}
if m.Content == "~>>join" {
c, err := s.State.Channel(m.ChannelID)
if err != nil {
log.Println("fail find channel")
return
}
g, err := s.State.Guild(c.GuildID)
if err != nil {
log.Println("fail find guild")
return
}
for _, vs := range g.VoiceStates {
if vs.UserID == m.Author.ID {
log.Println("trying to connect to channel")
err = s.ChannelVoiceJoinManual(c.GuildID, vs.ChannelID, false, false)
if err != nil {
log.Println(err)
} else {
log.Println("channel voice join succeeded")
}
}
}
} else if strings.HasPrefix(m.Content, "~>>play") {
query := m.Content[8:]
node, err := lavalink.BestNode()
if err != nil {
log.Println(err)
return
}
tracks, err := node.LoadTracks(query)
if err != nil {
log.Println(err)
return
}
if tracks.Type != gavalink.TrackLoaded {
log.Println("weird tracks type", tracks.Type)
}
track := tracks.Tracks[0].Data
err = player.Play(track)
if err != nil {
log.Println(err)
}
} else if m.Content == "~>>stop" {
err := player.Stop()
if err != nil {
log.Println(err)
}
} else if m.Content == "~>>pause" {
err := player.Pause(!player.Paused())
if err != nil {
log.Println(err)
}
} else if strings.HasPrefix(m.Content, "~>>volume") {
query := m.Content[10:]
vol, err := strconv.Atoi(query)
if err != nil {
log.Println(err)
return
}
err = player.Volume(vol)
if err != nil {
log.Println(err)
}
}
}
func voiceServerUpdate(s *discordgo.Session, event *discordgo.VoiceServerUpdate) {
log.Println("received VSU")
vsu := gavalink.VoiceServerUpdate{
Endpoint: event.Endpoint,
GuildID: event.GuildID,
Token: event.Token,
}
if p, err := lavalink.GetPlayer(event.GuildID); err == nil {
err = p.Forward(s.State.SessionID, vsu)
if err != nil {
log.Println(err)
}
return
}
node, err := lavalink.BestNode()
if err != nil {
log.Println(err)
return
}
handler := new(gavalink.DummyEventHandler)
player, err = node.CreatePlayer(event.GuildID, s.State.SessionID, vsu, handler)
if err != nil {
log.Println(err)
return
}
}