All Downloads are FREE. Search and download functionalities are using the official Maven repository.
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.
org.jsfr.json.compiler.JsonPathCompiler Maven / Gradle / Ivy
/*
* MIT License
*
* Copyright (c) 2019 WANG Lingsong
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
package org.jsfr.json.compiler;
import org.antlr.v4.runtime.BailErrorStrategy;
import org.antlr.v4.runtime.CharStreams;
import org.antlr.v4.runtime.CommonTokenStream;
import org.antlr.v4.runtime.tree.ParseTree;
import org.antlr.v4.runtime.tree.TerminalNode;
import org.jsfr.json.filter.*;
import org.jsfr.json.path.JsonPath;
import java.math.BigDecimal;
import java.util.InputMismatchException;
import java.util.regex.Pattern;
/**
* Created by Leo on 2015/4/1.
*/
public class JsonPathCompiler extends JsonPathBaseVisitor {
private JsonPath.Builder pathBuilder;
private FilterBuilder filterBuilder;
private JsonPath.Builder filterPathBuilder;
@Override
public Void visitPath(JsonPathParser.PathContext ctx) {
pathBuilder = JsonPath.Builder.start();
return super.visitPath(ctx);
}
private JsonPath.Builder currentPathBuilder() {
return filterPathBuilder != null ? filterPathBuilder : pathBuilder;
}
@Override
public Void visitSearchChild(JsonPathParser.SearchChildContext ctx) {
currentPathBuilder().scan().child(ctx.KEY().getText());
return super.visitSearchChild(ctx);
}
@Override
public Void visitSearch(JsonPathParser.SearchContext ctx) {
currentPathBuilder().scan();
return super.visitSearch(ctx);
}
@Override
public Void visitChildNode(JsonPathParser.ChildNodeContext ctx) {
currentPathBuilder().child(ctx.KEY().getText());
return super.visitChildNode(ctx);
}
@Override
public Void visitChildrenNode(JsonPathParser.ChildrenNodeContext ctx) {
int i = 0;
String[] strings = new String[ctx.QUOTED_STRING().size()];
for (TerminalNode node : ctx.QUOTED_STRING()) {
String quotedString = node.getText();
strings[i++] = removeQuote(quotedString);
}
currentPathBuilder().children(strings);
return super.visitChildren(ctx);
}
private String removeQuote(String quotedString) {
return quotedString.substring(1, quotedString.length() - 1);
}
@Override
public Void visitIndex(JsonPathParser.IndexContext ctx) {
currentPathBuilder().index(Integer.parseInt(ctx.NUM().getText()));
return super.visitIndex(ctx);
}
@Override
public Void visitIndexes(JsonPathParser.IndexesContext ctx) {
int i = 0;
Integer[] keys = new Integer[ctx.NUM().size()];
for (TerminalNode key : ctx.NUM()) {
keys[i++] = Integer.parseInt(key.getText());
}
currentPathBuilder().indexes(keys);
return super.visitIndexes(ctx);
}
@Override
public Void visitSlicing(JsonPathParser.SlicingContext ctx) {
Integer left = null;
Integer right;
Integer temp = null;
for (ParseTree node : ctx.children) {
if (node instanceof TerminalNode) {
TerminalNode tNode = (TerminalNode) node;
if (((TerminalNode) node).getSymbol().getType() == JsonPathParser.COLON) {
left = temp;
temp = null;
} else if (tNode.getSymbol().getType() == JsonPathParser.NUM) {
temp = Integer.parseInt(tNode.getText());
}
}
}
right = temp;
pathBuilder.slicing(left, right);
return super.visitSlicing(ctx);
}
@Override
public Void visitAnyChild(JsonPathParser.AnyChildContext ctx) {
currentPathBuilder().anyChild();
return super.visitAnyChild(ctx);
}
@Override
public Void visitAnyIndex(JsonPathParser.AnyIndexContext ctx) {
currentPathBuilder().anyIndex();
return super.visitAnyIndex(ctx);
}
@Override
public Void visitAny(JsonPathParser.AnyContext ctx) {
currentPathBuilder().any();
return super.visitAny(ctx);
}
@Override
public Void visitFilter(JsonPathParser.FilterContext ctx) {
filterBuilder = new FilterBuilder();
Void rst = super.visitFilter(ctx);
pathBuilder.arrayFilter(filterBuilder.build());
return rst;
}
@Override
public Void visitFilterExpr(JsonPathParser.FilterExprContext ctx) {
Void rst;
if (ctx.NegationOperator() != null) {
filterBuilder.startNegationPredicate();
rst = super.visitFilterExpr(ctx);
filterBuilder.endNegationAndPredicate();
}
else if (ctx.AndOperator() != null) {
filterBuilder.startAndPredicate();
rst = super.visitFilterExpr(ctx);
filterBuilder.endAndPredicate();
} else if (ctx.OrOperator() != null) {
filterBuilder.startOrPredicate();
rst = super.visitFilterExpr(ctx);
filterBuilder.endOrPredicate();
} else {
rst = super.visitFilterExpr(ctx);
}
return rst;
}
private JsonPath.Builder createFilterPathBuilder() {
return JsonPath.Builder.startFilterPath();
}
@Override
public Void visitFilterEqualNum(JsonPathParser.FilterEqualNumContext ctx) {
// JsonPath relativePath = JsonPath.Builder.start().child(ctx.KEY().getText()).build();
filterPathBuilder = createFilterPathBuilder();
Void rst = super.visitFilterEqualNum(ctx);
filterBuilder.append(new EqualityNumPredicate(filterPathBuilder.build(), new BigDecimal(ctx.NUM().getText())));
filterPathBuilder = null;
return rst;
}
@Override
public Void visitFilterEqualBool(JsonPathParser.FilterEqualBoolContext ctx) {
filterPathBuilder = createFilterPathBuilder();
Void rst = super.visitFilterEqualBool(ctx);
filterBuilder.append(new EqualityBoolPredicate(filterPathBuilder.build(), Boolean.parseBoolean(ctx.BOOL().getText())));
filterPathBuilder = null;
return rst;
}
@Override
public Void visitFilterExist(JsonPathParser.FilterExistContext ctx) {
filterPathBuilder = createFilterPathBuilder();
Void rst = super.visitFilterExist(ctx);
filterBuilder.append(new ExistencePredicate(filterPathBuilder.build()));
filterPathBuilder = null;
return rst;
}
@Override
public Void visitFilterGtNum(JsonPathParser.FilterGtNumContext ctx) {
filterPathBuilder = createFilterPathBuilder();
Void rst = super.visitFilterGtNum(ctx);
filterBuilder.append(new GreaterThanNumPredicate(filterPathBuilder.build(), new BigDecimal(ctx.NUM().getText())));
filterPathBuilder = null;
return rst;
}
@Override
public Void visitFilterLtNum(JsonPathParser.FilterLtNumContext ctx) {
filterPathBuilder = createFilterPathBuilder();
Void rst = super.visitFilterLtNum(ctx);
filterBuilder.append(new LessThanNumPredicate(filterPathBuilder.build(), new BigDecimal(ctx.NUM().getText())));
filterPathBuilder = null;
return rst;
}
@Override
public Void visitFilterEqualStr(JsonPathParser.FilterEqualStrContext ctx) {
filterPathBuilder = createFilterPathBuilder();
Void rst = super.visitFilterEqualStr(ctx);
filterBuilder.append(new EqualityStrPredicate(filterPathBuilder.build(), removeQuote(ctx.QUOTED_STRING().getText())));
filterPathBuilder = null;
return rst;
}
@Override
public Void visitFilterMatchRegex(JsonPathParser.FilterMatchRegexContext ctx) {
filterPathBuilder = createFilterPathBuilder();
Void rst = super.visitFilterMatchRegex(ctx);
filterBuilder.append(new MatchRegexPredicate(filterPathBuilder.build(), toPattern(ctx.REGEX().getText())));
filterPathBuilder = null;
return rst;
}
private static Pattern toPattern(String str) {
String[] split = str.split("(?
