mirror-filecoffee-filehost/controllers/file.controller.js

const initializeLocalStorage = require("../engines/local.engine");
const initializeS3Storage = require("../engines/s3.engine");

const storageMode = process.env.STORAGE_MODE || "local";
const fileNameLength = parseInt(process.env.FILE_NAME_LENGTH, 10) || 10;
const multerOptions = {
  limits: parseInt(process.env.FILE_MAX_SIZE_MB, 10) * 1024 * 1024,
};
const publicMulterOptions = {
  limits: parseInt(process.env.PUBLIC_UPLOAD_SIZE_LIMIT, 10) * 1024 * 1024,
};

let storageEngine;

if (storageMode === "local") {
  storageEngine = initializeLocalStorage(
    multerOptions,
    fileNameLength,
    process.env.LOCAL_UPLOAD_PATH,
  );
} else if (storageMode === "s3") {
  const s3Config = {
    accessKeyId: process.env.AWS_ACCESS_KEY_ID,
    secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY,
    region: process.env.AWS_REGION,
    bucketName: process.env.S3_BUCKET_NAME,
  };
  storageEngine = initializeS3Storage(multerOptions, fileNameLength, s3Config);
} else {
  throw new Error("Invalid STORAGE_MODE");
}

const uploadFile = (req, res) => {
  storageEngine.writeFile(req, res, () => {
    const fileHostDomain =
      process.env.FILEHOST_DOMAIN || `${req.protocol}://${req.get("host")}`;
    res.status(200).json({
      message: "File uploaded successfully",
      url: `${fileHostDomain}/u/${req.filePath}`,
    });
  });
};

const getFile = (req, res) => {
  const filename = req.params.filename;
  storageEngine.findFile(filename, res);
};

module.exports = { uploadFile, getFile };