"""
Additional validators for schema validation
"""

from pydantic import validator
import re


def validate_period_field(cls, v):
    """Validate period format"""
    if v:
        from app.utils.date_utils import validate_period_format
        if not validate_period_format(v):
            raise ValueError(f"Invalid period format: {v}. Expected format: Nd/Nm/Ny (e.g., 1d, 3m, 2y)")
    return v


def validate_quarters_field(cls, v):
    """Validate quarter format"""
    if v:
        for quarter in v:
            if not re.match(r'^\d{4}Q[1-4]$', quarter):
                raise ValueError(f"Invalid quarter format: {quarter}. Expected format: YYYYQN (e.g., 2020Q1)")
    return v


def validate_time_approaches(cls, v, values):
    """Ensure one of dates, quarters, or period is provided"""
    quarters = values.get('quarters')
    period = values.get('period')
    
    # Count non-None approaches
    approaches = [bool(v), bool(quarters), bool(period)]
    provided_count = sum(approaches)
    
    if provided_count == 0:
        raise ValueError("One of start_date/end_date, quarters, or period must be provided")
    if provided_count > 1:
        raise ValueError("Cannot specify multiple time approaches - use one of: date range, quarters, or period")
    return v


def validate_end_date_field(cls, v, values):
    """Validate end_date if using date-based approach"""
    start_date = values.get('start_date')
    quarters = values.get('quarters')
    period = values.get('period')
    
    if not quarters and not period:  # Using date-based approach
        if not v:
            raise ValueError("end_date is required when using date range approach")
        if start_date and v <= start_date:
            raise ValueError("end_date must be after start_date")
    return v