attempt to resolve song

api/src/party-types.ts

@@ -1,11 +1,12 @@
 import type { InferSelectModel } from "drizzle-orm";
-import type { party, partyMember, user } from "./db/schema";
+import type { party, partyMember, track, user } from "./db/schema";
 
 export type Party = Omit<InferSelectModel<typeof party>, "data"> & {
 	data: QuizState;
 };
 export type PartyMember = InferSelectModel<typeof partyMember>;
 export type User = InferSelectModel<typeof user>;
+export type Song = InferSelectModel<typeof track>;
 
 export type PartyMemberWithUser = PartyMember & { user: User | null };
 
@@ -24,6 +25,7 @@ export type PartyState = {
 
 export type PartySocketOutgoing =
 	| { type: "ping" }
+	| { type: "subscribe_party"; partyId: string }
 	| { type: "member_payload"; payload: unknown };
 
 type BaseQuestion = {
@@ -32,6 +34,7 @@ type BaseQuestion = {
 	startTimestamp: number;
 	endTimestamp: number;
 	points: number;
+	song?: Song;
 };
 
 export type Question =

api/src/party/audio-question-generator.ts

@@ -1,3 +1,4 @@
+import type { db as Db } from "../db";
 import type { Question } from "../party-types";
 import {
 	buildOptionsWithCorrect,
@@ -8,16 +9,19 @@ import {
 	getTopClusterTracks,
 	type PartyAnalytics,
 	pickRandom,
+	resolveQuestionSong,
 } from "./question-utils";
 
-export function buildAudioMetadataQuestion(
+export async function buildAudioMetadataQuestion(
+	dbClient: typeof Db,
 	analytics: PartyAnalytics,
 	index: number,
-): Question {
+): Promise<Question | null> {
 	type ChoiceQuestion = Extract<Question, { type: "choice" }>;
 	const questions: Array<
 		Omit<ChoiceQuestion, "startTimestamp" | "endTimestamp">
 	> = [];
+	const topSong = await resolveQuestionSong(dbClient, analytics);
 
 	const genreOptions = buildOrderedOptions(
 		getMostSharedGenreNames(analytics),
@@ -30,6 +34,7 @@ export function buildAudioMetadataQuestion(
 			options: genreOptions,
 			correct: 0,
 			points: 10,
+			song: topSong ?? undefined,
 		});
 	}
 
@@ -44,6 +49,7 @@ export function buildAudioMetadataQuestion(
 				options: artistOptions,
 				correct: 0,
 				points: 10,
+				song: topSong ?? undefined,
 			});
 		}
 	}
@@ -62,12 +68,18 @@ export function buildAudioMetadataQuestion(
 				options: trackOptions,
 				correct: 0,
 				points: 10,
+				song: topSong ?? undefined,
 			});
 		}
 	}
 
 	const randomTopTrack = pickRandom(topTracks);
 	if (randomTopTrack) {
+		const randomTrackSong = await resolveQuestionSong(dbClient, analytics, {
+			trackName: randomTopTrack.name,
+			artistNames: randomTopTrack.artists?.map((artist) => artist.name),
+			albumName: randomTopTrack.albumName,
+		});
 		const trackArtists =
 			randomTopTrack.artists?.map((artist) => artist.name) ?? [];
 		const allArtists = topArtists.length > 0 ? topArtists : trackArtists;
@@ -85,6 +97,7 @@ export function buildAudioMetadataQuestion(
 					options: artistOptions,
 					correct: 0,
 					points: 10,
+					song: randomTrackSong ?? topSong ?? undefined,
 				});
 			}
 
@@ -101,6 +114,7 @@ export function buildAudioMetadataQuestion(
 					options: trackNameOptions,
 					correct: 0,
 					points: 10,
+					song: randomTrackSong ?? topSong ?? undefined,
 				});
 			}
 		}
@@ -121,13 +135,14 @@ export function buildAudioMetadataQuestion(
 					options: albumOptions,
 					correct: 0,
 					points: 10,
+					song: randomTrackSong ?? topSong ?? undefined,
 				});
 			}
 		}
 	}
 
 	if (questions.length === 0) {
-		throw new Error("Question not found");
+		return null;
 	}
 
 	const question = questions[index % questions.length];

