# -*- coding: utf-8 -*-
# @Author  : YY

from typing import List
from typing_extensions import Annotated
from pydantic import BeforeValidator
import json
from datetime import datetime

from flask import Response

from ruoyi_common.constant import UserConstants
from ruoyi_common.base.model import AjaxResponse, TableResponse
from ruoyi_common.base.transformer import ids_to_str_list
# from ruoyi_common.domain.entity import SysCompoConf
from ruoyi_common.domain.entity import SysCompoConf
from ruoyi_common.domain.enum import BusinessType
from ruoyi_common.descriptor.serializer import JsonSerializer, BaseSerializer
from ruoyi_common.descriptor.validator import BodyValidator, QueryValidator, PathValidator, FileDownloadValidator
from ruoyi_common.utils import security_util as SecurityUtil
from ruoyi_system.service.module_conf.compo_info_conf import SysModuleCompoinfoConfService
from ruoyi_framework.descriptor.log import Log
from ruoyi_framework.descriptor.permission import HasPerm, PreAuthorize
from ... import reg


@reg.api.route("/system/moduleConf/compoConf/list", methods=["GET"])
@QueryValidator(is_page=True)
# @PreAuthorize(HasPerm("moduleConf:compoConf:list"))
@JsonSerializer()
def system_compoinfo_conf_list(dto:SysCompoConf):
    '''
        获取套餐配置列表
    '''
    rows = SysModuleCompoinfoConfService.select_compoinfo_conf_list(dto)
    table_response = TableResponse(rows=rows)
    return table_response



@reg.api.route("/system/moduleConf/compoConf/<string:id>", methods=["GET"])
@PathValidator()
@PreAuthorize(HasPerm("moduleConf:compoConf:query"))
@JsonSerializer()
def module_compoinfo_conf(id:str):
    '''
        获取套餐配置详情
    '''
    # 注意：如果需要数据权限控制，需要实现 check_compoinfo_conf_data_scope 方法
    # SysModuleCompoinfoConfService.check_compoinfo_conf_data_scope(id)
    compoinfoConf = SysModuleCompoinfoConfService.select_compoinfo_conf_by_id(id)
    ajax_response = AjaxResponse.from_success(data=compoinfoConf)
    return ajax_response


@reg.api.route("/system/moduleConf/compoConf", methods=["POST"])
@BodyValidator()
@PreAuthorize(HasPerm("moduleConf:compoConf:add"))
@Log(title="套餐配置管理",business_type=BusinessType.INSERT)
@JsonSerializer()
def system_compoinfo_conf_create(dto:SysCompoConf):
    '''
        新增套餐配置
    '''
    # if UserConstants.NOT_UNIQUE == SysModuleCompoinfoConfService.check_compoinfo_conf_name_unique(dto):
    #     ajax_response = AjaxResponse.from_error("套餐配置名称已存在")
    #     return ajax_response
    # 注意：SysCompoConf 不继承 AuditEntity，没有 create_by 等字段，所以不需要调用 create_by_user
    SysModuleCompoinfoConfService.insert_conf(dto)
    ajax_response = AjaxResponse.from_success()
    return ajax_response


@reg.api.route("/system/moduleConf/compoConf", methods=["PUT"])
@BodyValidator()
@PreAuthorize(HasPerm("moduleConf:compoConf:edit"))
@Log(title="套餐配置管理",business_type=BusinessType.UPDATE)
@JsonSerializer()
def system_compoinfo_conf_update(dto:SysCompoConf):
    '''
        修改套餐配置
    '''
    # SysModuleCompoinfoConfService.check_compoinfo_conf_data_scope(dto.compoinfoConf_id)
    # if UserConstants.UNIQUE != SysModuleCompoinfoConfService.check_compoinfo_conf_name_unique(dto):
    #     return AjaxResponse.from_error("套餐配置名称已存在")
    # 注意：SysCompoConf 不继承 AuditEntity，没有 update_by 等字段，所以不需要调用 update_by_user
    flag = SysModuleCompoinfoConfService.update_conf(dto)
    return AjaxResponse.from_success() if flag else AjaxResponse.from_error()


@reg.api.route("/system/moduleConf/compoConf/<ids>", methods=["DELETE"])
@PathValidator()
@PreAuthorize(HasPerm("moduleConf:compoConf:remove"))
@Log(title="套餐配置管理",business_type=BusinessType.DELETE)
@JsonSerializer()
def system_compoinfo_conf_delete(
    ids: Annotated[List[str], BeforeValidator(ids_to_str_list)]
):
    '''
        删除套餐配置
    '''
    # if SysModuleCompoinfoConfService.has_child_by_compoinfo_conf_id(id):
    #     return AjaxResponse.from_error("该套餐配置存在下级套餐配置，不允许删除")
    # if SysModuleCompoinfoConfService.check_compoinfo_conf_exist_user(id):
    #     return AjaxResponse.from_error("该套餐配置存在用户，不允许删除")
    # SysModuleCompoinfoConfService.check_compoinfo_conf_data_scope(id)
    flag = SysModuleCompoinfoConfService.delete_compoinfo_conf_by_id(ids)
    return AjaxResponse.from_success() if flag > 0 else AjaxResponse.from_error()


@reg.api.route("/system/moduleConf/compoConf/export", methods=["POST"])
@FileDownloadValidator()
@PreAuthorize(HasPerm("moduleConf:compoConf:export"))
@Log(title="套餐配置管理", business_type=BusinessType.EXPORT)
@BaseSerializer()
def system_compoinfo_conf_export(dto: SysCompoConf):
    '''
        导出套餐配置列表为JSON
    '''
    try:
        rows = SysModuleCompoinfoConfService.select_compoinfo_conf_list(dto)
        # 将Pydantic模型转换为字典
        data = []
        for row in rows:
            try:
                if hasattr(row, 'model_dump'):
                    data.append(row.model_dump())
                elif hasattr(row, 'dict'):
                    data.append(row.dict())
                else:
                    # 如果是字典类型，直接使用
                    data.append(dict(row) if hasattr(row, '__iter__') and not isinstance(row, str) else row)
            except Exception as e:
                # 如果序列化失败，尝试使用默认方法
                data.append(str(row))
        # 转换为JSON字符串
        json_str = json.dumps(data, ensure_ascii=False, indent=2, default=str)
        # 创建响应 - 使用application/octet-stream确保前端识别为文件下载
        response = Response(
            json_str.encode('utf-8'),
            mimetype='application/octet-stream',
            headers={
                'Content-Disposition': f'attachment; filename=compoConf_{datetime.now().strftime("%Y%m%d_%H%M%S")}.json',
                'Content-Type': 'application/json; charset=utf-8'
            }
        )
        return response
    except Exception as e:
        import traceback
        error_msg = f"导出失败: {str(e)}"
        error_json = json.dumps({"code": 500, "msg": error_msg}, ensure_ascii=False)
        error_response = Response(
            error_json,
            mimetype='application/json',
            status=500
        )
        return error_response