from fastapi import APIRouter, Depends, HTTPException, UploadFile, File
from sqlalchemy.orm import Session
from typing import List
import os
from ...core.database import get_db
from ...schemas.lottery import (
    SSQLottery, SSQLotteryCreate,
    DLTLottery, DLTLotteryCreate,
    LotteryQuery
)
from ...services.lottery import LotteryService

router = APIRouter()


@router.post("/ssq/", response_model=SSQLottery)
def create_ssq_lottery(
    lottery: SSQLotteryCreate,
    db: Session = Depends(get_db)
):
    return LotteryService.create_ssq_lottery(db, lottery)


@router.post("/dlt/", response_model=DLTLottery)
def create_dlt_lottery(
    lottery: DLTLotteryCreate,
    db: Session = Depends(get_db)
):
    return LotteryService.create_dlt_lottery(db, lottery)


@router.get("/ssq/", response_model=List[SSQLottery])
def get_ssq_lotteries(
    query: LotteryQuery = Depends(),
    db: Session = Depends(get_db)
):
    lotteries, total = LotteryService.get_ssq_lotteries(db, query)
    return lotteries


@router.get("/dlt/", response_model=List[DLTLottery])
def get_dlt_lotteries(
    query: LotteryQuery = Depends(),
    db: Session = Depends(get_db)
):
    lotteries, total = LotteryService.get_dlt_lotteries(db, query)
    return lotteries


@router.get("/ssq/statistics")
def get_ssq_statistics(db: Session = Depends(get_db)):
    return LotteryService.get_ssq_statistics(db)


@router.get("/dlt/statistics")
def get_dlt_statistics(db: Session = Depends(get_db)):
    return LotteryService.get_dlt_statistics(db)


@router.post("/ssq/import")
async def import_ssq_data(
    file: UploadFile = File(...),
    db: Session = Depends(get_db)
):
    if not file.filename.endswith('.json'):
        raise HTTPException(
            status_code=400, detail="Only JSON files are allowed")

    # 保存上传的文件
    file_path = f"temp_{file.filename}"
    try:
        with open(file_path, "wb") as buffer:
            content = await file.read()
            buffer.write(content)

        # 导入数据
        count = LotteryService.import_ssq_data(db, file_path)
        return {"message": f"Successfully imported {count} records"}
    finally:
        # 清理临时文件
        if os.path.exists(file_path):
            os.remove(file_path)


@router.post("/dlt/import")
async def import_dlt_data(
    file: UploadFile = File(...),
    db: Session = Depends(get_db)
):
    if not file.filename.endswith('.json'):
        raise HTTPException(
            status_code=400, detail="Only JSON files are allowed")

    # 保存上传的文件
    file_path = f"temp_{file.filename}"
    try:
        with open(file_path, "wb") as buffer:
            content = await file.read()
            buffer.write(content)

        # 导入数据
        count = LotteryService.import_dlt_data(db, file_path)
        return {"message": f"Successfully imported {count} records"}
    finally:
        # 清理临时文件
        if os.path.exists(file_path):
            os.remove(file_path)


@router.post("/update/ssq")
async def update_ssq_data(db: Session = Depends(get_db)):
    """更新双色球数据"""
    try:
        result = LotteryService.update_ssq_data(db)
        return {
            "success": True,
            "message": f"双色球数据更新成功，新增 {result.get('new_count', 0)} 条记录",
            "data": result
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"更新双色球数据失败: {str(e)}")


@router.post("/update/dlt")
async def update_dlt_data(db: Session = Depends(get_db)):
    """更新大乐透数据"""
    try:
        result = LotteryService.update_dlt_data(db)
        return {
            "success": True,
            "message": f"大乐透数据更新成功，新增 {result.get('new_count', 0)} 条记录",
            "data": result
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"更新大乐透数据失败: {str(e)}")


@router.post("/update/all")
async def update_all_lottery_data(db: Session = Depends(get_db)):
    """更新所有彩票数据"""
    try:
        result = LotteryService.update_all_lottery_data(db)
        return {
            "success": True,
            "message": "所有彩票数据更新完成",
            "data": result
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"更新彩票数据失败: {str(e)}")