io.github.butkoprojects.bots.preprocess.TelegramUpdateHandlerBeanPostProcessor Maven / Gradle / Ivy

Go to download
package io.github.butkoprojects.bots.preprocess;

import io.github.butkoprojects.bots.preprocess.annotation.*;
import io.github.butkoprojects.bots.preprocess.factory.ControllerFactory;
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.annotation.Annotation;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Component
public class TelegramUpdateHandlerBeanPostProcessor implements BeanPostProcessor {

    @Autowired
    private ControllerFactory factory;

    private final Map botControllerMap = new HashMap<>();

    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName ) throws BeansException {
        Class beanClass = bean.getClass();
        if ( beanClass.isAnnotationPresent( BotController.class ) ) {
            botControllerMap.put( beanName, beanClass );
        }
        return bean;
    }

    @Override
    public Object postProcessAfterInitialization( Object bean, String beanName ) throws BeansException {
        if ( !botControllerMap.containsKey( beanName ) ) return bean;

        Class original = botControllerMap.get( beanName );
        Arrays.stream( original.getDeclaredMethods() )
                .filter( method -> Arrays.stream( method.getAnnotations() )
                        .anyMatch( annotation -> Arrays.stream( annotation.annotationType().getAnnotations() )
                                .anyMatch( annotation2 -> annotation2.annotationType().equals( ControllerAnnotation.class ) ) ) )
                .forEach( ( Method method ) -> factory.generateController( bean, method ) );
        return bean;
    }
}