/*

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

 */

/**

 * @test

 * @bug 8173914 8188035

 * @summary JavaFileManager.setLocationForModule

 * @modules jdk.compiler/com.sun.tools.javac.api

 *          jdk.compiler/com.sun.tools.javac.main

 * @library /tools/lib

 * @build toolbox.JavacTask toolbox.TestRunner toolbox.ToolBox SetLocationForModule

 * @run main SetLocationForModule

 */

import java.io.FileNotFoundException;

import java.io.IOException;

import java.nio.file.Files;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.util.ArrayList;

import java.util.Collections;

import java.util.List;

import java.util.Objects;

import java.util.Set;

import javax.tools.JavaCompiler;

import javax.tools.JavaFileManager;

import javax.tools.JavaFileManager.Location;

import javax.tools.StandardJavaFileManager;

import javax.tools.StandardLocation;

import javax.tools.ToolProvider;

import toolbox.JavacTask;

import toolbox.TestRunner;

import toolbox.TestRunner.Test;

import toolbox.ToolBox;

public class SetLocationForModule extends TestRunner {

    public static void main(String... args) throws Exception {

        new SetLocationForModule().runTests(m -> new Object[] { Paths.get(m.getName()) });

    }

    public SetLocationForModule() {

        super(System.err);

    }

    private final JavaCompiler comp = ToolProvider.getSystemJavaCompiler();

    private final ToolBox tb = new ToolBox();

    @Test

    public void testBasic(Path base) throws IOException {

        try (StandardJavaFileManager fm = comp.getStandardFileManager(null, null, null)) {

            Location[] locns = {

                StandardLocation.SOURCE_PATH,

                StandardLocation.CLASS_PATH,

                StandardLocation.PLATFORM_CLASS_PATH,

            };

            // set a value

            Path out = Files.createDirectories(base.resolve("out"));

            for (Location locn : locns) {

                checkException("unsupported for location",

                        IllegalArgumentException.class,

                        "location is not an output location or a module-oriented location: " + locn,

                        () -> fm.setLocationForModule(locn, "m", List.of(out)));

            }

        }

    }

    @Test

    public void testModulePath(Path base) throws IOException {

        try (StandardJavaFileManager fm = comp.getStandardFileManager(null, null, null)) {

            Path src = base.resolve("src");

            Path src_m = src.resolve("m");

            tb.writeJavaFiles(src_m, "module m { }");

            Location locn = StandardLocation.MODULE_PATH;

            Path modules1 = Files.createDirectories(base.resolve("modules1"));

            new JavacTask(tb)

                    .outdir(modules1)

                    .options("--module-source-path", src.toString())

                    .files(tb.findJavaFiles(src))

                    .run();

            fm.setLocationFromPaths(locn, List.of(modules1));

            Location m = fm.getLocationForModule(locn, "m");

            checkEqual("default setting",

                    fm.getLocationAsPaths(m), modules1.resolve("m"));

            Path override1 = Files.createDirectories(base.resolve("override1"));

            fm.setLocationForModule(locn, "m", List.of(override1));

            checkEqual("override setting 1",

                    fm.getLocationAsPaths(m), override1);

            checkEqual("override setting 1b",

                       fm.getLocationAsPaths(fm.listLocationsForModules(locn).iterator().next().iterator().next()),

                       override1);

            try (StandardJavaFileManager fm2 = comp.getStandardFileManager(null, null, null)) {

                fm2.setLocationForModule(locn, "m", List.of(override1));

                checkEqual("override setting 2",

                           fm2.getLocationAsPaths(m), override1);

                checkEqual("override setting 2b",

                           override1);

            }

            Path override2 = Files.createDirectories(base.resolve("override2"));

            fm.setLocationFromPaths(m, List.of(override2));

            checkEqual("override setting 3",

                    fm.getLocationAsPaths(m), override2);

            Path modules2 = Files.createDirectories(base.resolve("modules2"));

            new JavacTask(tb)

                    .outdir(modules2)

                    .options("--module-source-path", src.toString())

                    .files(tb.findJavaFiles(src))

                    .run();

            fm.setLocationFromPaths(locn, List.of(modules2));

            m = fm.getLocationForModule(locn, "m");

            checkEqual("updated setting",

                    fm.getLocationAsPaths(m), modules2.resolve("m"));

        }

    }

