| package freemarker.ext.jsp; |
| |
| import java.beans.IntrospectionException; |
| import java.io.IOException; |
| import java.io.Writer; |
| import java.util.Map; |
| |
| import javax.servlet.jsp.JspContext; |
| import javax.servlet.jsp.JspException; |
| import javax.servlet.jsp.tagext.JspFragment; |
| import javax.servlet.jsp.tagext.JspTag; |
| import javax.servlet.jsp.tagext.SimpleTag; |
| import javax.servlet.jsp.tagext.Tag; |
| |
| import freemarker.core.Environment; |
| import freemarker.template.TemplateDirectiveBody; |
| import freemarker.template.TemplateDirectiveModel; |
| import freemarker.template.TemplateException; |
| import freemarker.template.TemplateModel; |
| import freemarker.template.TemplateModelException; |
| |
| /** |
| * @author Attila Szegedi |
| */ |
| class SimpleTagDirectiveModel extends JspTagModelBase implements TemplateDirectiveModel |
| { |
| protected SimpleTagDirectiveModel(Class tagClass) throws IntrospectionException { |
| super(tagClass); |
| if(!SimpleTag.class.isAssignableFrom(tagClass)) { |
| throw new IllegalArgumentException(tagClass.getName() + |
| " does not implement either the " + Tag.class.getName() + |
| " interface or the " + SimpleTag.class.getName() + |
| " interface."); |
| } |
| } |
| |
| public void execute(Environment env, Map args, TemplateModel[] outArgs, |
| final TemplateDirectiveBody body) |
| throws TemplateException, IOException { |
| try { |
| SimpleTag tag = (SimpleTag)getTagInstance(); |
| final FreeMarkerPageContext pageContext = PageContextFactory.getCurrentPageContext(); |
| pageContext.pushWriter(new JspWriterAdapter(env.getOut())); |
| try { |
| tag.setJspContext(pageContext); |
| JspTag parentTag = (JspTag)pageContext.peekTopTag(JspTag.class); |
| if(parentTag != null) { |
| tag.setParent(parentTag); |
| } |
| setupTag(tag, args, pageContext.getObjectWrapper()); |
| if(body != null) { |
| tag.setJspBody(new JspFragment() { |
| public JspContext getJspContext() { |
| return pageContext; |
| } |
| |
| public void invoke(Writer out) throws JspException, IOException { |
| try { |
| body.render(out == null ? pageContext.getOut() : out); |
| } |
| catch(TemplateException e) { |
| throw new JspException(e); |
| } |
| } |
| }); |
| pageContext.pushTopTag(tag); |
| try { |
| tag.doTag(); |
| } |
| finally { |
| pageContext.popTopTag(); |
| } |
| } |
| else { |
| tag.doTag(); |
| } |
| } |
| finally { |
| pageContext.popWriter(); |
| } |
| } |
| catch(TemplateException e) { |
| throw e; |
| } |
| catch(RuntimeException e) { |
| throw e; |
| } |
| catch(Exception e) { |
| throw new TemplateModelException(e); |
| } |
| } |
| } |