1 files changed, 106 insertions, 0 deletions
diff --git a/src/lib/Database/Supabase/userBadges.ts b/src/lib/Database/Supabase/userBadges.ts
new file mode 100644
index 00000000..eef8490c
--- /dev/null
+++ b/src/lib/Database/Supabase/userBadges.ts
@@ -0,0 +1,106 @@
+import { databaseTimeToDate } from '$lib/Utility/time';
+import supabase from '../supabase';
+
+export interface Badge {
+	post?: string;
+	image?: string;
+	description?: string | null;
+	id?: number;
+	time?: string;
+	category?: string | null;
+	hidden?: boolean;
+	source?: string | null;
+	designer?: string | null;
+	shadow_hidden?: boolean;
+	click_count?: number;
+}
+
+export const getUserBadges = async (userId: number): Promise<Badge[]> => {
+	const { data, error } = await supabase.from('user_badges').select('*').eq('user_id', userId);
+
+	if (error) return [];
+
+	return data.sort((a, b) =>
+		databaseTimeToDate((a as Badge).time ?? '').getTime() >
+		databaseTimeToDate((b as Badge).time ?? '').getTime()
+			? -1
+			: 1
+	) as Badge[];
+};
+
+export const addUserBadge = async (userId: number, badge: 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,
+			source,
+			designer
+		});
+	} else {
+		await supabase
+			.from('user_badges')
+			.insert({ user_id: userId, post, image, description, category, hidden, source, designer });
+	}
+};
+
+export const removeUserBadge = async (userId: number, id: number) => {
+	if (!isNaN(id)) await supabase.from('user_badges').delete().eq('id', id).eq('user_id', userId);
+};
+
+export const updateUserBadge = async (userId: number, id: number, badge: Badge) => {
+	if (badge.post === undefined || badge.image === undefined) return;
+
+	await supabase
+		.from('user_badges')
+		.update({
+			post: badge.post,
+			image: badge.image,
+			description: badge.description,
+			category: badge.category,
+			time: badge.time,
+			hidden: badge.hidden,
+			source: badge.source,
+			designer: badge.designer
+		})
+		.eq('id', id)
+		.eq('user_id', userId);
+};
+
+export const renameCategory = async (userId: number, oldName: string, newName: string) =>
+	await supabase
+		.from('user_badges')
+		.update({ category: newName })
+		.eq('category', oldName)
+		.eq('user_id', userId);
+
+export const removeAllUserBadges = async (userId: number) =>
+	await supabase.from('user_badges').delete().eq('user_id', userId);
+
+export const migrateCategory = async (userId: number, oldName: string, newName: string) =>
+	await supabase
+		.from('user_badges')
+		.update({ category: newName })
+		.eq('category', oldName)
+		.eq('user_id', userId);
+
+export const setShadowHidden = async (userId: number, shadowHide: boolean) =>
+	await supabase.from('user_badges').update({ shadow_hidden: shadowHide }).eq('user_id', userId);
+
+export const setShadowHiddenBadge = async (userId: number, id: number, shadowHide: boolean) =>
+	await supabase
+		.from('user_badges')
+		.update({ shadow_hidden: shadowHide })
+		.eq('id', id)
+		.eq('user_id', userId);
+
+export const incrementClickCount = async (id: number) =>
+	await supabase.rpc('user_badges_increment_click_count', { user_badge_id: id });