api/src/party/numeric-question-generator.ts

@@ -10,6 +10,7 @@ import {
 	getQuestionRange,
 	type PartyAnalytics,
 	type PartyQuestionMember,
+	resolveQuestionSong,
 } from "./question-utils";
 
 type NumericQuestion = Omit<
@@ -36,6 +37,9 @@ async function getAlbumReleaseYear({
 				with: { album: true },
 			})
 		: null;
+	const song = await resolveQuestionSong(db, analytics, {
+		trackName: track?.name ?? trackName ?? undefined,
+	});
 	const correct =
 		track?.album?.release_date?.getFullYear() ??
 		new Date().getFullYear() - 1 - index;
@@ -46,6 +50,7 @@ async function getAlbumReleaseYear({
 		correct,
 		range: getQuestionRange(correct, 10, 3),
 		points: 10,
+		song: song ?? undefined,
 	};
 }
 
@@ -60,6 +65,7 @@ async function countTopTrackListeners({
 		where: { name: trackName },
 	});
 	if (!dbTrack) return null;
+	const song = await resolveQuestionSong(db, analytics, { trackName });
 	const memberIds = members.map((m) => m.userId);
 	const entries = await db
 		.select({ userId: topTrackTable.userId })
@@ -77,6 +83,7 @@ async function countTopTrackListeners({
 		correct,
 		range: { min: 0, max: members.length },
 		points: 10,
+		song: song ?? undefined,
 	};
 }
 
@@ -91,6 +98,9 @@ async function countFavouriteArtistListeners({
 		where: { name: artistName },
 	});
 	if (!dbArtist) return null;
+	const song = await resolveQuestionSong(db, analytics, {
+		artistNames: [artistName],
+	});
 	const memberIds = members.map((m) => m.userId);
 	const entries = await db
 		.select({ userId: topArtistTable.userId })
@@ -108,12 +118,13 @@ async function countFavouriteArtistListeners({
 		correct,
 		range: { min: 0, max: members.length },
 		points: 10,
+		song: song ?? undefined,
 	};
 }
 
 export async function buildNumericQuestion(
 	input: BuildNumericQuestionInput,
-): Promise<Question> {
+): Promise<Question | null> {
 	const questions: NumericQuestion[] = [];
 
 	questions.push(await getAlbumReleaseYear(input));
@@ -125,6 +136,6 @@ export async function buildNumericQuestion(
 	if (artistQ) questions.push(artistQ);
 
 	const question = questions[input.index % questions.length] ?? questions[0];
-	if (!question) throw new Error("Question not found");
+	if (!question) return null;
 	return buildQuestionWindow(question);
 }

api/src/party/question-generator.ts

@@ -22,13 +22,23 @@ export async function generatePartyQuestion({
 	quizState,
 	analytics,
 	index,
-}: GenerateQuestionInput): Promise<Question> {
+}: GenerateQuestionInput): Promise<Question | null> {
 	const members = await fetchPartyMembers(dbClient, partyId);
 	const type: PartyQuestionType =
 		index === 2 ? "numeric" : index % 2 === 0 ? "audio-metadata" : "social";
-	return type === "audio-metadata"
-		? buildAudioMetadataQuestion(analytics, index)
-		: type === "numeric"
-			? buildNumericQuestion({ db: dbClient, analytics, index, members })
-			: buildSocialQuestion(quizState, analytics, members, index);
+
+	if (type === "audio-metadata") {
+		const q = await buildAudioMetadataQuestion(dbClient, analytics, index);
+		if (q) return q;
+	}
+	if (type === "numeric") {
+		const q = await buildNumericQuestion({
+			db: dbClient,
+			analytics,
+			index,
+			members,
+		});
+		if (q) return q;
+	}
+	return buildSocialQuestion(dbClient, quizState, analytics, members, index);
 }

api/src/party/question-utils.ts

@@ -1,4 +1,6 @@
+import type { InferSelectModel } from "drizzle-orm";
 import type { db as Db } from "../db";
