Refactor: Implement multi-stage generation

This commit is contained in:
gpt-engineer-app[bot]
2025-10-10 18:07:54 +00:00
parent a7dec173df
commit 93e908aef7
2 changed files with 75 additions and 48 deletions

View File

@@ -14,6 +14,7 @@ interface SeedOptions {
includeVersionChains?: boolean;
includeEscalated?: boolean;
includeExpiredLocks?: boolean;
stage?: 'companies' | 'parks' | 'rides' | 'photos';
}
interface SeedPlan {
@@ -171,7 +172,8 @@ Deno.serve(async (req) => {
includeConflicts = false,
includeVersionChains = false,
includeEscalated = false,
includeExpiredLocks = false
includeExpiredLocks = false,
stage
}: SeedOptions = await req.json();
console.info('=== SEED DATA GENERATION DEBUG ===');
@@ -360,22 +362,23 @@ Deno.serve(async (req) => {
};
// Create companies FIRST so parks can reference operators/owners
console.info('=== COMPANY GENERATION CHECK ===');
console.info('Entity types array:', JSON.stringify(entityTypes));
console.info('Plan calls for companies:', plan.companies);
const companyTypes = ['manufacturer', 'operator', 'designer', 'property_owner'];
// First, determine which company types are selected
const selectedCompanyTypes = companyTypes.filter(compType =>
entityTypes.includes(pluralizeCompanyType(compType))
);
console.info(`✓ Selected company types (${selectedCompanyTypes.length}):`, selectedCompanyTypes);
let companiesCreatedTotal = 0;
for (let typeIndex = 0; typeIndex < selectedCompanyTypes.length; typeIndex++) {
if (!stage || stage === 'companies') {
console.info('=== COMPANY GENERATION CHECK ===');
console.info('Entity types array:', JSON.stringify(entityTypes));
console.info('Plan calls for companies:', plan.companies);
const companyTypes = ['manufacturer', 'operator', 'designer', 'property_owner'];
// First, determine which company types are selected
const selectedCompanyTypes = companyTypes.filter(compType =>
entityTypes.includes(pluralizeCompanyType(compType))
);
console.info(`✓ Selected company types (${selectedCompanyTypes.length}):`, selectedCompanyTypes);
let companiesCreatedTotal = 0;
for (let typeIndex = 0; typeIndex < selectedCompanyTypes.length; typeIndex++) {
const compType = selectedCompanyTypes[typeIndex];
const pluralType = pluralizeCompanyType(compType);
@@ -440,13 +443,10 @@ Deno.serve(async (req) => {
companiesCreatedTotal++;
}
}
}
// Create parks
console.log(`\n=== PARK GENERATION ===`);
console.log(`entityTypes includes parks: ${entityTypes.includes('parks')}`);
console.log(`plan.parks: ${plan.parks}`);
if (entityTypes.includes('parks')) {
if ((!stage || stage === 'parks') && entityTypes.includes('parks')) {
for (let i = 0; i < plan.parks; i++) {
console.log(` Creating park ${i + 1}/${plan.parks}`);
const level = getPopulationLevel(fieldDensity, i);
@@ -543,7 +543,7 @@ Deno.serve(async (req) => {
// Create rides
if (entityTypes.includes('rides') && includeDependencies && createdParks.length > 0) {
if ((!stage || stage === 'rides') && entityTypes.includes('rides') && includeDependencies && createdParks.length > 0) {
for (let i = 0; i < plan.rides; i++) {
const level = getPopulationLevel(fieldDensity, i);
const shouldConflict = includeConflicts && createdRideSlugs.length > 0 && Math.random() < 0.15;
@@ -674,7 +674,7 @@ Deno.serve(async (req) => {
}
// Create ride models
if (entityTypes.includes('ride_models') && includeDependencies && createdSubmissionItems.manufacturer.length > 0) {
if ((!stage || stage === 'rides') && entityTypes.includes('ride_models') && includeDependencies && createdSubmissionItems.manufacturer.length > 0) {
for (let i = 0; i < plan.rideModels; i++) {
const level = getPopulationLevel(fieldDensity, i);
const { data: mfgData } = await supabase.from('companies').select('id').eq('slug', randomItem(createdCompanies.manufacturer)).maybeSingle();
@@ -731,7 +731,7 @@ Deno.serve(async (req) => {
}
// Create photo submissions
if (entityTypes.includes('photos') && plan.photos > 0) {
if ((!stage || stage === 'photos') && entityTypes.includes('photos') && plan.photos > 0) {
const { data: approvedParks } = await supabase.from('parks').select('id').limit(Math.min(20, plan.photos));
const { data: approvedRides } = await supabase.from('rides').select('id, park_id').limit(Math.min(20, plan.photos));
@@ -812,7 +812,8 @@ Deno.serve(async (req) => {
JSON.stringify({
success: true,
summary,
time: (executionTime / 1000).toFixed(2)
time: (executionTime / 1000).toFixed(2),
stage: stage || 'all'
}),
{ headers: { ...corsHeaders, 'Content-Type': 'application/json' } }
);