diff options
Diffstat (limited to 'src/core')
| -rw-r--r-- | src/core/lombok/core/PublicApiCreatorApp.java | 188 |
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(); + } + } +} |