ServiceMapperTask.java

/*
 * Copyright 2011 cruxframework.org.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package org.cruxframework.crux.tools.servicemap;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.Project;
import org.apache.tools.ant.Task;
import org.apache.tools.ant.taskdefs.Java;
import org.apache.tools.ant.types.Path;
import org.apache.tools.ant.types.Commandline.Argument;

/**
 * An ant task to call the ServiceMapper program 
 * 
 * @author Thiago da Rosa de Bustamante
 */
public class ServiceMapperTask extends Task
{
	private List<Path> classpath = new ArrayList<Path>();
	private List<Argument> args = new ArrayList<Argument>();
	private List<Argument> jvmArgs = new ArrayList<Argument>();
	private String inputCharset = "UTF-8";
	private boolean failOnError = true;
	private File projectDir;

	public void addClasspath(Path classpath)
	{
		this.classpath.add(classpath);
	}
	
	public void addArg(Argument arg)
	{
		this.args.add(arg);
	}

	public void addJvmarg(Argument jvmarg)
	{
		this.jvmArgs.add(jvmarg);
	}
	
	public void setInputCharset(String inputCharset)
	{
		this.inputCharset = inputCharset;
	}


	public void setFailOnError(Boolean failOnError)
	{
		this.failOnError = failOnError;
	}
	
	public void setProjectDir(File projectDir)
	{
		this.projectDir = projectDir;
	}

	/**
	 * @see org.apache.tools.ant.Task#execute()
	 */
	public void execute() throws BuildException
	{
		try 
		{
			generateMap();
		} 
		catch (Throwable e) 
		{
			log(e.getMessage(), Project.MSG_ERR);
			new BuildException(e.getMessage(), e);
		}
	}

	/**
	 * 
	 */
	private void generateMap()
	{
		log("Generating map...");
		Java javatask = (Java) getProject().createTask("java");
		javatask.setClassname(ServiceMapper.class.getCanonicalName());
		javatask.setFork(true);
		javatask.setFailonerror(true);

		javatask.createJvmarg().setValue("-Dfile.encoding="+this.inputCharset );
		
		for (Argument arg : jvmArgs)
		{
			if(arg != null)
			{
				javatask.createJvmarg().setValue(arg.getParts()[0]);
			}
		}

		addServiceMapperParameters(javatask);
		
		for (Argument arg : args)
		{
			if(arg != null)
			{
				javatask.createArg().setValue(arg.getParts()[0]);
			}
		}

		for (Path path : this.classpath)
		{
			javatask.setClasspath(path);
		}
		
		int resultCode = javatask.executeJava();
		if (resultCode != 0)
		{
			if (this.failOnError )
			{
				throw new ServiceMapperException("Crux Service Mapper returned errors.");
			}
			else
			{
				log("Crux Service Mapper returned errors.", Project.MSG_ERR);				
			}
		}
	}

	/**
	 * @param javatask
	 */
	private void addServiceMapperParameters(Java javatask)
	{
		if (projectDir == null)
		{
			throw new ServiceMapperException("Missing required parameter <projectDir>.");
		}
		else
		{
			javatask.createArg().setValue("projectDir");
			try
			{
				javatask.createArg().setValue(projectDir.getCanonicalPath());
			}
			catch (IOException e)
			{
				throw new ServiceMapperException("Invalid <ProjectDir>.", e);
			}
			javatask.createArg().setValue("-override");
		}
	}
}