import { Client } from "https://deno.land/x/postgres@v0.19.3/mod.ts";
import { parseTimestampFromPsql } from "lib/utils/formatTimestampToPostgre.ts";

export async function getNotCollectedSongs(client: Client) {
	const queryResult = await client.queryObject<{ aid: number }>(`
		SELECT lr.aid
		FROM labelling_result lr
		WHERE lr.label != 0
		AND NOT EXISTS (
			SELECT 1
			FROM songs s
			WHERE s.aid = lr.aid
		);
	`);
	return queryResult.rows.map((row) => row.aid);
}

export async function aidExistsInSongs(client: Client, aid: number) {
	const queryResult = await client.queryObject<{ exists: boolean }>(
		`
	    SELECT EXISTS (
	        SELECT 1
	        FROM songs
	        WHERE aid = $1
	    );
		`,
		[aid],
	);
	return queryResult.rows[0].exists;
}

export async function getSongsPublihsedAt(client: Client, aid: number) {
	const queryResult = await client.queryObject<{ published_at: string }>(
		`
		SELECT published_at
		FROM songs
		WHERE aid = $1;
		`,
		[aid],
	);
	if (queryResult.rows.length === 0) {
		return null;
	}
	return parseTimestampFromPsql(queryResult.rows[0].published_at);
}