    @Test

    public void testModuleSourcePath(Path base) throws IOException {

        try (StandardJavaFileManager fm = comp.getStandardFileManager(null, null, null)) {

            Location locn = StandardLocation.MODULE_SOURCE_PATH;

            Path src1 = Files.createDirectories(base.resolve("src1"));

            Path src1_m = src1.resolve("m");

            tb.writeJavaFiles(src1_m, "module m { }");

//            fm.setLocationFromPaths(locn, List.of(src1));

            fm.handleOption("--module-source-path", List.of(src1.toString()).iterator());

            Location m = fm.getLocationForModule(locn, "m");

            checkEqual("default setting",

                    fm.getLocationAsPaths(m), src1.resolve("m"));

            Path override1 = Files.createDirectories(base.resolve("override1"));

            tb.writeJavaFiles(override1, "module m { }");

            fm.setLocationForModule(locn, "m", List.of(override1));

            checkEqual("override setting 1",

                    fm.getLocationAsPaths(m), override1);

            checkEqual("override setting 1b",

                       fm.getLocationAsPaths(fm.listLocationsForModules(locn).iterator().next().iterator().next()),

                       override1);

            Path override2 = Files.createDirectories(base.resolve("override2"));

            tb.writeJavaFiles(override2, "module m { }");

            fm.setLocationFromPaths(m, List.of(override2));

            checkEqual("override setting 2",

                    fm.getLocationAsPaths(m), override2);

            Path src2 = Files.createDirectories(base.resolve("src2"));

            Path src2_m = src2.resolve("m");

            tb.writeJavaFiles(src2_m, "module m { }");

//            fm.setLocationFromPaths(locn, List.of(src2));

            fm.handleOption("--module-source-path", List.of(src2.toString()).iterator());

            m = fm.getLocationForModule(locn, "m");

            checkEqual("updated setting",

                    fm.getLocationAsPaths(m), src2.resolve("m"));

        }

    }

    @Test

    public void testOutput(Path base) throws IOException {

        try (StandardJavaFileManager fm = comp.getStandardFileManager(null, null, null)) {

            Location locn = StandardLocation.CLASS_OUTPUT;

            Path out1 = Files.createDirectories(base.resolve("out1"));

            fm.setLocationFromPaths(locn, List.of(out1));

            Location m = fm.getLocationForModule(locn, "m");

            checkEqual("default setting",

                    fm.getLocationAsPaths(m), out1.resolve("m"));

            Path override1 = Files.createDirectories(base.resolve("override1"));

            fm.setLocationForModule(locn, "m", List.of(override1));

            checkEqual("override setting 1",

                    fm.getLocationAsPaths(m), override1);

            checkEqual("override setting 1b",

                       fm.getLocationAsPaths(fm.listLocationsForModules(locn).iterator().next().iterator().next()),

                       override1);

            Path override2 = Files.createDirectories(base.resolve("override2"));

            fm.setLocationFromPaths(m, List.of(override2));

            checkEqual("override setting 2",

                    fm.getLocationAsPaths(m), override2);

            Path out2 = Files.createDirectories(base.resolve("out2"));

            fm.setLocationFromPaths(locn, List.of(out2));

            m = fm.getLocationForModule(locn, "m");

            checkEqual("updated setting",

                    fm.getLocationAsPaths(m), out2.resolve("m"));

        }

    }

    @Test

    public void testOutput_invalid(Path base) throws IOException {

        try (StandardJavaFileManager fm = comp.getStandardFileManager(null, null, null)) {

            Location locn = StandardLocation.CLASS_OUTPUT;

            // set a top default

            Path out1 = Files.createDirectories(base.resolve("out1"));

            fm.setLocationFromPaths(locn, List.of(out1));

            // getLocnForModule

            Location m = fm.getLocationForModule(locn, "m");

            checkEqual("default setting",

                    fm.getLocationAsPaths(m), out1.resolve("m"));

            checkException("empty arg list",

                    IllegalArgumentException.class, "empty path for directory",

                    () -> fm.setLocationFromPaths(m, Collections.emptyList()));

            Path out2 = Files.createDirectories(base.resolve("out2"));

            checkException("empty arg list",

                    IllegalArgumentException.class, "path too long for directory",

                    () -> fm.setLocationFromPaths(m, List.of(out2, out2)));

            Path notExist = base.resolve("notExist");

            checkException("not exist",

                    FileNotFoundException.class, notExist + ": does not exist",

                    () -> fm.setLocationFromPaths(m, List.of(notExist)));

            Path file = Files.createFile(base.resolve("file.txt"));

            checkException("not exist",

                    IOException.class, file + ": not a directory",

                    () -> fm.setLocationFromPaths(m, List.of(file)));

        }

    }

