#!/usr/bin/env python3
"""
Report Cookbook - Comprehensive Examples for Stellium Report Generation

This cookbook demonstrates all the ways to generate reports with Stellium,
from simple terminal output to beautiful PDF documents.

Run this script to generate example reports in examples/reports/

Usage:
    source ~/.zshrc && pyenv activate starlight
    python examples/report_cookbook.py

For full documentation, see docs/REPORTS.md

Note: PDF generation requires Typst (brew install typst or see https://typst.app)
"""

import os
from pathlib import Path

from stellium import ChartBuilder, MultiChartBuilder, ReportBuilder
from stellium.components import (
    DignityComponent,
    FixedStarsComponent,
    MidpointCalculator,
)
from stellium.engines import PlacidusHouses, WholeSignHouses
from stellium.engines.patterns import AspectPatternAnalyzer
from stellium.engines.releasing import ZodiacalReleasingAnalyzer

# Output directory for generated reports
SCRIPT_DIR = Path(__file__).resolve().parent
OUTPUT_DIR = SCRIPT_DIR / "reports"
os.makedirs(OUTPUT_DIR, exist_ok=True)


def section_header(title: str) -> None:
    """Print a formatted section header."""
    print(f"\n{'=' * 60}")
    print(f"  {title}")
    print(f"{'=' * 60}\n")


# =============================================================================
# PART 1: BASIC REPORTS
# =============================================================================


def example_1_minimal_terminal():
    """
    Example 1: Minimal Terminal Report

    The simplest possible report - just overview and positions,
    displayed directly in your terminal with Rich formatting.
    """
    section_header("Example 1: Minimal Terminal Report")

    chart = ChartBuilder.from_notable("Albert Einstein").with_aspects().calculate()

    print("Generating minimal report...\n")
    ReportBuilder().from_chart(chart).preset_minimal().render()


def example_2_standard_terminal():
    """
    Example 2: Standard Terminal Report

    A balanced report with positions, aspects, and houses.
    Great for everyday use.
    """
    section_header("Example 2: Standard Terminal Report")

    chart = ChartBuilder.from_notable("Marie Curie").with_aspects().calculate()

    print("Generating standard report...\n")
    ReportBuilder().from_chart(chart).preset_standard().render()


def example_3_detailed_terminal():
    """
    Example 3: Detailed Terminal Report

    Comprehensive report with moon phase, declinations, and dignities.
    """
    section_header("Example 3: Detailed Terminal Report")

    chart = (
        ChartBuilder.from_notable("Nikola Tesla")
        .with_aspects()
        .add_component(DignityComponent())
        .calculate()
    )

    print("Generating detailed report...\n")
    ReportBuilder().from_chart(chart).preset_detailed().render()


# =============================================================================
# PART 2: PDF REPORTS WITH TYPST
# =============================================================================


def example_4_basic_pdf():
    """
    Example 4: Basic PDF Report

    Generate a simple but beautiful PDF using Typst.
    Features Cinzel Decorative headings and Crimson Pro body text.
    """
    section_header("Example 4: Basic PDF Report")

    chart = ChartBuilder.from_notable("Frida Kahlo").with_aspects().calculate()

    output_file = OUTPUT_DIR / "basic_report.pdf"

    print(f"Generating PDF report to {output_file}...")

    ReportBuilder().from_chart(chart).preset_standard().render(
        format="pdf",
        file=str(output_file),
        show=False,
    )

    print(f"Done! Open {output_file} to see the result.")


def example_5_pdf_with_chart():
    """
    Example 5: PDF Report with Embedded Chart Wheel

    Generate a PDF with the natal chart wheel displayed on the title page.
    This is the most visually impressive output!
    """
    section_header("Example 5: PDF Report with Chart Wheel")

    chart = (
        ChartBuilder.from_notable("Carl Jung")
        .with_aspects()
        .add_component(DignityComponent())
        .calculate()
    )

    # Generate PDF with auto-generated chart image
    pdf_path = OUTPUT_DIR / "jung_full_report.pdf"
    print(f"Generating PDF report with chart to {pdf_path}...")

    (
        ReportBuilder()
        .from_chart(chart)
        .preset_detailed()
        .with_chart_image()  # Auto-generates chart wheel
        .with_title("Carl Jung - Natal Chart Analysis")
        .render(format="pdf", file=str(pdf_path))
    )

    print(f"Done! Open {pdf_path} to see the result.")


