import { createClient } from 'https://esm.sh/@supabase/supabase-js@2.57.4';

const corsHeaders = {
  'Access-Control-Allow-Origin': '*',
  'Access-Control-Allow-Headers': 'authorization, x-client-info, apikey, content-type',
};

interface SeedOptions {
  preset: 'small' | 'medium' | 'large' | 'stress';
  entityTypes: string[];
  fieldDensity?: 'mixed' | 'minimal' | 'standard' | 'maximum';
  includeDependencies?: boolean;
  includeConflicts?: boolean;
  includeVersionChains?: boolean;
  includeEscalated?: boolean;
  includeExpiredLocks?: boolean;
}

interface SeedPlan {
  parks: number;
  rides: number;
  companies: number;
  rideModels: number;
  photos: number;
}

const PRESETS: Record<string, SeedPlan> = {
  small: { parks: 5, rides: 10, companies: 3, rideModels: 2, photos: 5 },
  medium: { parks: 20, rides: 50, companies: 20, rideModels: 10, photos: 25 },
  large: { parks: 100, rides: 250, companies: 100, rideModels: 50, photos: 100 },
  stress: { parks: 400, rides: 1000, companies: 400, rideModels: 200, photos: 500 }
};

const CITIES = [
  { city: 'Orlando', state: 'Florida', country: 'USA' },
  { city: 'Anaheim', state: 'California', country: 'USA' },
  { city: 'Paris', state: 'Île-de-France', country: 'France' },
  { city: 'Tokyo', state: 'Tokyo', country: 'Japan' },
  { city: 'Berlin', state: 'Berlin', country: 'Germany' }
];

// Helper functions
function randomInt(min: number, max: number): number {
  return Math.floor(Math.random() * (max - min + 1)) + min;
}

function randomItem<T>(array: T[]): T {
  return array[randomInt(0, array.length - 1)];
}

function randomDate(startYear: number, endYear: number): string {
  const year = randomInt(startYear, endYear);
  const month = randomInt(1, 12);
  const day = randomInt(1, 28);
  return `${year}-${String(month).padStart(2, '0')}-${String(day).padStart(2, '0')}`;
}

function getPopulationLevel(fieldDensity: string, index: number): number {
  if (fieldDensity === 'mixed') {
    const rand = Math.random();
    if (rand < 0.1) return 0; // 10% minimal
    if (rand < 0.3) return 1; // 20% basic
    if (rand < 0.7) return 2; // 40% standard
    if (rand < 0.9) return 3; // 20% complete
    return 4; // 10% maximum
  }
  if (fieldDensity === 'minimal') return 0;
  if (fieldDensity === 'standard') return 2;
  if (fieldDensity === 'maximum') return 4;
  return 2;
}

