plugins/org.eclipse.bpmn2.modeler.runtime.jboss.jbpm/src/org/eclipse/bpmn2/modeler/runtime/jboss/jbpm5/wid/WIDLoader.java

package org.eclipse.bpmn2.modeler.runtime.jboss.jbpm5.wid;

import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.IResourceVisitor;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.Path;
import org.eclipse.jdt.core.IClasspathEntry;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.swt.graphics.ImageData;

/*
 * Class: Visits each file in the project to see if it's a *.conf/*.wid
 * and loads the file's contents into a String. If the given IProject is
 * a JavaProject, then all jars in the classpath are searched for *.conf/*.wid
 * files and their contents are loaded.
 * 
 * @author bfitzpat
 *
 */
public class WIDLoader {

	private static final int BUFFER_SIZE = 1024;
	private static final String WID_FOLDER = "/src/main/resources/META-INF";
	private static final String ICONS_FOLDER = "/src/main/resources/icons";
	
	private HashMap<String, WorkItemDefinition> projectWIDs = new HashMap<String, WorkItemDefinition>();
	private HashMap<String, ImageDescriptor> projectIcons = new HashMap<String, ImageDescriptor>();
	private HashMap<String, WorkItemDefinition> classpathWIDs = new HashMap<String, WorkItemDefinition>();
	private HashMap<String, ImageDescriptor> classpathIcons = new HashMap<String, ImageDescriptor>();
	
	private class WIDVisitor implements IResourceVisitor {

		@Override
		public boolean visit (IResource resource) throws CoreException {
			try {
				if (resource.getType() == IResource.FILE) {
					// only check the folder src/main/resources/META-INF for WID files
					if ("wid".equalsIgnoreCase(((IFile)resource).getFileExtension())) { //$NON-NLS-1$
						String folder = resource.getParent().getFullPath().toString();
						IProject project = resource.getProject();
						if ((project.getFullPath().toString() + WID_FOLDER).equals(folder)) {
							getProjectFileWIDs((IFile)resource);
						}
						return true;
					}
				}
				else if (resource.getType() == IResource.PROJECT) {
					// Check if this is a Java Project.
					// If so, check all jars in the classpath for WID files.
					IProject project = (IProject)resource;
					IJavaProject javaProject;
					javaProject = getJavaProject(project);
					if (javaProject!=null) {
						for (IClasspathEntry e : javaProject.getRawClasspath()) {
							if (e.getEntryKind() == IClasspathEntry.CPE_LIBRARY) {
								IPath path = e.getPath();
								String name = null;
								if (path.getDevice()!=null)
									name = e.getPath().toOSString();
								else
									name = project.getLocation().removeLastSegments(1).append(e.getPath()).toOSString();
								java.io.File jarFile = new java.io.File(name);
								getJarFileWIDs(jarFile);
							}
						}
					}
					return true;
				}
				else if (resource.getType() == IResource.FOLDER) {
					// skip over "bin" and "target" folders
					String name = resource.getName();
					if ("bin".equals(name) || "target".equals(name)) //$NON-NLS-1$ //$NON-NLS-2$
						return false;
				}
			} catch (IOException e1) {
				e1.printStackTrace();
			} catch (WIDException e2) {
				e2.printStackTrace();
			}
			return true;
		}
	}
	
	public void load(IProject project) throws CoreException {
		WIDVisitor visitor = new WIDVisitor();
		project.accept(visitor, IResource.DEPTH_INFINITE, false);
	}
	
	public HashMap<String, WorkItemDefinition> getProjectWIDs() {
		return projectWIDs;
	}

	public HashMap<String, WorkItemDefinition> getClasspathWIDs() {
		return classpathWIDs;
	}
	
	public HashMap<String, ImageDescriptor> getProjectIcons() {
		return projectIcons;
	}

	public HashMap<String, ImageDescriptor> getClasspathIcons() {
		return classpathIcons;
	}
	