def example_6_full_pdf():
    """
    Example 6: Complete PDF Report with All Sections

    Generate a comprehensive PDF with every available section.
    Requires DignityComponent and MidpointCalculator.
    """
    section_header("Example 6: Complete PDF Report")

    chart = (
        ChartBuilder.from_notable("Albert Einstein")
        .with_house_systems([PlacidusHouses(), WholeSignHouses()])
        .with_aspects()
        .add_component(DignityComponent())
        .add_component(MidpointCalculator())
        .add_component(FixedStarsComponent())
        .add_analyzer(AspectPatternAnalyzer())
        .add_analyzer(
            ZodiacalReleasingAnalyzer(lots=["Part of Fortune", "Part of Spirit"])
        )
        .calculate()
    )

    # Generate full PDF with auto-generated chart
    pdf_path = OUTPUT_DIR / "einstein_complete_report.pdf"
    print(f"Generating complete PDF report to {pdf_path}...")

    (
        ReportBuilder()
        .from_chart(chart)
        .preset_full()
        .with_chart_image()  # Auto-generates chart wheel
        .with_title("Albert Einstein - Complete Natal Analysis")
        .render(format="pdf", file=str(pdf_path))
    )

    print(f"Done! Open {pdf_path} to see the result.")


# =============================================================================
# PART 3: CUSTOM REPORTS
# =============================================================================


def example_7_custom_sections():
    """
    Example 7: Custom Report with Selected Sections

    Build a report with exactly the sections you want.
    """
    section_header("Example 7: Custom Report")

    chart = (
        ChartBuilder.from_notable("Leonardo da Vinci")
        .with_aspects()
        .add_component(MidpointCalculator())
        .calculate()
    )

    print("Generating custom report (positions + major aspects + midpoints)...\n")

    (
        ReportBuilder()
        .from_chart(chart)
        .with_chart_overview()
        .with_planet_positions(include_speed=True, include_house=True)
        .with_aspects(mode="major", sort_by="planet")
        .with_midpoints(mode="core", threshold=10)
        .with_midpoint_aspects()  # Show planets conjunct midpoints!
        .render()
    )


def example_7b_midpoint_aspects():
    """
    Example 7b: Midpoint Aspects Report

    The most important thing about midpoints is which planets aspect them.
    A planet conjunct a midpoint "activates" that midpoint's energy.

    For example, Mars conjunct the Sun/Moon midpoint brings action and
    energy to the core self (Sun) and emotions (Moon).

    Typically only conjunctions matter (with tight orbs of 1-2°).
    """
    section_header("Example 7b: Midpoint Aspects")

    chart = (
        ChartBuilder.from_notable("Albert Einstein")
        .with_aspects()
        .add_component(MidpointCalculator())
        .calculate()
    )

    print("Generating midpoint aspects report...\n")
    print("This shows which planets activate which midpoints.\n")

    (
        ReportBuilder()
        .from_chart(chart)
        .with_chart_overview()
        .with_midpoint_aspects()  # Conjunctions within 1.5°
        .render()
    )

    print("\n--- Now with hard aspects to core midpoints ---\n")

    (
        ReportBuilder()
        .from_chart(chart)
        .with_midpoint_aspects(
            mode="hard",  # Conjunction, square, opposition
            midpoint_filter="core",  # Sun/Moon/ASC/MC midpoints only
            orb=2.0,  # Slightly wider orb
        )
        .render()
    )


