borgwarehouse/pages/api/cron/storage.ts

import { NextApiRequest, NextApiResponse } from 'next';
import { ConfigService, ShellService } from '~/services';
import ApiResponse from '~/helpers/functions/apiResponse';
import { BorgWarehouseApiResponse } from '~/types';

export default async function handler(
  req: NextApiRequest,
  res: NextApiResponse<BorgWarehouseApiResponse>
) {
  if (!req.headers.authorization) {
    return ApiResponse.unauthorized(res);
  }

  const CRONJOB_KEY = process.env.CRONJOB_KEY;
  const ACTION_KEY = req.headers.authorization.split(' ')[1];

  if (req.method !== 'POST' || ACTION_KEY !== CRONJOB_KEY) {
    return ApiResponse.unauthorized(res);
  }

  try {
    //Check the repoList
    const repoList = await ConfigService.getRepoList();
    if (repoList.length === 0) {
      return ApiResponse.success(res, 'Storage cron executed. No repository to check.');
    }

    const storageUsed = await ShellService.getStorageUsed();

    //Update the storageUsed value of each repository
    const updatedRepoList = repoList.map((repo) => {
      const repoFiltered = storageUsed.find((x) => x.name === repo.repositoryName);
      if (!repoFiltered) return repo;
      return {
        ...repo,
        storageUsed: repoFiltered.size,
      };
    });

    await ConfigService.updateRepoList(updatedRepoList);
    return ApiResponse.success(res, 'Storage cron has been executed.');
  } catch (error) {
    console.log(error);
    if (error instanceof Error && error.message === 'The storage used service is already running') {
      return ApiResponse.conflict(res, error.message);
    } else {
      return ApiResponse.serverError(res);
    }
  }
}