import { v } from 'convex/values';
import { mutation, query } from './_generated/server';
import type { Id } from './_generated/dataModel';
import { getCurrentUserFunction } from './auth';

export const list = query({
	args: {},
	handler: async (ctx) => {
		return await ctx.db.query('atas').collect();
	}
});

export const get = query({
	args: { id: v.id('atas') },
	handler: async (ctx, args) => {
		return await ctx.db.get(args.id);
	}
});

export const getObjetos = query({
	args: { id: v.id('atas') },
	handler: async (ctx, args) => {
		const links = await ctx.db
			.query('atasObjetos')
			.withIndex('by_ataId', (q) => q.eq('ataId', args.id))
			.collect();

		const objetos = await Promise.all(links.map((link) => ctx.db.get(link.objetoId)));
		return objetos.filter((obj) => obj !== null);
	}
});

export const listByObjetoIds = query({
	args: {
		objetoIds: v.array(v.id('objetos'))
	},
	handler: async (ctx, args) => {
		if (args.objetoIds.length === 0) return [];

		// Buscar todos os vínculos ata-objeto para os objetos informados
		const links = [];
		for (const objetoId of args.objetoIds) {
			const partial = await ctx.db
				.query('atasObjetos')
				.withIndex('by_objetoId', (q) => q.eq('objetoId', objetoId as Id<'objetos'>))
				.collect();
			links.push(...partial);
		}

		const ataIds = Array.from(
			new Set(links.map((l) => l.ataId as Id<'atas'>))
		);

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

		const atas = await Promise.all(ataIds.map((id) => ctx.db.get(id)));
		return atas.filter((a): a is NonNullable<typeof a> => a !== null);
	}
});

export const create = mutation({
	args: {
		numero: v.string(),
		dataInicio: v.optional(v.string()),
		dataFim: v.optional(v.string()),
		empresaId: v.id('empresas'),
		pdf: v.optional(v.string()),
		numeroSei: v.string(),
		objetosIds: v.optional(v.array(v.id('objetos')))
	},
	handler: async (ctx, args) => {
		const user = await getCurrentUserFunction(ctx);
		if (!user) throw new Error('Unauthorized');

		const ataId = await ctx.db.insert('atas', {
			numero: args.numero,
			dataInicio: args.dataInicio,
			dataFim: args.dataFim,
			empresaId: args.empresaId,
			pdf: args.pdf,
			numeroSei: args.numeroSei,
			criadoPor: user._id,
			criadoEm: Date.now(),
			atualizadoEm: Date.now()
		});

		if (args.objetosIds) {
			for (const objetoId of args.objetosIds) {
				await ctx.db.insert('atasObjetos', {
					ataId,
					objetoId
				});
			}
		}

		return ataId;
	}
});

export const update = mutation({
	args: {
		id: v.id('atas'),
		numero: v.string(),
		dataInicio: v.optional(v.string()),
		dataFim: v.optional(v.string()),
		empresaId: v.id('empresas'),
		pdf: v.optional(v.string()),
		numeroSei: v.string(),
		objetosIds: v.optional(v.array(v.id('objetos')))
	},
	handler: async (ctx, args) => {
		const user = await getCurrentUserFunction(ctx);
		if (!user) throw new Error('Unauthorized');

		await ctx.db.patch(args.id, {
			numero: args.numero,
			dataInicio: args.dataInicio,
			dataFim: args.dataFim,
			empresaId: args.empresaId,
			pdf: args.pdf,
			numeroSei: args.numeroSei,
			atualizadoEm: Date.now()
		});

		if (args.objetosIds !== undefined) {
			// Remove existing links
			const existingLinks = await ctx.db
				.query('atasObjetos')
				.withIndex('by_ataId', (q) => q.eq('ataId', args.id))
				.collect();

			for (const link of existingLinks) {
				await ctx.db.delete(link._id);
			}

			// Add new links
			for (const objetoId of args.objetosIds) {
				await ctx.db.insert('atasObjetos', {
					ataId: args.id,
					objetoId
				});
			}
		}
	}
});

export const remove = mutation({
	args: {
		id: v.id('atas')
	},
	handler: async (ctx, args) => {
		const user = await getCurrentUserFunction(ctx);
		if (!user) throw new Error('Unauthorized');

		// Remove linked objects
		const links = await ctx.db
			.query('atasObjetos')
			.withIndex('by_ataId', (q) => q.eq('ataId', args.id))
			.collect();

		for (const link of links) {
			await ctx.db.delete(link._id);
		}

		await ctx.db.delete(args.id);
	}
});