org.jnario.ui.buildpath.JnarioLibClasspathAdder.java Source code

Introduction

Here is the source code for org.jnario.ui.buildpath.JnarioLibClasspathAdder.java

Source

/*******************************************************************************
 * Copyright (c) 2012 BMW Car IT 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
 *******************************************************************************/
package org.jnario.ui.buildpath;

import static com.google.common.collect.Sets.newHashSet;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import org.apache.log4j.Logger;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.SubMonitor;
import org.eclipse.jdt.core.IClasspathEntry;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.xtext.util.MergeableManifest;

/**
 * @author Sebastian Benz
 *
 */
public class JnarioLibClasspathAdder {

    private static final Logger LOG = Logger.getLogger(JnarioLibClasspathAdder.class);

    private static final String PLUGIN_NATURE = "org.eclipse.pde.PluginNature";

    public void addLibsToClasspath(IJavaProject javaProject, IProgressMonitor monitor) {
        try {
            SubMonitor progress = SubMonitor.convert(monitor, 2);
            IProject project = javaProject.getProject();
            if (!project.hasNature(PLUGIN_NATURE) || !addToPluginManifest(project, progress.newChild(1)))
                addToClasspath(javaProject, progress.newChild(1));
        } catch (Exception exc) {
            LOG.error("Error adding Xtend libs to classpath", exc);
        }
    }

    protected boolean addToClasspath(IJavaProject javaProject, IProgressMonitor monitor) throws JavaModelException {
        IClasspathEntry xtendContainerEntry = JavaCore
                .newContainerEntry(JnarioContainerInitializer.JNARIO_LIBRARY_PATH);
        IClasspathEntry[] rawClasspath = javaProject.getRawClasspath();
        IClasspathEntry[] newRawClasspath = new IClasspathEntry[rawClasspath.length + 1];
        for (int i = 0; i < rawClasspath.length; ++i) {
            IClasspathEntry entry = rawClasspath[i];
            if (entry.getEntryKind() == IClasspathEntry.CPE_CONTAINER
                    && entry.getPath().equals(xtendContainerEntry.getPath())) {
                return false;
            }
            newRawClasspath[i + 1] = entry;
        }
        newRawClasspath[0] = xtendContainerEntry;
        javaProject.setRawClasspath(newRawClasspath, monitor);
        return true;
    }

    protected boolean addToPluginManifest(IProject project, IProgressMonitor monitor)
            throws IOException, CoreException {
        IResource manifestFile = project.findMember("META-INF/MANIFEST.MF");
        if (manifestFile != null && manifestFile.isAccessible()
                && !manifestFile.getResourceAttributes().isReadOnly() && manifestFile instanceof IFile) {
            OutputStream output = null;
            InputStream input = null;
            try {
                MergeableManifest manifest = new MergeableManifest(((IFile) manifestFile).getContents());
                manifest.addRequiredBundles(newHashSet(JnarioContainerInitializer.BUNDLE_IDS_TO_INCLUDE));
                ByteArrayOutputStream out = new ByteArrayOutputStream();
                output = new BufferedOutputStream(out);
                manifest.write(output);
                ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray());
                input = new BufferedInputStream(in);
                ((IFile) manifestFile).setContents(input, true, true, monitor);
                return true;
            } finally {
                if (output != null)
                    output.close();
                if (input != null)
                    input.close();
            }
        }
        return false;
    }
}