from fastapi import APIRouter, Depends, HTTPException, Query
from sqlalchemy.orm import Session
from typing import List, Optional, Dict, Any
from core.database import get_db
from models.device import Device

router = APIRouter()

@router.post("/", summary="创建设备")
async def create_device(
    device_data: Dict[str, Any],
    db: Session = Depends(get_db)
):
    """创建新的设备"""
    db_device = Device(**device_data)
    db.add(db_device)
    db.commit()
    db.refresh(db_device)
    
    return {
        "id": db_device.id,
        "name": db_device.name,
        "device_type": db_device.device_type,
        "ip_address": db_device.ip_address,
        "port": db_device.port,
        "username": db_device.username,
        "password": db_device.password,
        "location": db_device.location,
        "status": db_device.status,
        "is_enabled": db_device.is_enabled,
        "description": db_device.description,
        "created_at": db_device.created_at,
        "updated_at": db_device.updated_at
    }

@router.get("/", summary="获取设备列表")
async def get_devices(
    skip: int = Query(0, ge=0, description="跳过记录数"),
    limit: int = Query(10, ge=1, le=100, description="返回记录数"),
    name: Optional[str] = Query(None, description="设备名称"),
    device_type: Optional[str] = Query(None, description="设备类型"),
    status: Optional[str] = Query(None, description="设备状态"),
    location: Optional[str] = Query(None, description="设备位置"),
    is_enabled: Optional[bool] = Query(None, description="是否启用"),
    db: Session = Depends(get_db)
):
    """获取设备列表，支持分页和筛选"""
    query = db.query(Device)
    
    if name:
        query = query.filter(Device.name.contains(name))
    if device_type:
        query = query.filter(Device.device_type == device_type)
    if status:
        query = query.filter(Device.status == status)
    if location:
        query = query.filter(Device.location.contains(location))
    if is_enabled is not None:
        query = query.filter(Device.is_enabled == is_enabled)
    
    total = query.count()
    devices = query.offset(skip).limit(limit).all()
    
    device_list = []
    for device in devices:
        device_list.append({
            "id": device.id,
            "name": device.name,
            "device_type": device.device_type,
            "ip_address": device.ip_address,
            "port": device.port,
            "username": device.username,
            "password": device.password,
            "location": device.location,
            "status": device.status,
            "is_enabled": device.is_enabled,
            "description": device.description,
            "created_at": device.created_at,
            "updated_at": device.updated_at
        })
    
    return {
        "devices": device_list,
        "total": total,
        "page": skip // limit + 1,
        "size": limit
    }

@router.get("/{device_id}", summary="获取设备详情")
async def get_device(
    device_id: int,
    db: Session = Depends(get_db)
):
    """根据ID获取设备详情"""
    device = db.query(Device).filter(Device.id == device_id).first()
    if not device:
        raise HTTPException(status_code=404, detail="设备不存在")
    
    return {
        "id": device.id,
        "name": device.name,
        "device_type": device.device_type,
        "ip_address": device.ip_address,
        "port": device.port,
        "username": device.username,
        "password": device.password,
        "location": device.location,
        "status": device.status,
        "is_enabled": device.is_enabled,
        "description": device.description,
        "created_at": device.created_at,
        "updated_at": device.updated_at
    }

@router.put("/{device_id}", summary="更新设备")
async def update_device(
    device_id: int,
    device_data: Dict[str, Any],
    db: Session = Depends(get_db)
):
    """更新设备信息"""
    db_device = db.query(Device).filter(Device.id == device_id).first()
    if not db_device:
        raise HTTPException(status_code=404, detail="设备不存在")
    
    for field, value in device_data.items():
        if hasattr(db_device, field):
            setattr(db_device, field, value)
    
    db.commit()
    db.refresh(db_device)
    
    return {
        "id": db_device.id,
        "name": db_device.name,
        "device_type": db_device.device_type,
        "ip_address": db_device.ip_address,
        "port": db_device.port,
        "username": db_device.username,
        "password": db_device.password,
        "location": db_device.location,
        "status": db_device.status,
        "is_enabled": db_device.is_enabled,
        "description": db_device.description,
        "created_at": db_device.created_at,
        "updated_at": db_device.updated_at
    }

@router.delete("/{device_id}", summary="删除设备")
async def delete_device(
    device_id: int,
    db: Session = Depends(get_db)
):
    """删除设备"""
    device = db.query(Device).filter(Device.id == device_id).first()
    if not device:
        raise HTTPException(status_code=404, detail="设备不存在")
    
    db.delete(device)
    db.commit()
    return {"message": "设备删除成功"}

@router.patch("/{device_id}/status", summary="更新设备状态")
async def update_device_status(
    device_id: int,
    status: str = Query(..., description="新状态"),
    db: Session = Depends(get_db)
):
    """更新设备状态"""
    device = db.query(Device).filter(Device.id == device_id).first()
    if not device:
        raise HTTPException(status_code=404, detail="设备不存在")
    
    device.status = status
    db.commit()
    db.refresh(device)
    
    return {
        "id": device.id,
        "name": device.name,
        "device_type": device.device_type,
        "ip_address": device.ip_address,
        "port": device.port,
        "username": device.username,
        "password": device.password,
        "location": device.location,
        "status": device.status,
        "is_enabled": device.is_enabled,
        "description": device.description,
        "created_at": device.created_at,
        "updated_at": device.updated_at
    }

@router.patch("/{device_id}/enable", summary="启用/禁用设备")
async def toggle_device_enabled(
    device_id: int,
    enabled: bool = Query(..., description="是否启用"),
    db: Session = Depends(get_db)
):
    """启用或禁用设备"""
    device = db.query(Device).filter(Device.id == device_id).first()
    if not device:
        raise HTTPException(status_code=404, detail="设备不存在")
    
    device.is_enabled = enabled
    db.commit()
    db.refresh(device)
    
    return {
        "id": device.id,
        "name": device.name,
        "device_type": device.device_type,
        "ip_address": device.ip_address,
        "port": device.port,
        "username": device.username,
        "password": device.password,
        "location": device.location,
        "status": device.status,
        "is_enabled": device.is_enabled,
        "description": device.description,
        "created_at": device.created_at,
        "updated_at": device.updated_at
    }

@router.get("/types/list", summary="获取设备类型列表")
async def get_device_types():
    """获取所有设备类型"""
    return {
        "types": [
            {"value": "camera", "label": "摄像头"},
            {"value": "sensor", "label": "传感器"},
            {"value": "gate", "label": "门禁"},
            {"value": "alarm", "label": "报警器"},
            {"value": "other", "label": "其他"}
        ]
    }

@router.get("/status/stats", summary="获取设备状态统计")
async def get_device_status_stats(db: Session = Depends(get_db)):
    """获取设备状态统计信息"""
    total = db.query(Device).count()
    online = db.query(Device).filter(Device.status == "online").count()
    offline = db.query(Device).filter(Device.status == "offline").count()
    error = db.query(Device).filter(Device.status == "error").count()
    
    return {
        "total": total,
        "online": online,
        "offline": offline,
        "error": error,
        "online_rate": round(online / total * 100, 2) if total > 0 else 0
    }