From cd0ee8157baf9b70a417ebb802cb708157e9bd07 Mon Sep 17 00:00:00 2001 From: "Gustavo \"Guz\" L. de Mello" Date: Tue, 13 Aug 2024 21:33:11 -0300 Subject: [PATCH] feat(cli): cli flags --- main.go | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/main.go b/main.go index 146c0d1..ca1b498 100644 --- a/main.go +++ b/main.go @@ -4,12 +4,27 @@ import ( "dislate/internals/discord" "dislate/internals/guilddb" "dislate/internals/translator" + "flag" "log" "os" "os/signal" "syscall" ) +type TranslationProvider string + +const ( + GOOGLE_TRANSLATE TranslationProvider = "google-translate" +) + +var translation_provider = flag.String("tprovider", string(GOOGLE_TRANSLATE), "Translation provider") +var database_file = flag.String("db", "file:./guild.db", "SQLite database file/location") +var discord_token = flag.String("token", os.Getenv("DISCORD_TOKEN"), "Discord bot authentication token") + +func init() { + flag.Parse() +} + func main() { log.Printf("Hello, world") @@ -31,7 +46,7 @@ func main() { return } - bot, err := discord.NewBot(os.Getenv(""), db, translator.NewMockTranslator()) + bot, err := discord.NewBot(*discord_token, db, translator.NewMockTranslator()) if err != nil { log.Printf("ERROR: failed to create discord bot: %s", err) return