huridocs/uwazi

import { Application, Request, Response, NextFunction } from 'express';
// eslint-disable-next-line node/no-restricted-import
import { createWriteStream } from 'fs';
// eslint-disable-next-line node/no-restricted-import
import fs from 'fs/promises';
import QueryString from 'qs';

import { legacyLogger } from 'api/log';
import { search } from 'api/search';
import { CSVExporter } from 'api/csv';
import settings from 'api/settings';
import captchaMiddleware from 'api/auth/captchaMiddleware';
import { csvExportParamsSchema } from 'shared/types/searchParameterSchema';
import { CsvExportBody } from 'shared/types/searchParameterType';
import { temporalFilesPath, generateFileName } from './filesystem';
import { validation } from '../utils';

export default (app: Application) => {
  const generateExportFileName = (databaseName: string = '') =>
    `${databaseName}-${new Date().toISOString()}.csv`;

  const removeTempFile = (filePath: string) => async () => {
    try {
      await fs.unlink(filePath);
    } catch (err) {
      legacyLogger.error(`Error unlinking exported file: ${filePath}`);
    }
  };

  app.post(
    '/api/export',
    async (req: Request, res: Response, next: NextFunction) =>
      req.user ? next() : captchaMiddleware()(req, res, next),
    validation.validateRequest(csvExportParamsSchema),
    // eslint-disable-next-line max-statements
    async (
      req: Request<any, any, CsvExportBody, QueryString.ParsedQs, Record<string, any>>,
      res: Response,
      next: NextFunction
    ) => {
      const temporalFilePath = temporalFilesPath(generateFileName({ originalname: 'export.csv' }));
      try {
        const query = req.body;

        const results = await search.search(query, req.language, req.user);
        // eslint-disable-next-line camelcase
        const { dateFormat = '', site_name } = await settings.get();

        const exporter = new CSVExporter();

        const fileStream = createWriteStream(temporalFilePath);
        const exporterOptions = { dateFormat, language: req.language };

        await exporter.export(results, fileStream, req.hostname, query.types, exporterOptions);

        res.download(temporalFilePath, generateExportFileName(site_name), err => {
          if (err) next(err);
          //eslint-disable-next-line @typescript-eslint/no-floating-promises
          removeTempFile(temporalFilePath)();
        });
      } catch (e) {
        await removeTempFile(temporalFilePath)();
        next(e);
      }
    }
  );
};