import { v } from "convex/values";
import { mutation, query } from "./_generated/server";
import { Id } from "./_generated/dataModel";

// Mutation para fazer upload de arquivo e obter o storage ID
export const generateUploadUrl = mutation({
  args: {},
  returns: v.string(),
  handler: async (ctx) => {
    return await ctx.storage.generateUploadUrl();
  },
});

// Mutation para atualizar um campo de documento do funcionário
export const updateDocumento = mutation({
  args: {
    funcionarioId: v.id("funcionarios"),
    campo: v.string(),
    storageId: v.union(v.id("_storage"), v.null()),
  },
  returns: v.null(),
  handler: async (ctx, args) => {
    const funcionario = await ctx.db.get(args.funcionarioId);
    if (!funcionario) {
      throw new Error("Funcionário não encontrado");
    }

    // Atualizar o campo específico do documento
    await ctx.db.patch(args.funcionarioId, {
      [args.campo]: args.storageId,
    });

    return null;
  },
});

// Query para obter URLs de todos os documentos de um funcionário
export const getDocumentosUrls = query({
  args: { funcionarioId: v.id("funcionarios") },
  returns: v.object({
    certidaoAntecedentesPF: v.union(v.string(), v.null()),
    certidaoAntecedentesJFPE: v.union(v.string(), v.null()),
    certidaoAntecedentesSDS: v.union(v.string(), v.null()),
    certidaoAntecedentesTJPE: v.union(v.string(), v.null()),
    certidaoImprobidade: v.union(v.string(), v.null()),
    rgFrente: v.union(v.string(), v.null()),
    rgVerso: v.union(v.string(), v.null()),
    cpfFrente: v.union(v.string(), v.null()),
    cpfVerso: v.union(v.string(), v.null()),
    situacaoCadastralCPF: v.union(v.string(), v.null()),
    tituloEleitorFrente: v.union(v.string(), v.null()),
    tituloEleitorVerso: v.union(v.string(), v.null()),
    comprovanteVotacao: v.union(v.string(), v.null()),
    carteiraProfissionalFrente: v.union(v.string(), v.null()),
    carteiraProfissionalVerso: v.union(v.string(), v.null()),
    comprovantePIS: v.union(v.string(), v.null()),
    certidaoRegistroCivil: v.union(v.string(), v.null()),
    certidaoNascimentoDependentes: v.union(v.string(), v.null()),
    cpfDependentes: v.union(v.string(), v.null()),
    reservistaDoc: v.union(v.string(), v.null()),
    comprovanteEscolaridade: v.union(v.string(), v.null()),
    comprovanteResidencia: v.union(v.string(), v.null()),
    comprovanteContaBradesco: v.union(v.string(), v.null()),
    declaracaoAcumulacaoCargo: v.union(v.string(), v.null()),
    declaracaoDependentesIR: v.union(v.string(), v.null()),
    declaracaoIdoneidade: v.union(v.string(), v.null()),
    termoNepotismo: v.union(v.string(), v.null()),
    termoOpcaoRemuneracao: v.union(v.string(), v.null()),
  }),
  handler: async (ctx, args) => {
    const funcionario = await ctx.db.get(args.funcionarioId);
    if (!funcionario) {
      throw new Error("Funcionário não encontrado");
    }

    // Tipo exato do retorno para alinhar com o validator
    type DocumentUrls = {
      certidaoAntecedentesPF: string | null;
      certidaoAntecedentesJFPE: string | null;
      certidaoAntecedentesSDS: string | null;
      certidaoAntecedentesTJPE: string | null;
      certidaoImprobidade: string | null;
      rgFrente: string | null;
      rgVerso: string | null;
      cpfFrente: string | null;
      cpfVerso: string | null;
      situacaoCadastralCPF: string | null;
      tituloEleitorFrente: string | null;
      tituloEleitorVerso: string | null;
      comprovanteVotacao: string | null;
      carteiraProfissionalFrente: string | null;
      carteiraProfissionalVerso: string | null;
      comprovantePIS: string | null;
      certidaoRegistroCivil: string | null;
      certidaoNascimentoDependentes: string | null;
      cpfDependentes: string | null;
      reservistaDoc: string | null;
      comprovanteEscolaridade: string | null;
      comprovanteResidencia: string | null;
      comprovanteContaBradesco: string | null;
      declaracaoAcumulacaoCargo: string | null;
      declaracaoDependentesIR: string | null;
      declaracaoIdoneidade: string | null;
      termoNepotismo: string | null;
      termoOpcaoRemuneracao: string | null;
    };

    const urls: DocumentUrls = {
      certidaoAntecedentesPF: null,
      certidaoAntecedentesJFPE: null,
      certidaoAntecedentesSDS: null,
      certidaoAntecedentesTJPE: null,
      certidaoImprobidade: null,
      rgFrente: null,
      rgVerso: null,
      cpfFrente: null,
      cpfVerso: null,
      situacaoCadastralCPF: null,
      tituloEleitorFrente: null,
      tituloEleitorVerso: null,
      comprovanteVotacao: null,
      carteiraProfissionalFrente: null,
      carteiraProfissionalVerso: null,
      comprovantePIS: null,
      certidaoRegistroCivil: null,
      certidaoNascimentoDependentes: null,
      cpfDependentes: null,
      reservistaDoc: null,
      comprovanteEscolaridade: null,
      comprovanteResidencia: null,
      comprovanteContaBradesco: null,
      declaracaoAcumulacaoCargo: null,
      declaracaoDependentesIR: null,
      declaracaoIdoneidade: null,
      termoNepotismo: null,
      termoOpcaoRemuneracao: null,
    };

    const campos: Array<keyof DocumentUrls> = [
      "certidaoAntecedentesPF",
      "certidaoAntecedentesJFPE",
      "certidaoAntecedentesSDS",
      "certidaoAntecedentesTJPE",
      "certidaoImprobidade",
      "rgFrente",
      "rgVerso",
      "cpfFrente",
      "cpfVerso",
      "situacaoCadastralCPF",
      "tituloEleitorFrente",
      "tituloEleitorVerso",
      "comprovanteVotacao",
      "carteiraProfissionalFrente",
      "carteiraProfissionalVerso",
      "comprovantePIS",
      "certidaoRegistroCivil",
      "certidaoNascimentoDependentes",
      "cpfDependentes",
      "reservistaDoc",
      "comprovanteEscolaridade",
      "comprovanteResidencia",
      "comprovanteContaBradesco",
      "declaracaoAcumulacaoCargo",
      "declaracaoDependentesIR",
      "declaracaoIdoneidade",
      "termoNepotismo",
      "termoOpcaoRemuneracao",
    ];

    for (const campo of campos) {
      const storageId = (funcionario as Record<string, unknown>)[campo as string] as
        | Id<"_storage">
        | undefined;
      if (storageId) {
        urls[campo] = await ctx.storage.getUrl(storageId);
      } else {
        urls[campo] = null;
      }
    }

    return urls;
  },
});

// Query para obter metadados de um documento
export const getDocumentoMetadata = query({
  args: { storageId: v.id("_storage") },
  returns: v.union(
    v.object({
      _id: v.id("_storage"),
      _creationTime: v.number(),
      contentType: v.optional(v.string()),
      sha256: v.string(),
      size: v.number(),
    }),
    v.null()
  ),
  handler: async (ctx, args) => {
    return await ctx.db.system.get(args.storageId);
  },
});