import { PDFPage } from 'pdf-lib';
import { PdfFile, fromPdfLib } from '../wrappers/PdfFile';

export async function scaleContent(file: PdfFile, scaleFactor: number|number[]): Promise<PdfFile> {
    const pdfDoc = await file.getAsPdfLib();
    const pages = pdfDoc.getPages();

    if (Array.isArray(scaleFactor)) {
        if (scaleFactor.length != pages.length) {
            throw new Error(`Number of given scale factors '${scaleFactor.length}' is not the same as the number of pages '${pages.length}'`)
        }
        for (let i=0; i<scaleFactor.length; i++) {
            scalePage(pages[i], scaleFactor[i]);
        }
    } else {
        pages.forEach(page => scalePage(page, scaleFactor));
    }

    return fromPdfLib(pdfDoc, file.filename);
};

function scalePage(page: PDFPage, scaleFactor: number) {
    const width = page.getWidth();
    const height = page.getHeight();

    // Scale content
    page.scaleContent(scaleFactor, scaleFactor);
    const scaled_diff = {
        width: Math.round(width - scaleFactor * width),
        height: Math.round(height - scaleFactor * height),
    };

    // Center content in new page format
    page.translateContent(Math.round(scaled_diff.width / 2), Math.round(scaled_diff.height / 2));
}