com.mysema.commons.freemarker.LinkDirective Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mysema-commons-freemarker

There is a newer version: 0.2.4

/*
 * Copyright (c) 2009 Mysema Ltd.
 * All rights reserved.
 * 
 */
package com.mysema.commons.freemarker;

import java.io.IOException;
import java.io.StringWriter;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;

import com.mysema.commons.web.ContextURL;

import freemarker.core.Environment;
import freemarker.template.TemplateDirectiveBody;
import freemarker.template.TemplateException;
import freemarker.template.TemplateModel;

/**
 * LinkDirective provides
 *
 * @author tiwe
 * @version $Id$
 */
public class LinkDirective extends FreeMarkerHelper{
    
    @SuppressWarnings("unchecked")
    public void execute(Environment env, Map params, TemplateModel[] loopVars,
            TemplateDirectiveBody body) throws TemplateException, IOException {
    	HttpServletRequest req = getRequest(env);
    	if (params.containsKey("self")){
    		String link = join(req.getRequestURI(), req.getQueryString());
    		env.getOut().write(link);
    	
    	}else if (params.containsKey("selfInContext")){	
    		String link = join(req.getRequestURI(), req.getQueryString());
    		if (req.getContextPath().length() > 1){
    			link = link.substring(req.getContextPath().length());
    		}
    		env.getOut().write(link);
    		
    	}else{
    		StringWriter writer = new StringWriter();
            body.render(writer);            
            env.getOut().write(ContextURL.resolve(req, writer.toString()));	
    	}       
    }
    
    private String join(String url, String queryString){
    	if (queryString != null){
    		return url + "?" + queryString;
    	}else{
    		return url;
    	}
    }
    
}