/*
 * Copyright (c) 2016, 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 7020499
 * @summary Verify that the close resource code works properly in all cases
 * @library /tools/lib
 * @modules jdk.compiler/com.sun.tools.javac.api
 *          jdk.compiler/com.sun.tools.javac.comp
 *          jdk.compiler/com.sun.tools.javac.main
 * @build toolbox.ToolBox TwrClose
 * @run main TwrClose
 */

import javax.tools.JavaFileManager;
import javax.tools.StandardLocation;
import javax.tools.ToolProvider;

import com.sun.tools.javac.comp.Lower;

import toolbox.JavacTask;
import toolbox.ToolBox;

public class TwrClose {

    public static void main(String... args) throws Exception {
        for (int i = 1; i < Lower.USE_CLOSE_RESOURCE_METHOD_THRESHOLD * 2; i++) {
            new TwrClose().compile(i);
        }
    }

    ToolBox tb = new ToolBox();
    JavaFileManager fm = ToolProvider.getSystemJavaCompiler()
                                     .getStandardFileManager(null, null, null);

    void compile(int trysCount) throws Exception {
        StringBuilder testInvocations = new StringBuilder();
        StringBuilder testMethods = new StringBuilder();

        for (int i = 0; i < trysCount; i++) {
            testInvocations.append(TEST_INVOCATIONS_TEMPLATE.replace("#N", Integer.toString(i)));
            testMethods.append(TEST_METHOD_TEMPLATE.replace("#N", Integer.toString(i)));
        }

        String sourceCode = FILE_TEMPLATE.replace("#TEST_INVOCATIONS", testInvocations.toString())
                                         .replace("#TEST_METHODS", testMethods.toString());

        System.err.println("analyzing:");
        System.err.println(sourceCode);

        try (ToolBox.MemoryFileManager mfm = new ToolBox.MemoryFileManager()) {
            new JavacTask(tb).fileManager(mfm)
                             .sources(sourceCode)
                             .run()
                             .writeAll();
            ClassLoader cl = new ClassLoader(TwrClose.class.getClassLoader()) {
                @Override
                protected Class<?> findClass(String name) throws ClassNotFoundException {
                    byte[] data = mfm.getFileBytes(StandardLocation.CLASS_OUTPUT, name);
                    if (data != null) {
                        return defineClass(name, data, 0, data.length);
                    }
                    return super.findClass(name);
                }
            };

            ((Runnable) cl.loadClass("Test").newInstance()).run();
        }
    }

    final String TEST_INVOCATIONS_TEMPLATE =
        "        test#N(false, false, Arrays.asList(\"close\"));\n" +
        "        test#N(false, true, Arrays.asList(\"close\", \"close-exception\"));\n" +
        "        test#N(true, false, Arrays.asList(\"close\", \"inTwr\"));\n" +
        "        test#N(true, true, Arrays.asList(\"close\", \"inTwr\", \"close-exception\"));\n";

    final String TEST_METHOD_TEMPLATE =
        "    private void test#N(boolean failInTwr, boolean failOnClose,\n" +
        "                        List<String> expectedMessages) {\n" +
        "        List<String> messages = new ArrayList<>();\n" +
        "        try {\n" +
        "            try (CloseableImpl c = new CloseableImpl(messages, failOnClose)) {\n" +
        "                if (failInTwr)\n" +
        "                    throw new IllegalStateException(\"inTwr\");\n" +
        "            }\n" +
        "        } catch (IllegalStateException ex) {\n" +
        "            messages.add(ex.getMessage());\n" +
        "            for (Throwable t : ex.getSuppressed()) {\n" +
        "                messages.add(t.getMessage());\n" +
        "            }\n" +
        "        }\n" +
        "        if (!expectedMessages.equals(messages))\n" +
        "            throw new AssertionError(\"Expected and actual messages differ; expectedMessages=\" +\n" +
        "                                     expectedMessages + \"; actual=\" + messages);\n" +
        "    }\n";

    final String FILE_TEMPLATE =
        "import java.util.*;\n" +
        "public class Test implements Runnable {\n" +
        "    public void run() {\n" +
        "#TEST_INVOCATIONS" +
        "    }\n" +
        "#TEST_METHODS" +
        "    static class CloseableImpl implements AutoCloseable {\n" +
        "        private final List<String> messages;\n" +
        "        private final boolean failOnClose;\n" +
        "        public CloseableImpl(List<String> messages, boolean failOnClose) {\n" +
        "            this.messages = messages;\n" +
        "            this.failOnClose = failOnClose;\n" +
        "        }\n" +
        "        @Override\n" +
        "        public void close() {\n" +
        "            messages.add(\"close\");\n" +
        "            if (failOnClose)\n" +
        "                throw new IllegalStateException(\"close-exception\");\n" +
        "        }\n" +
        "    }\n" +
        "}\n";
}