diff --git a/.drone.yml b/.drone.yml index 2024244..a04fdb2 100644 --- a/.drone.yml +++ b/.drone.yml @@ -9,6 +9,8 @@ steps: environment: DISCORD_TOKEN: from_secret: discord_token + DISCORD_BOT_CHANNEL: + from_secret: discord_bot_channel commands: - go get - go build diff --git a/buttsbot.go b/buttsbot.go index eee62a0..b1fc335 100644 --- a/buttsbot.go +++ b/buttsbot.go @@ -13,15 +13,20 @@ import ( var Session, _ = discordgo.New() var testPtr = flag.Bool("test", false, "Test mode") var bot_token string +var bot_channel string func init() { + flag.StringVar(&bot_token, "t", "", "Discord Authentication Token") + flag.StringVar(&bot_channel, "bc", "", "Bot status channel") + flag.Parse() if os.Getenv("DISCORD_TOKEN") != "" { Session.Token = "Bot " + os.Getenv("DISCORD_TOKEN") } else { - flag.StringVar(&bot_token, "t", "", "Discord Authentication Token") - flag.Parse() Session.Token = "Bot " + bot_token } + if os.Getenv("DISCORD_BOT_CHANNEL") != "" { + bot_channel = os.Getenv("DISCORD_BOT_CHANNEL") + } } func main() { @@ -43,10 +48,12 @@ 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