toml file support

Author: Jsinco

src/main/java/dev/jsinco/abstractjavafilelib/AbstractFileManager.java

@@ -3,6 +3,10 @@
 import org.slf4j.Logger;
 
 import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.nio.file.Files;
 
 public abstract class AbstractFileManager extends ConfigurationSection {
 
@@ -17,7 +21,29 @@ public AbstractFileManager(File file) {
     }
 
 
-    public abstract AbstractFileManager generateFile();
+    public AbstractFileManager generateFile() {
+        if (!file.exists()) {
+            try (InputStream inputStream = this.getClass().getClassLoader().getResourceAsStream(file.getName())) {
+                file.getParentFile().mkdirs();
+                file.createNewFile();
+
+                if (inputStream != null) {
+                    OutputStream outputStream = Files.newOutputStream(file.toPath());
+                    byte[] buffer = new byte[1024];
+                    int bytesRead;
+                    while ((bytesRead = inputStream.read(buffer)) != -1) {
+                        outputStream.write(buffer, 0, bytesRead);
+                    }
+                    outputStream.flush();
+                    outputStream.close();
+                }
+
+            } catch (IOException e) {
+                logger.error("Error creating file: " + e.getMessage());
+            }
+        }
+        return loadFile();
+    }
 
     public abstract AbstractFileManager loadFile();
 

src/main/java/dev/jsinco/abstractjavafilelib/ConfigurationSection.java

@@ -6,6 +6,7 @@
 import java.util.AbstractMap;
 import java.util.ArrayList;
 import java.util.Collections;
+import java.util.HashMap;
 import java.util.LinkedHashMap;
 import java.util.List;
 
@@ -32,12 +33,16 @@ public boolean contains(String key) {
 
     public ConfigurationSection getConfigurationSection(String key) {
         data.computeIfAbsent(key, k -> new LinkedHashMap<String, Object>());
-        if (data.get(key) instanceof LinkedHashMap<?, ?>) { // SNAKEYAML + GSON
-            return new ConfigurationSection((LinkedHashMap<String, Object>) data.get(key));
-        } else if (data.get(key) instanceof LinkedTreeMap<?, ?>) { // GSON
-            return new ConfigurationSection((LinkedTreeMap<String, Object>) data.get(key));
+        Object object = data.get(key);
+
+        if (object instanceof LinkedHashMap<?, ?>) { // SNAKEYAML + GSON
+            return new ConfigurationSection((LinkedHashMap<String, Object>) object);
+        } else if (object instanceof LinkedTreeMap<?, ?>) { // GSON
+            return new ConfigurationSection((LinkedTreeMap<String, Object>) object);
+        } else if (object instanceof HashMap<?,?>) {
+            return new ConfigurationSection((HashMap<String, Object>) object);
         }
-        return new ConfigurationSection((AbstractMap<String, Object>) data.get(key)); // UNKNOWN
+        return new ConfigurationSection((AbstractMap<String, Object>) object); // UNKNOWN
     }
 
     private Object getLastFromSection(String path) {

src/main/java/dev/jsinco/abstractjavafilelib/schemas/TomlConfig.java

@@ -1,4 +1,73 @@
 package dev.jsinco.abstractjavafilelib.schemas;
 
-public class TomlConfig {
+import com.moandjiezana.toml.Toml;
+import com.moandjiezana.toml.TomlWriter;
+import dev.jsinco.abstractjavafilelib.AbstractFileManager;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.HashMap;
+import java.util.LinkedHashMap;
+
+public class TomlConfig extends AbstractFileManager {
+
+    private final Toml toml = new Toml();
+
+    public TomlConfig(String fileName) {
+        this(new File(dataFolder, fileName), false);
+    }
+
+    public TomlConfig(String fileName, boolean isImaginary) {
+        this(new File(dataFolder, fileName), isImaginary);
+    }
+
+    public TomlConfig(File file, boolean isImaginary) {
+        super(file);
+        if (isImaginary) {
+            loadImaginaryFile();
+        } else {
+            generateFile();
+        }
+    }
+
+    @Override
+    public TomlConfig loadFile() {
+        if (isToml(file)) {
+            data = (HashMap<String, Object>) toml.read(file).toMap();
+        }
+        if (data == null) {
+            data = new LinkedHashMap<>();
+        }
+        return this;
+    }
+
+    @Override
+    public AbstractFileManager loadImaginaryFile() {
+        if (isToml(file)) {
+            try (InputStream inputStream = this.getClass().getClassLoader().getResourceAsStream(file.getName())) {
+                data = (HashMap<String, Object>) toml.read(inputStream).toMap();
+                if (data == null) {
+                    data = new LinkedHashMap<>();
+                }
+            } catch (IOException e) {
+                logger.error("Error loading imaginary file: " + e.getMessage());
+            }
+        }
+        return this;
+    }
+
+    @Override
+    public void save() {
+        TomlWriter writer = new TomlWriter();
+        try {
+            writer.write(data, file);
+        } catch (IOException e) {
+            logger.error("Error saving file: " + e.getMessage());
+        }
+    }
+
+    public static boolean isToml(File file) {
+        return file.isFile() && file.getName().endsWith(".toml");
+    }
 }