"""Schemas for Ban System integration in cabinet."""

from datetime import datetime
from typing import Any

from pydantic import BaseModel, Field


# === Status ===


class BanSystemStatusResponse(BaseModel):
    """Ban System integration status."""

    enabled: bool
    configured: bool


# === Stats ===


class BanSystemStatsResponse(BaseModel):
    """Overall Ban System statistics."""

    total_users: int = 0
    active_users: int = 0
    users_over_limit: int = 0
    total_requests: int = 0
    total_punishments: int = 0
    active_punishments: int = 0
    nodes_online: int = 0
    nodes_total: int = 0
    agents_online: int = 0
    agents_total: int = 0
    panel_connected: bool = False
    uptime_seconds: int | None = None


# === Users ===


class BanUserIPInfo(BaseModel):
    """User IP address information."""

    ip: str
    first_seen: datetime | None = None
    last_seen: datetime | None = None
    node: str | None = None
    request_count: int = 0
    country_code: str | None = None
    country_name: str | None = None
    city: str | None = None


class BanUserRequestLog(BaseModel):
    """User request log entry."""

    timestamp: datetime
    source_ip: str
    destination: str | None = None
    dest_port: int | None = None
    protocol: str | None = None
    action: str | None = None
    node: str | None = None


class BanUserListItem(BaseModel):
    """User in the list."""

    email: str
    unique_ip_count: int = 0
    total_requests: int = 0
    limit: int | None = None
    is_over_limit: bool = False
    blocked_count: int = 0
    last_seen: datetime | None = None


class BanUsersListResponse(BaseModel):
    """Paginated list of users."""

    users: list[BanUserListItem] = []
    total: int = 0
    offset: int = 0
    limit: int = 50


class BanUserDetailResponse(BaseModel):
    """Detailed user information."""

    email: str
    unique_ip_count: int = 0
    total_requests: int = 0
    limit: int | None = None
    is_over_limit: bool = False
    blocked_count: int = 0
    ips: list[BanUserIPInfo] = []
    recent_requests: list[BanUserRequestLog] = []
    network_type: str | None = None  # wifi, mobile, mixed


# === Punishments (Bans) ===


class BanPunishmentItem(BaseModel):
    """Punishment/ban entry."""

    id: int | None = None
    user_id: str
    uuid: str | None = None
    username: str
    reason: str | None = None
    punished_at: datetime
    enable_at: datetime | None = None
    ip_count: int = 0
    limit: int = 0
    enabled: bool = False
    enabled_at: datetime | None = None
    node_name: str | None = None


class BanPunishmentsListResponse(BaseModel):
    """List of active punishments."""

    punishments: list[BanPunishmentItem] = []
    total: int = 0


class BanHistoryResponse(BaseModel):
    """Punishment history."""

    items: list[BanPunishmentItem] = []
    total: int = 0


class BanUserRequest(BaseModel):
    """Request to ban a user."""

    username: str = Field(..., min_length=1)
    minutes: int = Field(default=30, ge=1)
    reason: str | None = Field(None, max_length=500)


class UnbanResponse(BaseModel):
    """Unban response."""

    success: bool
    message: str


# === Nodes ===


class BanNodeItem(BaseModel):
    """Node information."""

    name: str
    address: str | None = None
    is_connected: bool = False
    last_seen: datetime | None = None
    users_count: int = 0
    agent_stats: dict[str, Any] | None = None


class BanNodesListResponse(BaseModel):
    """List of nodes."""

    nodes: list[BanNodeItem] = []
    total: int = 0
    online: int = 0


# === Agents ===


class BanAgentItem(BaseModel):
    """Monitoring agent information."""

    node_name: str
    sent_total: int = 0
    dropped_total: int = 0
    batches_total: int = 0
    reconnects: int = 0
    failures: int = 0
    queue_size: int = 0
    queue_max: int = 0
    dedup_checked: int = 0
    dedup_skipped: int = 0
    filter_checked: int = 0
    filter_filtered: int = 0
    health: str = 'unknown'  # healthy, warning, critical
    is_online: bool = False
    last_report: datetime | None = None


