plugins/org.eclipse.jst.j2ee.ejb.annotations.xdoclet/src/org/eclipse/jst/j2ee/ejb/annotations/internal/xdoclet/XDocletWebAntProjectBuilder.java

package org.eclipse.jst.j2ee.ejb.annotations.internal.xdoclet;

import java.io.File;
import java.net.URL;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Properties;

import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.Path;
import org.eclipse.core.runtime.Platform;
import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.IJavaElement;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.IPackageFragment;
import org.eclipse.jdt.core.IPackageFragmentRoot;
import org.eclipse.jdt.core.IParent;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jst.j2ee.internal.J2EEVersionConstants;
import org.eclipse.jst.j2ee.web.componentcore.util.WebArtifactEdit;
import org.eclipse.wst.common.componentcore.internal.ComponentResource;
import org.eclipse.wst.common.componentcore.internal.StructureEdit;
import org.eclipse.wst.common.componentcore.internal.WorkbenchComponent;

public class XDocletWebAntProjectBuilder extends XDocletAntProjectBuilder {

	String contextRoot = ""; //$NON-NLS-1$

	public XDocletWebAntProjectBuilder() {
		super();
		templateUrl = Platform.getBundle(XDocletAnnotationPlugin.PLUGINID).getEntry("/templates/builder/xdocletweb.xml"); //$NON-NLS-1$
	}

	protected String getTaskName() {
		return "webdoclet"; //$NON-NLS-1$
	}

	protected HashMap createTemplates(String beanPath) {
		HashMap templates = new HashMap();
		templates.put("@servlets@", beanPath); //$NON-NLS-1$

		templates.put("@jboss@", addJbossTask(contextRoot)); //$NON-NLS-1$
		templates.put("@jonas@", addJonasTask(contextRoot)); //$NON-NLS-1$
		templates.put("@weblogic@", addWeblogicTask(contextRoot)); //$NON-NLS-1$
		templates.put("@websphere@", addWebSphereTask(contextRoot)); //$NON-NLS-1$

		return templates;
	}

	protected Properties createAntBuildProperties(IResource resource, IJavaProject javaProject,
			IPackageFragmentRoot packageFragmentRoot, String beanPath) {
		Properties properties = new Properties();
		StructureEdit moduleCore = null;
		WebArtifactEdit webEdit = null;
		try {
			moduleCore = StructureEdit.getStructureEditForRead(javaProject.getProject());
			WorkbenchComponent wbModule = null;
			ComponentResource[] moduleResources = moduleCore.findResourcesBySourcePath(resource.getProjectRelativePath());
			for (int i = 0; i < moduleResources.length; i++) {
				ComponentResource moduleResource = moduleResources[i];
				if (moduleResource != null)
					wbModule = moduleResource.getComponent();
				if (wbModule != null)
					break;
			}
			IProject proj = StructureEdit.getContainingProject(wbModule);
			webEdit = WebArtifactEdit.getWebArtifactEditForRead(proj);
			int j2eeVersion = 0;
			if (webEdit != null) {
				j2eeVersion = webEdit.getJ2EEVersion();
				// TODO
				contextRoot = webEdit.getServerContextRoot();
			}
			properties.put("web.module.webinf", getWebInfFolder(wbModule).toString()); //$NON-NLS-1$
			properties.put("web", contextRoot); //$NON-NLS-1$
			properties.put("web.project.dir", resource.getProject().getLocation().toString()); //$NON-NLS-1$
			properties.put("web.project.classpath", asClassPath(javaProject)); //$NON-NLS-1$
			properties.put("web.module.src", packageFragmentRoot.getResource().getProjectRelativePath().toString()); //$NON-NLS-1$
			properties.put("web.module.gen", packageFragmentRoot.getResource().getProjectRelativePath().toString()); //$NON-NLS-1$
			properties.put("web.bin.dir", this.getJavaProjectOutputContainer(javaProject).toString()); //$NON-NLS-1$
			properties.put("xdoclet.home", getPreferenceStore().getProperty(XDocletPreferenceStore.XDOCLETHOME)); //$NON-NLS-1$
			URL url = Platform.getBundle("org.apache.ant").getEntry("/"); //$NON-NLS-1$ //$NON-NLS-2$
			url = Platform.asLocalURL(url);
			File file = new File(url.getFile());
			properties.put("ant.home", file.getAbsolutePath()); //$NON-NLS-1$

			String servletLevel = J2EEVersionConstants.VERSION_2_2_TEXT;
			if (j2eeVersion == J2EEVersionConstants.J2EE_1_3_ID)
				servletLevel = J2EEVersionConstants.VERSION_2_3_TEXT;
			else if (j2eeVersion == J2EEVersionConstants.J2EE_1_4_ID)
				servletLevel = J2EEVersionConstants.VERSION_2_4_TEXT;

			properties.put("servlet.spec.version", servletLevel); //$NON-NLS-1$
			properties.put("java.class.path", ""); //$NON-NLS-1$ //$NON-NLS-2$
		} catch (Exception e) {
			Logger.logException(e);
		} finally {
			if (moduleCore != null)
				moduleCore.dispose();
			if (webEdit != null)
				webEdit.dispose();
		}
		return properties;
	}

