LibreChat/api/server/routes/files/images.js

const { z } = require('zod');
const fs = require('fs').promises;
const express = require('express');
const upload = require('./multer');
const { localStrategy } = require('~/server/services/Files');

const router = express.Router();

router.post('/', upload.single('file'), async (req, res) => {
  const file = req.file;
  const metadata = req.body;
  // TODO: add file size/type validation

  const uuidSchema = z.string().uuid();

  try {
    if (!file) {
      throw new Error('No file provided');
    }

    if (!metadata.file_id) {
      throw new Error('No file_id provided');
    }

    if (!metadata.width) {
      throw new Error('No width provided');
    }

    if (!metadata.height) {
      throw new Error('No height provided');
    }
    /* parse to validate api call */
    uuidSchema.parse(metadata.file_id);
    metadata.temp_file_id = metadata.file_id;
    metadata.file_id = req.file_id;
    await localStrategy({ res, file, metadata });
  } catch (error) {
    console.error('Error processing file:', error);
    try {
      await fs.unlink(file.path);
    } catch (error) {
      console.error('Error deleting file:', error);
    }
    res.status(500).json({ message: 'Error processing file' });
  }

  // do this if strategy is not local
  // finally {
  //   try {
  //     // await fs.unlink(file.path);
  //   } catch (error) {
  //     console.error('Error deleting file:', error);

  //   }
  // }
});

module.exports = router;