Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright (c) 2013-2024 Hutool Team and hutool.cn
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.dromara.hutool.setting.toml;
import org.dromara.hutool.setting.SettingException;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.ZonedDateTime;
import java.util.*;
/**
* TOML文件读取
* 来自：https://github.com/TheElectronWill/TOML-javalib
*
* 此类支持更加宽松的key，除了{@code A-Za-z0-9_- }，还支持' ','.', '[', ']' 和 '='
*
* @author TheElectronWill
*/
public class TomlReader {
private final String data;
private final boolean strictAsciiBareKeys;
private int pos = 0;// current position
private int line = 1;// current line
/**
* 构造
*
* 严格模式：[A-Za-z0-9_-]
* 宽松模式：所有字符但是不包括. [ ] # =
*
*
* @param data TOML数据
* @param strictAsciiBareKeys {@code true} 只允许严格的key格式, {@code false} 支持宽松格式.
*/
public TomlReader(final String data, final boolean strictAsciiBareKeys) {
this.data = data;
this.strictAsciiBareKeys = strictAsciiBareKeys;
}
/**
* 读取TOML
*
* @return TOML
*/
@SuppressWarnings("unchecked")
public Map read() {
final Map map = nextTableContent();
if (!hasNext() && pos > 0 && data.charAt(pos - 1) == '['){
throw new SettingException("Invalid table declaration at line " + line + ": it never ends");
}
while (hasNext()) {
char c = nextUseful(true);
final boolean twoBrackets;
if (c == '[') {
twoBrackets = true;
c = nextUseful(false);
} else {
twoBrackets = false;
}
pos--;
// --- Reads the key --
final List keyParts = new ArrayList<>(4);
boolean insideSquareBrackets = true;
while (insideSquareBrackets) {
if (!hasNext())
throw new SettingException("Invalid table declaration at line " + line + ": it never ends");
String name = null;
final char nameFirstChar = nextUseful(false);
switch (nameFirstChar) {
case '"': {
if (pos + 1 < data.length()) {
final char c2 = data.charAt(pos);
final char c3 = data.charAt(pos + 1);
if (c2 == '"' && c3 == '"') {
pos += 2;
name = nextBasicMultilineString();
}
}
if (name == null) {
name = nextBasicString();
}
break;
}
case '\'': {
if (pos + 1 < data.length()) {
final char c2 = data.charAt(pos);
final char c3 = data.charAt(pos + 1);
if (c2 == '\'' && c3 == '\'') {
pos += 2;
name = nextLiteralMultilineString();
}
}
if (name == null) {
name = nextLiteralString();
}
break;
}
default:
pos--;// to include the first (already read) non-space character
name = nextBareKey(']', '.').trim();
if (data.charAt(pos) == ']') {
if (!name.isEmpty())
keyParts.add(name);
insideSquareBrackets = false;
} else if (name.isEmpty()) {
throw new SettingException("Invalid empty key at line " + line);
}
pos++;// to go after the character we stopped at in nextBareKey()
break;
}
if (insideSquareBrackets)
keyParts.add(name.trim());
}
// -- Checks --
if (keyParts.isEmpty())
throw new SettingException("Invalid empty key at line " + line);
if (twoBrackets && next() != ']') {// 2 brackets at the start but only one at the end!
throw new SettingException("Missing character ']' at line " + line);
}
// -- Reads the value (table content) --
final Map value = nextTableContent();
// -- Saves the value --
Map valueMap = map;// the map that contains the value
for (int i = 0; i < keyParts.size() - 1; i++) {
final String part = keyParts.get(i);
final Object child = valueMap.get(part);
final Map childMap;
if (child == null) {// implicit table
childMap = new LinkedHashMap<>(4);
valueMap.put(part, childMap);
} else if (child instanceof Map) {// table
childMap = (Map) child;
} else {// array
final List> list = (List>) child;
childMap = list.get(list.size() - 1);
}
valueMap = childMap;
}
if (twoBrackets) {// element of a table array
final String name = keyParts.get(keyParts.size() - 1);
Collection> tableArray = (Collection>) valueMap.get(name);
if (tableArray == null) {
tableArray = new ArrayList<>(2);
valueMap.put(name, tableArray);
}
tableArray.add(value);
} else {// just a table
valueMap.put(keyParts.get(keyParts.size() - 1), value);
}
}
return map;
}
private boolean hasNext() {
return pos < data.length();
}
private char next() {
return data.charAt(pos++);
}
private char nextUseful(final boolean skipComments) {
char c = ' ';
while (hasNext() && (c == ' ' || c == '\t' || c == '\r' || c == '\n' || (c == '#' && skipComments))) {
c = next();
if (skipComments && c == '#') {
final int nextLinebreak = data.indexOf('\n', pos);
if (nextLinebreak == -1) {
pos = data.length();
} else {
pos = nextLinebreak + 1;
line++;
}
} else if (c == '\n') {
line++;
}
}
return c;
}
private char nextUsefulOrLinebreak() {
char c = ' ';
while (c == ' ' || c == '\t' || c == '\r') {
if (!hasNext())// fixes error when no '\n' at the end of the file
return '\n';
c = next();
}
if (c == '\n')
line++;
return c;
}
private Object nextValue(final char firstChar) {
switch (firstChar) {
case '+':
case '-':
case '0':
case '1':
case '2':
case '3':
case '4':
case '5':
case '6':
case '7':
case '8':
case '9':
return nextNumberOrDate(firstChar);
case '"':
if (pos + 1 < data.length()) {
final char c2 = data.charAt(pos);
final char c3 = data.charAt(pos + 1);
if (c2 == '"' && c3 == '"') {
pos += 2;
return nextBasicMultilineString();
}
}
return nextBasicString();
case '\'':
if (pos + 1 < data.length()) {
final char c2 = data.charAt(pos);
final char c3 = data.charAt(pos + 1);
if (c2 == '\'' && c3 == '\'') {
pos += 2;
return nextLiteralMultilineString();
}
}
return nextLiteralString();
case '[':
return nextArray();
case '{':
return nextInlineTable();
case 't':// Must be "true"
if (pos + 3 > data.length() || next() != 'r' || next() != 'u' || next() != 'e') {
throw new SettingException("Invalid value at line " + line);
}
return true;
case 'f':// Must be "false"
if (pos + 4 > data.length() || next() != 'a' || next() != 'l' || next() != 's' || next() != 'e') {
throw new SettingException("Invalid value at line " + line);
}
return false;
default:
throw new SettingException("Invalid character '" + toString(firstChar) + "' at line " + line);
}
}
private List