Deno.serve(async (req) => {
  if (req.method === 'OPTIONS') {
    return new Response(null, { headers: corsHeaders });
  }

  try {
    const supabaseUrl = Deno.env.get('SUPABASE_URL')!;
    const supabaseServiceKey = Deno.env.get('SUPABASE_SERVICE_ROLE_KEY')!;
    const supabase = createClient(supabaseUrl, supabaseServiceKey);

    const authHeader = req.headers.get('Authorization');
    if (!authHeader) {
      return new Response(JSON.stringify({ error: 'No authorization header' }), {
        status: 401,
        headers: { ...corsHeaders, 'Content-Type': 'application/json' }
      });
    }

    const token = authHeader.replace('Bearer ', '');
    const { data: { user }, error: userError } = await supabase.auth.getUser(token);
    
    if (userError || !user) {
      return new Response(JSON.stringify({ error: 'Unauthorized' }), {
        status: 401,
        headers: { ...corsHeaders, 'Content-Type': 'application/json' }
      });
    }

    const { data: isMod, error: modError } = await supabase.rpc('is_moderator', { _user_id: user.id });
    if (modError || !isMod) {
      return new Response(JSON.stringify({ error: 'Insufficient permissions. Moderator role required.' }), {
        status: 403,
        headers: { ...corsHeaders, 'Content-Type': 'application/json' }
      });
    }

    const { 
      preset = 'small', 
      entityTypes = [], 
      fieldDensity = 'mixed',
      includeDependencies = true, 
      includeConflicts = false, 
      includeVersionChains = false, 
      includeEscalated = false, 
      includeExpiredLocks = false 
    }: SeedOptions = await req.json();
    
    const plan = PRESETS[preset];
    if (!plan) {
      return new Response(JSON.stringify({ error: 'Invalid preset' }), {
        status: 400,
        headers: { ...corsHeaders, 'Content-Type': 'application/json' }
      });
    }

    const startTime = Date.now();
    const summary = { parks: 0, rides: 0, companies: 0, rideModels: 0, photos: 0, totalPhotoItems: 0, conflicts: 0, versionChains: 0 };
    const createdParks: string[] = [];
    const createdCompanies: Record<string, string[]> = { manufacturer: [], operator: [], designer: [], property_owner: [] };
    const createdParkSlugs: string[] = [];
    const createdRideSlugs: string[] = [];

    // Helper to create submission
    async function createSubmission(userId: string, type: string, itemData: any, options: { escalated?: boolean; expiredLock?: boolean } = {}) {
      const submissionId = crypto.randomUUID();
      const itemId = crypto.randomUUID();
      
      const contentData = {
        action: 'create',
        metadata: {
          is_test_data: true,
          generated_at: new Date().toISOString(),
          generator_version: '2.0.0',
          preset,
          fieldDensity
        }
      };

      const submissionData: any = {
        id: submissionId,
        user_id: userId,
        submission_type: type,
        status: 'pending',
        content: contentData,
        submitted_at: new Date().toISOString()
      };

      if (options.escalated) {
        submissionData.escalated = true;
        submissionData.escalation_reason = 'Test escalation for priority testing';
      }

      if (options.expiredLock) {
        submissionData.assigned_to = userId;
        submissionData.locked_until = new Date(Date.now() - 1000 * 60 * 30).toISOString();
      }

      const { error: subError } = await supabase.from('content_submissions').insert(submissionData);
      if (subError) throw subError;

      const { error: itemError } = await supabase.from('submission_items').insert({
        id: itemId,
        submission_id: submissionId,
        item_type: type,
        item_data: itemData,
        status: 'pending',
        order_index: 0
      });
      if (itemError) throw itemError;

      const typeTableMap: Record<string, string> = {
        park: 'park_submissions',
        ride: 'ride_submissions',
        manufacturer: 'company_submissions',
        operator: 'company_submissions',
        designer: 'company_submissions',
        property_owner: 'company_submissions',
        ride_model: 'ride_model_submissions'
      };

      const table = typeTableMap[type];
      if (table) {
        const typeData = { ...itemData, submission_id: submissionId };
        if (table === 'company_submissions') {
          typeData.company_type = type;
        }
        
        const { data: insertedData, error: typeError } = await supabase.from(table).insert(typeData).select('id').single();
        if (typeError) throw typeError;
        return { submissionId, typeId: insertedData?.id };
      }

      return { submissionId, typeId: null };
    }

    // Create parks
    if (entityTypes.includes('parks')) {
      for (let i = 0; i < plan.parks; i++) {
        const level = getPopulationLevel(fieldDensity, i);
        const shouldConflict = includeConflicts && createdParkSlugs.length > 0 && Math.random() < 0.15;
        const shouldVersionChain = includeVersionChains && createdParkSlugs.length > 0 && Math.random() < 0.15;
        
        let slug = `test-park-${i + 1}`;
        if (shouldConflict) {
          slug = randomItem(createdParkSlugs);
          summary.conflicts++;
        } else if (shouldVersionChain) {
          slug = randomItem(createdParkSlugs);
          summary.versionChains++;
        }

        const parkData: any = {
          name: shouldVersionChain ? `Test Park ${slug} (Updated)` : `Test Park ${i + 1}`,
          slug: slug,
          park_type: randomItem(['theme_park', 'amusement_park', 'water_park']),
          status: 'operating'
        };

        if (level >= 1) {
          parkData.opening_date = randomDate(1950, 2024);
          parkData.description = `A ${parkData.park_type === 'theme_park' ? 'themed' : 'exciting'} park for all ages with various attractions.`;
        }

        if (level >= 2) {
          parkData.website_url = `https://test-park-${i + 1}.example.com`;
          parkData.phone = `+1-555-${randomInt(100, 999)}-${randomInt(1000, 9999)}`;
        }

        if (level >= 3 && createdCompanies.operator.length > 0) {
          const { data: operatorData } = await supabase.from('companies').select('id').eq('slug', randomItem(createdCompanies.operator)).maybeSingle();
          if (operatorData) parkData.operator_id = operatorData.id;
          parkData.email = `info@test-park-${i + 1}.example.com`;
          parkData.card_image_id = `test-park-card-${i + 1}`;
          parkData.card_image_url = `https://imagedelivery.net/test/park-${i + 1}/card`;
        }

        if (level >= 4) {
          if (createdCompanies.property_owner.length > 0) {
            const { data: ownerData } = await supabase.from('companies').select('id').eq('slug', randomItem(createdCompanies.property_owner)).maybeSingle();
            if (ownerData) parkData.property_owner_id = ownerData.id;
          }
          if (Math.random() > 0.9) {
            parkData.closing_date = randomDate(2000, 2024);
            parkData.status = 'closed';
          }
          parkData.banner_image_id = `test-park-banner-${i + 1}`;
          parkData.banner_image_url = `https://imagedelivery.net/test/park-${i + 1}/banner`;
        }

        const options = {
          escalated: includeEscalated && Math.random() < 0.1,
          expiredLock: includeExpiredLocks && Math.random() < 0.1
        };

        await createSubmission(user.id, 'park', parkData, options);
        createdParks.push(slug);
        if (!shouldConflict && !shouldVersionChain) {
          createdParkSlugs.push(slug);
        }
        summary.parks++;
      }
    }

    // Create companies
    const companyTypes = ['manufacturer', 'operator', 'designer', 'property_owner'];
    for (const compType of companyTypes) {
      if (entityTypes.includes(compType + 's') || entityTypes.includes(compType === 'manufacturer' ? 'manufacturers' : compType === 'property_owner' ? 'property_owners' : compType + 's')) {
        const count = Math.floor(plan.companies / 4);
        for (let i = 0; i < count; i++) {
          const level = getPopulationLevel(fieldDensity, i);
          const companyData: any = {
            name: `Test ${compType.replace('_', ' ')} ${i + 1}`,
            slug: `test-${compType}-${i + 1}`,
            company_type: compType
          };

          if (level >= 1) {
            companyData.description = `Leading ${compType.replace('_', ' ')} in the amusement industry.`;
            companyData.person_type = compType === 'designer' && Math.random() > 0.5 ? 'individual' : 'company';
          }

          if (level >= 2) {
            companyData.founded_year = randomInt(1950, 2020);
            const location = randomItem(CITIES);
            companyData.headquarters_location = `${location.city}, ${location.country}`;
          }

          if (level >= 3) {
            companyData.website_url = `https://test-${compType}-${i + 1}.example.com`;
            companyData.logo_url = `https://imagedelivery.net/test/${compType}-${i + 1}/logo`;
          }

          if (level >= 4) {
            companyData.card_image_id = `test-${compType}-card-${i + 1}`;
            companyData.card_image_url = `https://imagedelivery.net/test/${compType}-${i + 1}/card`;
            companyData.banner_image_id = `test-${compType}-banner-${i + 1}`;
            companyData.banner_image_url = `https://imagedelivery.net/test/${compType}-${i + 1}/banner`;
          }

          await createSubmission(user.id, compType, companyData);
          createdCompanies[compType].push(`test-${compType}-${i + 1}`);
          summary.companies++;
        }
      }
    }

    // Create rides
    if (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;
        const shouldVersionChain = includeVersionChains && createdRideSlugs.length > 0 && Math.random() < 0.15;
        
        let slug = `test-ride-${i + 1}`;
        if (shouldConflict) {
          slug = randomItem(createdRideSlugs);
          summary.conflicts++;
        } else if (shouldVersionChain) {
          slug = randomItem(createdRideSlugs);
          summary.versionChains++;
        }

        const parkSlug = randomItem(createdParks);
        const { data: parkData } = await supabase.from('parks').select('id').eq('slug', parkSlug).maybeSingle();

        const category = randomItem(['roller_coaster', 'flat_ride', 'water_ride', 'dark_ride']);
        const rideData: any = {
          name: shouldVersionChain ? `Test Ride ${slug} (Updated)` : `Test Ride ${i + 1}`,
          slug: slug,
          category: category,
          status: 'operating',
          park_id: parkData?.id || null
        };

        if (level >= 1) {
          rideData.opening_date = randomDate(2000, 2024);
          rideData.description = `An exciting ${category.replace('_', ' ')} attraction.`;
          rideData.height_requirement = randomInt(100, 140);
        }

        if (level >= 2) {
          rideData.max_speed_kmh = randomInt(40, 120);
          rideData.max_height_meters = randomInt(20, 100);
          rideData.duration_seconds = randomInt(60, 240);
          rideData.capacity_per_hour = randomInt(500, 2000);
          rideData.intensity_level = randomItem(['family', 'moderate', 'high', 'extreme']);
        }

        if (level >= 3) {
          if (createdCompanies.manufacturer.length > 0) {
            const { data: mfgData } = await supabase.from('companies').select('id').eq('slug', randomItem(createdCompanies.manufacturer)).maybeSingle();
            if (mfgData) rideData.manufacturer_id = mfgData.id;
          }
          if (category === 'roller_coaster') {
            rideData.inversions = randomInt(0, 7);
            rideData.coaster_type = randomItem(['steel', 'wooden', 'hybrid']);
            rideData.seating_type = randomItem(['sit_down', 'inverted', 'flying', 'stand_up']);
          }
          rideData.card_image_id = `test-ride-card-${i + 1}`;
          rideData.card_image_url = `https://imagedelivery.net/test/ride-${i + 1}/card`;
        }

        if (level >= 4) {
          if (createdCompanies.designer.length > 0) {
            const { data: designerData } = await supabase.from('companies').select('id').eq('slug', randomItem(createdCompanies.designer)).maybeSingle();
            if (designerData) rideData.designer_id = designerData.id;
          }
          rideData.length_meters = randomInt(500, 2000);
          rideData.drop_height_meters = randomInt(10, 80);
          rideData.max_g_force = (Math.random() * 4 + 2).toFixed(1);
          rideData.banner_image_id = `test-ride-banner-${i + 1}`;
          rideData.banner_image_url = `https://imagedelivery.net/test/ride-${i + 1}/banner`;
        }

        const { submissionId, typeId } = await createSubmission(user.id, 'ride', rideData);
        
        // Add technical specs and stats for level 4
        if (level >= 4 && typeId && category === 'roller_coaster') {
          // Add coaster stats
          for (let s = 0; s < randomInt(2, 3); s++) {
            await supabase.from('ride_submission_coaster_statistics').insert({
              ride_submission_id: typeId,
              stat_name: randomItem(['Airtime Duration', 'Zero-G Time', 'Track Gauge']),
              stat_value: Math.random() * 100,
              unit: randomItem(['seconds', 'mm']),
              category: 'technical'
            });
          }

          // Add technical specs
          for (let t = 0; t < randomInt(3, 5); t++) {
            await supabase.from('ride_submission_technical_specifications').insert({
              ride_submission_id: typeId,
              spec_name: randomItem(['Lift System', 'Brake System', 'Train Count', 'Track Material']),
              spec_value: randomItem(['Chain lift', 'Magnetic brakes', '3 trains', 'Steel tubular']),
              spec_type: 'system',
              display_order: t
            });
          }

          // Add former name
          if (Math.random() > 0.7) {
            await supabase.from('ride_submission_name_history').insert({
              ride_submission_id: typeId,
              former_name: `Original Name ${i + 1}`,
              date_changed: randomDate(2010, 2020),
              reason: 'Rebranding',
              order_index: 0
            });
          }
        }

        if (!shouldConflict && !shouldVersionChain) {
          createdRideSlugs.push(slug);
        }
        summary.rides++;
      }
    }

    // Create ride models
    if (entityTypes.includes('ride_models') && includeDependencies && createdCompanies.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();

        const category = randomItem(['roller_coaster', 'flat_ride', 'water_ride']);
        const modelData: any = {
          name: `Test Model ${i + 1}`,
          slug: `test-model-${i + 1}`,
          category: category,
          manufacturer_id: mfgData?.id || null
        };

        if (level >= 1) {
          modelData.description = `Popular ${category.replace('_', ' ')} model.`;
          modelData.ride_type = randomItem(['spinning', 'launch', 'suspended', 'family']);
        }

        if (level >= 2) {
          modelData.card_image_id = `test-model-card-${i + 1}`;
          modelData.card_image_url = `https://imagedelivery.net/test/model-${i + 1}/card`;
        }

        if (level >= 3) {
          modelData.banner_image_id = `test-model-banner-${i + 1}`;
          modelData.banner_image_url = `https://imagedelivery.net/test/model-${i + 1}/banner`;
        }

        await createSubmission(user.id, 'ride_model', modelData);
        summary.rideModels++;
      }
    }

    // Create photo submissions
    if (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));

      for (let i = 0; i < plan.photos; i++) {
        const photoCount = randomInt(1, Math.min(10, Math.ceil(plan.photos / 50) + 3));
        const submissionId = crypto.randomUUID();
        const photoSubmissionId = crypto.randomUUID();
        
        let entityType = 'park';
        let entityId = null;
        let parentId = null;

        if (Math.random() > 0.5 && approvedParks && approvedParks.length > 0) {
          entityType = 'park';
          entityId = randomItem(approvedParks).id;
        } else if (approvedRides && approvedRides.length > 0) {
          entityType = 'ride';
          const ride = randomItem(approvedRides);
          entityId = ride.id;
          parentId = ride.park_id;
        }

        if (!entityId) continue;

        // Create content_submission
        await supabase.from('content_submissions').insert({
          id: submissionId,
          user_id: user.id,
          submission_type: 'photo',
          status: 'pending',
          content: {
            action: 'create',
            metadata: {
              is_test_data: true,
              generated_at: new Date().toISOString(),
              generator_version: '2.0.0',
              preset
            }
          },
          submitted_at: new Date().toISOString()
        });

        // Create photo_submission
        await supabase.from('photo_submissions').insert({
          id: photoSubmissionId,
          submission_id: submissionId,
          entity_type: entityType,
          entity_id: entityId,
          parent_id: parentId,
          title: Math.random() > 0.5 ? `${entityType} Photos Collection ${i + 1}` : null
        });

        // Create photo_submission_items
        for (let p = 0; p < photoCount; p++) {
          const imageId = `test-photo-${crypto.randomUUID()}`;
          await supabase.from('photo_submission_items').insert({
            photo_submission_id: photoSubmissionId,
            cloudflare_image_id: imageId,
            cloudflare_image_url: `https://imagedelivery.net/test/${imageId}/public`,
            caption: Math.random() > 0.3 ? `Test photo ${p + 1} - Great view of the ${entityType}` : null,
            title: Math.random() > 0.7 ? `Photo ${p + 1}` : null,
            filename: `photo-${p + 1}.jpg`,
            order_index: p,
            file_size: randomInt(500000, 5000000),
            mime_type: 'image/jpeg',
            date_taken: Math.random() > 0.5 ? randomDate(2015, 2024) : null
          });
          summary.totalPhotoItems++;
        }
        
        summary.photos++;
      }
    }

    const executionTime = Date.now() - startTime;

    return new Response(
      JSON.stringify({
        success: true,
        summary,
        time: (executionTime / 1000).toFixed(2)
      }),
      { headers: { ...corsHeaders, 'Content-Type': 'application/json' } }
    );

  } catch (error) {
    console.error('Seed error:', error);
    return new Response(
      JSON.stringify({ error: error.message }),
      { status: 500, headers: { ...corsHeaders, 'Content-Type': 'application/json' } }
    );
  }
});