From d2ef31b670296e471466607f226118d98004c8ed Mon Sep 17 00:00:00 2001
From: spiral <spiral@spiral.sh>
Date: Sun, 19 Mar 2023 11:18:09 -0400
Subject: [PATCH] add token2 to config

---
 lib/libpk/src/_config.rs                 |  3 +++
 services/api/src/middleware/ratelimit.rs | 12 +++++++-----
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/lib/libpk/src/_config.rs b/lib/libpk/src/_config.rs
index f49db243..ddf6af7a 100644
--- a/lib/libpk/src/_config.rs
+++ b/lib/libpk/src/_config.rs
@@ -31,6 +31,9 @@ pub struct ApiConfig {
     pub ratelimit_redis_addr: Option<String>,
 
     pub remote_url: String,
+
+    #[serde(default)]
+    pub temp_token2: Option<String>,
 }
 
 fn _metrics_default() -> bool {
diff --git a/services/api/src/middleware/ratelimit.rs b/services/api/src/middleware/ratelimit.rs
index daad827a..ceec27e3 100644
--- a/services/api/src/middleware/ratelimit.rs
+++ b/services/api/src/middleware/ratelimit.rs
@@ -19,9 +19,6 @@ lazy_static::lazy_static! {
     static ref LUA_SCRIPT_SHA: String = sha1_hash(LUA_SCRIPT);
 }
 
-// todo lol
-const TOKEN2: &'static str = "h";
-
 // this is awful but it works
 pub fn ratelimiter<F, T>(f: F) -> FromFnLayer<F, Option<RedisPool>, T> {
     let redis = libpk::config.api.ratelimit_redis_addr.as_ref().map(|val| {
@@ -67,10 +64,15 @@ pub async fn do_request_ratelimited<B>(
         let headers = request.headers().clone();
         let source_ip = header_or_unknown(headers.get("Fly-Client-IP"));
 
+        // https://github.com/rust-lang/rust/issues/53667
         let (rl_key, rate) = if let Some(header) = request.headers().clone().get("X-PluralKit-App")
         {
-            if header == TOKEN2 {
-                ("token2", 20)
+            if let Some(token2) = &libpk::config.api.temp_token2 {
+                if header.to_str().unwrap_or("invalid") == token2 {
+                    ("token2", 20)
+                } else {
+                    (source_ip, 2)
+                }
             } else {
                 (source_ip, 2)
             }