Use a different short domain for non devs

DuncteBot · Jul 23, 2021 · 3929525 · 3929525
1 parent 8fedcec
commit 3929525
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/src/main/java/ml/duncte123/skybot/utils/AirUtils.java b/src/main/java/ml/duncte123/skybot/utils/AirUtils.java
@@ -346,13 +346,17 @@ public static void setJDAContext(JDA jda) {
         ((JDAImpl) jda).setContext();
     }
 
-    @Nonnull
     public static PendingRequest<String> shortenUrl(String url, String googleKey, ObjectMapper mapper) {
+        return shortenUrl(url, googleKey, mapper, "duncte.bot");
+    }
+
+    @Nonnull
+    public static PendingRequest<String> shortenUrl(String url, String googleKey, ObjectMapper mapper, String prefix) {
         final ObjectNode json = mapper.createObjectNode();
 
         json.set("dynamicLinkInfo",
             mapper.createObjectNode()
-                .put("domainUriPrefix", "duncte.bot")
+                .put("domainUriPrefix", prefix)
                 .put("link", url)
         );
         json.set("suffix",

diff --git a/src/main/kotlin/ml/duncte123/skybot/commands/uncategorized/ShortenCommand.kt b/src/main/kotlin/ml/duncte123/skybot/commands/uncategorized/ShortenCommand.kt
@@ -23,6 +23,7 @@ import ml.duncte123.skybot.objects.command.Command
 import ml.duncte123.skybot.objects.command.CommandContext
 import ml.duncte123.skybot.utils.AirUtils.isURL
 import ml.duncte123.skybot.utils.AirUtils.shortenUrl
+import ml.duncte123.skybot.utils.CommandUtils.isDev
 
 class ShortenCommand : Command() {
 
@@ -42,7 +43,9 @@ class ShortenCommand : Command() {
             return
         }
 
-        shortenUrl(args[0], ctx.config.apis.googl, ctx.variables.jackson).async(
+        val prefix = if (isDev(ctx.author)) "duncte.bot" else "dunctebot.page.link"
+
+        shortenUrl(args[0], ctx.config.apis.googl, ctx.variables.jackson, prefix).async(
             {
                 sendMsg(ctx, "Here is your shortened url: <$it>")
             },