+import type { track as trackTable } from "../db/schema";
 
 export type PartyQuestionMember = {
 	userId: string;
@@ -23,6 +25,14 @@ export type PartyAnalytics = {
 	pairwise?: { userIdA: string; userIdB: string }[];
 } | null;
 
+export type AnalyticsTrack = {
+	name: string;
+	artists?: { name: string }[];
+	albumName?: string;
+	memberScores?: { userId: string; score: number }[];
+};
+export type QuestionSong = InferSelectModel<typeof trackTable>;
+
 export const QUESTION_DURATION_MS = 60_000;
 export const MIN_PARTY_SIZE = 2;
 export const MAX_PARTY_SIZE = 4;
@@ -100,15 +110,80 @@ export function getTopClusterArtists(analytics: PartyAnalytics): string[] {
 	);
 }
 
-export function getTopClusterTracks(analytics: PartyAnalytics): Array<{
-	name: string;
-	artists?: { name: string }[];
-	albumName?: string;
-	memberScores?: { userId: string; score: number }[];
-}> {
+export function getTopClusterTracks(
+	analytics: PartyAnalytics,
+): AnalyticsTrack[] {
 	return analytics?.storyClusters?.[0]?.tracks ?? [];
 }
 
+export function pickRelevantTrack(
+	analytics: PartyAnalytics,
+	hints: {
+		trackName?: string;
+		artistNames?: string[];
+		albumName?: string;
+	} = {},
+): AnalyticsTrack | null {
+	const tracks = getTopClusterTracks(analytics);
+	if (tracks.length === 0) return null;
+
+	const scored = tracks.map((track) => {
+		const trackArtistNames = track.artists?.map((artist) => artist.name) ?? [];
+		const matchesTrackName = hints.trackName && track.name === hints.trackName;
+		const matchesAlbum = hints.albumName && track.albumName === hints.albumName;
+		const matchesArtist =
+			hints.artistNames?.some((name) => trackArtistNames.includes(name)) ??
+			false;
+
+		return {
+			track,
+			score:
+				(matchesTrackName ? 4 : 0) +
+				(matchesAlbum ? 2 : 0) +
+				(matchesArtist ? 2 : 0),
+		};
+	});
+
+	const best = scored.sort((a, b) => b.score - a.score).at(0);
+	return best?.track ?? tracks[0] ?? null;
+}
+
+export async function resolveQuestionSong(
+	db: typeof Db,
+	analytics: PartyAnalytics,
+	hints: {
+		trackName?: string;
+		artistNames?: string[];
+		albumName?: string;
+	} = {},
+): Promise<QuestionSong | null> {
+	const trackHint = pickRelevantTrack(analytics, hints);
+	if (!trackHint?.name) return null;
+
+	const tracks = await db.query.track.findMany({
+		where: { name: trackHint.name },
+		with: { album: true, artists: true },
+	});
+	if (tracks.length === 0) return null;
+
+	const scoreTrack = (track: (typeof tracks)[number]) => {
+		const artistNames = track.artists?.map((artist) => artist.name) ?? [];
+		const matchesAlbum = trackHint.albumName
+			? track.album?.name === trackHint.albumName
+			: false;
+		const matchesArtist =
+			hints.artistNames?.some((name) => artistNames.includes(name)) ?? false;
+		return (matchesAlbum ? 2 : 0) + (matchesArtist ? 2 : 0);
+	};
+
+	const chosen = tracks
+		.slice()
+		.sort((a, b) => scoreTrack(b) - scoreTrack(a))[0];
+	if (!chosen) return null;
+	const { album: _album, artists: _artists, ...song } = chosen;
+	return song;
+}
+
 export function getTopTrackListener(
 	track: { memberScores?: { userId: string; score: number }[] },
 	members: PartyQuestionMember[],

api/src/party/social-question-generator.ts

@@ -1,3 +1,4 @@
+import type { db as Db } from "../db";
 import type { Question, QuizState } from "../party-types";
 import {
 	buildMemberOptions,
@@ -12,18 +13,21 @@ import {
 	type PartyAnalytics,
 	type PartyQuestionMember,
 	pickRandom,
+	resolveQuestionSong,
 } from "./question-utils";
 
-export function buildSocialQuestion(
+export async function buildSocialQuestion(
+	dbClient: typeof Db,
 	quizState: QuizState,
 	analytics: PartyAnalytics,
 	members: PartyQuestionMember[],
 	index: number,
-): Question {
+): Promise<Question | null> {
 	type ChoiceQuestion = Extract<Question, { type: "choice" }>;
 	const questions: Array<
 		Omit<ChoiceQuestion, "startTimestamp" | "endTimestamp">
 	> = [];
+	const topSong = await resolveQuestionSong(dbClient, analytics);
 
 	const hasMultipleMembers = members.length >= 2;
 	if (hasMultipleMembers && hasClearLeader(quizState)) {
@@ -34,6 +38,7 @@ export function buildSocialQuestion(
 			options: buildMemberOptions(leader, members),
 			correct: 0,
 			points: 10,
+			song: topSong ?? undefined,
 		});
 	}
 
@@ -45,6 +50,7 @@ export function buildSocialQuestion(
 			options: buildMemberOptions(diverse, members),
 			correct: 0,
 			points: 10,
+			song: topSong ?? undefined,
 		});
 
 		const aligned = getMostAlignedMember(analytics, members);
@@ -54,6 +60,7 @@ export function buildSocialQuestion(
 			options: buildMemberOptions(aligned, members),
 			correct: 0,
 			points: 10,
+			song: topSong ?? undefined,
 		});
 	}
 
@@ -62,12 +69,18 @@ export function buildSocialQuestion(
 	if (randomTrack && hasMultipleMembers) {
 		const topListener = getTopTrackListener(randomTrack, members);
 		if (topListener) {
+			const randomTrackSong = await resolveQuestionSong(dbClient, analytics, {
+				trackName: randomTrack.name,
+				artistNames: randomTrack.artists?.map((artist) => artist.name),
+				albumName: randomTrack.albumName,
+			});
 			questions.push({
 				type: "choice",
 				text: `Who listens the most to "${randomTrack.name}"?`,
 				options: buildMemberOptions(topListener, members),
 				correct: 0,
 				points: 10,
+				song: randomTrackSong ?? topSong ?? undefined,
 			});
 			questions.push({
 				type: "choice",
@@ -75,6 +88,7 @@ export function buildSocialQuestion(
 				options: buildMemberOptions(topListener, members),
 				correct: 0,
 				points: 10,
+				song: randomTrackSong ?? topSong ?? undefined,
 			});
 		}
 	}
@@ -93,13 +107,14 @@ export function buildSocialQuestion(
 					options: pairOptions,
 					correct: 0,
 					points: 10,
+					song: topSong ?? undefined,
 				});
 			}
 		}
 	}
 
 	if (questions.length === 0) {
-		throw new Error("Question not found");
+		return null;
 	}
 
 	const question = questions[index % questions.length];

