import multer from "multer";
import path from "path";
import response from "../../response.js";

const memoryStorage = multer.memoryStorage();

const fileFilter = (req, file, cb) => {
  const ext = path.extname(file.originalname).toLowerCase();
  if (ext === ".csv" || ext === ".xlsx") {
    cb(null, true);
  } else {
    cb(
      new Error("Invalid file type, only .csv and .xlsx files are allowed!"),
      false
    );
  }
};

const upload = multer({
  storage: memoryStorage,
  fileFilter,
  limits: { fileSize: 2 * 1024 * 1024 },
}).fields([{ name: "file", maxCount: 1 }]);

const handleCsvUpload = (req, res, next) => {
  upload(req, res, (err) => {
    if (err) {
      return response(400, null, err.message, res);
    }

    const files = req.files;
    const file = files?.file ? files.file[0] : null;

    if (!file) {
      return response(400, null, "No file uploaded!", res);
    }

    req.uploadedFile = {
      buffer: file.buffer,
      extension: path.extname(file.originalname).toLowerCase(),
    };
    next();
  });
};

export default handleCsvUpload;