feat(badges): source and designer fields

1 files changed, 18 insertions, 4 deletions

diff --git a/src/lib/Database/userBadges.ts b/src/lib/Database/userBadges.ts
index c6cb4919..36bf7955 100644
--- a/src/lib/Database/userBadges.ts
+++ b/src/lib/Database/userBadges.ts
@@ -9,6 +9,8 @@ export interface Badge {
 	time?: string;
 	category?: string;
 	hidden?: boolean;
+	source: string;
+	designer: string;
 }
 
 export const getUserBadges = async (userId: number): Promise<Badge[]> => {
@@ -25,18 +27,28 @@ export const getUserBadges = async (userId: number): Promise<Badge[]> => {
 };
 
 export const addUserBadge = async (userId: number, badge: Badge) => {
-	const { post, image, description, time, category, hidden } = badge;
+	const { post, image, description, time, category, hidden, source, designer } = badge;
 
 	if (post === undefined || image === undefined) return;
 
 	if (time) {
 		await supabase
 			.from('user_badges')
-			.insert({ user_id: userId, post, image, description, time, category, hidden });
+			.insert({
+				user_id: userId,
+				post,
+				image,
+				description,
+				time,
+				category,
+				hidden,
+				source,
+				designer
+			});
 	} else {
 		await supabase
 			.from('user_badges')
-			.insert({ user_id: userId, post, image, description, category, hidden });
+			.insert({ user_id: userId, post, image, description, category, hidden, source, designer });
 	}
 };
 
@@ -55,7 +67,9 @@ export const updateUserBadge = async (userId: number, id: number, badge: Badge)
 			description: badge.description,
 			category: badge.category,
 			time: badge.time,
-			hidden: badge.hidden
+			hidden: badge.hidden,
+			source: badge.source,
+			designer: badge.designer
 		})
 		.eq('id', id)
 		.eq('user_id', userId);