/**

 * 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.aot;

import java.io.BufferedReader;

import java.io.ByteArrayInputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

import java.net.URL;

import java.util.ArrayList;

import java.util.Enumeration;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import java.util.Set;

import java.util.jar.JarEntry;

import java.util.jar.JarFile;

import java.util.logging.Level;

import java.util.logging.Logger;

import java.util.zip.ZipEntry;

import org.apidesign.vm4brwsr.Bck2Brwsr;

/** Utilities to process JAR files and set a compiler

 * up.

 *

 * @since 0.9

 * @author Jaroslav Tulach

 */

public final class Bck2BrwsrJars {

    private static final Logger LOG = Logger.getLogger(Bck2BrwsrJars.class.getName());

    private Bck2BrwsrJars() {

    }

    /** Creates new compiler pre-configured from the content of 

     * provided JAR file. The compiler will compile all classes.

     * The system understands OSGi manifest entries and will export

     * all packages that are exported in the JAR file. The system

     * also recognizes META-INF/services and makes sure the class names

     * are not mangled.

     * 

     * @param c the compiler to {@link Bck2Brwsr#addClasses(java.lang.String...) add classes},

     *    {@link Bck2Brwsr#addResources(java.lang.String...) add resources} and

     *    {@link Bck2Brwsr#addExported(java.lang.String...) exported objects} to.

     *    Can be <code>null</code> - in such case an 

     *    {@link Bck2Brwsr#newCompiler() empty compiler} is constructed.

     * @param jar the file to process

     * @return newly configured compiler

     * @throws IOException if something goes wrong

     */

    public static Bck2Brwsr configureFrom(Bck2Brwsr c, File jar) throws IOException {

        if (jar.isDirectory()) {

            return configureDir(c, jar);

        }

        final JarFile jf = new JarFile(jar);

        final List<String> classes = new ArrayList<>();

        List<String> resources = new ArrayList<>();

        Set<String> exported = new HashSet<>();

        class JarRes extends EmulationResources implements Bck2Brwsr.Resources {

            JarRes() {

                super(classes);

            }

            @Override

            public InputStream get(String resource) throws IOException {

                InputStream is = getConverted(resource);

                if (is != null) {

                    return is;

                }

                is = jf.getInputStream(new ZipEntry(resource));

                return is == null ? super.get(resource) : is;

            }

        }

        JarRes jarRes = new JarRes();

        listJAR(jf, jarRes, resources, exported);

        String cp = jf.getManifest().getMainAttributes().getValue("Class-Path"); // NOI18N

        String[] classpath = cp == null ? new String[0] : cp.split(" ");

        if (c == null) {

            c = Bck2Brwsr.newCompiler();

        }

        return c

            .library(classpath)

            .addClasses(classes.toArray(new String[classes.size()]))

            .addExported(exported.toArray(new String[exported.size()]))

            .addResources(resources.toArray(new String[resources.size()]))

            .resources(jarRes);

    }

    private static void listJAR(

        JarFile j, EmulationResources classes,

        List<String> resources, Set<String> keep

    ) throws IOException {

        Enumeration<JarEntry> en = j.entries();

        while (en.hasMoreElements()) {

            JarEntry e = en.nextElement();

            final String n = e.getName();

            if (n.endsWith("/")) {

                continue;

            }

            if (n.startsWith("META-INF/maven/")) {

                continue;

            }

            int last = n.lastIndexOf('/');

            String pkg = n.substring(0, last + 1);

            if (pkg.startsWith("java/")) {

                keep.add(pkg);

            }

            if (n.endsWith(".class")) {

                classes.addClassResource(n);

            } else {

                resources.add(n);

                if (n.startsWith("META-INF/services/") && keep != null) {

                    BufferedReader r = new BufferedReader(new InputStreamReader(j.getInputStream(e)));

                    for (;;) {

                        String l = r.readLine();

                        if (l == null) {

                            break;

                        }

                        if (l.startsWith("#")) {

                            continue;

                        }

                        keep.add(l.replace('.', '/'));

                    }

                }

            }

        }

        String exp = j.getManifest().getMainAttributes().getValue("Export-Package");

        if (exp != null && keep != null) {

            for (String def : exp.split(",")) {

                for (String sep : def.split(";")) {

                    keep.add(sep.replace('.', '/') + "/");

                    break;

                }

            }

        }

    }

