From 87360672f053ebc08c8cda4871a611dec5e5032b Mon Sep 17 00:00:00 2001 From: pocketjawa Date: Sat, 27 Apr 2019 02:27:24 -0400 Subject: [PATCH] Moved startup status to ready handler. --- buttsbot.go | 21 ++++++++++++++------- 1 file changed, 14 insertions(+), 7 deletions(-) diff --git a/buttsbot.go b/buttsbot.go index b1fc335..17e85eb 100644 --- a/buttsbot.go +++ b/buttsbot.go @@ -38,6 +38,9 @@ func main() { return } + //Handlers + Session.AddHandler(ready) + //Open a connection to Discord err = Session.Open() if err != nil { @@ -47,15 +50,19 @@ func main() { //Wait for a CTRL-C log.Printf("Now running. Press CTRL-C to exit.") - if *testPtr == true { - Session.ChannelMessageSend(bot_channel, "Test complete: discord-buttsbot-go") - log.Printf("Test successful! Now quiting.") - Session.Close() - os.Exit(0) - } - Session.ChannelMessageSend(bot_channel, "This isn't a butt...") sc := make(chan os.Signal, 1) signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill) <-sc Session.Close() } + +func ready(s *discordgo.Session, event *discordgo.Ready) { + if *testPtr == true { + s.ChannelMessageSend(bot_channel, "Test complete: discord-buttsbot-go") + log.Printf("Test successful! Now quiting.") + s.Close() + os.Exit(0) + } + s.UpdateStatus(0, "with butts!") + s.ChannelMessageSend(bot_channel, "This isn't a butt...") +}