Improve and fix ProvaService.ts

Signed-off-by: Rodrigo Martins <rodrigo.sousa.577@gmail.com>

src/server/services/ReportGenerator.ts

@@ -5,6 +5,7 @@ import ptBR from 'date-fns/locale/pt-BR';
 import { prisma } from '../prismaClient';
 import * as fs from 'fs/promises';
 import * as path from 'path';
+import sharp from 'sharp';
 
 export class ReportGenerator {
   async generateReport(data: {
@@ -175,7 +176,7 @@ export class ReportGenerator {
   private async drawImagePage(page: any, data: any, helvetica: any, helveticaBold: any, pdfDoc: PDFDocument) {
     const height = page.getHeight();
     
-    // Título
+    // Título e informações básicas (mantidos como estão)
     page.drawText('Documento Original', {
       x: 50,
       y: height - 50,
@@ -184,7 +185,6 @@ export class ReportGenerator {
       color: rgb(0, 0, 0.7)
     });
     
-    // Informações básicas do arquivo
     page.drawText(`Arquivo: ${data.file.originalname}`, {
       x: 50,
       y: height - 80,
@@ -192,7 +192,6 @@ export class ReportGenerator {
       font: helvetica
     });
     
-    // Formatação segura de data de registro
     let timestampStr = 'Data não disponível';
     try {
       if (data.timestamp?.timestamp) {
@@ -212,7 +211,6 @@ export class ReportGenerator {
       font: helvetica
     });
     
-    // Data de modificação original
     if (data.metadata?.fileInfo?.modifiedDate || data.metadata?.file_specific?.modified) {
       const modifiedDate = new Date(data.metadata?.fileInfo?.modifiedDate || data.metadata?.file_specific?.modified);
       if (!isNaN(modifiedDate.getTime())) {
@@ -225,15 +223,55 @@ export class ReportGenerator {
       }
     }
     
-    // Incorporar a imagem
+    // Incorporar a imagem com tratamento de erro mais robusto
     try {
       let image;
+      let imageBuffer = data.file.buffer;
+      
+      // Se for JPEG, tentar pré-processar com sharp para garantir um formato válido
       if (data.file.mimetype === 'image/jpeg') {
-        image = await pdfDoc.embedJpg(data.file.buffer);
-      } else if (data.file.mimetype === 'image/png') {
+        try {
+          // Converter para PNG primeiro - isso resolve muitos problemas de JPEGs corrompidos
+          const processedBuffer = await sharp(data.file.buffer)
+            .png()  // Converter para PNG que é mais robusto
+            .toBuffer();
+            
+          // Usar PNG em vez de JPEG para evitar o erro SOI
+          image = await pdfDoc.embedPng(processedBuffer);
+          console.log('JPEG convertido para PNG para evitar problemas de formatação');
+        } catch (sharpError) {
+          console.error('Erro ao processar JPEG com sharp:', sharpError);
+          
+          // Se falhar, tentar abordagem alternativa
+          try {
+            // Tentar usar a versão processada se disponível
+            if (data.processedFile && data.processedFile.length > 0) {
+              image = await pdfDoc.embedPng(data.processedFile);
+              console.log('Usando versão processada da imagem');
+            } else {
+              throw new Error('Imagem processada não disponível');
+            }
+          } catch (fallbackError) {
+            console.error('Erro no fallback para imagem processada:', fallbackError);
+            throw new Error('Não foi possível processar a imagem JPEG');
+          }
+        }
+      } 
+      // Para PNG ou outros formatos
+      else if (data.file.mimetype === 'image/png') {
         image = await pdfDoc.embedPng(data.file.buffer);
-      } else {
-        image = await pdfDoc.embedPng(data.processedFile);
+      } 
+      // Para todos os outros tipos, tentar converter para PNG
+      else {
+        try {
+          const pngBuffer = await sharp(data.file.buffer)
+            .png()
+            .toBuffer();
+          image = await pdfDoc.embedPng(pngBuffer);
+        } catch (conversionError) {
+          console.error('Erro ao converter imagem para PNG:', conversionError);
+          throw new Error('Formato de imagem não suportado');
+        }
       }
       
       // Calcular dimensões mantendo proporção
@@ -286,13 +324,30 @@ export class ReportGenerator {
       });
     } catch (imageError) {
       console.error('Erro ao processar imagem:', imageError);
+      
+      // Mensagem de erro mais informativa na página
       page.drawText('Não foi possível exibir a imagem original.', {
         x: 50,
-        y: height / 2,
+        y: height / 2 + 20,
         size: 14,
         font: helveticaBold,
         color: rgb(0.8, 0, 0)
       });
+      
+      page.drawText(`Erro: ${imageError.message}`, {
+        x: 50,
+        y: height / 2 - 10,
+        size: 12,
+        font: helvetica,
+        color: rgb(0.7, 0, 0)
+      });
+      
+      page.drawText('O documento foi validado, mas a visualização não está disponível.', {
+        x: 50,
+        y: height / 2 - 30,
+        size: 12,
+        font: helvetica
+      });
     }
   }