#!/usr/bin/env python3
"""
Generate a huge demo family for Gramps testing
"""

import random
from datetime import datetime, timedelta

# Generate unique handles
def gen_handle(prefix, num):
    return f"_{prefix}{num:08d}"

# Generate a person
def gen_person(pid, first_name, surname, birth_year, death_year=None, gender="M"):
    handle = gen_handle("PERSON", pid)
    birth_handle = gen_handle("EVENT", pid * 10)
    death_handle = gen_handle("EVENT", pid * 10 + 1) if death_year else None
    
    person_xml = f"""    <person handle="{handle}" change="{int(datetime.now().timestamp())}" id="I{pid:04d}">
      <gender>{gender}</gender>
      <name type="Birth Name">
        <first>{first_name}</first>
        <surname>{surname}</surname>
      </name>
      <eventref hlink="{birth_handle}" role="Primary"/>
"""
    if death_handle:
        person_xml += f"""      <eventref hlink="{death_handle}" role="Primary"/>
"""
    person_xml += """    </person>
"""
    
    # Birth event
    birth_month = random.randint(1, 12)
    birth_day = random.randint(1, 28)
    birth_event = f"""    <event handle="{birth_handle}" change="{int(datetime.now().timestamp())}" id="E{pid*10:04d}">
      <type>Birth</type>
      <dateval val="{birth_year}-{birth_month:02d}-{birth_day:02d}"/>
      <description>Birth of {surname}, {first_name}</description>
    </event>
"""
    
    # Death event
    death_event = ""
    if death_handle and death_year:
        death_month = random.randint(1, 12)
        death_day = random.randint(1, 28)
        death_event = f"""    <event handle="{death_handle}" change="{int(datetime.now().timestamp())}" id="E{pid*10+1:04d}">
      <type>Death</type>
      <dateval val="{death_year}-{death_month:02d}-{death_day:02d}"/>
      <description>Death of {surname}, {first_name}</description>
    </event>
"""
    
    return person_xml, birth_event, death_event

# Generate a family
def gen_family(fid, father_handle, mother_handle, marriage_year, children_handles):
    handle = gen_handle("FAMILY", fid)
    marriage_handle = gen_handle("EVENT", fid * 100)
    
    family_xml = f"""    <family handle="{handle}" change="{int(datetime.now().timestamp())}" id="F{fid:04d}">
      <rel type="Married"/>
      <father hlink="{father_handle}"/>
      <mother hlink="{mother_handle}"/>
"""
    for child_handle in children_handles:
        family_xml += f"""      <childref hlink="{child_handle}"/>
"""
    family_xml += f"""      <eventref hlink="{marriage_handle}" role="Family"/>
    </family>
"""
    
    # Marriage event
    marriage_month = random.randint(1, 12)
    marriage_day = random.randint(1, 28)
    marriage_event = f"""    <event handle="{marriage_handle}" change="{int(datetime.now().timestamp())}" id="E{fid*100:04d}">
      <type>Marriage</type>
      <dateval val="{marriage_year}-{marriage_month:02d}-{marriage_day:02d}"/>
      <description>Marriage</description>
    </event>
"""
    
    return family_xml, marriage_event

# First names
male_names = ["James", "John", "Robert", "Michael", "William", "David", "Richard", "Joseph", 
              "Thomas", "Charles", "Daniel", "Matthew", "Anthony", "Mark", "Donald", "Steven",
              "Paul", "Andrew", "Joshua", "Kenneth", "Kevin", "Brian", "George", "Timothy",
              "Ronald", "Jason", "Edward", "Jeffrey", "Ryan", "Jacob", "Gary", "Nicholas",
              "Eric", "Jonathan", "Stephen", "Larry", "Justin", "Scott", "Brandon", "Benjamin"]

female_names = ["Mary", "Patricia", "Jennifer", "Linda", "Elizabeth", "Barbara", "Susan", 
                "Jessica", "Sarah", "Karen", "Nancy", "Lisa", "Betty", "Margaret", "Sandra",
                "Ashley", "Kimberly", "Emily", "Donna", "Michelle", "Dorothy", "Carol",
                "Amanda", "Melissa", "Deborah", "Stephanie", "Rebecca", "Sharon", "Laura",
                "Cynthia", "Kathleen", "Amy", "Angela", "Shirley", "Anna", "Brenda", "Pamela",
                "Emma", "Nicole", "Helen", "Samantha", "Katherine", "Christine", "Debra"]

surnames = ["Smith", "Johnson", "Williams", "Brown", "Jones", "Garcia", "Miller", "Davis",
            "Rodriguez", "Martinez", "Hernandez", "Lopez", "Wilson", "Anderson", "Thomas",
            "Taylor", "Moore", "Jackson", "Martin", "Lee", "Thompson", "White", "Harris",
            "Sanchez", "Clark", "Ramirez", "Lewis", "Robinson", "Walker", "Young", "Allen",
            "King", "Wright", "Scott", "Torres", "Nguyen", "Hill", "Flores", "Green", "Adams"]

