import asyncio

from fastapi import APIRouter, Request

from modules.link.schemas import LinkForm, LinkStatusForm
from core.middleware import check_demo
from core.middleware import permission_required
from modules.link import service as link

from core import R
from utils.request import parse_batch_ids

# 创建路由
router = APIRouter()


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


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


# 添加友链
@router.post('/add', summary='添加友链')
@permission_required("sys:link:add")
@check_demo
def add(request: Request, data: LinkForm):
    # 调用添加友链服务
    result = link.add_link(request, data)
    # 返回结果
    return result


# 更新友链
@router.put('/update', summary='更新友链')
@permission_required("sys:link:update")
@check_demo
def update(request: Request, data: LinkForm):
    # 调用更新友链服务方法
    result = link.update_link(request, data)
    # 返回结果
    return result


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


# 设置友链状态
@router.put('/status', summary='设置友链状态')
@permission_required("sys:link:status")
@check_demo
def status(request: Request, data: LinkStatusForm):
    # 调用设置友链状态服务方法
    result = link.update_link_status(request, data)
    # 返回结果
    return result


# 批量删除友链
@router.delete('/batchDelete', summary='批量删除友链')
@permission_required("sys:link: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(link.delete_link, id_str)