SeamHandler.java

/*************************************************************************************
 * Copyright (c) 2010-2011 Red Hat, Inc. and others.
 * 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:
 *     JBoss by Red Hat - Initial implementation.
 ************************************************************************************/
package org.jboss.tools.runtime.handlers;

import java.io.File;
import java.io.FileFilter;
import java.io.FilenameFilter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.jboss.tools.runtime.core.JBossRuntimeLocator;
import org.jboss.tools.runtime.core.RuntimeCoreActivator;
import org.jboss.tools.runtime.core.internal.RuntimeDetector;
import org.jboss.tools.runtime.core.model.AbstractRuntimeDetectorDelegate;
import org.jboss.tools.runtime.core.model.IRuntimeDetector;
import org.jboss.tools.runtime.core.model.IRuntimeDetectorDelegate;
import org.jboss.tools.runtime.core.model.RuntimeDefinition;
import org.jboss.tools.runtime.core.util.RuntimeJarUtil;
import org.jboss.tools.seam.core.SeamUtil;
import org.jboss.tools.seam.core.project.facet.SeamRuntime;
import org.jboss.tools.seam.core.project.facet.SeamRuntimeManager;
import org.jboss.tools.seam.core.project.facet.SeamVersion;

public class SeamHandler extends AbstractRuntimeDetectorDelegate {

	private final static String seamJarName = "jboss-seam.jar"; //$NON-NLS-1$
	private final static String seamVersionAttributeName = "Seam-Version"; //$NON-NLS-1$
	private static final String SEAM = "SEAM";  //$NON-NLS-1$
	
	private static File getSeamRoot(RuntimeDefinition runtimeDefinition) {
		String type = runtimeDefinition.getType();
		if (SEAM.equals(type)) {
			return runtimeDefinition.getLocation();
		}
		return null;
	}
	
	@Override
	public boolean initializeRuntime(RuntimeDefinition runtimeDefinition) {
		return initializeRuntime(runtimeDefinition, true);
	}
	
	public boolean initializeRuntime(RuntimeDefinition runtimeDefinition, boolean save) {
		if (runtimeDefinition.isEnabled()) {
			String type = runtimeDefinition.getType();
			if (SEAM.equals(type)) {
				boolean ret = addSeam(runtimeDefinition,
						runtimeDefinition.getLocation());
				initializeRuntimes(runtimeDefinition.getIncludedRuntimeDefinitions());
				if( save ) {
					SeamRuntimeManager.getInstance().save();
				}
				return ret;
			}
		}
		return false;
	}
	
	@Override
	public void initializeRuntimes(List<RuntimeDefinition> runtimeDefinitions) {
		for(RuntimeDefinition runtimeDefinition:runtimeDefinitions) {
			initializeRuntime(runtimeDefinition, false);
		}
		SeamRuntimeManager.getInstance().save();
	}

	private static boolean addSeam(RuntimeDefinition runtimeDefinition, File seamFile) {
		if (seamFile.exists() && seamFile.canRead() && seamFile.isDirectory()) {
			SeamVersion seamVersion = getSeamVersion(seamFile.getAbsolutePath());
			if (seamVersion != null) {
				String name = "Seam " + runtimeDefinition.getName() + " " + seamVersion; //$NON-NLS-1$ //$NON-NLS-2$
				List<String> names = SeamRuntimeManager.getInstance().getAllRuntimeNames();
				if (names != null && names.size() > 0) {
					int i = 2;
					String baseName = "Seam " + runtimeDefinition.getName(); //$NON-NLS-1$
					while (names.contains(name)) {
						name = baseName + " (" + i++ + ") " + seamVersion; //$NON-NLS-1$ //$NON-NLS-2$
					}
				}
				return addSeam(seamFile.getAbsolutePath(), seamVersion, name);
			}
		}
		return false;
	}

	private static boolean addSeam(String seamPath,SeamVersion seamVersion, String name) {
		if (!seamExists(seamPath)) {
			File seamFolder = new File(seamPath);
			if(seamFolder.exists() && seamFolder.isDirectory()) {
				SeamRuntime rt = new SeamRuntime();
				rt.setHomeDir(seamPath);
				rt.setName(name);
				rt.setDefault(true);
				rt.setVersion(seamVersion);
				SeamRuntimeManager.getInstance().addRuntime(rt);
				return true;
			}
		}
		return false;
	}

	private static SeamVersion getSeamVersion(String seamGenBuildPath) {
		if (seamGenBuildPath == null || seamGenBuildPath.trim().length() <= 0) {
			return null;
		}
		String fullVersion = SeamUtil.getSeamVersionFromManifest(seamGenBuildPath);
		if (fullVersion == null) {
			return null;	
		}
		String version = fullVersion.substring(0,3);
		SeamVersion seamVersion = null;
		if (version != null) {
			seamVersion = SeamVersion.findByString(version);
		}
		return seamVersion;
	}
	
	/**
	 * @param seamPath
	 * @return
	 */
	private static boolean seamExists(String seamPath) {
		SeamRuntime[] seamRuntimes = SeamRuntimeManager.getInstance().getRuntimes();
		for (SeamRuntime sr:seamRuntimes) {
			if (seamPath != null && seamPath.equals(sr.getHomeDir())) {
				return true;
			}
		}
		return false;
	}