def main():
    print("Generating huge demo family...")
    
    # Generate main family
    # Father: John Smith, born 1950, died 2010
    father_id = 1
    father_handle = gen_handle("PERSON", father_id)
    father_person, father_birth, father_death = gen_person(father_id, "John", "Smith", 1950, 2010, "M")
    
    # Mother: Mary Smith, born 1952, died 2015
    mother_id = 2
    mother_handle = gen_handle("PERSON", mother_id)
    mother_person, mother_birth, mother_death = gen_person(mother_id, "Mary", "Smith", 1952, 2015, "F")
    
    # Generate 15 children
    children = []
    child_handles = []
    child_events = []
    child_id = 3
    
    for i in range(15):
        gender = "M" if i % 2 == 0 else "F"
        first_name = random.choice(male_names if gender == "M" else female_names)
        birth_year = 1970 + (i * 2)  # Spread births from 1970 to 1998
        death_year = birth_year + random.randint(60, 90) if random.random() < 0.3 else None  # 30% chance of death
        
        child_handle = gen_handle("PERSON", child_id)
        child_person, child_birth, child_death = gen_person(child_id, first_name, "Smith", birth_year, death_year, gender)
        
        children.append(child_person)
        child_handles.append(child_handle)
        child_events.append(child_birth)
        if child_death:
            child_events.append(child_death)
        child_id += 1
    
    # Generate family
    family_id = 1
    family_xml, marriage_event = gen_family(family_id, father_handle, mother_handle, 1969, child_handles)
    
    # Generate grandchildren (children of first 5 children)
    grandchildren = []
    grandchild_events = []
    grandchild_id = child_id
    
    for i in range(5):  # First 5 children have children
        parent_handle = child_handles[i]
        parent_gender = "M" if i % 2 == 0 else "F"
        spouse_gender = "F" if parent_gender == "M" else "M"
        
        # Create spouse
        spouse_name = random.choice(female_names if spouse_gender == "F" else male_names)
        spouse_birth = 1970 + (i * 2) + random.randint(-2, 2)
        spouse_handle = gen_handle("PERSON", grandchild_id)
        spouse_person, spouse_birth_event, spouse_death_event = gen_person(
            grandchild_id, spouse_name, "Smith", spouse_birth, None, spouse_gender
        )
        grandchildren.append(spouse_person)
        grandchild_events.append(spouse_birth_event)
        if spouse_death_event:
            grandchild_events.append(spouse_death_event)
        grandchild_id += 1
        
        # Create 3-5 children per couple
        num_grandchildren = random.randint(3, 5)
        grandchild_handles = []
        for j in range(num_grandchildren):
            gchild_gender = "M" if j % 2 == 0 else "F"
            gchild_name = random.choice(male_names if gchild_gender == "M" else female_names)
            gchild_birth = 1995 + (i * 3) + j
            gchild_handle = gen_handle("PERSON", grandchild_id)
            gchild_person, gchild_birth_event, gchild_death_event = gen_person(
                grandchild_id, gchild_name, "Smith", gchild_birth, None, gchild_gender
            )
            grandchildren.append(gchild_person)
            grandchild_handles.append(gchild_handle)
            grandchild_events.append(gchild_birth_event)
            if gchild_death_event:
                grandchild_events.append(gchild_death_event)
            grandchild_id += 1
        
        # Create family for this couple
        family_id += 1
        fam_xml, fam_marriage = gen_family(family_id, parent_handle, spouse_handle, 1990 + i, grandchild_handles)
        family_xml += fam_xml
        child_events.append(fam_marriage)
    
    # Write XML file
    xml_content = f"""<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE database PUBLIC "-//Gramps//DTD Gramps XML 1.7.1//EN"
"http://gramps-project.org/xml/1.7.1/grampsxml.dtd">
<database xmlns="http://gramps-project.org/xml/1.7.1/">
  <header>
    <created date="{datetime.now().strftime('%Y-%m-%d')}" version="5.1.0"/>
    <researcher>
      <resname>Demo Family Generator</resname>
    </researcher>
  </header>
  <tags>
  </tags>
  <events>
{father_birth}
{father_death}
{mother_birth}
{mother_death}
{marriage_event}
"""
    
    for event in child_events:
        xml_content += event
    for event in grandchild_events:
        xml_content += event
    
    xml_content += """  </events>
  <people>
"""
    xml_content += father_person
    xml_content += mother_person
    for child in children:
        xml_content += child
    for grandchild in grandchildren:
        xml_content += grandchild
    
    xml_content += """  </people>
  <families>
"""
    xml_content += family_xml
    
    xml_content += """  </families>
</database>
"""
    
    with open("demo_family.gramps", "w", encoding="utf-8") as f:
        f.write(xml_content)
    
    print(f"Generated demo_family.gramps with:")
    print(f"  - 2 parents (John and Mary Smith)")
    print(f"  - 15 children")
    print(f"  - 5 spouses")
    print(f"  - ~20 grandchildren")
    print(f"  - Multiple families with marriage events")
    print(f"  - Birth and death events for all")

if __name__ == "__main__":
    main()