import asyncio

from fastapi import APIRouter, Request

from modules.cms.ad.schemas import AdForm, AdStatusForm
from core.middleware import check_demo
from core.middleware import permission_required
from modules.cms.ad import service as ad

from core import R
from utils.request import parse_batch_ids

# 创建路由
router = APIRouter()


# 查询分页数据
@router.get('/page', summary='查询广告分页数据')
@permission_required("sys:ad:page")
def page(request: Request):
    # 调用查询分页数据服务方法
    result = ad.get_ad_page(request)
    # 返回结果
    return result


# 查询广告详情
@router.get('/detail/{id}', summary='查询广告详情')
@permission_required("sys:ad:detail")
def detail(request: Request, id):
    # 调用查询广告详情服务方法
    data = ad.get_ad_detail(id)
    # 返回结果
    return R.ok(data=data)


# 添加广告
@router.post('/add', summary='添加广告')
@permission_required("sys:ad:add")
@check_demo
def add(request: Request, data: AdForm):
    # 调用添加广告服务
    result = ad.add_ad(request, data)
    # 返回结果
    return result


# 更新广告
@router.put('/update', summary='更新广告')
@permission_required("sys:ad:update")
@check_demo
def update(request: Request, data: AdForm):
    # 调用更新广告服务方法
    result = ad.update_ad(request, data)
    # 返回结果
    return result


# 删除广告
@router.delete('/delete/{id}', summary='删除广告')
@permission_required("sys:ad:delete")
@check_demo
def delete(request: Request, id):
    # 调用删除广告服务方法
    result = ad.delete_ad(id)
    # 返回结果
    return result


# 设置广告状态
@router.put('/status', summary='设置广告状态')
@permission_required("sys:ad:status")
@check_demo
def status(request: Request, data: AdStatusForm):
    # 调用设置广告状态服务方法
    result = ad.update_ad_status(request, data)
    # 返回结果
    return result


# 批量删除广告
@router.delete('/batchDelete', summary='批量删除广告')
@permission_required("sys:ad:batch_delete")
@check_demo
async def batch_delete(request: Request):
    """
    批量删除广告

    Request Body:
        [1, 2, 3]  # 广告ID数组
    """
    id_str, err = await parse_batch_ids(request)
    if err:
        return err
    return await asyncio.to_thread(ad.delete_ad, id_str)