Search in sources :

Example 1 with NumericConstant

use of org.apache.sling.scripting.sightly.compiler.expression.nodes.NumericConstant in project sling by apache.

the class DeadCodeRemoval method visit.

@Override
public void visit(VariableBinding.Start variableBindingStart) {
    Boolean truthValue = null;
    ExpressionNode node = variableBindingStart.getExpression();
    if (node instanceof StringConstant) {
        truthValue = CompileTimeObjectModel.toBoolean(((StringConstant) node).getText());
    }
    if (node instanceof BooleanConstant) {
        truthValue = ((BooleanConstant) node).getValue();
    }
    if (node instanceof NumericConstant) {
        truthValue = CompileTimeObjectModel.toBoolean(((NumericConstant) node).getValue());
    }
    if (node instanceof NullLiteral) {
        truthValue = CompileTimeObjectModel.toBoolean(null);
    }
    tracker.pushVariable(variableBindingStart.getVariableName(), truthValue);
    outStream.write(variableBindingStart);
}
Also used : ExpressionNode(org.apache.sling.scripting.sightly.compiler.expression.ExpressionNode) NumericConstant(org.apache.sling.scripting.sightly.compiler.expression.nodes.NumericConstant) BooleanConstant(org.apache.sling.scripting.sightly.compiler.expression.nodes.BooleanConstant) StringConstant(org.apache.sling.scripting.sightly.compiler.expression.nodes.StringConstant) NullLiteral(org.apache.sling.scripting.sightly.compiler.expression.nodes.NullLiteral)

Example 2 with NumericConstant

use of org.apache.sling.scripting.sightly.compiler.expression.nodes.NumericConstant in project sling by apache.

the class RepeatPlugin method invoke.

@Override
public PluginInvoke invoke(final Expression expression, final PluginCallInfo callInfo, final CompilerContext compilerContext) {
    return new DefaultPluginInvoke() {

        private String listVariable = compilerContext.generateVariable("collectionVar");

        private String collectionSizeVar = compilerContext.generateVariable("size");

        @Override
        public void beforeElement(PushStream stream, String tagName) {
            stream.write(new VariableBinding.Start(listVariable, expression.getRoot()));
            stream.write(new VariableBinding.Start(collectionSizeVar, new UnaryOperation(UnaryOperator.LENGTH, new Identifier(listVariable))));
            stream.write(new Conditional.Start(collectionSizeVar, true));
            String itemVariable = decodeItemVariable();
            String loopStatusVar = Syntax.itemLoopStatusVariable(itemVariable);
            String indexVariable = compilerContext.generateVariable("index");
            stream.write(new Loop.Start(listVariable, itemVariable, indexVariable));
            stream.write(new VariableBinding.Start(loopStatusVar, buildStatusObj(indexVariable, collectionSizeVar)));
        }

        @Override
        public void afterTagClose(PushStream stream, boolean isSelfClosing) {
            stream.write(NEW_LINE);
        }

        @Override
        public void afterElement(PushStream stream) {
            stream.write(VariableBinding.END);
            stream.write(Loop.END);
            stream.write(Conditional.END);
            stream.write(VariableBinding.END);
            stream.write(VariableBinding.END);
        }

        private String decodeItemVariable() {
            String[] args = callInfo.getArguments();
            if (args.length > 0) {
                return args[0];
            }
            return Syntax.DEFAULT_LIST_ITEM_VAR_NAME;
        }

        private MapLiteral buildStatusObj(String indexVar, String sizeVar) {
            HashMap<String, ExpressionNode> obj = new HashMap<>();
            Identifier indexId = new Identifier(indexVar);
            BinaryOperation firstExpr = new BinaryOperation(BinaryOperator.EQ, indexId, NumericConstant.ZERO);
            BinaryOperation lastExpr = new BinaryOperation(BinaryOperator.EQ, indexId, new BinaryOperation(BinaryOperator.SUB, new Identifier(sizeVar), NumericConstant.ONE));
            obj.put(INDEX, indexId);
            obj.put(COUNT, new BinaryOperation(BinaryOperator.ADD, indexId, NumericConstant.ONE));
            obj.put(FIRST, firstExpr);
            obj.put(MIDDLE, new UnaryOperation(UnaryOperator.NOT, new BinaryOperation(BinaryOperator.OR, firstExpr, lastExpr)));
            obj.put(LAST, lastExpr);
            obj.put(ODD, parityCheck(indexId, NumericConstant.ZERO));
            obj.put(EVEN, parityCheck(indexId, NumericConstant.ONE));
            return new MapLiteral(obj);
        }

        private ExpressionNode parityCheck(ExpressionNode numericExpression, NumericConstant expected) {
            return new BinaryOperation(BinaryOperator.EQ, new BinaryOperation(BinaryOperator.REM, numericExpression, NumericConstant.TWO), expected);
        }
    };
}
Also used : Loop(org.apache.sling.scripting.sightly.compiler.commands.Loop) UnaryOperation(org.apache.sling.scripting.sightly.compiler.expression.nodes.UnaryOperation) HashMap(java.util.HashMap) BinaryOperation(org.apache.sling.scripting.sightly.compiler.expression.nodes.BinaryOperation) Conditional(org.apache.sling.scripting.sightly.compiler.commands.Conditional) MapLiteral(org.apache.sling.scripting.sightly.compiler.expression.nodes.MapLiteral) Identifier(org.apache.sling.scripting.sightly.compiler.expression.nodes.Identifier) ExpressionNode(org.apache.sling.scripting.sightly.compiler.expression.ExpressionNode) NumericConstant(org.apache.sling.scripting.sightly.compiler.expression.nodes.NumericConstant) PushStream(org.apache.sling.scripting.sightly.impl.compiler.PushStream) VariableBinding(org.apache.sling.scripting.sightly.compiler.commands.VariableBinding)

