From 15d9bed12c6fcfe912ddcb491cd16d8b8b551288 Mon Sep 17 00:00:00 2001
From: soyoka <55011525+soy0ka@users.noreply.github.com>
Date: Thu, 6 Nov 2025 16:12:31 +0900
Subject: [PATCH] feat: add force option to updateServer query

---
 utils/Query.ts | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/utils/Query.ts b/utils/Query.ts
index 4e1dddb..dcbba38 100644
--- a/utils/Query.ts
+++ b/utils/Query.ts
@@ -780,11 +780,18 @@ async function updatedServer(id: string, data: ManageServer) {
  * @returns 2 - Limit of 10M servers
  * @returns 3 - Limit of 200 shards
  */
-async function updateServer(id: string, servers: number, shards: number) {
+async function updateServer(
+	id: string,
+	servers: number,
+	shards: number,
+	force = false
+): Promise<number | void> {
 	const bot = await get.bot.load(id)
-	if (bot.servers < 10000 && servers >= 10000) return 1
-	else if (bot.servers < 1000000 && servers >= 1000000) return 2
-	if (bot.shards < 200 && shards >= 200) return 3
+	if (!force) {
+		if (bot.servers < 10000 && servers >= 10000) return 1
+		else if (bot.servers < 1000000 && servers >= 1000000) return 2
+		if (bot.shards < 200 && shards >= 200) return 3
+	}
 	await knex('bots')
 		.update({
 			servers: servers === undefined ? bot.servers : servers,