    @Test

    public void testOutput_nested(Path base) throws IOException {

        try (StandardJavaFileManager fm = comp.getStandardFileManager(null, null, null)) {

            Location locn = StandardLocation.CLASS_OUTPUT;

            Path out1 = Files.createDirectories(base.resolve("out1"));

            fm.setLocationForModule(locn, "m", List.of(out1));

            Location m = fm.getLocationForModule(locn, "m");

            checkEqual("initial setting",

                    fm.getLocationAsPaths(m), out1);

            Path out2 = Files.createDirectories(base.resolve("out2"));

            checkException("create nested module",

                    UnsupportedOperationException.class, "not supported for CLASS_OUTPUT[m]",

                    () -> fm.setLocationForModule(m, "x", List.of(out2)));

        }

    }

    @Test

    public void testSystemModules(Path base) throws IOException {

        try (StandardJavaFileManager fm = comp.getStandardFileManager(null, null, null)) {

            Location locn = StandardLocation.SYSTEM_MODULES;

            Location javaCompiler = fm.getLocationForModule(locn, "java.compiler");

            // cannot easily verify default setting: could be jrt: or exploded image

            Path override1 = Files.createDirectories(base.resolve("override1"));

            fm.setLocationForModule(locn, "java.compiler", List.of(override1));

            checkEqual("override setting 1",

                    fm.getLocationAsPaths(javaCompiler), override1);

            checkEqual("override setting 1b",

                       fm.getLocationAsPaths(findLocation(fm, fm.listLocationsForModules(locn), "java.compiler")),

                       override1);

            Path override2 = Files.createDirectories(base.resolve("override2"));

            fm.setLocationFromPaths(javaCompiler, List.of(override2));

            checkEqual("override setting 2",

                    fm.getLocationAsPaths(javaCompiler), override2);

        }

    }

    private Location findLocation(JavaFileManager fm, Iterable<Set<Location>> locations, String moduleName) {

        for (Set<Location> locs : locations) {

            for (Location loc : locs) {

                try {

                    if (moduleName.equals(fm.inferModuleName(loc))) {

                        return loc;

                    }

                } catch (IOException ex) {

                    throw new IllegalStateException(ex);

                }

            }

        }

        throw new IllegalStateException();

    }

    @Test

    public void testTemplate(Path base) {

        // set a top default

        // getLocnForModule

        // set a value

        // getLocnForModule

        // reset

        // getLocationForModule

    }

    interface RunnableWithException {

        public void run() throws Exception;

    }

    void checkException(String message,

            Class<? extends Throwable> expectedException, String expectedMessage,

            RunnableWithException r) {

        try {

            r.run();

            error(message + ": expected exception not thrown: " + expectedException);

        } catch (Exception | Error t) {

            if (expectedException.isAssignableFrom(t.getClass())) {

                checkEqual("exception message",

                        t.getMessage(), expectedMessage);

            } else {

                error(message + ": unexpected exception\n"

                        + "expect: " + expectedException + "\n"

                        + " found: " + t);

            }

        }

    }

    void checkEqual(String message, Iterable<? extends Path> found, Path... expect) {

        List<Path> fList = asList(found);

        List<Path> eList = List.of(expect);

        if (!Objects.equals(fList, eList)) {

            error(message + ": lists not equal\n"

                    + "expect: " + eList + "\n"

                    + " found: " + fList);

        }

    }

    void checkEqual(String message, String found, String expect) {

        if (!Objects.equals(found, expect)) {

            error(message + ": strings not equal\n"

                    + "expect: " + expect + "\n"

                    + " found: " + found);

        }

    }

    List<Path> asList(Iterable<? extends Path> a) {

        List<Path> list = new ArrayList<>();

        for (Path p : a) {

            list.add(p);

        }

        return list;

    }

}