From 7fda4dc333b4942832cce5a76a5532f3d2ac528d Mon Sep 17 00:00:00 2001
From: RTAkland <me@rtast.cn>
Date: Fri, 11 Oct 2024 11:08:18 +0800
Subject: [PATCH] chore: add api rtast key

---
 src/main/kotlin/cn/rtast/fancybot/Const.kt                    | 3 ++-
 .../cn/rtast/fancybot/commands/misc/ShortLinkCommand.kt       | 4 +++-
 src/main/kotlin/cn/rtast/fancybot/entity/Config.kt            | 1 +
 src/main/kotlin/cn/rtast/fancybot/util/file/ConfigManager.kt  | 1 +
 src/main/resources/config.json                                | 3 ++-
 5 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/main/kotlin/cn/rtast/fancybot/Const.kt b/src/main/kotlin/cn/rtast/fancybot/Const.kt
index 38dde72..144d748 100644
--- a/src/main/kotlin/cn/rtast/fancybot/Const.kt
+++ b/src/main/kotlin/cn/rtast/fancybot/Const.kt
@@ -78,7 +78,8 @@ val DEFAULT_CONFIG = Config(
     cloudflareR2SecretKey = "222",
     cloudflareR2BucketName = "fancybot",
     cloudflareR2PublicUrl = "https://114514.com",
-    apiSpaceKey = "114514"
+    apiSpaceKey = "114514",
+    apiRtastKey = "114514"
 )
 
 val configManager = ConfigManager()
diff --git a/src/main/kotlin/cn/rtast/fancybot/commands/misc/ShortLinkCommand.kt b/src/main/kotlin/cn/rtast/fancybot/commands/misc/ShortLinkCommand.kt
index 34e3eeb..f6def79 100644
--- a/src/main/kotlin/cn/rtast/fancybot/commands/misc/ShortLinkCommand.kt
+++ b/src/main/kotlin/cn/rtast/fancybot/commands/misc/ShortLinkCommand.kt
@@ -9,6 +9,7 @@ package cn.rtast.fancybot.commands.misc
 
 import cn.rtast.fancybot.API_RTAST_URL
 import cn.rtast.fancybot.annotations.CommandDescription
+import cn.rtast.fancybot.configManager
 import cn.rtast.fancybot.entity.shortlink.ShortLinkPayload
 import cn.rtast.fancybot.entity.shortlink.ShortLinkResponse
 import cn.rtast.fancybot.util.Http
@@ -26,7 +27,8 @@ class ShortLinkCommand : BaseCommand() {
         fun String.makeShortLink(): String {
             val shortLink = Http.post<ShortLinkResponse>(
                 "$API_RTAST_URL/api/sl",
-                ShortLinkPayload(this).toJson()
+                ShortLinkPayload(this).toJson(),
+                params = mapOf("key" to configManager.apiRtastKey)
             )
             return "$API_RTAST_URL/${shortLink.id}"
         }
diff --git a/src/main/kotlin/cn/rtast/fancybot/entity/Config.kt b/src/main/kotlin/cn/rtast/fancybot/entity/Config.kt
index 0106aa2..33db765 100644
--- a/src/main/kotlin/cn/rtast/fancybot/entity/Config.kt
+++ b/src/main/kotlin/cn/rtast/fancybot/entity/Config.kt
@@ -46,4 +46,5 @@ data class Config(
     val cloudflareR2BucketName: String,
     val cloudflareR2PublicUrl: String,
     val apiSpaceKey: String,
+    val apiRtastKey: String
 )
\ No newline at end of file
diff --git a/src/main/kotlin/cn/rtast/fancybot/util/file/ConfigManager.kt b/src/main/kotlin/cn/rtast/fancybot/util/file/ConfigManager.kt
index 55205f7..dc6db6d 100644
--- a/src/main/kotlin/cn/rtast/fancybot/util/file/ConfigManager.kt
+++ b/src/main/kotlin/cn/rtast/fancybot/util/file/ConfigManager.kt
@@ -47,4 +47,5 @@ class ConfigManager : JsonFileHandler<Config>("config.json") {
     val cloudflareR2BucketName = config.cloudflareR2BucketName
     val cloudflareR2PublicUrl = config.cloudflareR2PublicUrl
     val apiSpaceKey = config.apiSpaceKey
+    val apiRtastKey = config.apiRtastKey
 }
\ No newline at end of file
diff --git a/src/main/resources/config.json b/src/main/resources/config.json
index b27a920..1ca9e8c 100644
--- a/src/main/resources/config.json
+++ b/src/main/resources/config.json
@@ -37,5 +37,6 @@
   "cloudflareR2SecretKey": "222",
   "cloudflareR2BucketName": "fancybot",
   "cloudflareR2PublicUrl": "https://114514.com",
-  "apiSpaceKey": "114514"
+  "apiSpaceKey": "114514",
+  "apiRtastKey": "114514"
 }
\ No newline at end of file