• Home
• com.glispa.combo
• combo-core
• 1.4.1
• source code
• TemplateWalker.java

×

Project download

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.

com.glispa.combo.TemplateWalker Maven / Gradle / Ivy

package com.glispa.combo;

import org.antlr.v4.runtime.tree.ParseTreeWalker;

import java.util.ArrayList;
import java.util.Collection;
import java.util.LinkedList;
import java.util.List;

/**
 * Internal class for walking through the parsed tree
 */
class TemplateWalker extends MacroBaseListener {

    private final MacroRegistry macroRegistry;
    private List> currentElements;
    private Collection> currentChain;

    public TemplateWalker(MacroRegistry macroRegistry) {
        this.macroRegistry = macroRegistry;
    }

    public List> visit(MacroParser.TemplateContext ctx) {
        currentElements = new ArrayList<>(ctx.getChildCount());
        ParseTreeWalker.DEFAULT.walk(this, ctx);
        return currentElements;
    }

    @Override
    public void enterMacroChain(MacroParser.MacroChainContext ctx) {
        currentChain = new LinkedList<>();
    }

    @Override
    public void enterMacro(MacroParser.MacroContext ctx) {
        MacroFactory macroFactory = macroRegistry.get(ctx.macroId().getText());
        List argContexts = ctx.arg();
        String[] args = new String[argContexts.size()];
        for (int i = 0; i < argContexts.size(); ++i) {
            args[i] = argContexts.get(i).getText();
        }
        currentChain.add(macroFactory.build(args));
    }

    @Override
    public void exitMacroChain(MacroParser.MacroChainContext ctx) {
        currentElements.add(new MacroChainTemplateElement<>(currentChain));
    }

    @Override
    public void exitText(MacroParser.TextContext ctx) {
        currentElements.add(new TextTemplateElement<>(ctx.getText()));
    }
}