Added processing of beans from xml file

Author: tepa46

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

@@ -1,5 +1,6 @@
 package analyzer;
 
+import org.springframework.context.annotation.ImportResource;
 import org.springframework.context.annotation.PropertySource;
 
 import java.lang.annotation.Annotation;
@@ -19,16 +20,15 @@ public ConfigurationManager(ClassLoader classLoader, Class userConfigurationClas
         this.userConfigurationClass = userConfigurationClass;
     }
 
-    public void patchPropertySourceAnnotation() throws Exception {
-
+    private void patchAnnotation(Class<?> type, String path) throws Exception {
         Class<?> proxyClass = classLoader.loadClass("java.lang.reflect.Proxy");
         Field hField = proxyClass.getDeclaredField("h");
         hField.setAccessible(true);
 
         //Annotation annotationProxy = userConfigurationClass.getAnnotations()[2];
         Optional<Annotation> propertySourceAnnotation =
                 Arrays.stream(userConfigurationClass.getAnnotations())
-                        .filter(el -> el.annotationType() == PropertySource.class)
+                        .filter(el -> el.annotationType() == type)
                         .findFirst();
         if (propertySourceAnnotation.isPresent()) {
             InvocationHandler annotationInvocationHandler = (InvocationHandler) (hField.get(propertySourceAnnotation.get()));
@@ -39,7 +39,15 @@ public void patchPropertySourceAnnotation() throws Exception {
             memberValuesField.setAccessible(true);
 
             Map<String, Object> memberValues = (Map<String, Object>) (memberValuesField.get(annotationInvocationHandler));
-            memberValues.put("value", "classpath:fakeapplication.properties");
+            memberValues.put("value", path);
         }
     }
+
+    public void patchPropertySourceAnnotation() throws Exception {
+        patchAnnotation(PropertySource.class, "classpath:fakeapplication.properties");
+    }
+
+    public void patchImportResourceAnnotation() throws Exception {
+        patchAnnotation(ImportResource.class, "classpath:fakeapplication.xml");
+    }
 }

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

@@ -4,7 +4,6 @@
 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;
@@ -13,17 +12,23 @@
 public class SpringAnalysisRunner {
 
     public static void main(String[] args) throws Exception {
-        //String arg0 = "D:\\Projects\\spring-starter-lesson-28\\build\\classes\\java\\main";
+        //String arg0 = "/Users/kirillshishin/IdeaProjects/spring-starter-lesson-28/build/classes/java/main";
         //String arg1 = "com.dmdev.spring.config.ApplicationConfiguration";
-        //String arg2 = "D:\\Projects\\spring-starter-lesson-28\\build\\resources\\main\\application.properties";
+        //String arg2 = "/Users/kirillshishin/IdeaProjects/spring-starter-lesson-28/src/main/resources/application.properties";
+        //String arg3 = "/Users/kirillshishin/IdeaProjects/spring-starter-lesson-28/src/main/resources/application.xml";
 
         ClassLoader classLoader = new URLClassLoader(new URL[]{Path.of(args[0]).toUri().toURL()});
         Class<?> userConfigurationClass = classLoader.loadClass(args[1]);
 
         ConfigurationManager configurationManager = new ConfigurationManager(classLoader, userConfigurationClass);
         PropertiesAnalyzer propertiesAnalyzer = new PropertiesAnalyzer(args[2]);
+        XmlBeanAnalyzer xmlBeanAnalyzer = new XmlBeanAnalyzer(args[3]);
+
+        xmlBeanAnalyzer.rewriteXmlFile();
+        // call action to update project tree
 
         configurationManager.patchPropertySourceAnnotation();
+        configurationManager.patchImportResourceAnnotation();
 
         SpringApplicationBuilder app = new SpringApplicationBuilder(SpringAnalysisRunner.class)
                 .sources(TestApplicationConfiguration.class, userConfigurationClass);

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

@@ -0,0 +1,66 @@
+package analyzer;
+
+import org.w3c.dom.Document;
+import org.w3c.dom.Element;
+import org.w3c.dom.NodeList;
+
+import javax.xml.parsers.DocumentBuilder;
+import javax.xml.parsers.DocumentBuilderFactory;
+import javax.xml.transform.Result;
+import javax.xml.transform.Source;
+import javax.xml.transform.Transformer;
+import javax.xml.transform.TransformerException;
+import javax.xml.transform.TransformerFactory;
+import javax.xml.transform.dom.DOMSource;
+import javax.xml.transform.stream.StreamResult;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+
+
+public class XmlBeanAnalyzer {
+    //change to auto path detect
+    private final String PATH = "utbot-spring-analyzer/src/main/resources";
+    private final String xmlFilePath;
+    private final String fakeXmlFilePath;
+
+
+    public XmlBeanAnalyzer(String xmlFilePath) {
+        this.xmlFilePath = xmlFilePath;
+        this.fakeXmlFilePath = getFakePath(xmlFilePath);
+    }
+
+    private String getFakePath(String filePath) {
+        Path path = Paths.get(filePath);
+        return Paths.get(PATH, "fake" + path.getFileName()).toString();
+    }
+
+    public void rewriteXmlFile() throws Exception {
+        DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
+        DocumentBuilder builder = factory.newDocumentBuilder();
+        Document doc = builder.parse(xmlFilePath);
+
+        deletePropertyPlaceholder(doc);
+        writeXmlFile(doc);
+    }
+
+    private void deletePropertyPlaceholder(Document doc) {
+        NodeList elements = doc.getElementsByTagName("context:property-placeholder");
+        int elementsCount = elements.getLength();
+
+        //https://stackoverflow.com/questions/26618400/how-to-use-multiple-property-placeholder-in-a-spring-xml-file
+        for (int i = 0; i < elementsCount; i++) {
+            Element element = (Element) elements.item(i);
+            element.getParentNode().removeChild(element);
+        }
+        doc.normalize();
+    }
+
+    private void writeXmlFile(Document doc) throws TransformerException {
+        Source source = new DOMSource(doc);
+        Result dest = new StreamResult(fakeXmlFilePath);
+
+        TransformerFactory tFactory = TransformerFactory.newInstance();
+        Transformer tFormer = tFactory.newTransformer();
+        tFormer.transform(source, dest);
+    }
+}