server.js

const express = require('express');
const multer = require('multer');
const cors = require('cors');
const path = require('path');
const fs = require('fs');
const crypto = require('crypto');
const pdfParse = require('pdf-parse');
const { PDFDocument } = require('pdf-lib');
const { processTickets, processTicketsList, processSklad } = require('./lib/pdfProcessor');

const app = express();
const PORT = process.env.PORT || 3000;

app.use(cors());
app.use(express.json());

const storage = multer.memoryStorage();
const upload = multer({
  storage,
  limits: { fileSize: 50 * 1024 * 1024 } // 50MB limit
});

const fontPath = path.join(__dirname, 'assets/fonts/LiberationSans-Regular.ttf');
const jobsStore = new Map();

/**
 * Auto-detect PDF type: 'sklad', 'tickets_list', or 'tickets'
 */
async function detectPdfType(buffer, filename = '') {
  const nameLower = (filename || '').toLowerCase();
  try {
    const pdfDoc = await PDFDocument.load(buffer);
    const pageCount = pdfDoc.getPageCount();
    const p0 = pdfDoc.getPage(0);
    const size = p0.getSize();

    // Small label size (e.g. 164 x 113 pt)
    if (size.width < 300 && size.height < 300) {
      return 'tickets';
    }

    const data = await pdfParse(buffer);
    const text = data.text || '';

    if (text.includes('Номер отправления') || nameLower.includes('list') || nameLower.includes('реестр') || nameLower.includes('список')) {
      return 'tickets_list';
    }
    if (text.includes('NoФотоТоварАртикулКол-во') || (text.includes('Склад:') && !text.includes('Номер отправления')) || nameLower.includes('sklad') || nameLower.includes('склад')) {
      return 'sklad';
    }
    if (pageCount > 20 || nameLower.includes('ticket') || nameLower.includes('этикетк')) {
      return 'tickets';
    }
    return 'unknown';
  } catch (err) {
    if (nameLower.includes('list')) return 'tickets_list';
    if (nameLower.includes('sklad')) return 'sklad';
    if (nameLower.includes('ticket')) return 'tickets';
    return 'unknown';
  }
}

// Endpoint to quickly auto-detect uploaded files before processing
app.post('/api/detect', upload.any(), async (req, res) => {
  try {
    const files = req.files || [];
    const results = [];

    for (const file of files) {
      const type = await detectPdfType(file.buffer, file.originalname);
      results.push({
        originalname: file.originalname,
        type: type
      });
    }

    res.json({ success: true, files: results });
  } catch (err) {
    res.status(500).json({ error: 'Ошибка авто-определения файлов: ' + err.message });
  }
});

