reupload

2021-04-06 21:14:05 +08:00
parent 05cfdb73a9
commit 6d14314bd9
126 changed files with 191394 additions and 0 deletions
@@ -0,0 +1,148 @@
+package com.loohp.limbo.file;
+
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.io.OutputStreamWriter;
+import java.io.PrintWriter;
+import java.io.Reader;
+import java.nio.charset.StandardCharsets;
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+import org.yaml.snakeyaml.DumperOptions;
+import org.yaml.snakeyaml.Yaml;
+import org.yaml.snakeyaml.representer.Representer;
+
+import com.loohp.limbo.utils.YamlOrder;
+
+public class FileConfiguration {
+	
+	private Map<String, Object> mapping;
+	private String header;
+	
+	public FileConfiguration(File file) throws IOException {
+		if (file.exists()) {
+			InputStreamReader reader = new InputStreamReader(new FileInputStream(file), StandardCharsets.UTF_8);
+			reloadConfig(reader);
+			reader.close();
+		} else {
+			mapping = new LinkedHashMap<>();
+		}
+	}
+	
+	public FileConfiguration(InputStream input){
+		reloadConfig(new InputStreamReader(input, StandardCharsets.UTF_8));
+	}
+	
+	public FileConfiguration(Reader reader){
+		reloadConfig(reader);
+	}
+	
+	public FileConfiguration reloadConfig(File file) throws FileNotFoundException {
+		return reloadConfig(new InputStreamReader(new FileInputStream(file), StandardCharsets.UTF_8));
+	}
+	
+	public FileConfiguration reloadConfig(InputStream input) {
+		return reloadConfig(new InputStreamReader(input, StandardCharsets.UTF_8));
+	}
+	
+	public FileConfiguration reloadConfig(Reader reader) {
+		Yaml yml = new Yaml();
+		mapping = yml.load(reader);
+		return this;
+	}
+	
+	public void setHeader(String header) {
+		this.header = header;
+	}
+	
+	@SuppressWarnings("unchecked")
+	public <T> T get(String key, Class<T> returnType) {
+		try {
+			String[] tree = key.split("\\.");
+			Map<String, Object> map = mapping;
+			for (int i = 0; i < tree.length - 1; i++) {
+				map = (Map<String, Object>) map.get(tree[i]);
+			}
+			if (returnType.equals(String.class)) {
+				return (T) map.get(tree[tree.length - 1]).toString();
+			}
+			return returnType.cast(map.get(tree[tree.length - 1]));
+		} catch (Exception e) {
+			return null;
+		}
+	}
+	
+	@SuppressWarnings("unchecked")
+	public <T> void set(String key, T value) {
+		String[] tree = key.split("\\.");
+		Map<String, Object> map = mapping;
+		for (int i = 0; i < tree.length - 1; i++) {
+			Map<String, Object> map1 = (Map<String, Object>) map.get(tree[i]);
+			if (map1 == null) {
+				map1 = new LinkedHashMap<>();
+				map.put(tree[i], map1);
+			}
+			map = map1;
+		}
+		if (value != null) {
+			map.put(tree[tree.length - 1], (T) value); 
+		} else {
+			map.remove(tree[tree.length - 1]);
+		}
+	}
+	
+	public String saveToString() throws IOException {
+		DumperOptions options = new DumperOptions();
+        options.setIndent(2);
+        options.setPrettyFlow(true);
+        options.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK);
+        Representer customRepresenter = new Representer();
+        YamlOrder customProperty = new YamlOrder();
+        customRepresenter.setPropertyUtils(customProperty);
+		Yaml yaml = new Yaml(customRepresenter, options);
+		
+		ByteArrayOutputStream out = new ByteArrayOutputStream();
+		PrintWriter pw = new PrintWriter(new OutputStreamWriter(out, StandardCharsets.UTF_8));
+		if (header != null) {
+			pw.println("#" + header.replace("\n", "\n#"));
+		}
+		yaml.dump(mapping, pw);
+		pw.flush();
+		pw.close();
+		
+		String str = new String(out.toByteArray(), StandardCharsets.UTF_8);
+		
+		return str;
+	}
+	
+	public void saveConfig(File file) throws IOException {
+		DumperOptions options = new DumperOptions();
+        options.setIndent(2);
+        options.setPrettyFlow(true);
+        options.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK);
+        Representer customRepresenter = new Representer();
+        YamlOrder customProperty = new YamlOrder();
+        customRepresenter.setPropertyUtils(customProperty);
+		Yaml yaml = new Yaml(customRepresenter, options);
+		
+		if (file.getParentFile() != null) {
+			file.getParentFile().mkdirs();
+		}
+		
+		PrintWriter pw = new PrintWriter(new OutputStreamWriter(new FileOutputStream(file), StandardCharsets.UTF_8));
+		if (header != null) {
+			pw.println("#" + header.replace("\n", "\n#"));
+		}
+		yaml.dump(mapping, pw);
+		pw.flush();
+		pw.close();
+	}
+
+}