/*
 * Copyright (c) 2009, 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 6906175 6915476 6915497
 * @summary Path-based JavaFileManager
 * @compile -g HelloPathWorld.java
 * @run main CompileTest
 */

import java.io.*;
import java.nio.file.*;
import java.util.*;
import java.util.jar.*;
import javax.tools.*;

import com.sun.tools.javac.nio.*;
import com.sun.tools.javac.util.Context;
import java.nio.file.spi.FileSystemProvider;


public class CompileTest {
    public static void main(String[] args) throws Exception {
        new CompileTest().run();
    }

    public void run() throws Exception {
        File rtDir = new File("rt.dir");
        File javaHome = new File(System.getProperty("java.home"));
        if (javaHome.getName().equals("jre"))
            javaHome = javaHome.getParentFile();
        File rtJar = new File(new File(new File(javaHome, "jre"), "lib"), "rt.jar");
        expand(rtJar, rtDir);

        String[] rtDir_opts = {
            "-bootclasspath", rtDir.toString(),
            "-classpath", "",
            "-sourcepath", "",
            "-extdirs", ""
        };
        test(rtDir_opts, "HelloPathWorld");

        if (isJarFileSystemAvailable()) {
            String[] rtJar_opts = {
                "-bootclasspath", rtJar.toString(),
                "-classpath", "",
                "-sourcepath", "",
                "-extdirs", ""
            };
            test(rtJar_opts, "HelloPathWorld");

            String[] default_opts = { };
            test(default_opts, "HelloPathWorld");

            // finally, a non-trivial program
            test(default_opts, "CompileTest");
        } else
            System.err.println("jar file system not available: test skipped");
    }

    void test(String[] opts, String className) throws Exception {
        count++;
        System.err.println("Test " + count + " " + Arrays.asList(opts) + " " + className);
        Path testSrcDir = Paths.get(System.getProperty("test.src"));
        Path testClassesDir = Paths.get(System.getProperty("test.classes"));
        Path classes = Paths.get("classes." + count);
        classes.createDirectory();

        Context ctx = new Context();
        PathFileManager fm = new JavacPathFileManager(ctx, true, null);
        JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
        List<String> options = new ArrayList<String>();
        options.addAll(Arrays.asList(opts));
        options.addAll(Arrays.asList(
                "-verbose", "-XDverboseCompilePolicy",
                "-d", classes.toString(),
                "-g"
        ));
        Iterable<? extends JavaFileObject> compilationUnits =
                fm.getJavaFileObjects(testSrcDir.resolve(className + ".java"));
        StringWriter sw = new StringWriter();
        PrintWriter out = new PrintWriter(sw);
        JavaCompiler.CompilationTask t =
                compiler.getTask(out, fm, null, options, null, compilationUnits);
        boolean ok = t.call();
        System.err.println(sw.toString());
        if (!ok) {
            throw new Exception("compilation failed");
        }

        File expect = new File("classes." + count + "/" + className + ".class");
        if (!expect.exists())
            throw new Exception("expected file not found: " + expect);
        // Note that we explicitly specify -g for compiling both the actual class and the expected class.
        // This isolates the expected class from javac options that might be given to jtreg.
        long expectedSize = new File(testClassesDir.toString(), className + ".class").length();
        long actualSize = expect.length();
        if (expectedSize != actualSize)
            throw new Exception("wrong size found: " + actualSize + "; expected: " + expectedSize);
    }

    boolean isJarFileSystemAvailable() {
        boolean result = false;
        for (FileSystemProvider fsp: FileSystemProvider.installedProviders()) {
            String scheme = fsp.getScheme();
            System.err.println("Provider: " + scheme + " " + fsp);
            if (scheme.equalsIgnoreCase("jar") || scheme.equalsIgnoreCase("zip"))
                result = true;
        }
        return result;
    }

    void expand(File jar, File dir) throws IOException {
        JarFile jarFile = new JarFile(jar);
        try {
            Enumeration<JarEntry> entries = jarFile.entries();
            while (entries.hasMoreElements()) {
                JarEntry je = entries.nextElement();
                if (!je.isDirectory()) {
                    copy(jarFile.getInputStream(je), new File(dir, je.getName()));
                }
            }
        } finally {
            jarFile.close();
        }
    }

    void copy(InputStream in, File dest) throws IOException {
        dest.getParentFile().mkdirs();
        OutputStream out = new BufferedOutputStream(new FileOutputStream(dest));
        try {
            byte[] data = new byte[8192];
            int n;
            while ((n = in.read(data, 0, data.length)) > 0)
                out.write(data, 0, n);
        } finally {
            out.close();
            in.close();
        }
    }

    void error(String message) {
        System.err.println("Error: " + message);
        errors++;
    }

    int errors;
    int count;
}