ClasspathLookupDirect.java

/*******************************************************************************
 * Copyright (c) 2015 Pivotal Software, 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 
 * (http://www.eclipse.org/legal/epl-v10.html), and the Eclipse Distribution 
 * License v1.0 (http://www.eclipse.org/org/documents/edl-v10.html). 
 *
 * Contributors:
 *     Pivotal Software, Inc. - initial API and implementation
*******************************************************************************/
package org.springframework.ide.eclipse.core.java.typehierarchy;

import java.io.File;
import java.io.InputStream;
import java.net.URL;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.springframework.ide.eclipse.core.SpringCore;

/**
 * A simple lookup mechansim for finding resources on a number of directories and zip/jar files.
 * This can be used to lookup resources from a classpath of a project within the IDE.
 * 
 * You have to close a ClasspathLookup after using it in order to release possible file locks on zip
 * files, for example.
 * 
 * @author Martin Lippert
 * @since 3.7.0
 */
public class ClasspathLookupDirect implements ClasspathLookup {
	
	private ClasspathElement[] cpElements;
	
	public ClasspathLookupDirect(URL[] urls) {
		List<ClasspathElement> locations = new ArrayList<ClasspathElement>();
		
		Set<URL> usedURLs = new HashSet<URL>();
		for (URL url : urls) {
			if (!usedURLs.contains(url)) {
				if (url.toString().endsWith(".jar")) {
					try {
						String path = url.toURI().getPath();
						locations.add(new ClasspathElementJar(path));
						usedURLs.add(url);
					} catch (Exception e) {
						SpringCore.log(e);
					}
				}
				else {
					try {
						File file = new File(url.toURI());
						locations.add(new ClasspathElementDirectory(file));
						usedURLs.add(url);
					} catch (Exception e) {
						SpringCore.log(e);
					}
				}
			}
		}
		
		this.cpElements = locations.toArray(new ClasspathElement[0]);
	}

	public ClasspathLookupDirect(ClasspathElement[] cpElements) {
		this.cpElements = cpElements;
	}

	public InputStream getStream(String fullyQualifiedClassFileName, String packageName, String className) {
		for (int i = 0; i < cpElements.length; i++) {
			InputStream stream = null;
			synchronized(cpElements[i]) {
				try {
					stream = cpElements[i].getStream(fullyQualifiedClassFileName, packageName, className);
					if (stream != null) {
						return stream;
					}
				} catch (Exception e) {
				}
			}
		}
		return null;
	}

	public void close() {
		for (int i = 0; i < cpElements.length; i++) {
			synchronized(cpElements[i]) {
				cpElements[i].cleanup();
			}
		}
	}

}