com.mitchellbosecke.pebble.extension.AbstractNodeVisitor Maven / Gradle / Ivy

Go to download
/*******************************************************************************
 * This file is part of Pebble.
 * 
 * Copyright (c) 2014 by Mitchell Bösecke
 * 
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 ******************************************************************************/
package com.mitchellbosecke.pebble.extension;

import com.mitchellbosecke.pebble.node.ArgumentsNode;
import com.mitchellbosecke.pebble.node.AutoEscapeNode;
import com.mitchellbosecke.pebble.node.BlockNode;
import com.mitchellbosecke.pebble.node.BodyNode;
import com.mitchellbosecke.pebble.node.ExtendsNode;
import com.mitchellbosecke.pebble.node.FlushNode;
import com.mitchellbosecke.pebble.node.ForNode;
import com.mitchellbosecke.pebble.node.IfNode;
import com.mitchellbosecke.pebble.node.ImportNode;
import com.mitchellbosecke.pebble.node.IncludeNode;
import com.mitchellbosecke.pebble.node.MacroNode;
import com.mitchellbosecke.pebble.node.NamedArgumentNode;
import com.mitchellbosecke.pebble.node.Node;
import com.mitchellbosecke.pebble.node.ParallelNode;
import com.mitchellbosecke.pebble.node.PositionalArgumentNode;
import com.mitchellbosecke.pebble.node.PrintNode;
import com.mitchellbosecke.pebble.node.RootNode;
import com.mitchellbosecke.pebble.node.SetNode;
import com.mitchellbosecke.pebble.node.TextNode;
import com.mitchellbosecke.pebble.node.expression.Expression;
import com.mitchellbosecke.pebble.template.PebbleTemplateImpl;
import com.mitchellbosecke.pebble.utils.Pair;

/**
 * A base node visitor that can be extended for the sake of using it's
 * navigational abilities.
 * 
 * @author Mitchell
 * 
 */
public class AbstractNodeVisitor implements NodeVisitor {

    protected PebbleTemplateImpl template;

    public void setTemplate(PebbleTemplateImpl template) {
        this.template = template;
    }

    /**
     * Default method used for unknown nodes such as nodes from a user provided
     * extension.
     */
    @Override
    public void visit(Node node) {
    }

    /*
     * OVERLOADED NODES (keep alphabetized)
     */
    @Override
    public void visit(ArgumentsNode node) {
        if (node.getNamedArgs() != null) {
            for (Node arg : node.getNamedArgs()) {
                arg.accept(this);
            }
        }
        if (node.getPositionalArgs() != null) {
            for (Node arg : node.getPositionalArgs()) {
                arg.accept(this);
            }
        }
    }

    @Override
    public void visit(AutoEscapeNode node) {
        node.getBody().accept(this);
    }

    @Override
    public void visit(BlockNode node) {
        node.getBody().accept(this);
    }

    @Override
    public void visit(BodyNode node) {
        for (Node child : node.getChildren()) {
            child.accept(this);
        }
    }

    @Override
    public void visit(ExtendsNode node) {
        node.getParentExpression().accept(this);
    }

    @Override
    public void visit(FlushNode node) {

    }

    @Override
    public void visit(ForNode node) {
        node.getIterable().accept(this);
        node.getBody().accept(this);
        if (node.getElseBody() != null) {
            node.getElseBody().accept(this);
        }
    }

    @Override
    public void visit(IfNode node) {
        for (Pair, BodyNode> pairs : node.getConditionsWithBodies()) {
            pairs.getLeft().accept(this);
            pairs.getRight().accept(this);
        }
        if (node.getElseBody() != null) {
            node.getElseBody().accept(this);
        }
    }

    @Override
    public void visit(ImportNode node) {
        node.getImportExpression().accept(this);
    }

    @Override
    public void visit(IncludeNode node) {
        node.getIncludeExpression().accept(this);
    }

    @Override
    public void visit(MacroNode node) {
        node.getBody().accept(this);
        node.getArgs().accept(this);
    }

    @Override
    public void visit(NamedArgumentNode node) {
        if (node.getValueExpression() != null) {
            node.getValueExpression().accept(this);
        }
    }

    @Override
    public void visit(ParallelNode node) {
        node.getBody().accept(this);
    }

    @Override
    public void visit(PositionalArgumentNode node) {
        node.getValueExpression().accept(this);
    }

    @Override
    public void visit(PrintNode node) {
        node.getExpression().accept(this);
    }

    @Override
    public void visit(RootNode node) {
        node.getBody().accept(this);
    }

    @Override
    public void visit(SetNode node) {
        node.getValue().accept(this);
    }

    @Override
    public void visit(TextNode node) {

    }

}