modify toml4j fork to automatically add new fields without overwriting previous ones and setup motd

proxy/src/main/java/dev/plex/toml/StringValueReaderWriter.java

@@ -0,0 +1,129 @@
+package dev.plex.toml;
+
+import java.net.URI;
+import java.net.URL;
+import java.util.concurrent.atomic.AtomicInteger;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+class StringValueReaderWriter implements ValueReader, dev.plex.toml.ValueWriter
+{
+  
+  static final StringValueReaderWriter STRING_VALUE_READER_WRITER = new StringValueReaderWriter();
+  private static final Pattern UNICODE_REGEX = Pattern.compile("\\\\[uU](.{4})");
+
+  static private final String[] specialCharacterEscapes = new String[93];
+
+  static {
+    specialCharacterEscapes['\b'] = "\\b";
+    specialCharacterEscapes['\t'] = "\\t";
+    specialCharacterEscapes['\n'] = "\\n";
+    specialCharacterEscapes['\f'] = "\\f";
+    specialCharacterEscapes['\r'] = "\\r";
+    specialCharacterEscapes['"'] = "\\\"";
+    specialCharacterEscapes['\\'] = "\\\\";
+  }
+
+  @Override
+  public boolean canRead(String s) {
+    return s.startsWith("\"");
+  }
+
+  @Override
+  public Object read(String s, AtomicInteger index, Context context) {
+    int startIndex = index.incrementAndGet();
+    int endIndex = -1;
+
+    for (int i = index.get(); i < s.length(); i = index.incrementAndGet()) {
+      char ch = s.charAt(i);
+      if (ch == '"' && s.charAt(i - 1) != '\\') {
+        endIndex = i;
+        break;
+      }
+    }
+
+    if (endIndex == -1) {
+      Results.Errors errors = new Results.Errors();
+      errors.unterminated(context.identifier.getName(), s.substring(startIndex - 1), context.line.get());
+      return errors;
+    }
+    
+    String raw = s.substring(startIndex, endIndex);
+    s = replaceUnicodeCharacters(raw);
+    s = replaceSpecialCharacters(s);
+    
+    if (s == null) {
+      Results.Errors errors = new Results.Errors();
+      errors.invalidValue(context.identifier.getName(), raw, context.line.get());
+      return errors;
+    }
+
+    return s;
+  }
+
+  String replaceUnicodeCharacters(String value) {
+    Matcher unicodeMatcher = UNICODE_REGEX.matcher(value);
+
+    while (unicodeMatcher.find()) {
+      value = value.replace(unicodeMatcher.group(), new String(Character.toChars(Integer.parseInt(unicodeMatcher.group(1), 16))));
+    }
+    return value;
+  }
+
+  String replaceSpecialCharacters(String s) {
+    for (int i = 0; i < s.length() - 1; i++) {
+      char ch = s.charAt(i);
+      char next = s.charAt(i + 1);
+
+      if (ch == '\\' && next == '\\') {
+        i++;
+      } else if (ch == '\\' && !(next == 'b' || next == 'f' || next == 'n' || next == 't' || next == 'r' || next == '"' || next == '\\')) {
+        return null;
+      }
+    }
+
+    return s.replace("\\n", "\n")
+      .replace("\\\"", "\"")
+      .replace("\\t", "\t")
+      .replace("\\r", "\r")
+      .replace("\\\\", "\\")
+      .replace("\\/", "/")
+      .replace("\\b", "\b")
+      .replace("\\f", "\f");
+  }
+
+  @Override
+  public boolean canWrite(Object value) {
+    return value instanceof String || value instanceof Character || value instanceof URL || value instanceof URI || value instanceof Enum;
+  }
+
+  @Override
+  public void write(Object value, dev.plex.toml.WriterContext context) {
+    context.write('"');
+    escapeUnicode(value.toString(), context);
+    context.write('"');
+  }
+
+  @Override
+  public boolean isPrimitiveType() {
+    return true;
+  }
+
+  private void escapeUnicode(String in, dev.plex.toml.WriterContext context) {
+    for (int i = 0; i < in.length(); i++) {
+      int codePoint = in.codePointAt(i);
+      if (codePoint < specialCharacterEscapes.length && specialCharacterEscapes[codePoint] != null) {
+        context.write(specialCharacterEscapes[codePoint]);
+      } else {
+        context.write(in.charAt(i));
+      }
+    }
+  }
+
+  private StringValueReaderWriter() {}
+
+  @Override
+  public String toString() {
+    return "string";
+  }
+}