	private void getProjectFileWIDs(IFile file) throws CoreException, IOException, WIDException {
		InputStream is = file.getContents();
		String content = inputStreamToString(is,null);
		HashMap<String, WorkItemDefinition> widMap = WIDParser.parse(content);
		for (WorkItemDefinition wid : widMap.values()) {
			String icon = wid.getIcon();
			if (icon!=null && !icon.isEmpty()) {
				getProjectFileIcon(file, icon);
				((WorkItemDefinitionImpl)wid).setDefinitionFile(file.getFullPath().toFile());
			}
		}
		projectWIDs.putAll(widMap);
	}
	
	private void getProjectFileIcon(IFile file, String icon) throws CoreException, IOException {
		IProject project = file.getProject();
		
		// search for icon file in /src/main/resources/icons
		IFile iconFile = project.getFile(new Path(ICONS_FOLDER + "/" + icon));
		
		InputStream is = null;
		try {
			if (iconFile.exists()) {
				is = iconFile.getContents();
				if (is != null) {
					ImageDescriptor imageDescriptor = ImageDescriptor.createFromImageData(new ImageData(is));
					projectIcons.put(icon, imageDescriptor);
					is.close();
				}
			}
		}
		finally {
			if (is!=null)
				is.close();
		}
	}
	
	private void getJarFileWIDs(java.io.File jarFile) throws IOException, WIDException {
		JarFile jar = new java.util.jar.JarFile(jarFile);
	    InputStream is = null;
		try {
//			System.out.println("Jar: "+jar.getName());
//			for (Entry<Object, Object> e : jar.getManifest().getMainAttributes().entrySet()) {
//				System.out.println("  "+e.getKey() + "=" + e.getValue());
//			}
			Enumeration<JarEntry> enumEntries = jar.entries();
			while (enumEntries.hasMoreElements()) {
			    JarEntry entry = enumEntries.nextElement();
			    if (!entry.isDirectory()) {
				    String name = entry.getName();
				    if (name.endsWith(".wid")) {
				    	is = jar.getInputStream(entry);
						if (is!=null) {
							String content = inputStreamToString(is,null);
							HashMap<String, WorkItemDefinition> widMap = WIDParser.parse(content);
							for (WorkItemDefinition wid : widMap.values()) {
								String icon = wid.getIcon();
								getJarFileIcon(jar, icon);
								((WorkItemDefinitionImpl)wid).setDefinitionFile(jarFile.getAbsoluteFile());
							}
							classpathWIDs.putAll(widMap);
							is.close();
						}
				    }
			    }
			}
		}
		finally {
			jar.close();
			if (is!=null)
				is.close();
		}
	}

	private void getJarFileIcon(JarFile jar, String icon) throws IOException {
		ImageDescriptor imageDescriptor = null;
		if (icon!=null && !icon.isEmpty() && !classpathIcons.containsKey(icon)) {
		    InputStream is = null;
			try {
				Enumeration<JarEntry> enumEntries = jar.entries();
				while (enumEntries.hasMoreElements()) {
				    JarEntry entry = enumEntries.nextElement();
				    if (!entry.isDirectory()) {
					    String name = entry.getName();
					    if (icon.equals(name)) {
					    	is = jar.getInputStream(entry);
							if (is!=null) {
								imageDescriptor = ImageDescriptor.createFromImageData(new ImageData(is));
								classpathIcons.put(icon, imageDescriptor);
								is.close();
							}
							break;
					    }
				    }
				}
			}
			finally {
				if (is!=null)
					is.close();
			}
		}
	}
	
	private IJavaProject getJavaProject(IProject project) throws IOException {
        if (project != null) {
            try {
                if (project.getNature("org.eclipse.jdt.core.javanature") != null) {
                    IJavaProject javaProject = JavaCore.create(project);
                    if (javaProject.exists()){
                        return javaProject;
                    }
                }
            } catch (CoreException e) {
            	e.printStackTrace();
            }
        }
        return null;
    }

	public static String inputStreamToString(InputStream inputStream, Charset charset) throws IOException {
		if (charset==null)
			charset = StandardCharsets.UTF_8;
	    StringBuilder builder = new StringBuilder();
	    InputStreamReader reader = new InputStreamReader(inputStream, charset);
	    char[] buffer = new char[BUFFER_SIZE];
	    int length;
	    while ((length = reader.read(buffer)) != -1) {
	        builder.append(buffer, 0, length);
	    }
	    return builder.toString();
	}
	
}