diff -r add357fd6c5c -r d382dacfd73f rt/vmtest/src/main/java/org/apidesign/bck2brwsr/vmtest/impl/GenerateZipProcessor.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/rt/vmtest/src/main/java/org/apidesign/bck2brwsr/vmtest/impl/GenerateZipProcessor.java Tue Feb 26 16:54:16 2013 +0100 @@ -0,0 +1,101 @@ +/** + * Back 2 Browser Bytecode Translator + * Copyright (C) 2012 Jaroslav Tulach + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, version 2 of the License. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. Look for COPYING file in the top folder. + * If not, see http://opensource.org/licenses/GPL-2.0. + */ +package org.apidesign.bck2brwsr.vmtest.impl; + +import java.io.ByteArrayInputStream; +import java.io.IOException; +import java.io.OutputStream; +import java.util.Set; +import java.util.jar.JarEntry; +import java.util.jar.JarOutputStream; +import java.util.jar.Manifest; +import javax.annotation.processing.AbstractProcessor; +import javax.annotation.processing.Filer; +import javax.annotation.processing.Processor; +import javax.annotation.processing.RoundEnvironment; +import javax.annotation.processing.SupportedAnnotationTypes; +import javax.lang.model.element.Element; +import javax.lang.model.element.ElementKind; +import javax.lang.model.element.PackageElement; +import javax.lang.model.element.TypeElement; +import javax.tools.Diagnostic; +import javax.tools.FileObject; +import javax.tools.StandardLocation; +import org.openide.util.lookup.ServiceProvider; + +/** + * + * @author Jaroslav Tulach + */ +@ServiceProvider(service = Processor.class) +@SupportedAnnotationTypes("org.apidesign.bck2brwsr.vmtest.impl.GenerateZip") +public class GenerateZipProcessor extends AbstractProcessor { + + @Override + public boolean process(Set annotations, RoundEnvironment roundEnv) { + for (Element e : roundEnv.getElementsAnnotatedWith(GenerateZip.class)) { + GenerateZip gz = e.getAnnotation(GenerateZip.class); + if (gz == null) { + continue; + } + PackageElement pe = findPackage(e); + try { + generateJar(pe, gz, e); + } catch (IOException ex) { + processingEnv.getMessager().printMessage( + Diagnostic.Kind.ERROR, + "Can't generate JAR " + gz.name() + ": " + ex.getMessage() + ); + } + } + return true; + } + + private static PackageElement findPackage(Element e) { + while (e.getKind() != ElementKind.PACKAGE) { + e = e.getEnclosingElement(); + } + return (PackageElement)e; + } + + private void generateJar(PackageElement pe, GenerateZip gz, Element e) throws IOException { + final Filer filer = processingEnv.getFiler(); + FileObject res = filer.createResource( + StandardLocation.CLASS_OUTPUT, + pe.getQualifiedName().toString(), + gz.name(), e + ); + OutputStream os = res.openOutputStream(); + JarOutputStream jar; + if (gz.manifest().isEmpty()) { + jar = new JarOutputStream(os); + } else { + Manifest mf = new Manifest(new ByteArrayInputStream(gz.manifest().getBytes("UTF-8"))); + jar = new JarOutputStream(os, mf); + } + String[] arr = gz.contents(); + for (int i = 0; i < arr.length; i += 2) { + JarEntry je = new JarEntry(arr[i]); + jar.putNextEntry(je); + jar.write(arr[i + 1].getBytes("UTF-8")); + jar.closeEntry(); + } + jar.close(); + } + +}