"""
Prompt template for strategy brainstorming.
"""
from typing import List
from models.input_models import EnrichedTelemetryWebhook, RaceContext
from utils.validators import TelemetryAnalyzer
from config import get_settings


def build_brainstorm_prompt_fast(
    enriched_telemetry: List[EnrichedTelemetryWebhook],
    race_context: RaceContext,
    strategy_history: List[dict] = None
) -> str:
    """Build a faster, more concise prompt for quicker responses (lap-level data)."""
    settings = get_settings()
    count = settings.strategy_count
    latest = max(enriched_telemetry, key=lambda x: x.lap)
    pit_window = latest.optimal_pit_window
    
    # Format position and competitive info
    position = race_context.driver_state.current_position
    gap_to_leader = race_context.driver_state.gap_to_leader
    gap_to_ahead = race_context.driver_state.gap_to_ahead
    comp_info = f"P{position}"
    if gap_to_ahead > 0:
        comp_info += f", {gap_to_ahead:.1f}s behind P{position-1}"
    if gap_to_leader > 0 and position > 1:
        comp_info += f", {gap_to_leader:.1f}s from leader"
    
    # Format strategy history (last 3 strategies)
    history_text = ""
    if strategy_history and len(strategy_history) > 0:
        recent_history = strategy_history[-3:]  # Last 3 strategies
        history_lines = []
        for h in recent_history:
            history_lines.append(f"Lap {h['lap']}: {h['strategy_name']} (Risk: {h['risk_level']})")
        history_text = f"\n\nPAST STRATEGIES:\n" + "\n".join(history_lines)
        history_text += f"\n\nREQUIREMENT: If changing from previous strategy '{recent_history[-1]['strategy_name']}', explain WHY the switch is necessary. If staying with same approach, explain continuity."
    else:
        history_text = "\n\nNOTE: This is the first strategy generation - no previous strategy to compare."
    
    if count == 1:
        # Ultra-fast mode: just generate 1 strategy
        return f"""Generate 1 F1 race strategy for {race_context.driver_state.driver_name} at {race_context.race_info.track_name}.

CURRENT: Lap {race_context.race_info.current_lap}/{race_context.race_info.total_laps}, {comp_info}, {race_context.driver_state.current_tire_compound} tires ({race_context.driver_state.tire_age_laps} laps old)

TELEMETRY: Tire deg {latest.tire_degradation_rate:.2f}, Cliff risk {latest.tire_cliff_risk:.2f}, Pace {latest.pace_trend}, Position trend {latest.position_trend}, Competitive pressure {latest.competitive_pressure:.2f}{history_text}

Generate 1 optimal strategy considering competitive position. Min 2 tire compounds required.

JSON: {{"strategies": [{{"strategy_id": 1, "strategy_name": "name", "stop_count": 1, "pit_laps": [32], "tire_sequence": ["medium", "hard"], "brief_description": "one sentence", "reasoning": "detailed explanation including strategy change rationale if applicable", "risk_level": "medium", "key_assumption": "main assumption"}}]}}"""
    
    elif count <= 5:
        # Fast mode: 2-5 strategies with different approaches
        return f"""Generate {count} diverse F1 race strategies for {race_context.driver_state.driver_name} at {race_context.race_info.track_name}.

CURRENT: Lap {race_context.race_info.current_lap}/{race_context.race_info.total_laps}, {comp_info}, {race_context.driver_state.current_tire_compound} tires ({race_context.driver_state.tire_age_laps} laps old)

TELEMETRY: Tire deg {latest.tire_degradation_rate:.2f}, Cliff risk {latest.tire_cliff_risk:.2f}, Pace {latest.pace_trend}, Delta {latest.performance_delta:+.2f}s, Position trend {latest.position_trend}, Competitive pressure {latest.competitive_pressure:.2f}

COMPETITIVE SITUATION: Gap to ahead {gap_to_ahead:.1f}s ({"DRS RANGE - attack opportunity!" if gap_to_ahead < 1.0 else "close battle" if gap_to_ahead < 3.0 else "need to push"}){history_text}

Generate {count} strategies balancing tire management with competitive pressure. Consider if aggressive undercut makes sense given gaps. Min 2 tire compounds each.

CRITICAL: Each strategy MUST include 'reasoning' field explaining the approach and, if applicable, why it differs from or continues the previous strategy.

JSON: {{"strategies": [{{"strategy_id": 1, "strategy_name": "Conservative Stay Out", "stop_count": 1, "pit_laps": [35], "tire_sequence": ["medium", "hard"], "brief_description": "extend current stint then hard tires to end", "reasoning": "Detailed explanation of this strategy including why we're switching from/continuing previous approach based on current telemetry and competitive situation", "risk_level": "low", "key_assumption": "tire cliff risk stays below 0.7"}}]}}"""
    
    return f"""Generate {count} F1 race strategies for {race_context.driver_state.driver_name} at {race_context.race_info.track_name}.

CURRENT: Lap {race_context.race_info.current_lap}/{race_context.race_info.total_laps}, {comp_info}, {race_context.driver_state.current_tire_compound} tires ({race_context.driver_state.tire_age_laps} laps old)

TELEMETRY: Tire deg {latest.tire_degradation_rate:.2f}, Cliff risk {latest.tire_cliff_risk:.2f}, Pace {latest.pace_trend}, Delta {latest.performance_delta:+.2f}s, Position trend {latest.position_trend}, Competitive pressure {latest.competitive_pressure:.2f}

COMPETITIVE: Gap ahead {gap_to_ahead:.1f}s, Position trending {latest.position_trend}{history_text}

Generate {count} diverse strategies considering both tire management AND competitive positioning. Min 2 compounds.

CRITICAL: Each strategy MUST include 'reasoning' field with detailed rationale and strategy continuity/change explanation.

JSON: {{"strategies": [{{"strategy_id": 1, "strategy_name": "name", "stop_count": 1, "pit_laps": [32], "tire_sequence": ["medium", "hard"], "brief_description": "one sentence", "reasoning": "Comprehensive explanation including strategy evolution context", "risk_level": "low|medium|high|critical", "key_assumption": "main assumption"}}]}}"""


