validators.py

"""Input validation functions"""

from typing import Any, List, Optional, Union
from game.core.exceptions import ValidationException


def validate_integer(value: str, min_val: Optional[int] = None, 
                    max_val: Optional[int] = None) -> int:
    """Validate and convert string to integer"""
    try:
        num = int(value)
        
        if min_val is not None and num < min_val:
            raise ValidationException(f"Value must be at least {min_val}")
        
        if max_val is not None and num > max_val:
            raise ValidationException(f"Value must be at most {max_val}")
        
        return num
    except ValueError:
        raise ValidationException(f"'{value}' is not a valid integer")


def validate_float(value: str, min_val: Optional[float] = None,
                  max_val: Optional[float] = None) -> float:
    """Validate and convert string to float"""
    try:
        num = float(value)
        
        if min_val is not None and num < min_val:
            raise ValidationException(f"Value must be at least {min_val}")
        
        if max_val is not None and num > max_val:
            raise ValidationException(f"Value must be at most {max_val}")
        
        return num
    except ValueError:
        raise ValidationException(f"'{value}' is not a valid number")


def validate_choice(value: str, choices: List[str], case_sensitive: bool = False) -> str:
    """Validate that value is in choices"""
    if not case_sensitive:
        value = value.lower()
        choices = [c.lower() for c in choices]
    
    if value not in choices:
        choice_str = ", ".join(choices)
        raise ValidationException(f"'{value}' is not valid. Choose from: {choice_str}")
    
    return value


def validate_money(amount: int, available: int, min_amount: int = 0) -> bool:
    """Validate money transaction"""
    if amount < min_amount:
        raise ValidationException(f"Amount must be at least ${min_amount}")
    
    if amount > available:
        raise ValidationException(
            f"Insufficient funds. Required: ${amount}, Available: ${available}"
        )
    
    return True


def validate_string(value: str, min_length: int = 1, 
                   max_length: Optional[int] = None,
                   allowed_chars: Optional[str] = None) -> str:
    """Validate string"""
    if len(value) < min_length:
        raise ValidationException(f"String must be at least {min_length} characters")
    
    if max_length is not None and len(value) > max_length:
        raise ValidationException(f"String must be at most {max_length} characters")
    
    if allowed_chars is not None:
        invalid_chars = [c for c in value if c not in allowed_chars]
        if invalid_chars:
            raise ValidationException(f"Invalid characters: {invalid_chars}")
    
    return value


def validate_range(value: int, min_val: int, max_val: int) -> int:
    """Validate value is within range"""
    if value < min_val or value > max_val:
        raise ValidationException(f"Value must be between {min_val} and {max_val}")
    return value


def validate_not_empty(value: str, field_name: str = "Value") -> str:
    """Validate string is not empty"""
    stripped = value.strip()
    if not stripped:
        raise ValidationException(f"{field_name} cannot be empty")
    return stripped


def validate_yes_no(value: str) -> bool:
    """Validate yes/no input"""
    normalized = value.lower()
    if normalized in ("y", "yes"):
        return True
    elif normalized in ("n", "no"):
        return False
    else:
        raise ValidationException("Please enter 'yes' or 'no'")


def validate_list(items: List[Any], min_length: int = 0,
                 max_length: Optional[int] = None) -> List[Any]:
    """Validate list"""
    if len(items) < min_length:
        raise ValidationException(f"List must have at least {min_length} items")
    
    if max_length is not None and len(items) > max_length:
        raise ValidationException(f"List must have at most {max_length} items")
    
    return items


def validate_in_list(value: Any, items: List[Any], field_name: str = "Value") -> Any:
    """Validate value is in list"""
    if value not in items:
        raise ValidationException(f"{field_name} must be one of: {items}")
    return value


def safe_input(prompt: str, validator_func=None, error_message: str = "Invalid input"):
    """Get input with validation"""
    while True:
        try:
            user_input = input(prompt).strip()
            if validator_func:
                return validator_func(user_input)
            return user_input
        except ValidationException as e:
            print(f"Error: {e}")
        except Exception as e:
            print(f"Error: {error_message}")