/*

 * Copyright (c) 2015, 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.

 *

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

 */

import java.nio.file.Path;

import java.nio.file.Paths;

import java.util.LinkedHashMap;

import java.util.List;

import java.util.Map;

import java.util.Arrays;

import java.io.IOException;

import java.lang.module.ModuleDescriptor;

import java.util.ArrayList;

import jdk.testlibrary.ProcessTools;

import jdk.testlibrary.OutputAnalyzer;

import org.testng.annotations.BeforeTest;

/**

 * @test

 * @bug 8078813

 * @library /lib/testlibrary

 * @library /java/security/modules

 * @build CompilerUtils JarUtils

 * @summary Test custom JAAS module with all possible modular option. The test

 *          includes different combination of JAAS client/login modules

 *          interaction with or without service description.

 * @run testng JaasModularClientTest

 */

public class JaasModularClientTest extends ModularTest {

    private static final Path S_SRC = SRC.resolve("TestLoginModule.java");

    private static final String S_PKG = "login";

    private static final String S_JAR_NAME = S_PKG + JAR_EXTN;

    private static final String S_DESCR_JAR_NAME = S_PKG + DESCRIPTOR

            + JAR_EXTN;

    private static final String MS_JAR_NAME = MODULAR + S_PKG + JAR_EXTN;

    private static final String MS_DESCR_JAR_NAME = MODULAR + S_PKG + DESCRIPTOR

            + JAR_EXTN;

    private static final Path C_SRC = SRC.resolve("JaasClient.java");

    private static final String C_PKG = "client";

    private static final String C_JAR_NAME = C_PKG + JAR_EXTN;

    private static final String MC_JAR_NAME = MODULAR + C_PKG + JAR_EXTN;

    private static final Path BUILD_DIR = Paths.get(".").resolve("build");

    private static final Path COMPILE_DIR = BUILD_DIR.resolve("bin");

    private static final Path S_BUILD_DIR = COMPILE_DIR.resolve(S_PKG);

    private static final Path S_WITH_META_DESCR_BUILD_DIR = COMPILE_DIR.resolve(

            S_PKG + DESCRIPTOR);

    private static final Path M_BASE_PATH = BUILD_DIR.resolve("mbase");

    private static final Path ARTIFACTS_DIR = BUILD_DIR.resolve("artifacts");

    private static final Path S_ARTIFACTS_DIR = ARTIFACTS_DIR.resolve(S_PKG);

    private static final Path S_JAR = S_ARTIFACTS_DIR.resolve(S_JAR_NAME);

    private static final Path S_WITH_DESCRIPTOR_JAR = S_ARTIFACTS_DIR.resolve(

            S_DESCR_JAR_NAME);

    private static final Path MS_JAR = S_ARTIFACTS_DIR.resolve(MS_JAR_NAME);

    private static final Path MS_WITH_DESCR_JAR = S_ARTIFACTS_DIR.resolve(

            MS_DESCR_JAR_NAME);

    private static final Path C_ARTIFACTS_DIR = ARTIFACTS_DIR.resolve(C_PKG);

    private static final Path C_JAR = C_ARTIFACTS_DIR.resolve(C_JAR_NAME);

    private static final Path MC_JAR = C_ARTIFACTS_DIR.resolve(MC_JAR_NAME);

    private static final String MAIN = C_PKG + ".JaasClient";

    private static final String S_INTERFACE

            = "javax.security.auth.spi.LoginModule";

    private static final String S_IMPL = S_PKG + ".TestLoginModule";

    private static final List<String> M_REQUIRED = Arrays.asList("java.base",

            "jdk.security.auth");

    private static final Path META_DESCR_PATH = Paths.get("META-INF")

            .resolve("services").resolve(S_INTERFACE);

    private static final Path S_META_DESCR_FPATH = S_WITH_META_DESCR_BUILD_DIR

            .resolve(META_DESCR_PATH);

    private static final boolean WITH_S_DESCR = true;

    private static final boolean WITHOUT_S_DESCR = false;

    private static final String NO_FAILURE = null;

    /**

     * Generates Test specific input parameters.

     */

    @Override

