package admin_role_api

import (
	"context"

	"tydata-server/app/main/api/internal/svc"
	"tydata-server/app/main/api/internal/types"
	"tydata-server/app/main/model"
	"tydata-server/common/xerr"

	"github.com/pkg/errors"
	"github.com/zeromicro/go-zero/core/logx"
)

type AdminRemoveRoleApiLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewAdminRemoveRoleApiLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminRemoveRoleApiLogic {
	return &AdminRemoveRoleApiLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *AdminRemoveRoleApiLogic) AdminRemoveRoleApi(req *types.AdminRemoveRoleApiReq) (resp *types.AdminRemoveRoleApiResp, err error) {
	// 1. 参数验证
	if req.RoleId <= 0 {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR),
			"角色ID必须大于0, roleId: %d", req.RoleId)
	}
	if len(req.ApiIds) == 0 {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR),
			"API ID列表不能为空")
	}

	// 2. 查询角色是否存在
	_, err = l.svcCtx.AdminRoleModel.FindOne(l.ctx, req.RoleId)
	if err != nil {
		if errors.Is(err, model.ErrNotFound) {
			return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR),
				"角色不存在, roleId: %d", req.RoleId)
		}
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR),
			"查询角色失败, err: %v, roleId: %d", err, req.RoleId)
	}

	// 3. 批量移除API权限
	successCount := 0
	for _, apiId := range req.ApiIds {
		if apiId <= 0 {
			continue
		}

		// 查询关联记录
		roleApi, err := l.svcCtx.AdminRoleApiModel.FindOneByRoleIdApiId(l.ctx, req.RoleId, apiId)
		if err != nil {
			if errors.Is(err, model.ErrNotFound) {
				continue // 不存在，跳过
			}
			logx.Errorf("查询角色API关联失败, err: %v, roleId: %d, apiId: %d", err, req.RoleId, apiId)
			continue
		}

		// 删除关联
		err = l.svcCtx.AdminRoleApiModel.DeleteSoft(l.ctx, nil, roleApi)
		if err != nil {
			logx.Errorf("删除角色API关联失败, err: %v, roleId: %d, apiId: %d", err, req.RoleId, apiId)
			continue
		}

		successCount++
	}

	// 4. 返回结果
	return &types.AdminRemoveRoleApiResp{Success: true}, nil
}