com.greenpepper.confluence.macros.GreenPepperImport Maven / Gradle / Ivy

Go to download
package com.greenpepper.confluence.macros;

import java.util.*;

import com.atlassian.confluence.content.render.xhtml.ConversionContext;
import com.atlassian.confluence.macro.Macro;
import com.atlassian.confluence.macro.MacroExecutionException;
import com.atlassian.confluence.renderer.radeox.macros.MacroUtils;
import com.atlassian.confluence.util.velocity.VelocityUtils;
import com.atlassian.renderer.RenderContext;
import com.atlassian.renderer.v2.macro.MacroException;
import com.greenpepper.confluence.utils.MacroParametersUtils;
import com.greenpepper.confluence.velocity.ConfluenceGreenPepper;
import org.apache.commons.lang.StringUtils;

/**
 * GreenPepperImport class.
 *
 * @author oaouattara
 * @version $Id: $Id
 */
public class GreenPepperImport extends AbstractGreenPepperMacro
{
	/** Constant IMPORTS_PARAM="imports" */
	public static final String IMPORTS_PARAM = "imports";

    @Override
    public BodyType getBodyType() {
        return BodyType.PLAIN_TEXT;
    }

    @Override
    public String execute(Map parameters, String body, ConversionContext context) throws MacroExecutionException {
        String[] imports = StringUtils.split(body, '\n');
        List importList = Arrays.asList(imports);
        importList.addAll(getImportList(parameters));
        return executeMacro(importList);
    }

    /** {@inheritDoc} */
	@SuppressWarnings("unchecked")
    public String execute(Map parameters, String body, RenderContext renderContext) throws MacroException 
	{
        List importList = getImportList(parameters);
        return executeMacro(importList);
    }

    private String executeMacro(Collection importList) {
        try
        {
            Map contextMap = MacroUtils.defaultVelocityContext();
            contextMap.put(IMPORTS_PARAM, importList);
            return VelocityUtils.getRenderedTemplate("/templates/greenpepper/confluence/macros/greenPepperImport.vm", contextMap);
        }
        catch (Exception e)
        {
            return getErrorView("greenpepper.import.macroid", e.getMessage());
        }
    }

    /** {@inheritDoc} */
    @SuppressWarnings("unchecked")
    public static String getErrorView(String macroId, String errorId)
    {
        Map contextMap = MacroUtils.defaultVelocityContext();
        contextMap.put("macroId", macroId);
        contextMap.put("errorId", errorId);
        return VelocityUtils.getRenderedTemplate("/templates/greenpepper/confluence/macros/greenPepperMacros-error.vm", contextMap);
    }

    private List getImportList(Map parameters)
    {
        List imports = new ArrayList();
        
        // v3
        int index = 0;
        String importParam = (String)parameters.get(String.valueOf(index));
        while(importParam != null)
        {
            imports.add(ConfluenceGreenPepper.clean(importParam));
            importParam = (String)parameters.get(String.valueOf(++index));
        }
        
        // v4
        String[] values = MacroParametersUtils.extractParameterMultiple(IMPORTS_PARAM, parameters);
        for (String value : values) {
            if (value != null) {
                imports.add(ConfluenceGreenPepper.clean(value));
            }
        }
        
        return imports;
    }

    
}