feat:初始化文件树

Author: ZhiFengJia

remote-core/src/main/java/com/jzf/remote/core/JavaFileObjectByteCode.java

@@ -20,7 +20,7 @@ public class JavaFileObjectByteCode extends SimpleJavaFileObject {
     private final ByteArrayOutputStream byteArrayOutputStream;
 
     protected JavaFileObjectByteCode(String className) {
-        super(URI.create("string:///" + className.replaceAll("\\.", "/") + Kind.CLASS.extension), Kind.CLASS);
+        super(URI.create("string:///" + className + Kind.CLASS.extension), Kind.CLASS);
         this.byteArrayOutputStream = new ByteArrayOutputStream();
     }
 

remote-core/src/main/java/com/jzf/remote/core/JavaFileObjectSource.java

@@ -16,7 +16,7 @@ public class JavaFileObjectSource extends SimpleJavaFileObject {
     private final CharSequence content;
 
     protected JavaFileObjectSource(String className, CharSequence content) {
-        super(URI.create("string:///" + className.replaceAll("\\.", "/") + Kind.SOURCE.extension), Kind.SOURCE);
+        super(URI.create("string:///" + className + Kind.SOURCE.extension), Kind.SOURCE);
         this.content = content;
     }
 

remote-web/pom.xml

@@ -54,6 +54,10 @@
             <artifactId>commons-io</artifactId>
             <version>2.11.0</version>
         </dependency>
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+        </dependency>
 
         <dependency>
             <groupId>org.springframework.boot</groupId>

remote-web/src/assembly/assembly.xml

@@ -1,4 +1,6 @@
-<assembly xmlns="http://maven.apache.org/plugins/maven-assembly-plugin/assembly/1.1.3" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/plugins/maven-assembly-plugin/assembly/1.1.3 http://maven.apache.org/xsd/assembly-1.1.3.xsd">
+<assembly xmlns="http://maven.apache.org/plugins/maven-assembly-plugin/assembly/1.1.3"
+          xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+          xsi:schemaLocation="http://maven.apache.org/plugins/maven-assembly-plugin/assembly/1.1.3 http://maven.apache.org/xsd/assembly-1.1.3.xsd">
     <!-- id 标识符，添加到生成文件名称的后缀符。如果指定id的话，目标文件则是 ${artifactId}-${id}.jar -->
     <id>distribution</id>
     <formats>

remote-web/src/main/java/com/jzf/remote/web/controller/RemoteDebugController.java

@@ -4,7 +4,6 @@
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
-import org.springframework.web.multipart.MultipartFile;
 
 /**
  * @author jiazhifeng
@@ -27,18 +26,13 @@ public class RemoteDebugController {
      * 2.监控服务的内存占用,线程堆栈信息等等.
      * 3.可以把这段调式代码想象成打包部署时就已经存在项目中一样.
      *
-     * @param file 待执行的java文件
+     * @param className  待编译的Java文件名称
+     * @param sourceCode 待编译的Java源码
      * @return java文件执行结果
      */
-    @PostMapping("/executeJavaSourceFile")
-    public String executeJavaSourceFile(MultipartFile file) throws Exception {
-        String className = file.getOriginalFilename().replace(".java", "");
-        String sourceCode = new String(file.getBytes());
-        return JavaFileExecuter.execute(className, sourceCode);
-    }
-
     @PostMapping("/executeJavaSourceCode")
     public String executeJavaSourceCode(String className, String sourceCode) throws Exception {
+        className = className.substring(className.lastIndexOf("\\") + 1, className.lastIndexOf("."));
         return JavaFileExecuter.execute(className, sourceCode);
     }
 }

remote-web/src/main/java/com/jzf/remote/web/controller/SourceCodeController.java

@@ -0,0 +1,35 @@
+package com.jzf.remote.web.controller;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.charset.Charset;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.List;
+import java.util.stream.Collectors;
+
+import com.jzf.remote.web.model.dto.TreeDTO;
+import com.jzf.remote.web.util.TreeUtils;
+import org.apache.commons.io.FileUtils;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+@RequestMapping("/project")
+public class projectController {
+
+    @PostMapping("/getFile")
+    public String getFile(String filePath) throws IOException {
+        return FileUtils.readFileToString(new File(TreeUtils.fileDirectory + filePath), Charset.forName("UTF-8"));
+    }
+
+    @GetMapping("/tree")
+    public List<TreeDTO> tree() {
+        TreeDTO root = new TreeDTO(null);
+        TreeUtils.generateTree(new File(TreeUtils.fileDirectory),root);
+        return root.getChildren();
+    }
+}

remote-web/src/main/java/com/jzf/remote/web/controller/projectController.java

@@ -0,0 +1,47 @@
+package com.jzf.remote.web.model.dto;
+
+import lombok.Data;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * @author jiazhifeng
+ * @date 2021/12/8 9:25
+ */
+@Data
+public class TreeDTO {
+    public static final String FOLDER_ICON = "jstree-folder";
+    public static final String FILE_ICON = "jstree-file";
+
+    private String id;
+    private String text;
+    private String icon;
+    private State state = new State();
+    private List<TreeDTO> children = new ArrayList<>();
+    private String li_attr;
+    private String a_attr;
+
+    public TreeDTO(String name){
+        this.id = name;
+        this.text = name;
+    }
+
+    public TreeDTO(String id,String text){
+        this.id = id;
+        this.text = text;
+    }
+
+    public TreeDTO(String id,String text,String icon){
+        this.id = id;
+        this.text = text;
+        this.icon = icon;
+    }
+
+    @Data
+    public static class State{
+        private boolean opened;
+        private boolean disabled;
+        private boolean selected;
+    }
+}

remote-web/src/main/java/com/jzf/remote/web/model/dto/TreeDTO.java

@@ -0,0 +1,32 @@
+package com.jzf.remote.web.util;
+
+import com.jzf.remote.web.model.dto.TreeDTO;
+
+import java.io.File;
+
+/**
+ * @author jiazhifeng
+ * @date 2021/12/8 10:59
+ */
+public class TreeUtils {
+    public static final String fileDirectory = System.getProperty("user.dir") + "/../source_code_files/";
+
+    public static void generateTree(File fileDir, TreeDTO tree) {
+        if (fileDir.exists()) {
+            File[] files = fileDir.listFiles();
+            if (files != null) {
+                for (File file : files) {
+                    String filePath = file.getPath().substring(fileDirectory.length() - 1);
+                    if (file.isDirectory()) {
+                        TreeDTO treeDTO = new TreeDTO(filePath, file.getName(), TreeDTO.FOLDER_ICON);
+                        tree.getChildren().add(treeDTO);
+                        generateTree(file, treeDTO);
+                    } else {
+                        tree.getChildren().add(new TreeDTO(filePath, file.getName(), TreeDTO.FILE_ICON));
+                    }
+                }
+            }
+        }
+    }
+
+}