api/src/routes/party-socket.ts

@@ -29,6 +29,40 @@ export const pubsub = {
 	},
 };
 
+async function subscribeWsToParty(
+	ws: {
+		subscribe: (topic: string) => void;
+		unsubscribe: (topic: string) => void;
+		send: (message: string) => void;
+	},
+	userId: string,
+) {
+	const membership = await getMemberRecord(db, userId);
+	if (!membership) return null;
+
+	const nextPartyId = membership.partyId;
+	const currentPartyId = socketPartyId.get(ws as object);
+	if (currentPartyId && currentPartyId !== nextPartyId) {
+		ws.unsubscribe(partyTopic(currentPartyId));
+	}
+
+	socketPartyId.set(ws as object, nextPartyId);
+	ws.subscribe(partyTopic(nextPartyId));
+
+	const snapshot = await getPartyStatus(nextPartyId);
+	if (snapshot) {
+		ws.send(
+			JSON.stringify({
+				type: "party_status",
+				party: snapshot.party,
+				members: snapshot.members,
+			} satisfies PartySocketEvent),
+		);
+	}
+
+	return nextPartyId;
+}
+
 export async function broadcastQuizState(
 	ws: { publish: (topic: string, message: string) => void },
 	partyId: string,
@@ -73,8 +107,8 @@ export const partySocketApp = new Elysia()
 
 					ws.subscribe(userTopic(user.id));
 
-					const membership = await getMemberRecord(db, user.id);
-					if (!membership) {
+					const subscribedPartyId = await subscribeWsToParty(ws, user.id);
+					if (!subscribedPartyId) {
 						ws.send(
 							JSON.stringify({
 								type: "party_status",
@@ -85,21 +119,7 @@ export const partySocketApp = new Elysia()
 						return;
 					}
 
-					socketPartyId.set(ws, membership.partyId);
-					ws.subscribe(partyTopic(membership.partyId));
-
-					const snapshot = await getPartyStatus(membership.partyId);
-					if (snapshot) {
-						ws.send(
-							JSON.stringify({
-								type: "party_status",
-								party: snapshot.party,
-								members: snapshot.members,
-							} as PartySocketEvent),
-						);
-
-						await broadcastQuizState(ws, membership.partyId);
-					}
+					await broadcastQuizState(ws, subscribedPartyId);
 				},
 				message: async (ws, message) => {
 					const data = ws.data;
@@ -121,6 +141,16 @@ export const partySocketApp = new Elysia()
 						return;
 					}
 
+					if (parsed.type === "subscribe_party") {
+						const payload = parsed as {
+							type: "subscribe_party";
+							partyId: string;
+						};
+						if (typeof payload.partyId !== "string") return;
+						await subscribeWsToParty(ws, user.id);
+						return;
+					}
+
 					if (parsed.type !== "member_payload") return;
 
 					const MAX_MEMBER_PAYLOAD_SIZE = 8_000;

api/src/routes/quiz.ts

@@ -3,8 +3,8 @@ import { eq } from "drizzle-orm";
 import Elysia, { t } from "elysia";
 import { betterAuthElysia } from "../auth";
 import { db } from "../db";
-import { party, partyMember } from "../db/schema";
-import { getMemberRecord } from "../party-data";
+import { party } from "../db/schema";
+import { getMemberRecord, getPartyStatus } from "../party-data";
 import type { QuizState } from "../party-types";
 import { QuizWorkflow, quizQueue } from "../workflows/quiz";
 import { pubsub } from "./party-socket";
@@ -49,22 +49,17 @@ export const quizRoutes = new Elysia()
 						})
 						.where(eq(party.id, params.partyId));
 
-					const members = await db
-						.select({
-							id: partyMember.id,
-							userId: partyMember.userId,
-						})
-						.from(partyMember)
-						.where(eq(partyMember.partyId, params.partyId));
-
-					pubsub.publish(
-						`party:${params.partyId}`,
-						JSON.stringify({
-							type: "party_status",
-							party: { status: "started" },
-							members,
-						}),
-					);
+					const status = await getPartyStatus(params.partyId);
+					if (status) {
+						pubsub.publish(
+							`party:${params.partyId}`,
+							JSON.stringify({
+								type: "party_status",
+								party: status.party,
+								members: status.members,
+							}),
+						);
+					}
 
 					return {
 						message: "Quiz started",

api/src/workflows/quiz.ts

@@ -155,13 +155,17 @@ export class QuizWorkflow extends ConfiguredInstance {
 			},
 		});
 		const analytics = (partyRecord?.analysisData ?? null) as PartyAnalytics;
-		return generatePartyQuestion({
+		const question = await generatePartyQuestion({
 			db,
 			partyId,
 			quizState,
 			analytics,
 			index,
 		});
+		if (!question) {
+			throw new Error("Failed to generate quiz question");
+		}
+		return question;
 	}
 
 	private static scoreRound(round: QuizRound): Array<[string, number]> {

esp32/src/main.rs

@@ -137,7 +137,7 @@ pub async fn main_loop() {
     println!("Main loop started");
     let mut last = (String::new(), String::new());
     loop {
-        embassy_time::Timer::after_millis(100).await;
+        embassy_time::Timer::after_millis(300).await;
         let mut state = STATE.lock().await;
         state.tick();
         let lines = state.render_lines();
@@ -198,7 +198,7 @@ async fn main(spawner: Spawner) -> ! {
                 p.GPIO26.degrade(),
                 GpioInputConfig::default().with_pull(Pull::Up),
             ),
-            1,
+            0,
         )
         .expect("spawn btn1"),
     );
@@ -208,7 +208,7 @@ async fn main(spawner: Spawner) -> ! {
                 p.GPIO25.degrade(),
                 GpioInputConfig::default().with_pull(Pull::Up),
             ),
-            2,
+            1,
         )
         .expect("spawn btn2"),
     );
@@ -218,7 +218,7 @@ async fn main(spawner: Spawner) -> ! {
                 p.GPIO33.degrade(),
                 GpioInputConfig::default().with_pull(Pull::Up),
             ),
-            3,
+            2,
         )
         .expect("spawn btn3"),
     );
