Message bot channel during startup and testing.
parent
944d6137d4
commit
c6151323ee
|
@ -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
|
||||
|
|
11
buttsbot.go
11
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
|
||||
|
|
Loading…
Reference in New Issue