Fixed issue in yamlEditor

Simplix-Softworks · Feb 26, 2020 · cf46a14 · cf46a14
1 parent a958ae1
commit cf46a14
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 5 deletions.
diff --git a/src/main/java/de/leonhard/storage/Yaml.java b/src/main/java/de/leonhard/storage/Yaml.java
@@ -20,6 +20,7 @@
 import java.io.FileReader;
 import java.io.IOException;
 import java.io.InputStream;
+import java.util.Arrays;
 import java.util.List;
 import java.util.Map;
 
@@ -140,15 +141,23 @@ protected void write(final FileData data) throws IOException {
 	// Specific utility methods for YAML
 	// ----------------------------------------------------------------------------------------------------
 
-	public List<String> getHeader() {
+	public final List<String> getHeader() {
 		return yamlEditor.readHeader();
 	}
 
-	public void setHeader(final List<String> header) {
+	public final void setHeader(final List<String> header) {
 		yamlEditor.setHeader(header);
 	}
 
-	public void addHeader(final List<String> toAdd) {
+	public final void setHeader(final String... header) {
+		setHeader(Arrays.asList(header));
+	}
+
+	public final void addHeader(final List<String> toAdd) {
 		yamlEditor.addHeader(toAdd);
 	}
+
+	public final void addHeader(final String... header) {
+		addHeader(Arrays.asList(header));
+	}
 }
diff --git a/src/main/java/de/leonhard/storage/internal/editor/yaml/YamlEditor.java b/src/main/java/de/leonhard/storage/internal/editor/yaml/YamlEditor.java
@@ -59,7 +59,8 @@ public void setHeader(final List<String> header) {
 
 		// Adding new header in front
 		for (int i = 0; i < header.size(); i++) {
-			lines.add(i, header.get(i));
+			final String toAdd = header.get(i);
+			lines.add(i, toAdd.startsWith("#") ? toAdd : "#" + toAdd);
 		}
 
 		// Write to file
@@ -69,7 +70,8 @@ public void setHeader(final List<String> header) {
 	public void addHeader(final List<String> header) {
 		final List<String> lines = read();
 		for (int i = 0; i < header.size(); i++) {
-			lines.add(i, header.get(i));
+			final String toAdd = header.get(i);
+			lines.add(i, toAdd.startsWith("#") ? toAdd : "#" + toAdd);
 		}
 
 		//Write to file