examples/complex/scripts/v4/seed-with-db.ts - develop - githubmirror/strapi

examples/complex/scripts/v4/seed-with-db.ts
93 строки2 KB
Ben Irvin
test(migration): version migration testing framework (#26060)
06 авг 2026, 15:39
06 авг 2026, 15:39ac0c928
#!/usr/bin/env node

const { spawn } = require('child_process');
const path = require('path');

const PROJECT_DIR = path.resolve(__dirname, '..');
const DOCKER_COMPOSE_FILE = '__DOCKER_COMPOSE_FILE__';
const {
  getContainerId,
  isContainerRunning,
  startContainer,
  assertPostgresReady,
  assertMysqlReady,
  assertMariadbReady,
  getDatabaseEnv,
} = require('./db-utils');

const dbType = process.argv[2];
let multiplierArgIndex = 3;
if (process.argv[multiplierArgIndex] === '--') {
  multiplierArgIndex = 4;
}
const multiplier = process.argv[multiplierArgIndex] || '1';

if (!dbType || !['postgres', 'mysql', 'mariadb', 'sqlite'].includes(dbType)) {
  console.error('Error: Database type is required');
  console.error(
    'Usage: node --import tsx scripts/seed-with-db.ts <postgres|mysql|mariadb|sqlite> [multiplier]'
  );
  process.exit(1);
}

const readinessCheckers = {
  postgres: assertPostgresReady,
  mysql: assertMysqlReady,
  mariadb: assertMariadbReady,
};

function ensureContainerRunning(serviceName) {
  if (dbType === 'sqlite') return;

  const assertReady = readinessCheckers[dbType];
  const containerId = getContainerId(DOCKER_COMPOSE_FILE, PROJECT_DIR, serviceName);
  if (containerId && isContainerRunning(containerId)) {
    console.log(`✅ ${serviceName} container is already running`);
    assertReady(containerId);
    return;
  }

  console.log(`Starting ${serviceName} container...`);
  try {
    startContainer(DOCKER_COMPOSE_FILE, PROJECT_DIR, serviceName);
    console.log(`✅ ${serviceName} container started`);
    const newContainerId = getContainerId(DOCKER_COMPOSE_FILE, PROJECT_DIR, serviceName);
    assertReady(newContainerId);
  } catch (error) {
    console.error(`Error starting ${serviceName} container: ${error.message}`);
    process.exit(1);
  }
}

function runSeed() {
  if (['postgres', 'mysql', 'mariadb'].includes(dbType)) {
    ensureContainerRunning(dbType);
  }

  const env = getDatabaseEnv(dbType);

  console.log(`\n🌱 Seeding database (${dbType}) with multiplier: ${multiplier}...\n`);

  const isWindows = process.platform === 'win32';
  const seedProcess = spawn(
    isWindows ? 'node.exe' : 'node',
    ['--import', 'tsx', 'scripts/seed.ts', multiplier],
    {
      cwd: PROJECT_DIR,
      env,
      stdio: 'inherit',
      shell: !isWindows,
    }
  );

  seedProcess.on('exit', (code) => {
    process.exit(code || 0);
  });

  seedProcess.on('error', (error) => {
    console.error('Error running seed script:', error);
    process.exit(1);
  });
}

runSeed();