def example_8_aspect_focused():
    """
    Example 8: Aspect-Focused Report

    A report that emphasizes planetary relationships.
    """
    section_header("Example 8: Aspect-Focused Report")

    chart = ChartBuilder.from_notable("Mozart").with_aspects().calculate()

    output_file = OUTPUT_DIR / "mozart_aspects.pdf"

    print(f"Generating aspect-focused PDF to {output_file}...")

    (
        ReportBuilder()
        .from_chart(chart)
        .with_chart_overview()
        .with_aspects(mode="all", orbs=True, sort_by="orb")
        .with_chart_image()
        .with_title("Wolfgang Mozart - Aspects Analysis")
        .render(format="pdf", file=str(output_file))
    )

    print(f"Done! Open {output_file} to see the result.")


def example_9_positions_only():
    """
    Example 9: Positions-Only Report

    Focus on placements without aspects - useful for quick reference.
    """
    section_header("Example 9: Positions-Only Report")

    chart = (
        ChartBuilder.from_notable("Queen Elizabeth II")
        .with_house_systems([PlacidusHouses(), WholeSignHouses()])
        .calculate()
    )

    print("Generating positions-only report...\n")

    (
        ReportBuilder()
        .from_chart(chart)
        .with_chart_overview()
        .with_planet_positions(
            include_speed=True, include_house=True, house_systems="all"
        )
        .with_declinations()
        .with_house_cusps(systems="all")
        .render()
    )


def example_9b_fixed_stars_report():
    """
    Example 9b: Fixed Stars Report

    Include fixed star positions in your report. Fixed stars add another
    layer of traditional astrological interpretation to your charts.

    Stars are organized in tiers:
    - Tier 1: Royal Stars (Aldebaran, Regulus, Antares, Fomalhaut)
    - Tier 2: Major Stars (Sirius, Algol, Spica, etc.)
    - Tier 3: Extended Stars
    """
    section_header("Example 9b: Fixed Stars Report")

    # Calculate chart with fixed stars
    chart = (
        ChartBuilder.from_notable("Aleister Crowley")
        .with_aspects()
        .add_component(FixedStarsComponent())  # All 26 stars
        .calculate()
    )

    print("Generating report with all fixed stars...\n")

    (
        ReportBuilder()
        .from_chart(chart)
        .with_chart_overview()
        .with_planet_positions()
        .with_fixed_stars()  # All stars, sorted by zodiac position
        .render()
    )


def example_9c_royal_stars_only():
    """
    Example 9c: Royal Stars Only

    The four Royal Stars of Persia are the most important fixed stars.
    This example shows how to include only these stars in your report.
    """
    section_header("Example 9c: Royal Stars Only")

    # Calculate with just royal stars (faster, less clutter)
    chart = (
        ChartBuilder.from_notable("Albert Einstein")
        .with_aspects()
        .add_component(FixedStarsComponent(royal_only=True))
        .calculate()
    )

    print("Generating report with Royal Stars only...\n")

    (
        ReportBuilder()
        .from_chart(chart)
        .with_chart_overview()
        .with_planet_positions()
        .with_fixed_stars(tier=1)  # Only Tier 1 (Royal)
        .render()
    )


def example_9d_fixed_stars_pdf():
    """
    Example 9d: Fixed Stars PDF Report

    Generate a beautiful PDF with fixed stars included.
    Stars sorted by brightness (magnitude).
    """
    section_header("Example 9d: Fixed Stars PDF")

    chart = (
        ChartBuilder.from_notable("Nostradamus")
        .with_house_systems([PlacidusHouses()])
        .with_aspects()
        .add_component(FixedStarsComponent(tier=2, include_higher_tiers=True))
        .add_component(DignityComponent())
        .calculate()
    )

    # Generate PDF with fixed stars
    pdf_path = OUTPUT_DIR / "nostradamus_fixed_stars.pdf"
    print(f"Generating fixed stars PDF to {pdf_path}...")

    (
        ReportBuilder()
        .from_chart(chart)
        .with_chart_overview()
        .with_planet_positions()
        .with_fixed_stars(sort_by="magnitude")  # Brightest first
        .with_aspects(mode="major")
        .with_dignities()
        .with_chart_image()
        .with_title("Nostradamus - Natal Chart with Fixed Stars")
        .render(format="pdf", file=str(pdf_path))
    )

    print(f"Done! Open {pdf_path} to see the result.")


