package com.ard.work.aspect;

import com.ard.common.exception.ControlException;
import com.ard.common.utils.SecurityUtils;
import com.ard.work.aspect.annotation.CheckTakeoffAltitude;
import com.ard.work.aspect.annotation.CheckControlRight;
import com.ard.work.domain.cache.UavTelemetryDataCache;
import com.ard.work.domain.dto.telemetry.TelemetryData;
import com.ard.work.service.control.UavControlService;
import com.ard.work.domain.message.UavControlType;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.lang.reflect.Field;

/**
 * AOP 切面：拦截 @RequireUavControl，检查用户是否拥有 UAV 控制权限
 *
 * 注意切面顺序：同类方法注解按字母ABCD顺序，不同类注解按order(1)顺序 越小优先级越高
 */
@Aspect
@Component
public class UavControlAspect {

    @Resource
    private UavControlService uavControlService;

    /**
     * 前置通知：控制权检测
     */
    @Before("@annotation(checkControlRight)")
    public void checkUavControl(JoinPoint joinPoint, CheckControlRight checkControlRight) {
        // 获取当前用户 ID
        Long userId = SecurityUtils.getUserId();

        // 获取控制类型
        UavControlType controlType = checkControlRight.value();

        // 获取方法参数
        Object[] args = joinPoint.getArgs();

        // 遍历参数，使用反射获取每个参数中的 boxSn
        for (Object arg : args) {
            String boxSn = getBoxSn(arg);  // 调用之前实现的 getBoxSn 方法
            if (boxSn != null) {
                // 检查当前用户是否拥有该设备的控制权限
                Long currentController = uavControlService.getCurrentController(boxSn,
                        controlType.name().toLowerCase());
                if (currentController == null || !currentController.equals(userId)) {
                    throw new ControlException("无权执行该操作，需要 " + controlType + " 控制权限");
                }
            }
        }
    }

    /**
     * 前置通知：高度检查
     */
    @Before("@annotation(checkTakeoffAltitude)")
    public void takeoffAltitudeCheck(JoinPoint joinPoint, CheckTakeoffAltitude checkTakeoffAltitude) {
        // 获取最小高度
        double minAltitude = checkTakeoffAltitude.minAltitude();
        // 获取方法参数
        Object[] args = joinPoint.getArgs();

        // 遍历参数，使用反射获取每个参数中的 boxSn
        for (Object arg : args) {
            String boxSn = getBoxSn(arg);  // 调用之前实现的 getBoxSn 方法
            if (boxSn != null) {
                TelemetryData uavTelemetryData = UavTelemetryDataCache.get(boxSn);
                if (uavTelemetryData == null) {
                    throw new ControlException("遥测数据异常，无法执行！");
                }
                // 检查高度
                if (uavTelemetryData.getAltitude() < minAltitude) {
                    throw new ControlException("高度太低，无法执行！最低高度: " + minAltitude);
                }
            }
        }
    }

    /**
     * 通过反射获取对象中的 boxSn 字段
     *
     * @param dto 要反射的对象
     * @return boxSn 字段的值，如果没有找到则返回 null
     */
    public static String getBoxSn(Object dto) {
        if (dto == null) {
            return null; // 如果对象为空，直接返回 null
        }

        try {
            // 获取类的所有字段
            Field[] fields = dto.getClass().getDeclaredFields();
            for (Field field : fields) {
                // 设置字段可访问
                field.setAccessible(true);

                // 判断是否是 boxSn 字段
                if ("boxSn".equals(field.getName())) {
                    // 检查字段类型是否是 String 类型
                    if (field.getType().equals(String.class)) {
                        // 获取并返回 boxSn 字段的值
                        return (String) field.get(dto);
                    } else {
                        // 如果字段类型不是 String，返回 null
                        return null;
                    }
                }
            }
        } catch (IllegalAccessException e) {
            // 捕获异常并打印日志
            System.err.println("反射访问失败: " + e.getMessage());
            e.printStackTrace();
        }

        return null; // 如果没有找到 boxSn 字段，返回 null
    }
}