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

/**
 * @test
 * @bug 8072480 8277106
 * @summary Unit test for CreateSymbols
 * @modules java.compiler
 *          jdk.compiler/com.sun.tools.javac.api
 *          jdk.compiler/com.sun.tools.javac.jvm
 *          jdk.compiler/com.sun.tools.javac.main
 *          jdk.compiler/com.sun.tools.javac.util
 *          jdk.jdeps/com.sun.tools.classfile
 * @clean *
 * @run main/othervm CreateSymbolsTest
 */

import java.io.IOException;
import java.net.URL;
import java.net.URLClassLoader;
import java.nio.file.FileVisitResult;
import java.nio.file.FileVisitor;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.ArrayList;
import java.util.List;
import javax.tools.JavaCompiler;
import javax.tools.StandardJavaFileManager;
import javax.tools.ToolProvider;

public class CreateSymbolsTest {

    static final String CREATE_SYMBOLS_NAME = "symbolgenerator.CreateSymbols";

    public static void main(String... args) throws Exception {
        new CreateSymbolsTest().doRun();
    }

    void doRun() throws Exception {
        Path testClasses = Paths.get(System.getProperty("test.classes"));
        Path compileDir = testClasses.resolve("data");
        deleteRecursively(compileDir);
        Files.createDirectories(compileDir);
        Path createSymbols = findFile("../../make/langtools/src/classes/build/tools/symbolgenerator/CreateSymbols.java");

        if (createSymbols == null) {
            System.err.println("Warning: cannot find CreateSymbols, skipping.");
            return ;
        }

        Path createTestImpl = findFile("tools/javac/platform/createsymbols/CreateSymbolsTestImpl.java");

        if (createTestImpl == null) {
            throw new AssertionError("Warning: cannot find CreateSymbolsTestImpl, skipping.");
        }

        Path toolBox = findFile("tools/lib/toolbox/");

        if (toolBox == null) {
            throw new AssertionError("Warning: cannot find ToolBox, skipping.");
        }

        JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();

        try (StandardJavaFileManager fm = compiler.getStandardFileManager(null, null, null)) {
            List<Path> files = new ArrayList<>();

            files.add(createSymbols);
            files.add(createTestImpl);

            files.add(toolBox.resolve("AbstractTask.java"));
            files.add(toolBox.resolve("JavacTask.java"));
            files.add(toolBox.resolve("Task.java"));
            files.add(toolBox.resolve("ToolBox.java"));

            Boolean res =
                    compiler.getTask(null,
                                      null,
                                      null,
                                      List.of("-d",
                                              compileDir.toAbsolutePath().toString(),
                                              "-g",
                                              "--add-modules", "jdk.jdeps",
                                              "--add-exports", "jdk.compiler/com.sun.tools.javac.api=ALL-UNNAMED",
                                              "--add-exports", "jdk.compiler/com.sun.tools.javac.file=ALL-UNNAMED",
                                              "--add-exports", "jdk.compiler/com.sun.tools.javac.jvm=ALL-UNNAMED",
                                              "--add-exports", "jdk.compiler/com.sun.tools.javac.main=ALL-UNNAMED",
                                              "--add-exports", "jdk.compiler/com.sun.tools.javac.util=ALL-UNNAMED",
                                              "--add-exports", "jdk.jdeps/com.sun.tools.classfile=ALL-UNNAMED"),
                                      null,
                                      fm.getJavaFileObjectsFromPaths(files)
                                    ).call();
            if (!res) {
                throw new IllegalStateException("Cannot compile test.");
            }
        }

        URLClassLoader cl = new URLClassLoader(new URL[] {testClasses.toUri().toURL(), compileDir.toUri().toURL()});
        Class<?> createSymbolTest = cl.loadClass("CreateSymbolsTestImpl");

        createSymbolTest.getMethod("main", String[].class).invoke(null, (Object) new String[0]);
    }

    Path findFile(String path) {
        Path testSrc = Paths.get(System.getProperty("test.src", "."));

        for (Path d = testSrc; d != null; d = d.getParent()) {
            if (Files.exists(d.resolve("TEST.ROOT"))) {
                Path file = d.resolve(path);
                if (Files.exists(file)) {
                    return file;
                }
            }
        }

        return null;
    }

    void deleteRecursively(Path dir) throws IOException {
        Files.walkFileTree(dir, new FileVisitor<Path>() {
            @Override
            public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs) throws IOException {
                return FileVisitResult.CONTINUE;
            }
            @Override public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
                Files.delete(file);
                return FileVisitResult.CONTINUE;
            }
            @Override public FileVisitResult visitFileFailed(Path file, IOException exc) throws IOException {
                return FileVisitResult.CONTINUE;
            }
            @Override public FileVisitResult postVisitDirectory(Path dir, IOException exc) throws IOException {
                Files.delete(dir);
                return FileVisitResult.CONTINUE;
            }
        });
    }
}