class BanAgentsSummary(BaseModel):
    """Agents summary statistics."""

    total_agents: int = 0
    online_agents: int = 0
    total_sent: int = 0
    total_dropped: int = 0
    avg_queue_size: float = 0.0
    healthy_count: int = 0
    warning_count: int = 0
    critical_count: int = 0


class BanAgentsListResponse(BaseModel):
    """List of agents."""

    agents: list[BanAgentItem] = []
    summary: BanAgentsSummary | None = None
    total: int = 0
    online: int = 0


# === Traffic ===


class BanTrafficStats(BaseModel):
    """Traffic statistics."""

    total_bytes: int = 0
    upload_bytes: int = 0
    download_bytes: int = 0
    total_users: int = 0
    violators_count: int = 0


class BanTrafficUserItem(BaseModel):
    """User traffic information."""

    username: str
    email: str | None = None
    total_bytes: int = 0
    upload_bytes: int = 0
    download_bytes: int = 0
    limit_bytes: int | None = None
    is_over_limit: bool = False


class BanTrafficViolationItem(BaseModel):
    """Traffic limit violation entry."""

    id: int | None = None
    username: str
    email: str | None = None
    violation_type: str
    description: str | None = None
    bytes_used: int = 0
    bytes_limit: int = 0
    detected_at: datetime
    resolved: bool = False


class BanTrafficViolationsResponse(BaseModel):
    """List of traffic violations."""

    violations: list[BanTrafficViolationItem] = []
    total: int = 0


class BanTrafficTopItem(BaseModel):
    """Top user by traffic."""

    username: str
    bytes_total: int = 0
    bytes_limit: int | None = None
    over_limit: bool = False


class BanTrafficResponse(BaseModel):
    """Full traffic statistics response."""

    enabled: bool = False
    stats: dict[str, Any] | None = None
    top_users: list[BanTrafficTopItem] = []
    recent_violations: list[BanTrafficViolationItem] = []


# === Settings ===


class BanSettingDefinition(BaseModel):
    """Setting definition with value."""

    key: str
    value: Any
    type: str  # bool, int, str, list
    min_value: int | None = None
    max_value: int | None = None
    editable: bool = True
    description: str | None = None
    category: str | None = None


class BanSettingsResponse(BaseModel):
    """All settings response."""

    settings: list[BanSettingDefinition] = []


class BanSettingUpdateRequest(BaseModel):
    """Request to update a setting."""

    value: Any


class BanWhitelistRequest(BaseModel):
    """Request to add/remove from whitelist."""

    username: str = Field(..., min_length=1)


# === Reports ===


class BanReportTopViolator(BaseModel):
    """Top violator in report."""

    username: str
    count: int = 0


class BanReportResponse(BaseModel):
    """Period report response."""

    period_hours: int = 24
    current_users: int = 0
    current_ips: int = 0
    punishment_stats: dict[str, Any] | None = None
    top_violators: list[BanReportTopViolator] = []


# === Health ===


class BanHealthComponent(BaseModel):
    """Health component status."""

    name: str
    status: str  # healthy, degraded, unhealthy
    message: str | None = None
    details: dict[str, Any] | None = None


class BanHealthResponse(BaseModel):
    """Health status response."""

    status: str  # healthy, degraded, unhealthy
    uptime: int | None = None
    components: list[BanHealthComponent] = []


class BanHealthDetailedResponse(BaseModel):
    """Detailed health response."""

    status: str
    uptime: int | None = None
    components: dict[str, Any] = {}


# === Agent History ===


class BanAgentHistoryItem(BaseModel):
    """Agent history item."""

    timestamp: datetime
    sent_total: int = 0
    dropped_total: int = 0
    queue_size: int = 0
    batches_total: int = 0


class BanAgentHistoryResponse(BaseModel):
    """Agent history response."""

    node: str
    hours: int = 24
    records: int = 0
    delta: dict[str, Any] | None = None
    first: dict[str, Any] | None = None
    last: dict[str, Any] | None = None
    history: list[BanAgentHistoryItem] = []