org.eclipse.modisco.doc/src/org/eclipse/modisco/doc/internal/GenerateApiJavadocTask.java

/*******************************************************************************
 * Copyright (c) 2011, 2019 Mia-Software and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v2.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v20.html
 *
 * Contributors:
 *    Nicolas Bros (Mia-Software) - initial API and implementation
 *******************************************************************************/
package org.eclipse.modisco.doc.internal;

import java.io.File;
import java.io.FileInputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Map.Entry;
import java.util.jar.Attributes;
import java.util.jar.Manifest;
import java.util.regex.Pattern;

import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.Project;
import org.apache.tools.ant.taskdefs.Javadoc;
import org.apache.tools.ant.types.FileSet;

public class GenerateApiJavadocTask extends Javadoc {

	private String sourceTreePath = null;
	private Pattern pattern = Pattern.compile(".*"); //$NON-NLS-1$

	public void setSourceTreePath(final String path) {
		this.sourceTreePath = path;
	}

	public void setPattern(final String pattern) {
		this.pattern = Pattern.compile(pattern);
	}

	@Override
	public void execute() throws BuildException {
		if (this.sourceTreePath == null) {
			throw new BuildException("sourceTreePath must be set"); //$NON-NLS-1$
		}

		getProject().log("sourceTreePath = " + this.sourceTreePath, Project.MSG_DEBUG); //$NON-NLS-1$

		List<File> apiPackages = new ArrayList<File>();

		File root = new File(this.sourceTreePath);
		if (!root.isAbsolute()) {
			root = new File(getProject().getBaseDir(), this.sourceTreePath);
		}
		if (!root.exists()) {
			throw new BuildException("root path doesn't exist: " + root.getPath()); //$NON-NLS-1$
		}
		try {
			for (String childPath : root.list()) {
				boolean matches = this.pattern.matcher(childPath).matches();
				if (matches) {
					getProject().log("pattern matches: " + childPath, Project.MSG_DEBUG); //$NON-NLS-1$
				} else {
					getProject().log("pattern rejects: " + childPath, Project.MSG_DEBUG); //$NON-NLS-1$
				}
				if (matches) {
					File file = new File(root, childPath);
					if (file.isDirectory()) {
						File manifestFile = new File(file, "META-INF/MANIFEST.MF"); //$NON-NLS-1$
						if (manifestFile.exists()) {
							FileInputStream fileInputStream;
							fileInputStream = new FileInputStream(manifestFile);
							Manifest manifest = new Manifest(fileInputStream);

							Attributes mainAttributes = manifest.getMainAttributes();
							for (Entry<Object, Object> entry : mainAttributes.entrySet()) {
								String name = entry.getKey().toString();
								if ("Export-Package".equals(name)) { //$NON-NLS-1$
									String value = (String) entry.getValue();
									String[] packageEntries = value.split(","); //$NON-NLS-1$
									for (String packageEntry : packageEntries) {
										String[] parts = packageEntry.split(";"); //$NON-NLS-1$
										String packageName = parts[0];
										boolean internal = false;
										for (int i = 1; i < parts.length; i++) {
											if (parts[i].contains("x-internal") || parts[i].contains("x-friends")) { //$NON-NLS-1$ //$NON-NLS-2$
												internal = true;
											}
										}
										if (!internal) {
											File packageDir = new File(file.getPath(), "src/" + packageName.replace('.', '/')); //$NON-NLS-1$
											if (packageDir.exists()) {
												apiPackages.add(packageDir);
											}
										}
									}
								}
							}
						}
					}
				}
			}
		} catch (Exception e) {
			throw new BuildException("Error reading manifests", e); //$NON-NLS-1$
		}

		for (File apiPackage : apiPackages) {
			String[] children = apiPackage.list();
			boolean packageImpl = false;
			boolean factoryImpl = false;
			for (String child : children) {
				if (child.endsWith("PackageImpl.java")) { //$NON-NLS-1$
					packageImpl = true;
				}
				if (child.endsWith("FactoryImpl.java")) { //$NON-NLS-1$
					factoryImpl = true;
				}
			}
			if (packageImpl && factoryImpl) {
				getProject().log("excluding metamodel implementation package: " + apiPackage.getPath(), Project.MSG_INFO); //$NON-NLS-1$
				continue;
			}

			if (apiPackage.exists()) {
				System.out.println(apiPackage);

				FileSet fileSet = new FileSet();
				fileSet.setProject(getProject());
				fileSet.setDir(apiPackage);
				fileSet.setIncludes("*.java"); //$NON-NLS-1$
				System.out.println(fileSet);
				addFileset(fileSet);
			} else {
				getProject().log("package doesn't exist: " + apiPackage, Project.MSG_ERR); //$NON-NLS-1$
			}
		}

		super.execute();
	}
}