Example 3 with NumericConstant

use of org.apache.sling.scripting.sightly.compiler.expression.nodes.NumericConstant in project sling by apache.

the class ExpressionWrapperTest method testURIOptionsRemoval.

@Test
public void testURIOptionsRemoval() {
    Interpolation interpolation = new Interpolation();
    Map<String, ExpressionNode> options = new HashMap<>();
    options.put(URIManipulationFilter.SCHEME, new StringConstant("https"));
    options.put(URIManipulationFilter.DOMAIN, new StringConstant("www.example.org"));
    options.put(URIManipulationFilter.PREPEND_PATH, new StringConstant("/before"));
    options.put(URIManipulationFilter.PATH, new StringConstant("/path"));
    options.put(URIManipulationFilter.APPEND_PATH, new StringConstant("/after"));
    List<ExpressionNode> selectors = new ArrayList<>();
    selectors.add(new StringConstant("a"));
    selectors.add(new StringConstant("b"));
    options.put(URIManipulationFilter.SELECTORS, new ArrayLiteral(selectors));
    options.put(URIManipulationFilter.EXTENSION, new StringConstant("html"));
    options.put(URIManipulationFilter.PREPEND_SUFFIX, new StringConstant("/pre"));
    options.put(URIManipulationFilter.APPEND_SUFFIX, new StringConstant("/after"));
    options.put(URIManipulationFilter.FRAGMENT, new StringConstant("rewrite"));
    Map<String, ExpressionNode> query = new HashMap<>();
    query.put("q", new StringConstant("sightly"));
    query.put("array", new ArrayLiteral(new ArrayList<ExpressionNode>() {

        {
            add(new NumericConstant(1));
            add(new NumericConstant(2));
            add(new NumericConstant(3));
        }
    }));
    options.put(URIManipulationFilter.QUERY, new MapLiteral(query));
    options.put(URIManipulationFilter.REMOVE_QUERY, new StringConstant("array"));
    interpolation.addExpression(new Expression(new StringConstant("http://www.example.com/resource.selector.extension/suffix#fragment?param=value"), options));
    ExpressionWrapper wrapper = new ExpressionWrapper(filters);
    Expression result = wrapper.transform(interpolation, MarkupContext.TEXT, ExpressionContext.TEXT);
    List<ExpressionNode> xssArguments = runOptionsAndXSSAssertions(result, 0);
    RuntimeCall join = (RuntimeCall) xssArguments.get(0);
    assertEquals(RuntimeFunction.URI_MANIPULATION, join.getFunctionName());
}
Also used : HashMap(java.util.HashMap) ArrayList(java.util.ArrayList) RuntimeCall(org.apache.sling.scripting.sightly.compiler.expression.nodes.RuntimeCall) ArrayLiteral(org.apache.sling.scripting.sightly.compiler.expression.nodes.ArrayLiteral) Interpolation(org.apache.sling.scripting.sightly.impl.compiler.frontend.Interpolation) MapLiteral(org.apache.sling.scripting.sightly.compiler.expression.nodes.MapLiteral) Expression(org.apache.sling.scripting.sightly.compiler.expression.Expression) ExpressionNode(org.apache.sling.scripting.sightly.compiler.expression.ExpressionNode) NumericConstant(org.apache.sling.scripting.sightly.compiler.expression.nodes.NumericConstant) StringConstant(org.apache.sling.scripting.sightly.compiler.expression.nodes.StringConstant) ExpressionWrapper(org.apache.sling.scripting.sightly.impl.compiler.frontend.ExpressionWrapper) Test(org.junit.Test)

