/*

 * Copyright (c) 2016, 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.tools.jlink.internal.packager;

import jdk.tools.jlink.Jlink;

import jdk.tools.jlink.plugin.Plugin;

import java.io.File;

import java.io.IOException;

import java.nio.file.Path;

import java.util.ArrayList;

import java.util.Collections;

import java.util.List;

import java.util.Map;

import java.util.Set;

/**

 * AppRuntimeImageBuilder is a private API used only by the Java Packager to generate

 * a Java runtime image using jlink. AppRuntimeImageBuilder encapsulates the

 * arguments that jlink requires to generate this image. To create the image call the

 * build() method.

 */

public final class AppRuntimeImageBuilder {

    private Path outputDir = null;

    private List<Path> modulePath = null;

    private Set<String> addModules = null;

    private Set<String> limitModules = null;

    private String excludeFileList = null;

    private Map<String, String> userArguments = null;

    private Boolean stripNativeCommands = null;

    public AppRuntimeImageBuilder() {}

    public void setOutputDir(Path value) {

        outputDir = value;

    }

    public void setModulePath(List<Path> value) {

        modulePath = value;

    }

    public void setAddModules(Set<String> value) {

        addModules = value;

    }

    public void setLimitModules(Set<String> value) {

        limitModules = value;

    }

    public void setExcludeFileList(String value) {

        excludeFileList = value;

    }

    public void setStripNativeCommands(boolean value) {

        stripNativeCommands = value;

    }

    public void setUserArguments(Map<String, String> value) {

        userArguments = value;

    }

    public void build() throws IOException {

        // jlink main arguments

        Jlink.JlinkConfiguration jlinkConfig = new Jlink.JlinkConfiguration(

            new File("").toPath(), // Unused

            modulePath, addModules, limitModules);

        // plugin configuration

        List<Plugin> plugins = new ArrayList<Plugin>();

        if (stripNativeCommands) {

            plugins.add(Jlink.newPlugin(

                        "strip-native-commands",

                        Collections.singletonMap("strip-native-commands", "on"),

                        null));

        }

        if (excludeFileList != null && !excludeFileList.isEmpty()) {

            plugins.add(Jlink.newPlugin(

                        "exclude-files",

                        Collections.singletonMap("exclude-files", excludeFileList),

                        null));

        }

        // add user supplied jlink arguments

        for (Map.Entry<String, String> entry : userArguments.entrySet()) {

            String key = entry.getKey();

            String value = entry.getValue();

            plugins.add(Jlink.newPlugin(key,

                                        Collections.singletonMap(key, value),

                                        null));

        }

        plugins.add(Jlink.newPlugin("installed-modules", Collections.emptyMap(), null));

        // build the image

        Jlink.PluginsConfiguration pluginConfig = new Jlink.PluginsConfiguration(

            plugins, new DefaultImageBuilder(outputDir), null);

        Jlink jlink = new Jlink();

        jlink.build(jlinkConfig, pluginConfig);

    }

}