# =============================================================================
# PART 4: COMPARISON REPORTS (SYNASTRY & TRANSITS)
# =============================================================================


def example_10_synastry_report():
    """
    Example 10: Synastry Report

    Generate a relationship comparison report with side-by-side tables.
    """
    section_header("Example 10: Synastry Report")

    # Calculate individual charts
    person1 = ChartBuilder.from_notable("John Lennon").with_aspects().calculate()
    person2 = ChartBuilder.from_notable("Yoko Ono").with_aspects().calculate()

    # Create synastry comparison
    synastry = (
        MultiChartBuilder.synastry(
            person1, person2, label1="John Lennon", label2="Yoko Ono"
        )
        .with_cross_aspects()
        .calculate()
    )

    # Generate synastry PDF with auto-generated biwheel
    pdf_path = OUTPUT_DIR / "lennon_ono_synastry.pdf"
    print(f"Generating synastry PDF to {pdf_path}...")

    (
        ReportBuilder()
        .from_chart(synastry)
        .preset_synastry()
        .with_chart_image()  # Auto-generates biwheel
        .with_title("John Lennon & Yoko Ono - Synastry")
        .render(format="pdf", file=str(pdf_path))
    )

    print(f"Done! Open {pdf_path} to see the result.")


def example_11_transit_report():
    """
    Example 11: Transit Report

    Show current planetary transits to a natal chart.
    """
    section_header("Example 11: Transit Report")

    from datetime import datetime

    # Natal chart
    natal = ChartBuilder.from_notable("Albert Einstein").with_aspects().calculate()

    # Transit time (use a specific date for reproducibility)
    transit_time = datetime(2025, 1, 1, 12, 0)

    # Create transit comparison (MultiChartBuilder.transit creates the transit chart internally)
    transits = (
        MultiChartBuilder.transit(
            natal,
            transit_time,
            natal_label="Natal",
            transit_label="Transits 2025-01-01",
        )
        .with_cross_aspects()
        .calculate()
    )

    # Generate transit PDF
    pdf_path = OUTPUT_DIR / "einstein_transits.pdf"
    print(f"Generating transit PDF to {pdf_path}...")

    (
        ReportBuilder()
        .from_chart(transits)
        .preset_transit()
        .with_title("Albert Einstein - Transits for Jan 1, 2025")
        .render(format="pdf", file=str(pdf_path))
    )

    print(f"Done! Open {pdf_path} to see the result.")


def example_12_custom_synastry():
    """
    Example 12: Custom Synastry Report

    Build a detailed synastry report with specific sections.
    """
    section_header("Example 12: Custom Synastry Report")

    person1 = (
        ChartBuilder.from_notable("Prince William")
        .with_aspects()
        .add_component(DignityComponent())
        .calculate()
    )
    person2 = (
        ChartBuilder.from_notable("Kate Middleton")
        .with_aspects()
        .add_component(DignityComponent())
        .calculate()
    )

    synastry = (
        MultiChartBuilder.synastry(
            person1, person2, label1="Prince William", label2="Kate Middleton"
        )
        .with_cross_aspects()
        .calculate()
    )

    # Custom detailed synastry report with auto-generated biwheel
    pdf_path = OUTPUT_DIR / "royal_synastry_detailed.pdf"
    print(f"Generating detailed synastry PDF to {pdf_path}...")

    (
        ReportBuilder()
        .from_chart(synastry)
        .with_chart_overview()
        .with_planet_positions(include_house=True, include_speed=True)
        .with_cross_aspects(mode="all", sort_by="orb")
        .with_house_cusps()
        .with_chart_image()
        .with_title("Prince William & Kate Middleton - Detailed Synastry")
        .render(format="pdf", file=str(pdf_path))
    )

    print(f"Done! Open {pdf_path} to see the result.")


# =============================================================================
# PART 5: OUTPUT FORMATS
# =============================================================================


