bentopdf/src/js/logic/psd-to-pdf-page.ts

import { showLoader, hideLoader, showAlert } from '../ui.js';
import { downloadFile, formatBytes } from '../utils/helpers.js';
import { state } from '../state.js';
import { createIcons, icons } from 'lucide';
import { PyMuPDF } from '@bentopdf/pymupdf-wasm';
import { getWasmBaseUrl } from '../config/wasm-cdn-config.js';

const ACCEPTED_EXTENSIONS = ['.psd'];
const FILETYPE_NAME = 'PSD';

let pymupdf: PyMuPDF | null = null;

async function ensurePyMuPDF(): Promise<PyMuPDF> {
    if (!pymupdf) {
        pymupdf = new PyMuPDF(getWasmBaseUrl('pymupdf'));
        await pymupdf.load();
    }
    return pymupdf;
}

document.addEventListener('DOMContentLoaded', () => {
    const fileInput = document.getElementById('file-input') as HTMLInputElement;
    const dropZone = document.getElementById('drop-zone');
    const processBtn = document.getElementById('process-btn');
    const fileDisplayArea = document.getElementById('file-display-area');
    const fileControls = document.getElementById('file-controls');
    const addMoreBtn = document.getElementById('add-more-btn');
    const clearFilesBtn = document.getElementById('clear-files-btn');
    const backBtn = document.getElementById('back-to-tools');

    if (backBtn) {
        backBtn.addEventListener('click', () => {
            window.location.href = import.meta.env.BASE_URL;
        });
    }

    const updateUI = async () => {
        if (!fileDisplayArea || !processBtn || !fileControls) return;
        if (state.files.length > 0) {
            fileDisplayArea.innerHTML = '';
            for (let index = 0; index < state.files.length; index++) {
                const file = state.files[index];
                const fileDiv = document.createElement('div');
                fileDiv.className = 'flex items-center justify-between bg-gray-700 p-3 rounded-lg text-sm';
                const infoContainer = document.createElement('div');
                infoContainer.className = 'flex flex-col overflow-hidden';
                const nameSpan = document.createElement('div');
                nameSpan.className = 'truncate font-medium text-gray-200 text-sm mb-1';
                nameSpan.textContent = file.name;
                const metaSpan = document.createElement('div');
                metaSpan.className = 'text-xs text-gray-400';
                metaSpan.textContent = formatBytes(file.size);
                infoContainer.append(nameSpan, metaSpan);
                const removeBtn = document.createElement('button');
                removeBtn.className = 'ml-4 text-red-400 hover:text-red-300 flex-shrink-0';
                removeBtn.innerHTML = '<i data-lucide="trash-2" class="w-4 h-4"></i>';
                removeBtn.onclick = () => {
                    state.files = state.files.filter((_, i) => i !== index);
                    updateUI();
                };
                fileDiv.append(infoContainer, removeBtn);
                fileDisplayArea.appendChild(fileDiv);
            }
            createIcons({ icons });
            fileControls.classList.remove('hidden');
            processBtn.classList.remove('hidden');
        } else {
            fileDisplayArea.innerHTML = '';
            fileControls.classList.add('hidden');
            processBtn.classList.add('hidden');
        }
    };

    const resetState = () => {
        state.files = [];
        updateUI();
    };

    const convert = async () => {
        if (state.files.length === 0) {
            showAlert('No Files', `Please select at least one ${FILETYPE_NAME} file.`);
            return;
        }
        try {
            showLoader('Loading engine...');
            const mupdf = await ensurePyMuPDF();

            if (state.files.length === 1) {
                const file = state.files[0];
                showLoader(`Converting ${file.name}...`);
                const pdfBlob = await mupdf.imageToPdf(file, { imageType: 'psd' });
                const baseName = file.name.replace(/\.[^/.]+$/, '');
                downloadFile(pdfBlob, `${baseName}.pdf`);
                hideLoader();
                showAlert('Conversion Complete', `Successfully converted ${file.name} to PDF.`, 'success', () => resetState());
            } else {
                showLoader('Converting multiple files...');
                const pdfBlob = await mupdf.imagesToPdf(state.files);
                downloadFile(pdfBlob, 'psd_to_pdf.pdf');
                hideLoader();
                showAlert('Conversion Complete', `Successfully converted ${state.files.length} PSD files to a single PDF.`, 'success', () => resetState());
            }
        } catch (err) {
            hideLoader();
            const message = err instanceof Error ? err.message : 'Unknown error';
            console.error(`[${FILETYPE_NAME}ToPDF] Error:`, err);
            showAlert('Error', `An error occurred during conversion. Error: ${message}`);
        }
    };

    const handleFileSelect = (files: FileList | null) => {
        if (files && files.length > 0) {
            const validFiles = Array.from(files).filter(file => {
                const ext = '.' + file.name.split('.').pop()?.toLowerCase();
                return ACCEPTED_EXTENSIONS.includes(ext);
            });
            if (validFiles.length > 0) {
                state.files = [...state.files, ...validFiles];
                updateUI();
            }
        }
    };

    if (fileInput && dropZone) {
        fileInput.addEventListener('change', (e) => handleFileSelect((e.target as HTMLInputElement).files));
        dropZone.addEventListener('dragover', (e) => { e.preventDefault(); dropZone.classList.add('bg-gray-700'); });
        dropZone.addEventListener('dragleave', (e) => { e.preventDefault(); dropZone.classList.remove('bg-gray-700'); });
        dropZone.addEventListener('drop', (e) => { e.preventDefault(); dropZone.classList.remove('bg-gray-700'); handleFileSelect(e.dataTransfer?.files ?? null); });
        fileInput.addEventListener('click', () => { fileInput.value = ''; });
    }
    if (addMoreBtn) addMoreBtn.addEventListener('click', () => fileInput.click());
    if (clearFilesBtn) clearFilesBtn.addEventListener('click', resetState);
    if (processBtn) processBtn.addEventListener('click', convert);
});