package com.mes.s7.enhanced;

import com.github.xingshuangs.iot.protocol.s7.serializer.S7Variable;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.stereotype.Component;

import java.lang.reflect.Field;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Proxy;
import java.util.Map;

/**
 * @Author : zhoush
 * @Date: 2025/7/31 9:19
 * @Description:
 */
@Component
public class S7AnnotationProcessor implements BeanPostProcessor {
    @Autowired
    private S7AddressProperties addressProperties;

    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
        Class<?> clazz = bean.getClass();
        // 处理类上的注解
        processClassAnnotations(clazz);
        // 处理字段上的注解
        for (Field field : clazz.getDeclaredFields()) {
            processFieldAnnotations(field);
        }
        return bean;
    }

    private void processClassAnnotations(Class<?> clazz) {
        if (clazz.isAnnotationPresent(S7Variable.class)) {
            S7Variable annotation = clazz.getAnnotation(S7Variable.class);
            updateAnnotationValue(annotation, determineAddress(annotation.address()));
        }
    }

    private void processFieldAnnotations(Field field) {
        if (field.isAnnotationPresent(S7Variable.class)) {
            S7Variable annotation = field.getAnnotation(S7Variable.class);
            updateAnnotationValue(annotation, determineAddress(annotation.address()));
        }
    }

    private String determineAddress(String addressKey) {
        // 如果addressKey是配置中的key，则从配置获取
        if (addressProperties.getAddress().containsKey(addressKey)) {
            return addressProperties.getAddress().get(addressKey);
        }

        // 尝试直接解析为数字
        try {
            return addressKey;
        } catch (NumberFormatException e) {
            // 返回默认地址
            return "";
        }
    }

    @SuppressWarnings("unchecked")
    private void updateAnnotationValue(S7Variable annotation, String newAddress) {
        try {
            InvocationHandler handler = Proxy.getInvocationHandler(annotation);
            Field memberValuesField = handler.getClass().getDeclaredField("memberValues");
            memberValuesField.setAccessible(true);
            Map<String, Object> memberValues = (Map<String, Object>) memberValuesField.get(handler);
            memberValues.put("address", newAddress);
        } catch (Exception e) {
            throw new RuntimeException("Failed to update S7Variable annotation", e);
        }
    }
}