Version: 2.0 Update

src/main/java/burp/yaml/LoadConfigFile.java

@@ -0,0 +1,86 @@
+package burp.yaml;
+
+import org.jetbrains.annotations.NotNull;
+import org.yaml.snakeyaml.Yaml;
+
+import java.io.*;
+import java.util.HashMap;
+import java.util.Map;
+
+/*
+ * @author LinChen
+ */
+
+public class LoadConfigFile {
+    private static Yaml yaml = new Yaml();
+    private static final String SettingPath = "Setting.yml";
+    private static final String ConfigPath = "Config.yml";
+
+    public LoadConfigFile(){
+        init();
+    }
+
+    // 初始化配置
+    public void init(){
+        File yamlSetting = new File(SettingPath);
+        if (!(yamlSetting.exists() && yamlSetting.isFile())) {
+            Map<String,Object> r = new HashMap<>();
+            r.put("configPath", ConfigPath);
+            r.put("excludeSuffix", getExcludeSuffix());
+            try{
+                Writer ws = new OutputStreamWriter(new FileOutputStream(SettingPath),"UTF-8");
+                yaml.dump(r, ws);
+            }catch (Exception ex){
+                ex.printStackTrace();
+            }
+        }
+    }
+
+    public String getExcludeSuffix(){
+        try {
+            InputStream inorder = new FileInputStream(SettingPath);
+            Map<String,Object> r;
+            r = yaml.load(inorder);
+            return r.get("excludeSuffix").toString();
+        } catch (FileNotFoundException e) {
+            e.printStackTrace();
+            return "css|jpeg|gif|jpg|png|pdf|rar|zip|docx|doc|svg|jpeg|ico|woff|woff2|ttf|otf";
+        }
+    }
+
+    public String getConfigPath(){
+        try {
+            InputStream inorder = new FileInputStream(SettingPath);
+            Map<String,Object> r;
+            r = yaml.load(inorder);
+            return r.get("configPath").toString();
+        } catch (FileNotFoundException e) {
+            e.printStackTrace();
+            return ConfigPath;
+        }
+    }
+
+    public void setExcludeSuffix(@NotNull String excludeSuffix){
+        Map<String,Object> r = new HashMap<>();
+        r.put("excludeSuffix", excludeSuffix);
+        r.put("configPath", getConfigPath());
+        try{
+            Writer ws = new OutputStreamWriter(new FileOutputStream(SettingPath),"UTF-8");
+            yaml.dump(r, ws);
+        }catch (Exception ex){
+            ex.printStackTrace();
+        }
+    }
+
+    public void setConfigPath(@NotNull String filePath){
+        Map<String,Object> r = new HashMap<>();
+        r.put("configPath", filePath);
+        r.put("excludeSuffix", getExcludeSuffix());
+        try{
+            Writer ws = new OutputStreamWriter(new FileOutputStream(SettingPath),"UTF-8");
+            yaml.dump(r, ws);
+        }catch (Exception ex){
+            ex.printStackTrace();
+        }
+    }
+}