InjectLicensingMojo.java

/*******************************************************************************
 * Copyright (c) 2015 Kaloyan Raev.
 * 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:
 *    Kaloyan Raev - initial API and implementation
 *******************************************************************************/
package org.eclipse.licensing.eclipse_licensing_plugin;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Enumeration;
import java.util.List;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
import java.util.jar.Manifest;

import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugins.annotations.Mojo;
import org.apache.maven.plugins.annotations.Parameter;
import org.apache.maven.project.MavenProject;
import org.apache.maven.shared.utils.StringUtils;
import org.eclipse.tycho.ArtifactType;
import org.eclipse.tycho.core.utils.TychoProjectUtils;

/**
 * Injects the core licensing classes into the licensed plugin.
 */
@Mojo(name = "inject-licensing")
public class InjectLicensingMojo extends AbstractMojo {

	@Parameter(property = "project.build.outputDirectory", required = true)
	protected File classesDirectory;

	@Parameter(property = "project", readonly = true)
	private MavenProject project;

	public void execute() throws MojoExecutionException {
		try {
			extractLicensingBaseFiles();
			updateManifest();
		} catch (IOException e) {
			throw new MojoExecutionException(e.getMessage(), e);
		}
	}

	private void extractLicensingBaseFiles() throws IOException {
		JarFile jar = new JarFile(getLicensingBasePlugin());
		Enumeration<JarEntry> entries = jar.entries();
		while (entries.hasMoreElements()) {
			JarEntry entry = entries.nextElement();
			String entryName = entry.getName();
			if (entryName.startsWith("org")) {
				File file = new File(classesDirectory, entryName);
				if (entry.isDirectory()) {
					file.mkdir();
				} else {
					InputStream is = jar.getInputStream(entry);
					FileOutputStream fos = new FileOutputStream(file);
					while (is.available() > 0) {
						fos.write(is.read());
					}
					fos.close();
					is.close();
				}
			}
		}
		jar.close();
	}

	private File getLicensingBasePlugin() {
		return TychoProjectUtils
				.getDependencyArtifacts(project)
				.getArtifact(ArtifactType.TYPE_ECLIPSE_PLUGIN,
						"org.eclipse.licensing.base", null).getLocation();
	}

	private void updateManifest() throws IOException {
		File manifestFile = new File(project.getBasedir(),
				"META-INF/MANIFEST.MF");
		Manifest manifest = new Manifest(new FileInputStream(manifestFile));
		String[] requiredBundles = manifest.getMainAttributes()
				.getValue("Require-Bundle").split(",");
		List<String> requiredList = new ArrayList<String>(
				Arrays.asList(requiredBundles));
		requiredList.remove("org.eclipse.licensing.base");
		requiredList.add("org.apache.commons.codec");
		requiredList.add("org.eclipse.osgi");
		String newRequiredBundles = StringUtils.join(requiredList.toArray(),
				",");
		manifest.getMainAttributes().putValue("Require-Bundle",
				newRequiredBundles);
		manifest.write(new FileOutputStream(manifestFile));
	}

}