1 files changed, 188 insertions, 0 deletions
diff --git a/src/core/lombok/core/PublicApiCreatorApp.java b/src/core/lombok/core/PublicApiCreatorApp.java
new file mode 100644
index 00000000..24bcf83d
--- /dev/null
+++ b/src/core/lombok/core/PublicApiCreatorApp.java
@@ -0,0 +1,188 @@
+/*
+ * Copyright (C) 2012 The Project Lombok Authors.
+ * 
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ * 
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ * 
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ * THE SOFTWARE.
+ */
+package lombok.core;
+
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.util.ArrayList;
+import java.util.Enumeration;
+import java.util.List;
+import java.util.jar.JarEntry;
+import java.util.jar.JarFile;
+import java.util.jar.JarOutputStream;
+import java.util.zip.ZipEntry;
+
+import lombok.Lombok;
+import lombok.installer.IdeFinder;
+import lombok.patcher.inject.LiveInjector;
+
+import org.mangosdk.spi.ProviderFor;
+
+@ProviderFor(LombokApp.class)
+public class PublicApiCreatorApp extends LombokApp {
+	@Override public String getAppName() {
+		return "publicApi";
+	}
+	
+	@Override public String getAppDescription() {
+		return "Creates a small lombok-api.jar with the annotations and other public API\n" +
+				"classes of all lombok features. This is primarily useful to include in your\n" +
+				"android projects.";
+	}
+	
+	@Override public int runApp(List<String> rawArgs) throws Exception {
+		String loc = ".";
+		switch (rawArgs.size()) {
+		case 0: break;
+		case 1: loc = rawArgs.get(0); break;
+		default:
+			System.err.println("Supply 1 path to specify the directory where lombok-api.jar will be created. No path means the current directory is used.");
+			return 1;
+		}
+		
+		File out = new File(loc, "lombok-api.jar");
+		int errCode = 0;
+		try {
+			errCode = writeApiJar(out);
+		} catch (Exception e) {
+			System.err.println("ERROR: Creating " + canonical(out) + " failed: ");
+			e.printStackTrace();
+			return 1;
+		}
+		
+		return errCode;
+	}
+	
+	/**
+	 * Returns a File object pointing to our own jar file. Will obviously fail if the installer was started via
+	 * a jar that wasn't accessed via the file-system, or if its started via e.g. unpacking the jar.
+	 */
+	private static File findOurJar() {
+		return new File(LiveInjector.findPathJar(IdeFinder.class));
+	}
+	
+	private int writeApiJar(File outFile) throws Exception {
+		File selfRaw = findOurJar();
+		if (selfRaw == null) {
+			System.err.println("The publicApi option only works if lombok is a jar.");
+			return 2;
+		}
+		
+		List<String> toCopy = new ArrayList<String>();
+		JarFile self = new JarFile(selfRaw);
+		try {
+			Enumeration<JarEntry> entries = self.entries();
+			
+			while (entries.hasMoreElements()) {
+				JarEntry entry = entries.nextElement();
+				String name = entry.getName();
+				if (!name.startsWith("lombok/")) continue;
+				if (name.endsWith("/package-info.class")) continue;
+				if (!name.endsWith(".class")) continue;
+				
+				String subName = name.substring(7, name.length() - 6);
+				int firstSlash = subName.indexOf('/');
+				if (firstSlash == -1) {
+					// direct member of the lombok package.
+					toCopy.add(name);
+					continue;
+				}
+				String topPkg = subName.substring(0, firstSlash);
+				if ("extern".equals(topPkg) || "experimental".equals(topPkg)) {
+					toCopy.add(name);
+				}
+			}
+		} finally {
+			self.close();
+		}
+		
+		if (toCopy.isEmpty()) {
+			System.out.println("Not generating lombok-api.jar: No lombok api classes required!");
+			return 1;
+		}
+		
+		OutputStream out = new FileOutputStream(outFile);
+		boolean success = false;
+		try {
+			JarOutputStream jar = new JarOutputStream(out);
+			for (String resourceName : toCopy) {
+				InputStream in = Lombok.class.getResourceAsStream("/" + resourceName);
+				try {
+					if (in == null) {
+						throw new Fail(String.format("api class %s cannot be found", resourceName));
+					}
+					writeIntoJar(jar, resourceName, in);
+				} finally {
+					if (in != null) in.close();
+				}
+			}
+			jar.close();
+			out.close();
+			
+			System.out.println("Successfully created: " + canonical(outFile));
+			
+			return 0;
+		} catch (Throwable t) {
+			try { out.close();} catch (Throwable ignore) {}
+			if (!success) outFile.delete();
+			if (t instanceof Fail) {
+				System.err.println(t.getMessage());
+				return 1;
+			} else if (t instanceof Exception) {
+				throw (Exception)t;
+			} else if (t instanceof Error) {
+				throw (Error)t;
+			} else {
+				throw new Exception(t);
+			}
+		}
+	}
+	
+	private void writeIntoJar(JarOutputStream jar, String resourceName, InputStream in) throws IOException {
+		jar.putNextEntry(new ZipEntry(resourceName));
+		byte[] b = new byte[65536];
+		while (true) {
+			int r = in.read(b);
+			if (r == -1) break;
+			jar.write(b, 0, r);
+		}
+		jar.closeEntry();
+		in.close();
+	}
+	
+	private static class Fail extends Exception {
+		Fail(String message) {
+			super(message);
+		}
+	}
+	
+	private static String canonical(File out) {
+		try {
+			return out.getCanonicalPath();
+		} catch (Exception e) {
+			return out.getAbsolutePath();
+		}
+	}
+}