From 73b57750a074e4060db16604705aa8ab1605c4e6 Mon Sep 17 00:00:00 2001
From: TomatoCake <60300461+DEVTomatoCake@users.noreply.github.com>
Date: Fri, 23 Aug 2024 18:01:35 +0200
Subject: [PATCH] Fix updating vanity URL & retrieving widget.json

---
 src/api/routes/guilds/#guild_id/vanity-url.ts | 21 +++++++------------
 .../routes/guilds/#guild_id/widget.json.ts    |  5 +++++
 2 files changed, 12 insertions(+), 14 deletions(-)

diff --git a/src/api/routes/guilds/#guild_id/vanity-url.ts b/src/api/routes/guilds/#guild_id/vanity-url.ts
index a64ae2c9..34595ca6 100644
--- a/src/api/routes/guilds/#guild_id/vanity-url.ts
+++ b/src/api/routes/guilds/#guild_id/vanity-url.ts
@@ -1,17 +1,17 @@
 /*
 	Spacebar: A FOSS re-implementation and extension of the Discord.com backend.
 	Copyright (C) 2023 Spacebar and Spacebar Contributors
-	
+
 	This program is free software: you can redistribute it and/or modify
 	it under the terms of the GNU Affero General Public License as published
 	by the Free Software Foundation, either version 3 of the License, or
 	(at your option) any later version.
-	
+
 	This program is distributed in the hope that it will be useful,
 	but WITHOUT ANY WARRANTY; without even the implied warranty of
 	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 	GNU Affero General Public License for more details.
-	
+
 	You should have received a copy of the GNU Affero General Public License
 	along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
@@ -108,30 +108,23 @@ router.patch(
 		});
 
 		if (!guild.features.includes("ALIASABLE_NAMES")) {
-			await Invite.update(
-				{ guild_id },
-				{
-					code: code,
-				},
-			);
-
-			return res.json({ code });
+			await Invite.delete({ guild_id, vanity_url: true });
 		}
 
 		await Invite.create({
 			vanity_url: true,
-			code: code,
+			code,
 			temporary: false,
 			uses: 0,
 			max_uses: 0,
 			max_age: 0,
 			created_at: new Date(),
-			expires_at: new Date(),
 			guild_id: guild_id,
 			channel_id: id,
+			flags: 0,
 		}).save();
 
-		return res.json({ code: code });
+		return res.json({ code });
 	},
 );
 
diff --git a/src/api/routes/guilds/#guild_id/widget.json.ts b/src/api/routes/guilds/#guild_id/widget.json.ts
index 6c272e81..5d9f28c2 100644
--- a/src/api/routes/guilds/#guild_id/widget.json.ts
+++ b/src/api/routes/guilds/#guild_id/widget.json.ts
@@ -57,6 +57,10 @@ router.get(
 			where: { id: guild_id },
 			select: {
 				channel_ordering: true,
+				widget_channel_id: true,
+				widget_enabled: true,
+				presence_count: true,
+				name: true,
 			},
 		});
 		if (!guild.widget_enabled) throw DiscordApiErrors.EMBED_DISABLED;
@@ -82,6 +86,7 @@ router.get(
 				created_at: new Date(),
 				guild_id,
 				channel_id: guild.widget_channel_id,
+				flags: 0,
 			}).save();
 		}