toml4j/src/main/java/com/moandjiezana/toml/RegexParser.java

package com.moandjiezana.toml;

import static com.moandjiezana.toml.ValueAnalysis.INVALID;

import java.util.List;
import java.util.regex.Pattern;

import org.parboiled.Parboiled;
import org.parboiled.parserunners.BasicParseRunner;
import org.parboiled.support.ParsingResult;

public class RegexParser {
  private static final Pattern MULTILINE_ARRAY_REGEX = Pattern.compile("\\s*\\[([^\\]]*)");
  private static final Pattern MULTILINE_ARRAY_REGEX_END = Pattern.compile("\\s*\\]");

  private final Results results = new Results();

  public Results run(String tomlString) {
    if (tomlString.isEmpty()) {
      return results;
    }

    String[] lines = tomlString.split("[\\n\\r]");
    StringBuilder multilineBuilder = new StringBuilder();
    boolean multiline = false;

    String key = null;
    String value = null;

    for (int i = 0; i < lines.length; i++) {
      String line = lines[i];

      if (line != null) {
        line = line.trim();
      }

      if (isComment(line) || line.isEmpty()) {
        continue;
      }

      if (isTableArray(line)) {
        String tableName = getTableArrayName(line);
        if (tableName != null) {
          results.startTableArray(tableName);
          String afterTableName = line.substring(tableName.length() + 4);
          if (!isComment(afterTableName)) {
            results.errors.append("Invalid table array definition: " + line + "\n\n");
          }
        } else {
          results.errors.append("Invalid table array definition: " + line + "\n\n");
        }

        continue;
      }

      if (!multiline && isTable(line)) {
        String tableName = getTableName(line);
        if (tableName != null) {
//        Matcher matcher = TABLE_REGEX.matcher(line);
//        matcher.matches();
//        String tableName = matcher.group(1);
          results.startTables(tableName);
          String afterTableName = line.substring(tableName.length() + 2);
          if (!isComment(afterTableName)) {
            results.errors.append("Invalid table definition: " + line + "\n\n");
          }
        } else {
          results.errors.append("Invalid table definition: " + line + "\n\n");
        }

        continue;
      }

      String[] pair = line.split("=");

      if (!multiline && MULTILINE_ARRAY_REGEX.matcher(pair[1].trim()).matches()) {
        multiline = true;
        key = pair[0].trim();
        multilineBuilder.append(removeComment(pair[1]));
        continue;
      }


      if (multiline) {
        String lineWithoutComment = removeComment(line);
        multilineBuilder.append(lineWithoutComment);
        if (MULTILINE_ARRAY_REGEX_END.matcher(lineWithoutComment).matches()) {
          multiline = false;
          value = multilineBuilder.toString();
          multilineBuilder.delete(0, multilineBuilder.length() - 1);
        } else {
          continue;
        }
      } else {
        key = pair[0].trim();
        value = pair[1].trim();
      }

      if (!isKeyValid(key)) {
        results.errors.append("Invalid key name: " + key);
        continue;
      }

      ValueAnalysis lineAnalysis = new ValueAnalysis(value);

      Object convertedValue = lineAnalysis.getValue();

      if (convertedValue != INVALID) {
        results.addValue(key, convertedValue);
      } else {
        results.errors.append("Invalid key/value: " + key + " = " + value);
      }
    }

    return results;
  }

  private boolean isTableArray(String line) {
    return line.startsWith("[[");
  }
  
  private String getTableArrayName(String line) {
    ParboiledParser parser = Parboiled.createParser(ParboiledParser.class);
    ParsingResult<List<Object>> parsingResult = new BasicParseRunner<List<Object>>(parser.TableArray()).run(line);

    if (parsingResult.resultValue == null) {
      return null;
    }

    return (String) parsingResult.resultValue.get(0);
  }

  private boolean isTable(String line) {
    return line.startsWith("[");
  }

  private String getTableName(String line) {
    ParboiledParser parser = Parboiled.createParser(ParboiledParser.class);
    ParsingResult<List<Object>> parsingResult = new BasicParseRunner<List<Object>>(parser.Table()).run(line);

    if (parsingResult.resultValue == null) {
      return null;
    }

    return (String) parsingResult.resultValue.get(0);
  }

  private boolean isKeyValid(String key) {
    if (key.contains(".") || key.contains("#") || key.trim().isEmpty()) {
      return false;
    }

    return true;
  }

  private boolean isComment(String line) {
    if (line == null || line.isEmpty()) {
      return true;
    }

    char[] chars = line.toCharArray();

    for (char c : chars) {
      if (Character.isWhitespace(c)) {
        continue;
      }

      return c == '#';
    }

    return false;
  }

  private String removeComment(String line) {
    line = line.trim();
    if (line.startsWith("\"")) {
      int startOfComment = line.indexOf('#', line.lastIndexOf('"'));
      if (startOfComment > -1) {
        return line.substring(0, startOfComment - 1).trim();
      }
    } else {
      int startOfComment = line.indexOf('#');
      if (startOfComment > -1) {
        return line.substring(0, startOfComment - 1).trim();
      }
    }

    return line;
  }
}