From e53143bd64cd1e9a0a10a6150342a3b9651d2d8b Mon Sep 17 00:00:00 2001 From: Aleksei Gryczewski Date: Wed, 29 Jun 2022 18:22:58 -0400 Subject: [PATCH] Automated the token selection process. --- src/main/java/dev/salmonllama/fsbot/Main.java | 12 +++++++++++- .../dev/salmonllama/fsbot/config/SecretManager.java | 1 + 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/src/main/java/dev/salmonllama/fsbot/Main.java b/src/main/java/dev/salmonllama/fsbot/Main.java index 27acc71..3a65bf9 100644 --- a/src/main/java/dev/salmonllama/fsbot/Main.java +++ b/src/main/java/dev/salmonllama/fsbot/Main.java @@ -28,7 +28,17 @@ public class Main { FSDB.init(); - new DiscordApiBuilder().setToken(SecretManager.DISCORD_TOKEN.getPlainText()).login().thenAccept(api -> { + String token; + if (System.getenv("ENVIRONMENT") != null) + { + token = SecretManager.DISCORD_TOKEN_PROD.getPlainText(); + } + else + { + token = SecretManager.DISCORD_TOKEN.getPlainText(); + } + + new DiscordApiBuilder().setToken(token).login().thenAccept(api -> { @SuppressWarnings("unused") Guthix guthix = new Guthix(api); diff --git a/src/main/java/dev/salmonllama/fsbot/config/SecretManager.java b/src/main/java/dev/salmonllama/fsbot/config/SecretManager.java index 00226ac..8a8f2c2 100644 --- a/src/main/java/dev/salmonllama/fsbot/config/SecretManager.java +++ b/src/main/java/dev/salmonllama/fsbot/config/SecretManager.java @@ -6,6 +6,7 @@ import com.google.cloud.secretmanager.v1.SecretManagerServiceClient; public enum SecretManager { DISCORD_TOKEN ("projects/fashionscapers-212707/secrets/fs_discord_token/versions/1"), + DISCORD_TOKEN_PROD ("projects/fashionscapers-212707/secrets/fs_discord_token_prod/versions/1"), IMGUR_ID ("projects/fashionscapers-212707/secrets/fs_imgur_client_id/versions/1"), IMGUR_BEARER ("projects/fashionscapers-212707/secrets/fs_imgur_bearer_token/versions/1") ;