	protected String constructAnnotatedClassList(IPackageFragmentRoot root, IResource changedBean) {

		List webClasses = new ArrayList();
		getAllAnnotatedWebClasses(root, webClasses);
		String beans = ""; //$NON-NLS-1$
		try {
			Iterator iterator = webClasses.iterator();
			while (iterator.hasNext()) {
				ICompilationUnit cu = (ICompilationUnit) iterator.next();
				IResource bean;
				bean = cu.getCorrespondingResource();
				IPath path = bean.getProjectRelativePath();
				path = makeRelativeTo(path, root);
				beans += "\t<include name=\"" + path.toString() + "\" />\n"; //$NON-NLS-1$ //$NON-NLS-2$

			}
		} catch (JavaModelException e) {
			Logger.logException(e);
		}
		return beans;
	}

	/**
	 * @param root
	 * @return
	 */
	private void getAllAnnotatedWebClasses(IParent root, List list) {
		IJavaElement[] elements = null;
		try {
			elements = root.getChildren();
			if (elements == null)
				return;
			for (int i = 0; i < elements.length; i++) {
				IJavaElement element = elements[i];

				if (XDoxletAnnotationUtil.isXDocletAnnotatedWebClass(element)) {
					list.add(element);
				} else if (element.getElementType() == IJavaElement.PACKAGE_FRAGMENT) {
					getAllAnnotatedWebClasses((IPackageFragment) element, list);
				}
			}
		} catch (JavaModelException e) {
			Logger.logException(e);
		}
	}

	protected void refreshProjects(IProject project, IProgressMonitor monitor) throws CoreException {
		if (project != null)
			project.refreshLocal(IResource.DEPTH_INFINITE, monitor);

	}

	protected IPath getWebInfFolder(WorkbenchComponent webModule) {
		ComponentResource[] webXML = webModule.findResourcesByRuntimePath(new Path("/WEB-INF/web.xml"));
		if (webXML.length > 0)
			return webXML[0].getSourcePath().removeLastSegments(1);
		return null;
	}

	private String addJbossTask(String contextRoot) {
		if (!getPreferenceStore().isPropertyActive(XDocletPreferenceStore.WEB_JBOSS))
			return ""; //$NON-NLS-1$
		return "<jbosswebxml version=\"" + getPreferenceStore().getProperty(XDocletPreferenceStore.WEB_JBOSS + "_VERSION") + "\""
				+ "    contextroot=\"" + contextRoot + "\"" + "    xmlencoding=\"UTF-8\"" + "    destdir=\"\\${web.dd.dir}\""
				+ "    validatexml=\"false\"" + "    mergedir=\"\\${web.dd.dir}\" />";

	}

	private String addJonasTask(String contextRoot) {
		if (!getPreferenceStore().isPropertyActive(XDocletPreferenceStore.WEB_JONAS))
			return ""; //$NON-NLS-1$
		return "<jonaswebxml version=\"" + getPreferenceStore().getProperty(XDocletPreferenceStore.WEB_JONAS + "_VERSION") + "\""
				+ "    contextroot=\"" + contextRoot + "\"" + "    xmlencoding=\"UTF-8\"" + "    destdir=\"\\${web.dd.dir}\""
				+ "    validatexml=\"false\"" + "    mergedir=\"\\${web.dd.dir}\" />";

	}

	private String addWeblogicTask(String contextRoot) {
		if (!getPreferenceStore().isPropertyActive(XDocletPreferenceStore.WEB_WEBLOGIC))
			return ""; //$NON-NLS-1$
		return "<weblogicwebxml version=\"" + getPreferenceStore().getProperty(XDocletPreferenceStore.WEB_WEBLOGIC + "_VERSION")
				+ "\"" + "    contextRoot=\"" + contextRoot + "\"" + "    xmlencoding=\"UTF-8\"" + "    destdir=\"\\${web.dd.dir}\""
				+ "    validatexml=\"false\"" + "    mergedir=\"\\${web.dd.dir}\" />";

	}

	private String addWebSphereTask(String contextRoot) {
		if (!getPreferenceStore().isPropertyActive(XDocletPreferenceStore.WEB_WEBSPHERE))
			return ""; //$NON-NLS-1$
		return "<webspherewebxml " + "    xmlencoding=\"UTF-8\"" + "    destdir=\"\\${web.dd.dir}\"" + "    validateXML=\"false\""
				+ "    mergedir=\"\\${web.dd.dir}\" />";
	}

}