profilarr/src/lib/server/pcd/queries/entityTests/list.ts

/**
 * Entity test list queries
 */

import type { PCDCache } from '../../cache.ts';

/**
 * Get all test entities with their releases
 */
export async function list(cache: PCDCache) {
	const db = cache.kb;

	// 1. Get all test entities
	const entities = await db
		.selectFrom('test_entities')
		.select(['id', 'type', 'tmdb_id', 'title', 'year', 'poster_path', 'created_at', 'updated_at'])
		.orderBy('title')
		.execute();

	if (entities.length === 0) return [];

	// 2. Get all releases for all entities
	const allReleases = await db
		.selectFrom('test_releases')
		.select([
			'id',
			'entity_type',
			'entity_tmdb_id',
			'title',
			'size_bytes',
			'languages',
			'indexers',
			'flags',
			'created_at',
			'updated_at'
		])
		.orderBy('entity_type')
		.orderBy('entity_tmdb_id')
		.orderBy('title')
		.execute();

	// Build releases map using composite key
	const releasesMap = new Map<string, typeof allReleases>();
	for (const release of allReleases) {
		const key = `${release.entity_type}-${release.entity_tmdb_id}`;
		if (!releasesMap.has(key)) {
			releasesMap.set(key, []);
		}
		releasesMap.get(key)!.push(release);
	}

	// Build the final result
	return entities.map((entity) => ({
		...entity,
		type: entity.type as 'movie' | 'series',
		releases: (releasesMap.get(`${entity.type}-${entity.tmdb_id}`) || []).map((r) => ({
			id: r.id,
			title: r.title,
			size_bytes: r.size_bytes !== null ? Number(r.size_bytes) : null,
			languages: JSON.parse(r.languages) as string[],
			indexers: JSON.parse(r.indexers) as string[],
			flags: JSON.parse(r.flags) as string[],
			created_at: r.created_at,
			updated_at: r.updated_at
		}))
	}));
}