Example 4 with NumericConstant

use of org.apache.sling.scripting.sightly.compiler.expression.nodes.NumericConstant in project sling by apache.

the class ExpressionWrapperTest method testJoinOptionsRemoval.

@Test
public void testJoinOptionsRemoval() {
    Interpolation interpolation = new Interpolation();
    Map<String, ExpressionNode> options = new HashMap<>();
    options.put(JoinFilter.JOIN_OPTION, new StringConstant(";"));
    List<ExpressionNode> array = new ArrayList<>();
    array.add(new NumericConstant(0));
    array.add(new NumericConstant(1));
    interpolation.addExpression(new Expression(new ArrayLiteral(array), options));
    ExpressionWrapper wrapper = new ExpressionWrapper(filters);
    Expression result = wrapper.transform(interpolation, MarkupContext.TEXT, ExpressionContext.TEXT);
    List<ExpressionNode> xssArguments = runOptionsAndXSSAssertions(result, 0);
    RuntimeCall join = (RuntimeCall) xssArguments.get(0);
    assertEquals(RuntimeFunction.JOIN, join.getFunctionName());
}
Also used : HashMap(java.util.HashMap) ArrayList(java.util.ArrayList) RuntimeCall(org.apache.sling.scripting.sightly.compiler.expression.nodes.RuntimeCall) ArrayLiteral(org.apache.sling.scripting.sightly.compiler.expression.nodes.ArrayLiteral) Interpolation(org.apache.sling.scripting.sightly.impl.compiler.frontend.Interpolation) Expression(org.apache.sling.scripting.sightly.compiler.expression.Expression) ExpressionNode(org.apache.sling.scripting.sightly.compiler.expression.ExpressionNode) NumericConstant(org.apache.sling.scripting.sightly.compiler.expression.nodes.NumericConstant) StringConstant(org.apache.sling.scripting.sightly.compiler.expression.nodes.StringConstant) ExpressionWrapper(org.apache.sling.scripting.sightly.impl.compiler.frontend.ExpressionWrapper) Test(org.junit.Test)

Example 5 with NumericConstant

use of org.apache.sling.scripting.sightly.compiler.expression.nodes.NumericConstant in project sling by apache.

the class ListPlugin method invoke.

