/**

 * Back 2 Browser Bytecode Translator

 * Copyright (C) 2012 Jaroslav Tulach <jaroslav.tulach@apidesign.org>

 *

 * 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.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 <jtulach@netbeans.org>

 */

@ServiceProvider(service = Processor.class)

@SupportedAnnotationTypes("org.apidesign.bck2brwsr.vmtest.impl.GenerateZip")

public class GenerateZipProcessor extends AbstractProcessor {

    @Override

    public boolean process(Set<? extends TypeElement> 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 {

        FileObject res = processingEnv.getFiler().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();

        }

    }

}