    static byte[] readFrom(InputStream is) throws IOException {

        int expLen = is.available();

        if (expLen < 1) {

            expLen = 1;

        }

        byte[] arr = new byte[expLen];

        int pos = 0;

        for (;;) {

            int read = is.read(arr, pos, arr.length - pos);

            if (read == -1) {

                break;

            }

            pos += read;

            if (pos == arr.length) {

                byte[] tmp = new byte[arr.length * 2];

                System.arraycopy(arr, 0, tmp, 0, arr.length);

                arr = tmp;

            }

        }

        if (pos != arr.length) {

            byte[] tmp = new byte[pos];

            System.arraycopy(arr, 0, tmp, 0, pos);

            arr = tmp;

        }

        return arr;

    }

    static class EmulationResources implements Bck2Brwsr.Resources {

        private final List<String> classes;

        private final Map<String,byte[]> converted = new HashMap<>();

        private final BytecodeProcessor proc;

        protected EmulationResources(List<String> classes) {

            this.classes = classes;

            BytecodeProcessor p;

            try {

                Class<?> bpClass = Class.forName("org.apidesign.bck2brwsr.aot.RetroLambda");

                p = (BytecodeProcessor) bpClass.newInstance();

            } catch (Throwable t) {

                p = null;

            }

            this.proc = p;

        }

        protected final InputStream getConverted(String name) throws IOException {

            byte[] arr = converted.get(name);

            if (arr != null) {

                return new ByteArrayInputStream(arr);

            }

            return null;

        }

        @Override

        public InputStream get(String name) throws IOException {

            InputStream is = getConverted(name);

            if (is != null) {

                return is;

            }

            Enumeration<URL> en = Bck2BrwsrJars.class.getClassLoader().getResources(name);

            URL u = null;

            while (en.hasMoreElements()) {

                u = en.nextElement();

            }

            if (u == null) {

                LOG.log(Level.WARNING, "Cannot find {0}", name);

                return null;

            }

            if (u.toExternalForm().contains("/rt.jar!")) {

                LOG.log(Level.WARNING, "{0}No bootdelegation for ", name);

                return null;

            }

            return u.openStream();

        }

        private void addClassResource(String n) throws IOException {

            if (proc != null) {

                try (InputStream is = this.get(n)) {

                    Map<String, byte[]> conv = proc.process(n, readFrom(is), this);

                    if (conv != null) {

                        boolean found = false;

                        for (Map.Entry<String, byte[]> entrySet : conv.entrySet()) {

                            String res = entrySet.getKey();

                            byte[] bytes = entrySet.getValue();

                            if (res.equals(n)) {

                                found = true;

                            }

                            assert res.endsWith(".class") : "Wrong resource: " + res;

                            converted.put(res, bytes);

                            classes.add(res.substring(0, res.length() - 6));

                        }

                        if (!found) {

                            throw new IOException("Cannot find " + n + " among " + conv);

                        }

                        return;

                    }

                }

            }

            classes.add(n.substring(0, n.length() - 6));

        }

    }

    private static Bck2Brwsr configureDir(Bck2Brwsr c, final File dir) throws IOException {

        List<String> arr = new ArrayList<>();

        List<String> classes = new ArrayList<>();

        class DirRes extends EmulationResources {

            public DirRes(List<String> classes) {

                super(classes);

            }

            @Override

            public InputStream get(String name) throws IOException {

                InputStream is = super.get(name);

                if (is != null) {

                    return is;

                }

                File r = new File(dir, name.replace('/', File.separatorChar));

                if (r.exists()) {

                    return new FileInputStream(r);

                }

                return null;

            }

        }

        DirRes dirRes = new DirRes(classes);

        listDir(dir, null, dirRes, arr);

        if (c == null) {

            c = Bck2Brwsr.newCompiler();

        }

        return c

        .addRootClasses(classes.toArray(new String[0]))

        .addResources(arr.toArray(new String[0]))

        .library()

        //.obfuscation(ObfuscationLevel.FULL)

        .resources(dirRes);

    }

    private static void listDir(

        File f, String pref, EmulationResources res, List<String> resources

    ) throws IOException {

        File[] arr = f.listFiles();

        if (arr == null) {

            if (f.getName().endsWith(".class")) {

                res.addClassResource(pref + f.getName());

            } else {

                resources.add(pref + f.getName());

            }

        } else {

            for (File ch : arr) {

                listDir(ch, pref == null ? "" : pref + f.getName() + "/", res, resources);

            }

        }

    }

}