def build_brainstorm_prompt(
    enriched_telemetry: List[EnrichedTelemetryWebhook],
    race_context: RaceContext,
    strategy_history: List[dict] = None
) -> str:
    """
    Build the brainstorm prompt for Gemini.
    
    Args:
        enriched_telemetry: Recent enriched telemetry data
        race_context: Current race context
        strategy_history: List of previous strategies for context
        
    Returns:
        Formatted prompt string
    """
    # Get latest telemetry
    latest = max(enriched_telemetry, key=lambda x: x.lap)
    
    # Format telemetry data (lap-level)
    telemetry_data = []
    for t in sorted(enriched_telemetry, key=lambda x: x.lap, reverse=True)[:10]:
        telemetry_data.append({
            "lap": t.lap,
            "tire_degradation_rate": round(t.tire_degradation_rate, 3),
            "pace_trend": t.pace_trend,
            "tire_cliff_risk": round(t.tire_cliff_risk, 3),
            "optimal_pit_window": t.optimal_pit_window,
            "performance_delta": round(t.performance_delta, 2),
            "competitive_pressure": round(t.competitive_pressure, 3),
            "position_trend": t.position_trend
        })
    
    # Format competitors
    competitors_data = []
    for c in race_context.competitors:
        competitors_data.append({
            "position": c.position,
            "driver": c.driver,
            "tire_compound": c.tire_compound,
            "tire_age_laps": c.tire_age_laps,
            "gap_seconds": round(c.gap_seconds, 1)
        })
    
    # Format strategy history
    history_section = ""
    if strategy_history and len(strategy_history) > 0:
        history_section = "\n\nSTRATEGY HISTORY (Recent decisions):\n"
        for h in strategy_history[-5:]:  # Last 5 strategies
            history_section += f"- Lap {h['lap']}: {h['strategy_name']} (Risk: {h['risk_level']}) - {h.get('brief_description', 'N/A')}\n"
        history_section += f"\nCONTEXT: Previous strategy was '{strategy_history[-1]['strategy_name']}'. If recommending a change, clearly explain WHY. If continuing same approach, explain the continuity.\n"
    
    prompt = f"""You are an expert F1 strategist. Generate 20 diverse race strategies based on lap-level telemetry AND competitive positioning.

LAP-LEVEL TELEMETRY METRICS:
- tire_degradation_rate: 0-1 (higher = worse tire wear)
- tire_cliff_risk: 0-1 (probability of hitting tire cliff)
- pace_trend: "improving", "stable", or "declining"
- position_trend: "gaining", "stable", or "losing" positions
- competitive_pressure: 0-1 (combined metric from position and gaps)
- optimal_pit_window: [start_lap, end_lap] recommended pit range
- performance_delta: seconds vs baseline (negative = slower)

RACE STATE:
Track: {race_context.race_info.track_name}
Current Lap: {race_context.race_info.current_lap} / {race_context.race_info.total_laps}
Weather: {race_context.race_info.weather_condition}
Track Temperature: {race_context.race_info.track_temp_celsius}°C

DRIVER STATE:
Driver: {race_context.driver_state.driver_name}
Position: P{race_context.driver_state.current_position}
Current Tires: {race_context.driver_state.current_tire_compound} ({race_context.driver_state.tire_age_laps} laps old)
Fuel Remaining: {race_context.driver_state.fuel_remaining_percent}%

COMPETITORS:
{competitors_data}

ENRICHED TELEMETRY (Last {len(telemetry_data)} laps, newest first):
{telemetry_data}{history_section}

KEY INSIGHTS:
- Latest tire degradation rate: {latest.tire_degradation_rate:.3f}
- Latest tire cliff risk: {latest.tire_cliff_risk:.3f}
- Latest pace trend: {latest.pace_trend}
- Optimal pit window: Laps {latest.optimal_pit_window[0]}-{latest.optimal_pit_window[1]}
- Laps remaining: {race_context.race_info.total_laps - race_context.race_info.current_lap}

TASK: Generate exactly 20 diverse strategies.

DIVERSITY: Conservative (1-stop), Standard (balanced), Aggressive (undercut), Reactive (competitor), Contingency (safety car)

RULES:
- Pit laps: {race_context.race_info.current_lap + 1} to {race_context.race_info.total_laps - 1}
- Min 2 tire compounds (F1 rule)
- Consider optimal pit window and tire cliff risk

For each strategy provide:
- strategy_id: 1-20
- strategy_name: Short descriptive name
- stop_count: 1, 2, or 3
- pit_laps: [array of lap numbers]
- tire_sequence: [array of compounds: "soft", "medium", "hard"]
- brief_description: One sentence rationale
- reasoning: DETAILED explanation (3-5 sentences) including: (1) Why this strategy fits current conditions, (2) How it addresses telemetry trends, (3) Strategy evolution - why changing from or continuing previous approach
- risk_level: "low", "medium", "high", or "critical"
- key_assumption: Main assumption this strategy relies on

CRITICAL: The 'reasoning' field must include strategy continuity/change rationale relative to past decisions.

OUTPUT FORMAT (JSON only, no markdown):
{{
  "strategies": [
    {{
      "strategy_id": 1,
      "strategy_name": "Conservative 1-Stop",
      "stop_count": 1,
      "pit_laps": [32],
      "tire_sequence": ["medium", "hard"],
      "brief_description": "Extend mediums to lap 32, safe finish on hards",
      "reasoning": "Current tire degradation at 0.45 suggests we can safely extend to lap 32 before hitting the cliff. This maintains our conservative approach from lap 3 as conditions haven't changed significantly - tire deg is stable and we're not under immediate competitive pressure. The hard tire finish provides low-risk race completion.",
      "risk_level": "low",
      "key_assumption": "Tire degradation stays below 0.85 until lap 32"
    }}
  ]
}}"""
    
    return prompt