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...") +}