import fs from "fs/promises"; import path from "path"; import { PDFDocument, rgb, StandardFonts } from "pdf-lib"; import puppeteer from "puppeteer"; import { prisma } from "../prismaClient"; /** * Serviço para geração e manipulação de arquivos PDF */ export class PdfService { /** * Cria um documento PDF básico * @returns Documento PDF */ async criarDocumento() { return await PDFDocument.create(); } /** * Carrega um documento PDF existente * @param pdfBytes - Bytes do PDF * @returns Documento PDF */ async carregarDocumento(pdfBytes: Uint8Array) { return await PDFDocument.load(pdfBytes); } /** * Adiciona uma página ao documento * @param doc - Documento PDF * @param tamanho - Tamanho da página (padrão: A4) * @returns Página adicionada */ adicionarPagina(doc: any, tamanho = [595.28, 841.89]) { return doc.addPage(tamanho); } /** * Salva o documento como buffer * @param doc - Documento PDF * @returns Buffer do PDF */ async salvarComoBuffer(doc: any): Promise { const pdfBytes = await doc.save(); return Buffer.from(pdfBytes); } /** * Carrega uma fonte padrão * @param doc - Documento PDF * @param fonte - Nome da fonte padrão * @returns Fonte carregada */ async carregarFonte(doc: any, fonte: StandardFonts) { return await doc.embedFont(fonte); } /** * Desenha texto em uma página * @param pagina - Página do PDF * @param texto - Texto a ser desenhado * @param x - Posição X * @param y - Posição Y * @param fonte - Fonte a ser usada * @param tamanho - Tamanho da fonte * @param cor - Cor do texto (padrão: preto) */ desenharTexto( pagina: any, texto: string, x: number, y: number, fonte: any, tamanho: number, cor = rgb(0, 0, 0), ) { pagina.drawText(texto, { x, y, font: fonte, size: tamanho, color: cor, }); } /** * Cria um cabeçalho para o documento * @param pagina - Página do PDF * @param titulo - Título do cabeçalho * @param fonte - Fonte a ser usada * @param margemSuperior - Margem superior * @param margem - Margem lateral * @returns Posição Y após o cabeçalho */ criarCabecalho( pagina: any, titulo: string, fonte: any, margemSuperior: number, margem: number, ): number { const yPos = pagina.getHeight() - margemSuperior; this.desenharTexto(pagina, titulo, margem, yPos, fonte, 18, rgb(0, 0, 0.7)); // Linha separadora pagina.drawLine({ start: { x: margem, y: yPos - 10 }, end: { x: pagina.getWidth() - margem, y: yPos - 10 }, thickness: 1, color: rgb(0, 0, 0.5), }); return yPos - 30; // Retorna a posição Y após o cabeçalho } /** * Cria um rodapé para o documento * @param pagina - Página do PDF * @param texto - Texto do rodapé * @param fonte - Fonte a ser usada * @param margem - Margem */ criarRodape(pagina: any, texto: string, fonte: any, margem: number) { const yPos = margem / 2; this.desenharTexto( pagina, texto, margem, yPos, fonte, 8, rgb(0.5, 0.5, 0.5), ); } private async getConfiguracao() { try { // Buscar configuração ou usar valores padrão const config = await prisma.configuracaoDocumento.findFirst({ orderBy: { criadoEm: "desc" }, select: { id: true, headerImageUrl: true, footerImageUrl: true, logoUrl: true, timbradoUrl: true, timbradoMarginTop: true, timbradoMarginBottom: true, timbradoMarginLeft: true, timbradoMarginRight: true, criadoEm: true, }, }); console.log("Configuração encontrada:", config); // Debug // Retornar configuração padrão se não existir return { cabecalhoHtml: config?.cabecalhoHtml || `

{{escritorio.nome}}

`, rodapeHtml: config?.rodapeHtml || `
{{escritorio.endereco}}
Tel: {{escritorio.telefone}} | Email: {{escritorio.email}}
`, logoUrl: config?.logoUrl || "", timbradoUrl: config?.timbradoUrl || "", timbradoMarginTop: config?.timbradoMarginTop || 100, timbradoMarginBottom: config?.timbradoMarginBottom || 50, timbradoMarginLeft: config?.timbradoMarginLeft || 50, timbradoMarginRight: config?.timbradoMarginRight || 50, }; } catch (error) { console.error("Erro ao buscar configuração:", error); throw new Error("Falha ao buscar configuração"); } } private gerarHTML(content: string, config: any) { const timbradoUrl = config.timbradoUrl ? `${process.env.BACKEND_URL || "http://localhost:5000"}${config.timbradoUrl}` : ""; console.log("Gerando PDF com timbrado:", timbradoUrl); return `
${content}
`; } async gerarPDF(conteudo: string, escritorio: any) { try { // 1. Buscar configuração do template const config = await prisma.configuracaoDocumento.findFirst({ orderBy: { criadoEm: "desc" }, }); // 2. Gerar PDF do conteúdo const browser = await puppeteer.launch({ headless: "new", args: ["--no-sandbox"], }); const page = await browser.newPage(); // Usar template padrão ou fallback para o template básico const template = config?.templatePadrao || `
${conteudo}
`; // Processar o template substituindo variáveis const templateProcessado = template.replace( /\{\{(.+?)\}\}/g, (match, key) => { return escritorio[key.trim()] || match; }, ); await page.setContent(templateProcessado); const margens = config?.margensTemplate || { top: "35mm", bottom: "29mm", left: "25mm", right: "25mm", }; const contentPdfBuffer = await page.pdf({ format: "A4", margin: margens, displayHeaderFooter: false, printBackground: true, }); await browser.close(); // 2. Carregar os PDFs const templatePath = path.join( process.cwd(), "public", "uploads", "Timbrado_2025.pdf", ); const templateBytes = await fs.readFile(templatePath); const templateDoc = await PDFDocument.load(templateBytes); const contentDoc = await PDFDocument.load(contentPdfBuffer); // 3. Criar documento final const pdfDoc = await PDFDocument.create(); // 4. Copiar template e conteúdo const [templatePage] = await templateDoc.copyPages(templateDoc, [0]); const contentPages = await contentDoc.getPages(); // 5. Para cada página do conteúdo for (let i = 0; i < contentPages.length; i++) { // Primeiro: adicionar template const page = await pdfDoc.embedPage(templatePage); const contentPage = await pdfDoc.embedPage(contentPages[i]); // Segundo: criar nova página const newPage = pdfDoc.addPage([page.width, page.height]); // Terceiro: desenhar template e conteúdo newPage.drawPage(page); newPage.drawPage(contentPage); } // 6. Gerar PDF final const pdfBytes = await pdfDoc.save(); return Buffer.from(pdfBytes); } catch (error) { console.error("Erro ao gerar PDF:", error); throw new Error("Falha ao gerar PDF"); } } }