com.baomidou.framework.velocity.directive.OverrideDirective Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-wind Show documentation

spring-wind extension of spring framework.

The newest version!

package com.baomidou.framework.velocity.directive;

import java.io.IOException;
import java.io.Writer;

import org.apache.velocity.context.InternalContextAdapter;
import org.apache.velocity.exception.MethodInvocationException;
import org.apache.velocity.exception.ParseErrorException;
import org.apache.velocity.exception.ResourceNotFoundException;
import org.apache.velocity.runtime.directive.Directive;
import org.apache.velocity.runtime.parser.node.Node;
import org.apache.velocity.runtime.parser.node.SimpleNode;

public class OverrideDirective extends Directive {

	@Override
	public String getName() {
		return "override";
	}


	@Override
	public int getType() {
		return BLOCK;
	}


	@Override
	public boolean render( InternalContextAdapter context, Writer writer, Node node )
		throws IOException, ResourceNotFoundException, ParseErrorException, MethodInvocationException {

		String name = DirectiveUtil.getRequiredArgument(context, node, 0, getName());
		OverrideNodeWrapper override = (OverrideNodeWrapper) context.get(DirectiveUtil.getOverrideVariableName(name));
		if ( override == null ) {
			Node body = node.jjtGetChild(1);
			context.put(DirectiveUtil.getOverrideVariableName(name), new OverrideNodeWrapper(body));
		} else {
			OverrideNodeWrapper current = new OverrideNodeWrapper(node.jjtGetChild(1));
			DirectiveUtil.setParentForTop(current, override);
		}
		return true;
	}


	@SuppressWarnings("unused")
	private boolean isOverrided( InternalContextAdapter context, String name ) {
		return context.get(DirectiveUtil.getOverrideVariableName(name)) != null;
	}

	static class OverrideNodeWrapper extends SimpleNode {

		Node current;

		OverrideNodeWrapper parentNode;


		public OverrideNodeWrapper( Node node ) {
			super(1);
			this.current = node;
		}


		public boolean render( InternalContextAdapter context, Writer writer )
			throws IOException, MethodInvocationException, ParseErrorException, ResourceNotFoundException {
			OverrideNodeWrapper preNode = (OverrideNodeWrapper) context.get(DirectiveUtil.OVERRIDE_CURRENT_NODE);
			try {
				context.put(DirectiveUtil.OVERRIDE_CURRENT_NODE, this);
				return current.render(context, writer);
			} finally {
				context.put(DirectiveUtil.OVERRIDE_CURRENT_NODE, preNode);
			}
		}
	}

}