@@ -228,7 +228,7 @@ async fn main(spawner: Spawner) -> ! {
                 p.GPIO32.degrade(),
                 GpioInputConfig::default().with_pull(Pull::Up),
             ),
-            4,
+            3,
         )
         .expect("spawn btn4"),
     );

web/src/hooks/use-party-socket.ts

@@ -1,5 +1,8 @@
 import { useCallback, useEffect, useMemo, useRef, useState } from "react";
-import type { PartySocketEvent } from "../../../api/src/party-types";
+import type {
+	PartySocketEvent,
+	PartySocketOutgoing,
+} from "../../../api/src/party-types";
 
 type Handler = (event: PartySocketEvent) => void;
 
@@ -24,6 +27,12 @@ export function usePartySocket({
 	const reconnectAttemptRef = useRef(0);
 	const handlerRef = useRef(onMessage);
 
+	const send = useCallback((message: PartySocketOutgoing) => {
+		const ws = wsRef.current;
+		if (!ws || ws.readyState !== WebSocket.OPEN) return;
+		ws.send(JSON.stringify(message));
+	}, []);
+
 	useEffect(() => {
 		handlerRef.current = onMessage;
 	}, [onMessage]);
@@ -42,7 +51,6 @@ export function usePartySocket({
 
 			ws.onmessage = (event) => {
 				const parsed = JSON.parse(event.data) as PartySocketEvent;
-				console.log(parsed);
 				handlerRef.current?.(parsed);
 			};
 
@@ -120,8 +128,9 @@ export function usePartySocket({
 			isConnected: connectionState === "connected",
 			isConnecting: connectionState === "connecting",
 			isReconnecting: connectionState === "reconnecting",
+			send,
 		}),
-		[connectionState],
+		[connectionState, send],
 	);
 
 	return state;

web/src/hooks/use-party.ts

@@ -6,6 +6,7 @@ import {
 	useContext,
 	useEffect,
 	useMemo,
+	useRef,
 	useState,
 } from "react";
 import type {
@@ -75,6 +76,11 @@ export function PartyProvider({ children }: { children: ReactNode }) {
 		if (!url) return null;
 		return url;
 	}, []);
+	const subscribedPartyIdRef = useRef<string | null>(null);
+	const wsState = usePartySocket({
+		apiUrl: userId ? apiUrl : null,
+		onMessage: userId ? handleMessage : null,
+	});
 
 	const resetParty = useCallback(() => {
 		setState(emptyPartyState);
@@ -84,10 +90,21 @@ export function PartyProvider({ children }: { children: ReactNode }) {
 		if (!userId) resetParty();
 	}, [resetParty, userId]);
 
-	const wsState = usePartySocket({
-		apiUrl: userId ? apiUrl : null,
-		onMessage: userId ? handleMessage : null,
-	});
+	useEffect(() => {
+		if (wsState.connectionState !== "connected") {
+			subscribedPartyIdRef.current = null;
+			return;
+		}
+
+		if (!userId || !state.party?.id) return;
+		if (subscribedPartyIdRef.current === state.party.id) return;
+
+		wsState.send({
+			type: "subscribe_party",
+			partyId: state.party.id,
+		});
+		subscribedPartyIdRef.current = state.party.id;
+	}, [state.party?.id, userId, wsState.connectionState, wsState.send]);
 
 	const value = useMemo(
 		() => ({