/*
 * Copyright (c) 2016, 2017, 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.io.BufferedWriter;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.spi.ToolProvider;

import jdk.test.lib.compiler.CompilerUtils;
import jdk.testlibrary.OutputAnalyzer;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

import static org.testng.Assert.assertTrue;
import static jdk.testlibrary.ProcessTools.*;

/**
 * @test
 * @bug 8168205
 * @summary Test the default class path if -Djava.class.path is set
 * @library /lib/testlibrary /test/lib
 * @modules jdk.compiler
 *          jdk.jartool
 * @build jdk.test.lib.compiler.CompilerUtils jdk.testlibrary.*
 * @run testng JavaClassPathTest
 */

public class JavaClassPathTest {
    private static final Path SRC_DIR = Paths.get(System.getProperty("test.src"),
                                                  "src");
    private static final Path MODS_DIR = Paths.get("mods");
    private static final Path LIB_DIR = Paths.get("lib");
    private static final String TEST_MODULE = "m";
    private static final String TEST_MAIN = "jdk.test.Main";

    @BeforeTest
    public void setup() throws Exception {
        boolean compiled = CompilerUtils.compile(SRC_DIR.resolve(TEST_MODULE),
                                                 MODS_DIR.resolve(TEST_MODULE));
        assertTrue(compiled, "module " + TEST_MODULE + " did not compile");

        // add the class and a resource to the current working directory
        Path file = Paths.get("jdk/test/Main.class");
        Files.createDirectories(file.getParent());
        Files.copy(MODS_DIR.resolve(TEST_MODULE).resolve(file), file);

        Path res = Paths.get("jdk/test/res.properties");
        Files.createFile(res);

        ToolProvider jartool = ToolProvider.findFirst("jar").orElseThrow(
            () -> new RuntimeException("jar tool not found")
        );

        Path jarfile = LIB_DIR.resolve("m.jar");
        Files.createDirectories(LIB_DIR);
        assertTrue(jartool.run(System.out, System.err, "cfe",
                               jarfile.toString(), TEST_MAIN,
                               file.toString()) == 0);

        Path manifest = LIB_DIR.resolve("manifest");
        try (BufferedWriter writer = Files.newBufferedWriter(manifest)) {
            writer.write("CLASS-PATH: lib/m.jar");
        }
        jarfile = LIB_DIR.resolve("m1.jar");
        assertTrue(jartool.run(System.out, System.err, "cfme",
                               jarfile.toString(), manifest.toString(), TEST_MAIN,
                               file.toString()) == 0);
    }

    @DataProvider(name = "classpath")
    public Object[][] classpath() {
        return new Object[][]{
            // true indicates that class path default to current working directory
            { List.of(),                          "." },
            { List.of("-cp", ""),                 "" },
            { List.of("-cp", "."),                "." },
            { List.of("-Djava.class.path"),       "." },
            { List.of("-Djava.class.path="),      ""  },
            { List.of("-Djava.class.path=."),     "." },
        };
    }

    @Test(dataProvider = "classpath")
    public void testUnnamedModule(List<String> options, String expected)
        throws Throwable
    {
        List<String> args = new ArrayList<>(options);
        args.add(TEST_MAIN);
        args.add(Boolean.toString(true));
        args.add(expected);

        assertTrue(execute(args).getExitValue() == 0);
    }

    @DataProvider(name = "moduleAndClassPath")
    public Object[][] moduleAndClassPath() {
        return new Object[][]{
            // true indicates that class path default to current working directory
            { "",                              ""  },
            { "-Djava.class.path",             ""  },
            { "-Djava.class.path=",            ""  },
        };
    }

    @Test(dataProvider = "moduleAndClassPath")
    public void testNamedModule(String option, String expected) throws Throwable {
        List<String> args = new ArrayList<>();
        if (!option.isEmpty()) {
            args.add(option);
        }
        args.add("--module-path");
        args.add(MODS_DIR.toString());
        args.add("-m");
        args.add(TEST_MODULE + "/" + TEST_MAIN);
        // not default to CWD
        args.add(Boolean.toString(false));
        args.add(expected);


        assertTrue(execute(args).getExitValue() == 0);
    }

    @Test
    public void testClassPath() throws Throwable {
        List<String> args = new ArrayList<>();
        args.add("-Djava.class.path=.");
        args.add("--module-path");
        args.add(MODS_DIR.toString());
        args.add("-m");
        args.add(TEST_MODULE + "/" + TEST_MAIN);
        args.add(Boolean.toString(true));
        args.add(".");

        assertTrue(execute(args).getExitValue() == 0);
    }

    @Test
    public void testJAR() throws Throwable {
        String jarfile = LIB_DIR.resolve("m.jar").toString();
        List<String> args = new ArrayList<>();
        args.add("-jar");
        args.add(jarfile);
        args.add(Boolean.toString(false));
        args.add(jarfile);

        assertTrue(execute(args).getExitValue() == 0);
    }

    /*
     * Test CLASS-PATH attribute in manifest
     */
    @Test
    public void testClassPathAttribute() throws Throwable {
        String jarfile = LIB_DIR.resolve("m1.jar").toString();

        List<String> args = new ArrayList<>();
        args.add("-jar");
        args.add(jarfile);
        args.add(Boolean.toString(false));
        args.add(jarfile);

        assertTrue(execute(args).getExitValue() == 0);

        args.clear();
        args.add("-cp");
        args.add(jarfile);
        args.add(TEST_MAIN);
        args.add(Boolean.toString(false));
        args.add(jarfile);

        assertTrue(execute(args).getExitValue() == 0);
    }

    private OutputAnalyzer execute(List<String> options) throws Throwable {
        ProcessBuilder pb = createJavaProcessBuilder(
            options.stream()
                   .map(this::autoQuote)
                   .toArray(String[]::new)
        );

        Map<String,String> env = pb.environment();
        // remove CLASSPATH environment variable
        String value = env.remove("CLASSPATH");
        return executeCommand(pb)
                    .outputTo(System.out)
                    .errorTo(System.out);
    }

    private static final boolean IS_WINDOWS
        = System.getProperty("os.name").startsWith("Windows");

    /*
     * Autoquote empty string argument on Windows
     */
    private String autoQuote(String arg) {
        if (IS_WINDOWS && arg.isEmpty()) {
            return "\"\"";
        }
        return arg;
    }
}