ontime/apps/server/src/utils/upload.ts

import multer from 'multer';
import path from 'path';
import fs from 'fs';
import { rm } from 'fs/promises';

import { getAppDataPath, publicDir } from '../setup/index.js';

import { ensureDirectory } from './fileManagement.js';

function generateNewFileName(filePath: string, callback: (newName: string) => void) {
  const baseName = path.basename(filePath, path.extname(filePath));
  const extension = path.extname(filePath);
  let counter = 1;

  const checkExistence = (newPath: string) => {
    fs.access(newPath, fs.constants.F_OK, (err) => {
      if (err) {
        // File with the new name does not exist, use this name
        callback(path.basename(newPath));
      } else {
        // File exists, increment the counter and try again
        newPath = path.join(path.dirname(filePath), `${baseName} (${++counter})${extension}`);
        checkExistence(newPath);
      }
    });
  };

  const newPath = path.join(path.dirname(filePath), `${baseName} (${counter})${extension}`);
  checkExistence(newPath);
}

// Define multer storage object
export const storage = multer.diskStorage({
  destination: function (_req, file, cb) {
    const appDataPath = getAppDataPath();
    if (appDataPath === '') {
      throw new Error('Could not resolve public folder for platform');
    }

    ensureDirectory(publicDir.uploadsDir);

    const filePath = path.join(publicDir.uploadsDir, file.originalname);

    // Check if file already exists
    fs.access(filePath, fs.constants.F_OK, (err) => {
      if (err) {
        // File does not exist, can safely proceed to this destination
        cb(null, publicDir.uploadsDir);
      } else {
        generateNewFileName(filePath, (newName) => {
          file.originalname = newName;
          cb(null, publicDir.uploadsDir);
        });
      }
    });
  },
  filename: function (_, file, cb) {
    cb(null, file.originalname);
  },
});

/**
 * Clears the directory that holds the uploads
 */
export async function clearUploadfolder() {
  try {
    await rm(publicDir.uploadsDir, { recursive: true });
  } catch (_) {
    // we dont care that there was no folder
  }
}