cleanup

Author: adamrefaey

backend/src/document-processor/controllers/document-processor.controller.ts

@@ -48,11 +48,17 @@ export class DocumentProcessorController {
     }
 
     // Validate file type
-    const validMimeTypes = ['image/jpeg', 'image/png', 'image/tiff', 'application/pdf'];
+    const validMimeTypes = [
+      'image/jpeg',
+      'image/png',
+      'image/heic',
+      'image/heif',
+      'application/pdf',
+    ];
 
     if (!validMimeTypes.includes(file.mimetype)) {
       throw new BadRequestException(
-        `Invalid file type: ${file.mimetype}. Supported types: JPEG, PNG, TIFF, and PDF.`,
+        `Invalid file type: ${file.mimetype}. Supported types: JPEG, PNG, HEIC, HEIF, and PDF.`,
       );
     }
 
@@ -373,8 +379,8 @@ export class DocumentProcessorController {
       
       <form id="uploadForm" enctype="multipart/form-data">
         <div class="form-group">
-          <label for="file">Select File (PDF, JPEG, PNG, TIFF):</label>
-          <input type="file" id="file" name="file" accept=".pdf,.jpg,.jpeg,.png,.tiff">
+          <label for="file">Select File (PDF, JPEG, PNG, HEIC, HEIF):</label>
+          <input type="file" id="file" name="file" accept=".pdf,.jpg,.jpeg,.png,.heic,.heif">
         </div>
         
         <div id="filePreview"></div>

backend/src/document-processor/services/aws-textract.service.spec.ts

@@ -5,7 +5,6 @@ import { vi, describe, it, expect, beforeEach } from 'vitest';
 
 // Mock the security.utils module
 vi.mock('../../utils/security.utils', () => ({
-  validateFileSecurely: vi.fn(),
   RateLimiter: vi.fn().mockImplementation(() => ({
     tryRequest: vi.fn().mockReturnValue(true),
   })),

backend/src/document-processor/services/aws-textract.service.ts

@@ -1,7 +1,7 @@
 import { Injectable, Logger, BadRequestException } from '@nestjs/common';
 import { ConfigService } from '@nestjs/config';
 import { TextractClient, AnalyzeDocumentCommand, Block } from '@aws-sdk/client-textract';
-import { validateFileSecurely, RateLimiter } from '../../utils/security.utils';
+import { RateLimiter } from '../../utils/security.utils';
 import { createHash } from 'crypto';
 
 export interface ExtractedTextResult {
@@ -88,24 +88,18 @@ export class AwsTextractService {
     try {
       const startTime = Date.now();
 
-      // 1. Rate limiting check
       if (!this.rateLimiter.tryRequest(userId)) {
         throw new BadRequestException('Too many requests. Please try again later.');
       }
 
-      // 2. Validate file securely
-      validateFileSecurely(fileBuffer);
-
       // Add diagnostic information about the document being processed
       this.logger.debug('Processing document', {
         fileSize: `${(fileBuffer.length / 1024).toFixed(2)} KB`,
         contentHashPrefix: createHash('sha256').update(fileBuffer).digest('hex').substring(0, 10),
       });
 
-      // 3. Process document
       const result = await this.processDocument(fileBuffer);
 
-      // 4. Calculate processing time
       const processingTime = Date.now() - startTime;
 
       this.logger.log(`Document processed in ${processingTime}ms`, {