Introduce Spring beans analyzer project

Author: tepa46

settings.gradle.kts

@@ -69,3 +69,5 @@ if (goIde.split(",").contains(ideType)) {
     include("utbot-cli-go")
     include("utbot-intellij-go")
 }
+
+include("utbot-spring-analyzer")

utbot-spring-analyzer/build.gradle.kts

@@ -0,0 +1,15 @@
+plugins {
+    id("org.springframework.boot") version "2.7.8"
+    id("io.spring.dependency-management") version "1.1.0"
+    id("java")
+}
+
+java {
+    sourceCompatibility = JavaVersion.VERSION_11
+    targetCompatibility = JavaVersion.VERSION_11
+}
+
+dependencies {
+    implementation("org.springframework.boot:spring-boot-starter")
+    implementation("org.springframework.boot:spring-boot-starter-web")
+}

utbot-spring-analyzer/src/main/java/analyzer/SpringBeansAnalyzer.java

@@ -0,0 +1,31 @@
+package analyzer;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.builder.SpringApplicationBuilder;
+import org.springframework.context.ApplicationContextException;
+
+import java.io.IOException;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.nio.file.Path;
+
+
+@SpringBootApplication
+public class SpringBeansAnalyzer {
+
+    public static void main(String[] args) throws ClassNotFoundException, IOException {
+        ClassLoader classLoader = new URLClassLoader(new URL[]{Path.of(args[0]).toUri().toURL()});
+        Class<?> userConfigurationClass = classLoader.loadClass(args[1]);
+
+        SpringApplication app = new SpringApplicationBuilder(SpringBeansAnalyzer.class)
+                .sources(TestApplicationConfiguration.class, userConfigurationClass)
+                .build();
+
+        try {
+            app.run();
+        } catch (ApplicationContextException e) {
+            System.out.println("Bean analysis finished successfully");
+        }
+    }
+}

utbot-spring-analyzer/src/main/java/analyzer/TestApplicationConfiguration.java

@@ -0,0 +1,14 @@
+package analyzer;
+
+import org.springframework.beans.factory.config.BeanFactoryPostProcessor;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+public class TestApplicationConfiguration {
+
+    @Bean
+    public static BeanFactoryPostProcessor utBotBeanFactoryPostProcessor() {
+        return new UtBotBeanFactoryPostProcessor();
+    }
+}

utbot-spring-analyzer/src/main/java/analyzer/UtBotBeanFactoryPostProcessor.java

@@ -0,0 +1,77 @@
+package analyzer;
+
+import org.springframework.beans.BeansException;
+import org.springframework.beans.factory.annotation.AnnotatedBeanDefinition;
+import org.springframework.beans.factory.config.BeanDefinition;
+import org.springframework.beans.factory.config.BeanFactoryPostProcessor;
+import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
+import org.springframework.beans.factory.support.BeanDefinitionRegistry;
+import org.springframework.core.PriorityOrdered;
+import org.springframework.core.type.MethodMetadata;
+
+import java.io.File;
+import java.io.FileWriter;
+import java.util.ArrayList;
+import java.util.Arrays;
+
+public class UtBotBeanFactoryPostProcessor implements BeanFactoryPostProcessor, PriorityOrdered {
+
+    @Override
+    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
+
+        System.out.println("Started post-processing bean factory in UtBot");
+
+        ArrayList<String> beanClassNames = new ArrayList<>();
+        for (String beanDefinitionName : beanFactory.getBeanDefinitionNames()) {
+            BeanDefinition beanDefinition = beanFactory.getBeanDefinition(beanDefinitionName);
+
+            if (beanDefinition instanceof AnnotatedBeanDefinition) {
+                MethodMetadata factoryMethodMetadata = ((AnnotatedBeanDefinition) beanDefinition).getFactoryMethodMetadata();
+                if (factoryMethodMetadata != null) {
+                    beanClassNames.add(factoryMethodMetadata.getReturnTypeName());
+                }
+            } else {
+                String className = beanDefinition.getBeanClassName();
+                if (className == null) {
+                    className = beanFactory.getBean(beanDefinitionName).getClass().getName();
+                }
+                beanClassNames.add(className);
+            }
+        }
+        for (String beanDefinitionName : beanFactory.getBeanDefinitionNames()) {
+            BeanDefinitionRegistry beanRegistry = (BeanDefinitionRegistry) beanFactory;
+            beanRegistry.removeBeanDefinition(beanDefinitionName);
+        }
+
+        writeToFile(beanClassNames);
+    }
+
+    private void writeToFile(ArrayList<String> beanClassNames) {
+        try {
+            File springBeansFile = new File("SpringBeans.txt");
+            FileWriter fileWriter = new FileWriter(springBeansFile);
+
+            Object[] distinctClassNames = beanClassNames.stream()
+                    .distinct()
+                    .toArray();
+            Arrays.sort(distinctClassNames);
+
+            for (Object beanClassName : distinctClassNames) {
+                fileWriter.append(beanClassName.toString());
+                fileWriter.append("\n");
+            }
+
+            fileWriter.flush();
+            fileWriter.close();
+        } catch (Throwable e) {
+            System.out.println("Storing bean information failed");
+        } finally {
+            System.out.println("Finished post-processing bean factory in UtBot");
+        }
+    }
+
+    @Override
+    public int getOrder() {
+        return PriorityOrdered.HIGHEST_PRECEDENCE;
+    }
+}