/*

 * Copyright (c) 2015, 2018, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code 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

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package jdk.packager.internal;

import java.io.File;

import java.io.IOException;

import java.io.StringReader;

import java.nio.file.Files;

import java.nio.file.Path;

import java.text.MessageFormat;

import java.util.Collection;

import java.util.Collections;

import java.util.EnumSet;

import java.util.HashMap;

import java.util.HashSet;

import java.util.Iterator;

import java.util.LinkedHashMap;

import java.util.LinkedHashSet;

import java.util.List;

import java.util.Map;

import java.util.Properties;

import java.util.ResourceBundle;

import java.util.Set;

import jdk.packager.internal.builders.AbstractAppImageBuilder;

import jdk.tools.jlink.internal.packager.AppRuntimeImageBuilder;

public final class JLinkBundlerHelper {

    private static final String JRE_MODULES_FILENAME =

            "jdk/packager/internal/resources/jre.list";

    private static final String SERVER_JRE_MODULES_FILENAME =

            "jdk/packager/internal/resources/server.jre.list";

    private JLinkBundlerHelper() {}

    @SuppressWarnings("unchecked")

    public static final BundlerParamInfo<Boolean> DETECT_MODULES =

            new StandardBundlerParam<>(

                    I18N.getString("param.detect-modules.name"),

                    I18N.getString("param.detect-modules.description"),

                    "detect-modules",

                    Boolean.class,

                    p -> Boolean.FALSE,

                    (s, p) -> Boolean.valueOf(s));

    @SuppressWarnings("unchecked")

    public static final BundlerParamInfo<Map<String, String>> JLINK_OPTIONS =

            new StandardBundlerParam<>(

                    I18N.getString("param.jlink-options.name"),

                    I18N.getString("param.jlink-options.description"),

                    "jlinkOptions",

                    (Class<Map<String, String>>) (Object) Map.class,

                    p -> Collections.emptyMap(),

                    (s, p) -> {

                        try {

                            Properties props = new Properties();

                            props.load(new StringReader(s));

                            return new LinkedHashMap<>((Map)props);

                        } catch (IOException e) {

                            return new LinkedHashMap<>();

                        }

                    });

    @SuppressWarnings("unchecked")

    public static final BundlerParamInfo<String> JLINK_BUILDER =

            new StandardBundlerParam<>(

                    I18N.getString("param.jlink-builder.name"),

                    I18N.getString("param.jlink-builder.description"),

                    "jlink.builder",

                    String.class,

                    null,

                    (s, p) -> s);

    @SuppressWarnings("unchecked")

    public static final BundlerParamInfo<Integer> DEBUG =

            new StandardBundlerParam<>(

                    "",

                    "",

                    "-J-Xdebug",

                    Integer.class,

                    p -> null,

                    (s, p) -> {

                        return Integer.valueOf(s);

                    });

    public static String listOfPathToString(List<Path> value) {

        String result = "";

        for (Path path : value) {

            if (result.length() > 0) {

                result += File.pathSeparator;

            }

            result += path.toString();

        }

        return result;

    }

    public static String setOfStringToString(Set<String> value) {

        String result = "";

        for (String element : value) {

            if (result.length() > 0) {

                result += ",";

            }

            result += element;

        }

        return result;

    }

    public static File getMainJar(Map<String, ? super Object> params) {

        File result = null;

        if (fileset != null) {

            String filename = fileset.getIncludedFiles().iterator().next();

            if (result == null || !result.exists()) {

                if (srcdir != null) {

                    result = new File(srcdir + File.separator + filename);

                }

            }

        }

        return result;

    }

    public static String getMainClass(Map<String, ? super Object> params) {

        String result = "";

        String mainModule = StandardBundlerParam.MODULE.fetchFrom(params);

        if (mainModule != null)  {

            int index = mainModule.indexOf("/");

            if (index > 0) {

                result = mainModule.substring(index + 1);

            }

        } else {

            result = StandardBundlerParam.MAIN_CLASS.fetchFrom(params);

        }

        return result;

    }

    public static String getMainModule(Map<String, ? super Object> params) {

        String result = "";

        String mainModule = StandardBundlerParam.MODULE.fetchFrom(params);

        if (mainModule != null) {

            int index = mainModule.indexOf("/");

            if (index > 0) {

                result = mainModule.substring(0, index);

            }

            else {

                result = mainModule;

            }

        }

        return result;

    }

    public static String getJDKVersion(Map<String, ? super Object> params) {

        String result = "";

        Path javaBasePath = findPathOfModule(modulePath, "java.base.jmod");

        Set<String> addModules = getRedistributableModules(modulePath,

                StandardBundlerParam.ADD_MODULES.fetchFrom(params),

                limitModules, JRE_MODULES_FILENAME);

        if (javaBasePath != null && javaBasePath.toFile().exists()) {

        }

        return result;

    }

    public static Path getJDKHome(Map<String, ? super Object> params) {

        Path result = null;

        Path javaBasePath = findPathOfModule(modulePath, "java.base.jmod");

        if (javaBasePath != null && javaBasePath.toFile().exists()) {

            result = javaBasePath.getParent();

            // On a developer build the JDK Home isn't where we expect it

            // relative to the jmods directory. Do some extra

            // processing to find it.

            if (result != null) {

                boolean found = false;

                Path bin = result.resolve("bin");

                if (Files.exists(bin)) {

                    Path javaExe = bin.resolve("java" + exe);

                    if (Files.exists(javaExe)) {

                        found = true;

                    }

                }

                if (!found) {

                    result = result.resolve(".." + File.separator + "jdk");

                }

            }

        }

        return result;

    }

    private static Set<String> getRedistributableModules(List<Path> modulePath,

            Set<String> addModules, Set<String> limitModules, String filename) {

        return removeInvalidModules(modulePath, moduleHelper.modules());

    }

    public static void execute(Map<String, ? super Object> params,

        Path outputDir = imageBuilder.getRoot();

        String excludeFileList = imageBuilder.getExcludeFileList();

        File mainJar = getMainJar(params);

        Module.ModuleType mainJarType = Module.ModuleType.Unknown;

        if (mainJar != null) {

            mainJarType = new Module(mainJar).getModuleType();

            // user specified only main class, all jars will be on the classpath

            mainJarType = Module.ModuleType.UnnamedJar;

        }

        // Modules

        // The default for an unnamed jar is ALL_DEFAULT with the

        // non-redistributable modules removed.

        if (mainJarType == Module.ModuleType.UnnamedJar) {

            addModules.add(ModuleHelper.ALL_RUNTIME);

            String mainModule = getMainModule(params);

            addModules.add(mainModule);

            // Error if any of the srcfiles are modular jars.

            if (!modularJars.isEmpty()) {

                throw new Exception(MessageFormat.format(I18N.getString(

            }

        }

        Set<String> redistModules = getRedistributableModules(

                modulePath, addModules, limitModules, JRE_MODULES_FILENAME);

        addModules.addAll(redistModules);

        if (imageBuilder.getPlatformSpecificModulesFile() != null) {

                modulePath, imageBuilder.getPlatformSpecificModulesFile());

            addModules.addAll(platformModules);    

        }

        if (StandardBundlerParam.ECHO_MODE.fetchFrom(params)) {

            Log.info("\nECHO-MODE: Running jlink [ ");

            Log.info("--output = " + outputDir.toString());

            Log.info("--module-path = " + modulePath.toString());

            Log.info("--add-modules = " + addModules.toString());

            Log.info("--limit-modules = " + limitModules.toString());

            Log.info("--exclude-files = " + excludeFileList);

            Log.info("--strip-native-commands = " + stripNativeCommands);

            Log.info(" ]\n" + outputDir.toString());

        }

        AppRuntimeImageBuilder appRuntimeBuilder = new AppRuntimeImageBuilder();

        appRuntimeBuilder.setOutputDir(outputDir);

        appRuntimeBuilder.setModulePath(modulePath);

        appRuntimeBuilder.setAddModules(addModules);

        appRuntimeBuilder.setLimitModules(limitModules);

        appRuntimeBuilder.setExcludeFileList(excludeFileList);

        appRuntimeBuilder.setStripNativeCommands(stripNativeCommands);

        appRuntimeBuilder.setUserArguments(new HashMap<String,String>());

        appRuntimeBuilder.build();

        imageBuilder.prepareApplicationFiles();

    }

    public static void generateServerJre(Map<String, ? super Object> params,

        Path outputDir = imageBuilder.getRoot();

        addModules.add(ModuleHelper.ALL_RUNTIME);

        addModules.addAll(redistModules);

        if (imageBuilder.getPlatformSpecificModulesFile() != null) {

            addModules.addAll(platformModules);    

        }

        if (StandardBundlerParam.ECHO_MODE.fetchFrom(params)) {

            Log.info("\nECHO-MODE: Running jlink [ ");

            Log.info("--output = " + outputDir.toString());

            Log.info("--module-path = " + modulePath.toString());

            Log.info("--add-modules = " + addModules.toString());

            Log.info("--limit-modules = " + limitModules.toString());

            Log.info("--strip-native-commands = " + stripNativeCommands);

            Log.info(" ]\n" + outputDir.toString());

        }

        AppRuntimeImageBuilder appRuntimeBuilder = new AppRuntimeImageBuilder();

        appRuntimeBuilder.setOutputDir(outputDir);

        appRuntimeBuilder.setModulePath(modulePath);

        appRuntimeBuilder.setAddModules(addModules);

        appRuntimeBuilder.setLimitModules(limitModules);

        appRuntimeBuilder.setStripNativeCommands(stripNativeCommands);

        appRuntimeBuilder.setExcludeFileList("");

        appRuntimeBuilder.setUserArguments(new HashMap<String,String>());

        appRuntimeBuilder.build();

        imageBuilder.prepareServerJreFiles();

    }

    // Returns the path to the JDK modules in the user defined module path.

        Path result = null;

        for (Path path : modulePath) {

            Path moduleNamePath = path.resolve(moduleName);

            if (Files.exists(moduleNamePath)) {

                result = path;

                break;

            }

        }

        return result;

    }

    private static Set<String> getResourceFileJarList(

            Map<String, ? super Object> params, Module.JarType Query) {

        Set<String> files = new LinkedHashSet();

        String srcdir = StandardBundlerParam.SOURCE_DIR.fetchFrom(params);

            for (String resource : appResources.getIncludedFiles()) {

                if (resource.endsWith(".jar")) {

                    String filename = srcdir + File.separator + resource;

                    switch (Query) {

                        case All: {

                            files.add(filename);

                            break;

                        }

                        case ModularJar: {

                            Module module = new Module(new File(filename));

                                files.add(filename);

                            }

                            break;

                        }

                        case UnnamedJar: {

                            Module module = new Module(new File(filename));

                                files.add(filename);

                            }

                            break;

                        }

                    }

                }

            }

        }

        return files;

    }

        Set<String> result = new LinkedHashSet();

        ModuleManager mm = new ModuleManager(modulePath);

        HashMap<String, Module> validModules = new HashMap<>();

        for (Module module : lmodules) {

            validModules.put(module.getModuleName(), module);

        }

        for (String name : modules) {

            if (validModules.containsKey(name)) {

                result.add(name);

            }

            else {

            }

        }

        return result;

    }

    private static class ModuleHelper {

        // The token for "all modules on the module path".

        private static final String ALL_MODULE_PATH = "ALL-MODULE-PATH";

        // The token for "all redistributable runtime modules".

        public static final String ALL_RUNTIME = "ALL-RUNTIME";

        private final Set<String> modules = new HashSet<>();

        private enum Macros {None, AllModulePath, AllRuntime}

        public ModuleHelper(List<Path> paths, Set<String> roots,

                Set<String> limitMods, String filename) {

            Macros macro = Macros.None;

                String module = iterator.next();

                switch (module) {

                    case ALL_MODULE_PATH:

                        iterator.remove();

                        macro = Macros.AllModulePath;

                        break;

                    case ALL_RUNTIME:

                        iterator.remove();

                        macro = Macros.AllRuntime;

                        break;

                    default:

                        this.modules.add(module);

                }

            }

            switch (macro) {

                case AllModulePath:

                    this.modules.addAll(getModuleNamesFromPath(paths));

                    break;

                case AllRuntime: