fredy/lib/services/demoCleanup.js

import { setInterval } from 'node:timers';
import { removeJobsByUserName } from './storage/jobStorage.js';
import { config } from '../utils.js';
import { getUsers } from './storage/userStorage.js';
import logger from './logger.js';

/**
 * if we are running in demo environment, we have to cleanup the db files (specifically the jobs table)
 */
export function cleanupDemoAtMidnight() {
  const now = new Date();
  const millisUntilMidnightUTC =
    (24 - now.getUTCHours()) * 60 * 60 * 1000 -
    now.getUTCMinutes() * 60 * 1000 -
    now.getUTCSeconds() * 1000 -
    now.getUTCMilliseconds();

  cleanup();
  setTimeout(() => {
    setInterval(
      () => {
        cleanup();
      },
      24 * 60 * 60 * 1000,
    );
  }, millisUntilMidnightUTC);
}

function cleanup() {
  if (config.demoMode) {
    const demoUser = getUsers(false).find((user) => user.username === 'demo');
    if (demoUser == null) {
      logger.error('Demo user not found, cannot remove Jobs');
      return;
    }
    removeJobsByUserName(demoUser.id);
  }
}