    public Object[][] getTestInput() {

        List<List<Object>> params = new ArrayList<>();

        String[] args = new String[]{};

        params.add(Arrays.asList(MODULE_TYPE.EXPLICIT, MODULE_TYPE.EXPLICIT,

                WITH_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.EXPLICIT, MODULE_TYPE.EXPLICIT,

                WITHOUT_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.EXPLICIT, MODULE_TYPE.AUTO,

                WITH_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.EXPLICIT, MODULE_TYPE.AUTO,

        params.add(Arrays.asList(MODULE_TYPE.EXPLICIT, MODULE_TYPE.UNNAMED,

                WITH_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.EXPLICIT, MODULE_TYPE.UNNAMED,

                WITHOUT_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.AUTO, MODULE_TYPE.EXPLICIT,

                WITH_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.AUTO, MODULE_TYPE.EXPLICIT,

                WITHOUT_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.AUTO, MODULE_TYPE.AUTO,

                WITH_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.AUTO, MODULE_TYPE.AUTO,

        params.add(Arrays.asList(MODULE_TYPE.AUTO, MODULE_TYPE.UNNAMED,

                WITH_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.AUTO, MODULE_TYPE.UNNAMED,

                WITHOUT_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.UNNAMED, MODULE_TYPE.EXPLICIT,

                WITH_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.UNNAMED, MODULE_TYPE.EXPLICIT,

                WITHOUT_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.UNNAMED, MODULE_TYPE.AUTO,

                WITH_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.UNNAMED, MODULE_TYPE.AUTO,

        params.add(Arrays.asList(MODULE_TYPE.UNNAMED, MODULE_TYPE.UNNAMED,

                WITH_S_DESCR, NO_FAILURE, args));

        params.add(Arrays.asList(MODULE_TYPE.UNNAMED, MODULE_TYPE.UNNAMED,

                WITHOUT_S_DESCR, NO_FAILURE, args));

        return params.stream().map(p -> p.toArray()).toArray(Object[][]::new);

    }

    /**

     * Pre-compile and generate the artifacts required to run this test before

     * running each test cases.

     */

    @BeforeTest

    public void buildArtifacts() {

        boolean done = true;

        try {

            done = CompilerUtils.compile(S_SRC, S_BUILD_DIR);

            done &= CompilerUtils.compile(S_SRC, S_WITH_META_DESCR_BUILD_DIR);

            done &= createMetaInfServiceDescriptor(S_META_DESCR_FPATH, S_IMPL);

            generateJar(true, MODULE_TYPE.EXPLICIT, MS_JAR, S_BUILD_DIR, false);

            generateJar(true, MODULE_TYPE.EXPLICIT, MS_WITH_DESCR_JAR,

                    S_WITH_META_DESCR_BUILD_DIR, false);

            generateJar(true, MODULE_TYPE.UNNAMED, S_JAR, S_BUILD_DIR, false);

            generateJar(true, MODULE_TYPE.UNNAMED, S_WITH_DESCRIPTOR_JAR,

                    S_WITH_META_DESCR_BUILD_DIR, false);

            System.out.format("%nArtifacts generated successfully? %s", done);

            if (!done) {

                throw new RuntimeException("Artifact generation failed");

            }

        } catch (IOException e) {

            throw new RuntimeException(e);

        }

    }

    /**

     * Generate modular/regular jar based on module type for this test.

     */

    private void generateJar(boolean isService, MODULE_TYPE moduleType,

            Path jar, Path compilePath, boolean depends) throws IOException {

        ModuleDescriptor mDescriptor = null;

        if (isService) {

            mDescriptor = generateModuleDescriptor(isService, moduleType, S_PKG,

                    S_PKG, S_INTERFACE, S_IMPL, null, M_REQUIRED, depends);

        } else {

            mDescriptor = generateModuleDescriptor(isService, moduleType, C_PKG,

                    C_PKG, S_INTERFACE, null, S_PKG, M_REQUIRED, depends);

        }

        generateJar(mDescriptor, jar, compilePath);

    }

    /**

     * Holds Logic for the test client. This method will get called with each

     * test parameter.

     */

    @Override

    public OutputAnalyzer executeTestClient(MODULE_TYPE cModuleType,

            Path cJarPath, MODULE_TYPE sModuletype, Path sJarPath,

            String... args) throws Exception {

        OutputAnalyzer output = null;

        try {

            if (!(cModuleType == MODULE_TYPE.UNNAMED

                    && sModuletype == MODULE_TYPE.UNNAMED)) {

                copyJarsToModuleBase(cModuleType, cJarPath, M_BASE_PATH);

                copyJarsToModuleBase(sModuletype, sJarPath, M_BASE_PATH);

            }

            System.out.format("%nExecuting java client with required"

                    + " custom service in class/module path.");

            Path cmBasePath = (cModuleType != MODULE_TYPE.UNNAMED

                    || sModuletype != MODULE_TYPE.UNNAMED) ? M_BASE_PATH : null;

            String cPath = buildClassPath(cModuleType, cJarPath, sModuletype,

                    sJarPath);

            output = ProcessTools.executeTestJava(

                            args)).outputTo(System.out).errorTo(System.out);

        } finally {

            cleanModuleBasePath(M_BASE_PATH);

        }

        return output;

    }

    /**

     * Decide the pre-generated client/service jar path for each test case

     * based on client/service module type.

     */

    @Override

            boolean addMetaDesc, boolean dependsOnServiceModule) {

            if (moduleType == MODULE_TYPE.EXPLICIT) {

                if (addMetaDesc) {

                    return MS_WITH_DESCR_JAR;

                } else {

                    return MS_JAR;

                }

            } else {

                if (addMetaDesc) {

                    return S_WITH_DESCRIPTOR_JAR;

                } else {

                    return S_JAR;

                }

            }

        } else {

            if (moduleType == MODULE_TYPE.EXPLICIT) {

                if (dependsOnServiceModule) {

                    return MC_JAR;

                } else {

                }

            } else {

                return C_JAR;

            }

        }

    }

}