/*

 * Copyright (c) 2006, 2010, 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 6929404

 * @summary Filer.getResource(SOURCE_PATH, ...) does not work when -sourcepath contains >1 entry

 */

import java.io.*;

import java.security.*;

import java.util.*;

import javax.annotation.processing.*;

import javax.lang.model.*;

import javax.lang.model.element.*;

import javax.tools.*;

import javax.tools.Diagnostic.Kind;

import javax.tools.JavaCompiler.CompilationTask;

public class TestGetResource2 {

    public static void main(String[] args) throws Exception {

        new TestGetResource2().run();

    }

    void run() throws Exception {

        JavaCompiler javac = ToolProvider.getSystemJavaCompiler();

        CodeSource cs = javac.getClass().getProtectionDomain().getCodeSource();

        if (cs == null) {

            System.err.println("got compiler from " +

                ClassLoader.getSystemResource(javac.getClass().getName().replace(".", "/")+".class"));

        } else {

            System.err.println("got compiler from " + cs.getLocation());

        }

        testSingleSourceDir(javac);

        testCompositeSourcePath(javac);

        testMissingResource(javac);

    }

    private void testSingleSourceDir(JavaCompiler javac) throws Exception {

        System.err.println("testSingleSourceDir");

        File tmpdir = new File("testSingleSourceDir");

        File srcdir = new File(tmpdir, "src");

        File destdir = new File(tmpdir, "dest");

        write(srcdir, "pkg/X.java", "package pkg; class X {}");

        write(srcdir, "resources/file.txt", "hello");

        destdir.mkdirs();

        CompilationTask task = javac.getTask(null, null, null,

                Arrays.asList("-sourcepath", srcdir.toString(), "-d", destdir.toString()),

                Collections.singleton("pkg.X"), null);

        task.setProcessors(Collections.singleton(new AnnoProc()));

        boolean result = task.call();

        System.err.println("javac result with single source dir: " + result);

        expect(result, true);

    }

    private void testCompositeSourcePath(JavaCompiler javac) throws Exception {

        System.err.println("testCompositeSearchPath");

        File tmpdir = new File("testCompositeSourcePath");

        File srcdir = new File(tmpdir, "src");

        File rsrcdir = new File(tmpdir, "rsrc");

        File destdir = new File(tmpdir, "dest");

        write(srcdir, "pkg/X.java", "package pkg; class X {}");

        write(rsrcdir, "resources/file.txt", "hello");

        destdir.mkdirs();

        CompilationTask task = javac.getTask(null, null, null,

                Arrays.asList("-sourcepath", srcdir + File.pathSeparator + rsrcdir, "-d", destdir.toString()),

                Collections.singleton("pkg.X"), null);

        task.setProcessors(Collections.singleton(new AnnoProc()));

        boolean result = task.call();

        System.err.println("javac result with composite source path: " + result);

        expect(result, true);

    }

    private void testMissingResource(JavaCompiler javac) throws Exception {

        System.err.println("testMissingResource");

        File tmpdir = new File("testMissingResource");

        File srcdir = new File(tmpdir, "src");

        File destdir = new File(tmpdir, "dest");

        write(srcdir, "pkg/X.java", "package pkg; class X {}");

        destdir.mkdirs();

        CompilationTask task = javac.getTask(null, null, null,

                Arrays.asList("-sourcepath", srcdir.toString(), "-d", destdir.toString()),

                Collections.singleton("pkg.X"), null);

        task.setProcessors(Collections.singleton(new AnnoProc()));

        boolean result = task.call();

        System.err.println("javac result when missing resource: " + result);

        expect(result, false);

        if (errors > 0)

            throw new Exception(errors + " errors occurred");

    }

    static class AnnoProc extends JavacTestingAbstractProcessor {

        public @Override boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {

            if (roundEnv.processingOver()) {

                return false;

            }

            try {

                FileObject resource = filer.getResource(StandardLocation.SOURCE_PATH, "resources", "file.txt");

                try {

                    resource.openInputStream().close();

                    messager.printMessage(Kind.NOTE, "found: " + resource.toUri());

                    return true;

                } catch (IOException x) {

                    messager.printMessage(Kind.ERROR, "could not read: " + resource.toUri());

                    x.printStackTrace();

                }

            } catch (IOException x) {

                messager.printMessage(Kind.ERROR, "did not find resource");

                x.printStackTrace();

            }

            return false;

        }

    }

    private File write(File dir, String path, String contents) throws IOException {

        File f = new File(dir, path);

        f.getParentFile().mkdirs();

        Writer w = new FileWriter(f);

        try {

            w.write(contents);

        } finally {

            w.close();

        }

        return f;

    }

    void expect(boolean val, boolean expect) {

        if (val != expect)

            error("Unexpected value: " + val + "; expected: " + expect);

    }

    void error(String msg) {

        System.err.println(msg);

        errors++;

    }

    int errors = 0;

}