feat(hololive): stream pinning

1 files changed, 27 insertions, 3 deletions

diff --git a/src/lib/Database/userConfiguration.ts b/src/lib/Database/userConfiguration.ts
index 3a01411e..f1e3ef95 100644
--- a/src/lib/Database/userConfiguration.ts
+++ b/src/lib/Database/userConfiguration.ts
@@ -2,14 +2,16 @@ import supabase from './supabase';
 
 interface UserConfiguration {
 	user_id: number;
-	configuration: JSON;
+	configuration: object;
 	created_at: string;
 	updated_at: string;
+	pinned_hololive_streams: string[];
 }
 
 interface NewUserConfiguration {
-	configuration: JSON;
+	configuration: object;
 	updated_at?: string;
+	pinned_hololive_streams?: string[];
 }
 
 export const getUserConfiguration = async (userId: number) => {
@@ -24,13 +26,17 @@ export const getUserConfiguration = async (userId: number) => {
 };
 
 export const setUserConfiguration = async (userId: number, configuration: NewUserConfiguration) => {
+	const userConfiguration = await getUserConfiguration(userId);
 	const { data, error } = await supabase
 		.from('user_configuration')
 		.upsert(
 			{
 				user_id: userId,
 				configuration: configuration.configuration,
-				updated_at: configuration.updated_at || new Date().toISOString()
+				updated_at: configuration.updated_at || new Date().toISOString(),
+				pinned_hololive_streams:
+					configuration.pinned_hololive_streams ||
+					(userConfiguration ? userConfiguration.pinned_hololive_streams : [])
 			},
 			{ onConflict: 'user_id' }
 		)
@@ -40,3 +46,21 @@ export const setUserConfiguration = async (userId: number, configuration: NewUse
 
 	return data[0].configuration as UserConfiguration;
 };
+
+export const toggleHololiveStreamPinning = async (userId: number, streamId: string) => {
+	const userConfiguration = await getUserConfiguration(userId);
+
+	if (!userConfiguration) return null;
+
+	const pinnedStreams = userConfiguration.pinned_hololive_streams;
+	const index = pinnedStreams.indexOf(streamId);
+
+	if (index === -1) pinnedStreams.push(streamId);
+	else pinnedStreams.splice(index, 1);
+
+	return setUserConfiguration(userId, {
+		configuration: userConfiguration.configuration,
+		updated_at: new Date().toISOString(),
+		pinned_hololive_streams: pinnedStreams
+	});
+};