#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from typing import Annotated

from fastapi import APIRouter, Query

from backend.app.common.casbin_rbac import DependsRBAC
from backend.app.common.response.response_schema import response_base
from backend.app.schemas.dept import CreateDept, GetAllDept, UpdateDept
from backend.app.services.dept_service import DeptService
from backend.app.utils.serializers import select_to_json

router = APIRouter()


@router.get('/{pk}', summary='获取部门详情', dependencies=[DependsRBAC])
async def get_dept(pk: int):
    dept = await DeptService.get(pk=pk)
    data = GetAllDept(**select_to_json(dept))
    return await response_base.success(data=data)


@router.get('', summary='获取所有部门展示树', dependencies=[DependsRBAC])
async def get_all_depts(
    level: Annotated[int | None, Query()] = None,
    name: Annotated[str | None, Query()] = None,
    leader: Annotated[str | None, Query()] = None,
    phone: Annotated[str | None, Query()] = None,
    status: Annotated[int | None, Query()] = None,
):
    dept = await DeptService.get_dept_tree(name=name, leader=leader, phone=phone, status=status)
    return await response_base.success(data=dept)


@router.post('', summary='创建部门', dependencies=[DependsRBAC])
async def create_dept(obj: CreateDept):
    await DeptService.create(obj=obj)
    return await response_base.success()


@router.put('/{pk}', summary='更新部门', dependencies=[DependsRBAC])
async def update_dept(pk: int, obj: UpdateDept):
    count = await DeptService.update(pk=pk, obj=obj)
    if count > 0:
        return await response_base.success()
    return await response_base.fail()


@router.delete('{pk}', summary='删除部门', dependencies=[DependsRBAC])
async def delete_dept(pk: int):
    count = await DeptService.delete(pk=pk)
    if count > 0:
        return await response_base.success()
    return await response_base.fail()