com.mitchellbosecke.pebble.extension.escaper.EscaperExtension Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pebble Show documentation

Templating engine for Java.

There is a newer version: 2.4.0

/*******************************************************************************
 * 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.escaper;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.mitchellbosecke.pebble.extension.AbstractExtension;
import com.mitchellbosecke.pebble.extension.Filter;
import com.mitchellbosecke.pebble.extension.NodeVisitor;
import com.mitchellbosecke.pebble.tokenParser.AutoEscapeTokenParser;
import com.mitchellbosecke.pebble.tokenParser.TokenParser;

public class EscaperExtension extends AbstractExtension {

    private final EscapeFilter filter;

    private final EscaperNodeVisitor visitor;

    public EscaperExtension() {
        this.filter = new EscapeFilter();
        this.visitor = new EscaperNodeVisitor();
    }

    @Override
    public Map getFilters() {
        Map filters = new HashMap<>();
        filters.put("escape", filter);
        filters.put("raw", new RawFilter());
        return filters;
    }

    @Override
    public List getTokenParsers() {
        List parsers = new ArrayList<>();
        parsers.add(new AutoEscapeTokenParser());
        return parsers;
    }

    @Override
    public List getNodeVisitors() {
        List visitors = new ArrayList<>();
        visitors.add(visitor);
        return visitors;
    }

    /**
     * Sets the default escaping strategy.
     * 
     * @param strategy
     */
    public void setDefaultStrategy(String strategy) {
        filter.setDefaultStrategy(strategy);
    }

    public void setAutoEscaping(boolean auto) {
        visitor.pushAutoEscapeState(auto);
    }

    public void addSafeFilter(String filter) {
        visitor.addSafeFilter(filter);
    }

}