	@Override
	public RuntimeDefinition getRuntimeDefinition(File root,
			IProgressMonitor monitor) {
		if (monitor.isCanceled() || root == null) {
			return null;
		}
		String seamVersion = getSeamVersionFromManifest(root.getAbsolutePath());
		if (seamVersion != null) {
			return new RuntimeDefinition(root.getName(), seamVersion, SEAM, root.getAbsoluteFile());
		}
		return null;
	}

	private static String getSeamVersionFromManifest(String seamHome) {
		File seamHomeFolder = new File(seamHome);
		if (seamHomeFolder == null || !seamHomeFolder.isDirectory()) {
			return null;
		}
		String[] seamFiles = seamHomeFolder.list(new FilenameFilter() {
			
			public boolean accept(File dir, String name) {
				if ("seam-gen".equals(name)) { //$NON-NLS-1$
					return true;
				}
				if ("lib".equals(name)) { //$NON-NLS-1$
					return true;
				}
				return false;
			}
		});
		if (seamFiles == null || seamFiles.length != 2) {
			return null;
		}
		File jarFile = new File(seamHome, "lib/" + seamJarName); //$NON-NLS-1$
		if(!jarFile.isFile()) {
			jarFile = new File(seamHome, seamJarName);
			if(!jarFile.isFile()) {
				return null;
			}
		}
		String[] attributes = new String[]{seamVersionAttributeName, RuntimeJarUtil.IMPLEMENTATION_VERSION};
		return RuntimeJarUtil.getImplementationVersion(jarFile, attributes);
	}

	@Override
	public boolean exists(RuntimeDefinition runtimeDefinition) {
		if (runtimeDefinition == null || runtimeDefinition.getLocation() == null) {
			return false;
		}
		File seamRoot = getSeamRoot(runtimeDefinition);
		if (seamRoot == null || !seamRoot.isDirectory()) {
			return false;
		}
		String path = seamRoot.getAbsolutePath();
		return seamExists(path);
	}

	private static File getLocation(RuntimeDefinition runtimeDefinitions) {
		String type = runtimeDefinitions.getType();
		String version = runtimeDefinitions.getVersion();
		if ("EAP".equals(type) && version != null && version.startsWith("6") ) {//$NON-NLS-1$ //$NON-NLS-2$
			return runtimeDefinitions.getLocation();
		}
		if ("SOA_P".equals(type) || "EAP".equals(type) || "EPP".equals(type)) { //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
			return new File(runtimeDefinitions.getLocation(), "jboss-as");//$NON-NLS-1$
		}
		if ("SOA_P_STD".equals(type)) { //$NON-NLS-1$
			return new File(runtimeDefinitions.getLocation(),"jboss-esb"); //$NON-NLS-1$					
		}
		if("EWP".equals(type)) { //$NON-NLS-1$
				return new File(runtimeDefinitions.getLocation(),"jboss-as-web"); //$NON-NLS-1$
		}
		if ("AS".equals(type) || "EAP_STD".equals(type)) {  //$NON-NLS-1$//$NON-NLS-2$
			return runtimeDefinitions.getLocation();
		}
		return null;
	}
	@Override
	public void computeIncludedRuntimeDefinition(
			RuntimeDefinition runtimeDefinition) {
		runtimeDefinition.getIncludedRuntimeDefinitions().clear();
		List<RuntimeDefinition> runtimeDefinitions = runtimeDefinition
				.getIncludedRuntimeDefinitions();
		
		final File location = getLocation(runtimeDefinition);
		File[] directories = runtimeDefinition.getLocation().listFiles(
				new FileFilter() {
					public boolean accept(File file) {
						if (!file.isDirectory() || file.equals(location)) {
							return false;
						}
						return true;
					}
				});
		List<RuntimeDefinition> definitions = new ArrayList<RuntimeDefinition>();
		JBossRuntimeLocator locator = new JBossRuntimeLocator();
		Set<IRuntimeDetector> seamDetectors = new HashSet<IRuntimeDetector>();
		Set<IRuntimeDetector> runtimeDetectors = RuntimeCoreActivator.getDefault().getRuntimeDetectors();
		for (IRuntimeDetector runtimeDetector : runtimeDetectors) {
			if (runtimeDetector instanceof RuntimeDetector) {
				IRuntimeDetectorDelegate delegate = ((RuntimeDetector) runtimeDetector).getDelegate();
				if (delegate instanceof SeamHandler) {
					seamDetectors.add(runtimeDetector);
				break;
			}
			}
		}
		for (File directory : directories) {
			locator.searchDirectory(directory, definitions, 1, seamDetectors, new NullProgressMonitor());
			for (RuntimeDefinition definition:definitions) {
				definition.setParent(runtimeDefinition);
			}
		}
		for (RuntimeDefinition definition:definitions) {
			if (!runtimeDefinitions.contains(definition)) {
				runtimeDefinitions.add(definition);
			}
		}
	}

	@Override
	public String getVersion(RuntimeDefinition runtimeDefinition) {
		return runtimeDefinition.getVersion();
	}

}