@Override
public PluginInvoke invoke(final Expression expression, final PluginCallInfo callInfo, final CompilerContext compilerContext) {
    return new DefaultPluginInvoke() {

        private String listVariable = compilerContext.generateVariable("collectionVar");

        private String collectionSizeVar = compilerContext.generateVariable("size");

        @Override
        public void beforeElement(PushStream stream, String tagName) {
            stream.write(new VariableBinding.Start(listVariable, expression.getRoot()));
            stream.write(new VariableBinding.Start(collectionSizeVar, new UnaryOperation(UnaryOperator.LENGTH, new Identifier(listVariable))));
            stream.write(new Conditional.Start(collectionSizeVar, true));
        }

        @Override
        public void beforeChildren(PushStream stream) {
            String itemVariable = decodeItemVariable();
            String loopStatusVar = Syntax.itemLoopStatusVariable(itemVariable);
            String indexVariable = compilerContext.generateVariable("index");
            stream.write(new Loop.Start(listVariable, itemVariable, indexVariable));
            stream.write(new VariableBinding.Start(loopStatusVar, buildStatusObj(indexVariable, collectionSizeVar)));
        }

        @Override
        public void afterChildren(PushStream stream) {
            stream.write(VariableBinding.END);
            stream.write(Loop.END);
        }

        @Override
        public void afterElement(PushStream stream) {
            stream.write(Conditional.END);
            stream.write(VariableBinding.END);
            stream.write(VariableBinding.END);
        }

        private String decodeItemVariable() {
            String[] args = callInfo.getArguments();
            if (args.length > 0) {
                return args[0];
            }
            return Syntax.DEFAULT_LIST_ITEM_VAR_NAME;
        }

        private MapLiteral buildStatusObj(String indexVar, String sizeVar) {
            HashMap<String, ExpressionNode> obj = new HashMap<>();
            Identifier indexId = new Identifier(indexVar);
            BinaryOperation firstExpr = new BinaryOperation(BinaryOperator.EQ, indexId, NumericConstant.ZERO);
            BinaryOperation lastExpr = new BinaryOperation(BinaryOperator.EQ, indexId, new BinaryOperation(BinaryOperator.SUB, new Identifier(sizeVar), NumericConstant.ONE));
            obj.put(INDEX, indexId);
            obj.put(COUNT, new BinaryOperation(BinaryOperator.ADD, indexId, NumericConstant.ONE));
            obj.put(FIRST, firstExpr);
            obj.put(MIDDLE, new UnaryOperation(UnaryOperator.NOT, new BinaryOperation(BinaryOperator.OR, firstExpr, lastExpr)));
            obj.put(LAST, lastExpr);
            obj.put(ODD, parityCheck(indexId, NumericConstant.ZERO));
            obj.put(EVEN, parityCheck(indexId, NumericConstant.ONE));
            return new MapLiteral(obj);
        }

        private ExpressionNode parityCheck(ExpressionNode numericExpression, NumericConstant expected) {
            return new BinaryOperation(BinaryOperator.EQ, new BinaryOperation(BinaryOperator.REM, numericExpression, NumericConstant.TWO), expected);
        }
    };
}
Also used : Loop(org.apache.sling.scripting.sightly.compiler.commands.Loop) UnaryOperation(org.apache.sling.scripting.sightly.compiler.expression.nodes.UnaryOperation) HashMap(java.util.HashMap) BinaryOperation(org.apache.sling.scripting.sightly.compiler.expression.nodes.BinaryOperation) Conditional(org.apache.sling.scripting.sightly.compiler.commands.Conditional) MapLiteral(org.apache.sling.scripting.sightly.compiler.expression.nodes.MapLiteral) Identifier(org.apache.sling.scripting.sightly.compiler.expression.nodes.Identifier) ExpressionNode(org.apache.sling.scripting.sightly.compiler.expression.ExpressionNode) NumericConstant(org.apache.sling.scripting.sightly.compiler.expression.nodes.NumericConstant) PushStream(org.apache.sling.scripting.sightly.impl.compiler.PushStream) VariableBinding(org.apache.sling.scripting.sightly.compiler.commands.VariableBinding)

Aggregations

ExpressionNode (org.apache.sling.scripting.sightly.compiler.expression.ExpressionNode)5 NumericConstant (org.apache.sling.scripting.sightly.compiler.expression.nodes.NumericConstant)5 HashMap (java.util.HashMap)4 MapLiteral (org.apache.sling.scripting.sightly.compiler.expression.nodes.MapLiteral)3 StringConstant (org.apache.sling.scripting.sightly.compiler.expression.nodes.StringConstant)3 ArrayList (java.util.ArrayList)2 Conditional (org.apache.sling.scripting.sightly.compiler.commands.Conditional)2 Loop (org.apache.sling.scripting.sightly.compiler.commands.Loop)2 VariableBinding (org.apache.sling.scripting.sightly.compiler.commands.VariableBinding)2 Expression (org.apache.sling.scripting.sightly.compiler.expression.Expression)2 ArrayLiteral (org.apache.sling.scripting.sightly.compiler.expression.nodes.ArrayLiteral)2 BinaryOperation (org.apache.sling.scripting.sightly.compiler.expression.nodes.BinaryOperation)2 Identifier (org.apache.sling.scripting.sightly.compiler.expression.nodes.Identifier)2 RuntimeCall (org.apache.sling.scripting.sightly.compiler.expression.nodes.RuntimeCall)2 UnaryOperation (org.apache.sling.scripting.sightly.compiler.expression.nodes.UnaryOperation)2 PushStream (org.apache.sling.scripting.sightly.impl.compiler.PushStream)2 ExpressionWrapper (org.apache.sling.scripting.sightly.impl.compiler.frontend.ExpressionWrapper)2 Interpolation (org.apache.sling.scripting.sightly.impl.compiler.frontend.Interpolation)2 Test (org.junit.Test)2 BooleanConstant (org.apache.sling.scripting.sightly.compiler.expression.nodes.BooleanConstant)1