import { NextRequest, NextResponse } from 'next/server';
import path from 'path';
import fs from 'fs';
import { pool } from '@/lib/db-config';

export async function GET(
  request: NextRequest,
  context: { params: Promise<{ id: string; docId: string }> }
) {
  const { id: applicationId, docId: documentId } = await context.params;

  if (!applicationId || !documentId) {
    return NextResponse.json({ 
      success: false, 
      error: 'Application ID and Document ID are required' 
    }, { status: 400 });
  }

  try {
    // Query database for document
    const documentQuery = await pool.query(`
      SELECT * FROM documents 
      WHERE id = $1 AND application_id = $2
    `, [documentId, applicationId]);
    
    if (documentQuery.rowCount === 0) {
      return NextResponse.json({ 
        success: false, 
        error: 'Document not found' 
      }, { status: 404 });
    }
    
    const document = documentQuery.rows[0];
    const filePath = document.path;

 // Check if file exists
    if (!fs.existsSync(filePath)) {
      return NextResponse.json({ 
        success: false, 
        error: 'File not found on server' 
      }, { status: 404 });
    }
    
    // Read file and serve it
    const fileBuffer = fs.readFileSync(filePath);
    
    // Determine content type based on file extension
    const fileExtension = path.extname(filePath).toLowerCase();
    let contentType = 'application/octet-stream'; // Default
    
    if (['.jpg', '.jpeg'].includes(fileExtension)) contentType = 'image/jpeg';
    else if (fileExtension === '.png') contentType = 'image/png';
    else if (fileExtension === '.pdf') contentType = 'application/pdf';
    else if (['.doc', '.docx'].includes(fileExtension)) contentType = 'application/msword';
    
    // Return file as response
    return new NextResponse(fileBuffer, {
      headers: {
        'Content-Type': contentType,
        'Content-Disposition': `inline; filename="${document.original_name || path.basename(filePath)}"`,
      },
    });
  } catch (error: any) {
    console.error('Error retrieving document file:', error);
    return NextResponse.json({ 
      success: false, 
      error: error.message 
    }, { status: 500 });
  }
} 

/documents/route.ts

/documents/[docId]/route.ts

{ params }: { params: { id: string } }

{ params }: { params: Promise<{ id: string, docId: string }> }