using Microsoft.AspNetCore.Mvc.Controllers;
using Microsoft.OpenApi.Models;
using Swashbuckle.AspNetCore.SwaggerGen;
using System;
using System.Collections.Generic;
using System.ComponentModel.DataAnnotations;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace FlexJobApi.Core
{
    public class CustomOperationIdFilter : IOperationFilter
    {
        public void Apply(OpenApiOperation operation, OperationFilterContext context)
        {
            // 获取控制器和动作信息（Furion 中通过 ActionDescriptor 获取）
            if (context.ApiDescription.ActionDescriptor is not ControllerActionDescriptor actionDescriptor)
                return;

            // 自定义 OperationId 规则：例如“控制器名_动作名”
            var controllerName = actionDescriptor.ControllerName;
            var actionName = actionDescriptor.ActionName;

            // 设置最终的 OperationId
            operation.OperationId = $"{controllerName}-{actionName}";

            // 获取接口方法的返回类型（可能是 Task<ApiResult<T>> 或 ApiResult<T>）
            var returnType = context.MethodInfo.ReturnType;

            // 步骤 1：如果是 Task 或 Task<ApiResult<T>>，提取内部类型
            if (returnType.IsGenericType && returnType.GetGenericTypeDefinition() == typeof(Task<>))
            {
                returnType = returnType.GetGenericArguments()[0]; // 提取 Task<T> 中的 T
            }

            // 步骤 3：替换 Swagger 响应类型为 Data 的类型
            operation.Responses.Clear();
            operation.Responses.Add("200", new OpenApiResponse
            {
                Description = "操作成功",
                Content = new Dictionary<string, OpenApiMediaType>
                {
                    ["application/json"] = new OpenApiMediaType
                    {
                        Schema = context.SchemaGenerator.GenerateSchema(returnType, context.SchemaRepository)
                    }
                }
            });
        }
    }
}