def example_13_plain_text():
    """
    Example 13: Plain Text Output

    Generate reports as plain ASCII text - great for logs or email.
    """
    section_header("Example 13: Plain Text Output")

    chart = ChartBuilder.from_notable("Isaac Newton").with_aspects().calculate()

    output_file = OUTPUT_DIR / "newton_report.txt"
    print(f"Generating plain text report to {output_file}...")

    ReportBuilder().from_chart(chart).preset_standard().render(
        format="plain_table",
        file=str(output_file),
        show=False,
    )

    print("Done! Contents:\n")

    # Show first 40 lines
    with open(output_file) as f:
        lines = f.readlines()[:40]
        print("".join(lines))
        if len(lines) == 40:
            print("... (truncated)")


def example_14_save_and_display():
    """
    Example 14: Save to File AND Display in Terminal

    You can do both at once!
    """
    section_header("Example 14: Save and Display")

    chart = ChartBuilder.from_notable("Galileo Galilei").with_aspects().calculate()

    output_file = OUTPUT_DIR / "galileo_report.txt"

    print(f"Generating report (saving to {output_file} AND displaying)...\n")

    ReportBuilder().from_chart(chart).preset_minimal().render(
        format="rich_table",
        file=str(output_file),
        show=True,  # Display in terminal
    )

    print(f"\nAlso saved to {output_file}")


# =============================================================================
# PART 6: BATCH PROCESSING
# =============================================================================


def example_15_batch_reports():
    """
    Example 15: Batch Report Generation

    Generate reports for multiple people at once.
    """
    section_header("Example 15: Batch Report Generation")

    scientists = [
        "Albert Einstein",
        "Marie Curie",
        "Nikola Tesla",
        "Isaac Newton",
        "Galileo Galilei",
    ]

    print(f"Generating reports for {len(scientists)} scientists...\n")

    for name in scientists:
        # Calculate chart
        chart = ChartBuilder.from_notable(name).with_aspects().calculate()

        # Safe filename
        filename = name.lower().replace(" ", "_")

        # Generate PDF with auto-generated chart
        pdf_path = OUTPUT_DIR / f"{filename}_report.pdf"
        (
            ReportBuilder()
            .from_chart(chart)
            .preset_standard()
            .with_chart_image()
            .with_title(f"{name} - Natal Chart")
            .render(format="pdf", file=str(pdf_path))
        )

        print(f"  Created {pdf_path}")

    print(f"\nDone! Generated {len(scientists)} reports in {OUTPUT_DIR}")


# =============================================================================
# MAIN
# =============================================================================


def main():
    """Run selected examples."""
    print("\n" + "=" * 60)
    print("  STELLIUM REPORT COOKBOOK")
    print("=" * 60)
    print(f"\nOutput directory: {OUTPUT_DIR}\n")

    # Uncomment the examples you want to run:

    # --- Part 1: Basic Reports (Terminal) ---
    example_1_minimal_terminal()
    example_2_standard_terminal()
    example_3_detailed_terminal()

    # --- Part 2: PDF Reports ---
    example_4_basic_pdf()
    example_5_pdf_with_chart()
    example_6_full_pdf()

    # --- Part 3: Custom Reports ---
    example_7_custom_sections()
    example_7b_midpoint_aspects()
    example_8_aspect_focused()
    example_9_positions_only()
    example_9b_fixed_stars_report()
    example_9c_royal_stars_only()
    example_9d_fixed_stars_pdf()

    # --- Part 4: Comparison Reports ---
    example_10_synastry_report()
    example_11_transit_report()
    example_12_custom_synastry()

    # --- Part 5: Output Formats ---
    example_13_plain_text()
    example_14_save_and_display()

    # --- Part 6: Batch Processing ---
    example_15_batch_reports()

    print("\n" + "=" * 60)
    print("  COOKBOOK COMPLETE")
    print("=" * 60)
    print(f"\nGenerated files are in: {OUTPUT_DIR}")
    print("See docs/REPORTS.md for full documentation.\n")


if __name__ == "__main__":
    main()