app.post('/api/process', upload.any(), async (req, res) => {
  try {
    const files = req.files || [];
    if (files.length === 0) {
      return res.status(400).json({ error: 'Пожалуйста, загрузите хотя бы один PDF-файл (sklad.pdf, tickets_list.pdf или tickets.pdf).' });
    }

    let skladFile = null;
    let ticketsFile = null;
    let ticketsListFile = null;

    // Detect type for each uploaded file
    for (const file of files) {
      const detectedType = await detectPdfType(file.buffer, file.originalname);
      if (detectedType === 'sklad' && !skladFile) {
        skladFile = file;
      } else if (detectedType === 'tickets' && !ticketsFile) {
        ticketsFile = file;
      } else if (detectedType === 'tickets_list' && !ticketsListFile) {
        ticketsListFile = file;
      } else {
        const name = (file.originalname || '').toLowerCase();
        if (name.includes('sklad') && !skladFile) skladFile = file;
        else if (name.includes('list') && !ticketsListFile) ticketsListFile = file;
        else if (name.includes('ticket') && !ticketsFile) ticketsFile = file;
      }
    }

    const sortDirection = req.body.sortDirection || 'desc';
    let sortedSkladBuf = null;
    let sortedTicketsBuf = null;
    let sortedTicketsListBuf = null;
    const warnings = [];

    if (skladFile) {
      console.log('Processing sklad.pdf...');
      sortedSkladBuf = await processSklad(skladFile.buffer, fontPath, { sortDirection });
    }

    if (ticketsListFile) {
      console.log('Processing tickets_list.pdf...');
      sortedTicketsListBuf = await processTicketsList(ticketsListFile.buffer, fontPath);
    }

    if (ticketsFile && ticketsListFile) {
      console.log('Processing tickets.pdf + tickets_list.pdf...');
      sortedTicketsBuf = await processTickets(ticketsFile.buffer, ticketsListFile.buffer, fontPath);
    } else if (ticketsFile && !ticketsListFile) {
      warnings.push('Файл этикеток загружен, но для нанесения артикулов нужен также файл реестра (tickets_list.pdf).');
    }

    if (!sortedSkladBuf && !sortedTicketsBuf && !sortedTicketsListBuf) {
      const errMsg = warnings.length > 0
        ? warnings.join(' ')
        : 'Не удалось определить тип загруженных файлов. Убедитесь, что вы загружаете валидные PDF-документы Ozon.';
      return res.status(400).json({ error: errMsg });
    }

    const jobId = crypto.randomUUID();
    jobsStore.set(jobId, {
      tickets: sortedTicketsBuf,
      ticketsList: sortedTicketsListBuf,
      sklad: sortedSkladBuf,
      timestamp: Date.now()
    });

    // Cleanup old jobs (> 1 hour)
    for (const [id, data] of jobsStore.entries()) {
      if (Date.now() - data.timestamp > 60 * 60 * 1000) {
        jobsStore.delete(id);
      }
    }

    res.json({
      success: true,
      jobId,
      skladAvailable: !!sortedSkladBuf,
      ticketsAvailable: !!sortedTicketsBuf,
      ticketsListAvailable: !!sortedTicketsListBuf,
      downloadTicketsUrl: sortedTicketsBuf ? `/api/download/${jobId}/tickets` : null,
      downloadTicketsListUrl: sortedTicketsListBuf ? `/api/download/${jobId}/tickets_list` : null,
      downloadSkladUrl: sortedSkladBuf ? `/api/download/${jobId}/sklad` : null,
      warnings: warnings.length > 0 ? warnings : null,
      detectedFiles: {
        sklad: skladFile ? skladFile.originalname : null,
        tickets: ticketsFile ? ticketsFile.originalname : null,
        tickets_list: ticketsListFile ? ticketsListFile.originalname : null
      }
    });
  } catch (err) {
    console.error('Error processing PDFs:', err);
    res.status(500).json({ error: 'Ошибка при обработке PDF файлов: ' + err.message });
  }
});

app.get('/api/download/:jobId/:fileType', (req, res) => {
  const { jobId, fileType } = req.params;
  const job = jobsStore.get(jobId);

  if (!job) {
    return res.status(404).send('Запрошенные файлы не найдены или истек срок их хранения.');
  }

  if (fileType === 'tickets' && job.tickets) {
    res.setHeader('Content-Type', 'application/pdf');
    res.setHeader('Content-Disposition', 'attachment; filename="sorted_tickets.pdf"');
    return res.send(job.tickets);
  } else if (fileType === 'tickets_list' && job.ticketsList) {
    res.setHeader('Content-Type', 'application/pdf');
    res.setHeader('Content-Disposition', 'attachment; filename="sorted_tickets_list.pdf"');
    return res.send(job.ticketsList);
  } else if (fileType === 'sklad' && job.sklad) {
    res.setHeader('Content-Type', 'application/pdf');
    res.setHeader('Content-Disposition', 'attachment; filename="sorted_sklad.pdf"');
    return res.send(job.sklad);
  } else {
    return res.status(400).send('Некорректный тип файла или файл недоступен.');
  }
});

// Serve frontend static build if present
const staticDistPath = path.join(__dirname, 'dist');
if (fs.existsSync(staticDistPath)) {
  app.use(express.static(staticDistPath));
  app.get('*', (req, res) => {
    res.sendFile(path.join(staticDistPath, 'index.html'));
  });
}

app.listen(PORT, () => {
  console.log(`🚀 Ozon PDF Sorter Server running on port ${PORT}`);
});