import { getDbClient } from '@/app/lib/dbClient';
import { BillingLocation } from '@/app/lib/db-types';
import { notFound } from 'next/navigation';

export async function GET(_request: Request, { params:{ id } }: { params: { id:string } }) {
    // Parse locationID-billID format
    const [locationID, billID] = id.split('-');

    if (!locationID || !billID) {
        notFound();
    }

    const dbClient = await getDbClient();
    const location = await dbClient.collection<BillingLocation>("lokacije")
        .findOne({ _id: locationID }, {
            projection: {
                // Don't load bill attachments, only proof of payment
                "bills._id": 1,
                "bills.proofOfPayment": 1,
            }
        });

    if(!location) {
        notFound();
    }

    // Find the specific bill
    const bill = location.bills.find(b => b._id === billID);

    if(!bill?.proofOfPayment) {
        notFound();
    }

    // Convert fileContentsBase64 from Base64 string to binary
    const fileContentsBuffer = Buffer.from(bill.proofOfPayment.fileContentsBase64, 'base64');

    // Convert fileContentsBuffer to format that can be sent to the client
    const fileContents = new Uint8Array(fileContentsBuffer);

    return new Response(fileContents, {
        status: 200,
        headers: {
            'Content-Type': 'application/pdf',
            'Content-Disposition': `attachment; filename="${bill.proofOfPayment.fileName}"`,
            'Last-Modified': `${bill.proofOfPayment.fileLastModified}`
        }
    });
}