plugins/org.eclipse.gmf.xpand.ant/src-ant/org/eclipse/gmf/internal/xpand/ant/AbstractTemplateTask.java

/**
 * Copyright (c) 2009 Borland Software Corp.
 * 
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Alexander Shatalin (Borland) - initial API and implementation
 */
package org.eclipse.gmf.internal.xpand.ant;

import java.net.MalformedURLException;
import java.util.ArrayList;
import java.util.StringTokenizer;

import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.Task;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import org.eclipse.gmf.internal.xpand.EPackageRegistryBasedURIResourceMap;

public abstract class AbstractTemplateTask extends Task {

	private final InputSupport myInput;

	private final ResourceSetImpl myResourceSet;

	private String[] myTemplateRoots;

	private MetamodelRegistry myMetamodelRegistry;

	public AbstractTemplateTask() {
		ResourceSetImpl resourceSet = new ResourceSetImpl();
		resourceSet.setURIResourceMap(new EPackageRegistryBasedURIResourceMap(resourceSet.getURIConverter()));
		myResourceSet = resourceSet;
		myInput = new InputSupport(myResourceSet);
	}

	public void setTemplateRoot(String root) {
		ArrayList<String> roots = new ArrayList<String>();
		for (StringTokenizer st = new StringTokenizer(root, ";, "); st.hasMoreTokens();) {
			roots.add(st.nextToken().trim());
		}
		myTemplateRoots = roots.toArray(new String[roots.size()]);
	}

	protected String[] getTemplateRoots() {
		return myTemplateRoots;
	}

	public void setInputURI(String uri) {
		try {
			myInput.setURI(URI.createURI(uri));
		} catch (IllegalArgumentException ex) {
			throw new BuildException(ex, getLocation());
		}
	}

	public void setBareInput(String input) {
		myInput.setBareInput(input);
	}

	public void addMetamodelRegistry(MetamodelRegistry metamodelRegistry) {
		if (myMetamodelRegistry != null) {
			throw new BuildException("Only one metamodelregistry nested element supported", getLocation());
		}
		myMetamodelRegistry = metamodelRegistry;
	}

	protected InputSupport getInput() {
		return myInput;
	}

	protected XpandFacade createFacade() throws BuildException {
		XpandFacade xf = new XpandFacade(myResourceSet);
		if (myTemplateRoots != null && myTemplateRoots.length > 0) {
			try {
				for (String r : myTemplateRoots) {
					xf.addLocation(r);
				}
			} catch (MalformedURLException ex) {
				throw new BuildException(ex, getLocation());
			}
		}
		return registerMetamodels(xf);
	}

	private XpandFacade registerMetamodels(XpandFacade xf) {
		if (myMetamodelRegistry != null) {
			for (Metamodel metamodel : myMetamodelRegistry.getMetamodels()) {
				xf.registerMetamodel(metamodel.getNsUri(), metamodel.getUri());
			}
		}
		xf.setSchemaLocations(getInput().getInputSchemaLocations());
		return xf;
	}

}