import { spawnSync } from 'node:child_process';
import { mkdir, readFile, readdir, rm, writeFile } from 'node:fs/promises';
import { dirname, join, resolve } from 'node:path';
import process from 'node:process';

const outputArgument = process.argv[2];
const timestamp = new Date()
  .toISOString()
  .replaceAll(':', '-')
  .replace(/\.\d{3}Z$/, 'Z');
const outputPath = resolve(
  outputArgument || `./data/backups/shd-stage-bootstrap-${timestamp}.json`,
);
const temporaryBackupPath = `${outputPath}.database.json`;
const databaseBackup = spawnSync(
  process.execPath,
  ['scripts/db-backup.mjs', temporaryBackupPath],
  {
    cwd: process.cwd(),
    stdio: 'inherit',
    env: process.env,
  },
);
if (databaseBackup.error) throw databaseBackup.error;
if (databaseBackup.status !== 0)
  throw new Error(
    'Database backup failed; no staging bootstrap bundle was created.',
  );

async function collectFiles(directory, pattern) {
  try {
    const names = (await readdir(directory))
      .filter((name) => pattern.test(name))
      .sort();
    return await Promise.all(
      names.map(async (name) => ({
        name,
        contentBase64: (await readFile(join(directory, name))).toString(
          'base64',
        ),
      })),
    );
  } catch (error) {
    if (error?.code === 'ENOENT') return [];
    throw error;
  }
}

try {
  const database = JSON.parse(await readFile(temporaryBackupPath, 'utf8'));
  const avatars = await collectFiles(
    resolve(process.env.TDR_AVATAR_STORAGE_PATH || './data/avatars'),
    /^[0-9a-f-]{36}\.webp$/i,
  );
  const gameIcons = await collectFiles(
    resolve(process.env.TDR_GAME_ICON_STORAGE_PATH || './data/game-icons'),
    /^[a-z_]+-[0-9a-f-]{36}\.webp$/i,
  );
  const bundle = {
    format: 'shd-build-stage-bootstrap',
    version: 1,
    createdAt: new Date().toISOString(),
    database,
    files: { avatars, gameIcons },
  };
  await mkdir(dirname(outputPath), { recursive: true });
  await writeFile(outputPath, `${JSON.stringify(bundle, null, 2)}\n`, {
    encoding: 'utf8',
    flag: 'wx',
  });
  console.log(
    `Staging bootstrap bundle includes ${avatars.length} avatars and ${gameIcons.length} game-data icons.`,
  );
  console.log(`Bootstrap bundle: ${outputPath}`);
} finally {
  await rm(temporaryBackupPath, { force: true });
}