/*

 * Copyright (c) 2018, 2019, 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 8214031

 * @summary Test the CharacterRangeTable generated for switch expressions

 * @library /tools/lib

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

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

 *          jdk.jdeps/com.sun.tools.javap

 * @build toolbox.Assert toolbox.ToolBox toolbox.JavacTask

 * @run main CRT

 */

import java.nio.file.Path;

import java.nio.file.Paths;

import java.util.stream.Collectors;

import toolbox.JavacTask;

import toolbox.JavapTask;

import toolbox.Task.OutputKind;

import toolbox.ToolBox;

public class CRT {

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

        new CRT().run();

    }

    private static final String SOURCE_VERSION = Integer.toString(Runtime.version().feature());

    private ToolBox tb = new ToolBox();

    private void run() throws Exception {

        doTest("    private String convert(int i) {\n" +

               "        String res;" +

               "        switch (i) {\n" +

               "            case 0: res = \"a\"; break;\n" +

               "            default: res = \"\"; break;\n" +

               "        };\n" +

               "        return res;\n" +

               "    }\n",

               "CharacterRangeTable:\n" +

               "             0,  0,    c24,    c25,    8        //  0,  0,    3:36,    3:37, flow-controller\n" +

               "            20, 22,   1015,   101f,    1        // 20, 22,    4:21,    4:31, statement\n" +

               "            23, 25,   1020,   1026,    1        // 23, 25,    4:32,    4:38, statement\n" +

               "            20, 25,   1015,   1026,   10        // 20, 25,    4:21,    4:38, flow-target\n" +

               "            26, 28,   1416,   141f,    1        // 26, 28,    5:22,    5:31, statement\n" +

               "            29, 31,   1420,   1426,    1        // 29, 31,    5:32,    5:38, statement\n" +

               "            26, 31,   1416,   1426,   10        // 26, 31,    5:22,    5:38, flow-target\n" +

               "             0, 31,    c1c,   180a,    1        //  0, 31,    3:28,    6:10, statement\n" +

               "            32, 33,   1c09,   1c14,    1        // 32, 33,    7:09,    7:20, statement\n" +

               "             0, 33,    823,   2006,    2        //  0, 33,    2:35,    8:06, block\n");

        doTest("    private String convert(int i) {\n" +

               "        return switch (i) {\n" +

               "            case 0 -> \"a\";\n" +

               "            default -> \"\";\n" +

               "        };\n" +

               "    }\n",

               "CharacterRangeTable:\n" +

               "             0,  0,    c18,    c19,    8        //  0,  0,    3:24,    3:25, flow-controller\n" +

               "            20, 24,   1017,   101b,   11        // 20, 24,    4:23,    4:27, statement, flow-target\n" +

               "            25, 29,   1418,   141b,   11        // 25, 29,    5:24,    5:27, statement, flow-target\n" +

               "             0, 30,    c09,   180b,    1        //  0, 30,    3:09,    6:11, statement\n" +

               "             0, 30,    823,   1c06,    2        //  0, 30,    2:35,    7:06, block");

        doTest("    private boolean convert(int i) {\n" +

               "        return switch (i) {\n" +

               "            case 0 -> true;\n" +

               "            default -> false;\n" +

               "        } && i == 0;\n" +

               "    }\n",

               "CharacterRangeTable:\n" +

               "             0,  0,    c18,    c19,    8        //  0,  0,    3:24,    3:25, flow-controller\n" +

               "            20, 22,   1017,   101c,   11        // 20, 22,    4:23,    4:28, statement, flow-target\n" +

               "            23, 25,   1418,   141e,   11        // 23, 25,    5:24,    5:30, statement, flow-target\n" +

               "             0, 25,    c10,   180a,    8        //  0, 25,    3:16,    6:10, flow-controller\n" +

               "            26, 26,   180e,   1814,   10        // 26, 26,    6:14,    6:20, flow-target\n" +

               "             0, 35,    c09,   1815,    1        //  0, 35,    3:09,    6:21, statement\n" +

               "             0, 35,    824,   1c06,    2        //  0, 35,    2:36,    7:06, block\n");

        doTest("    private boolean convert(int i) {\n" +

               "        return i >= 0 ? i == 0\n" +

               "                        ? true\n" +

               "                        : false\n" +

               "                      : i == -1\n" +

               "                        ? false\n" +

               "                        : true;\n" +

               "    }\n",

               "CharacterRangeTable:\n" +

               "             0,  0,    c10,    c16,    8        //  0,  0,    3:16,    3:22, flow-controller\n" +

               "             1,  3,    c10,    c16,  100        //  1,  3,    3:16,    3:22, branch-false\n" +

               "             4,  4,    c19,    c1f,    8        //  4,  4,    3:25,    3:31, flow-controller\n" +

               "             5,  7,    c19,    c1f,  100        //  5,  7,    3:25,    3:31, branch-false\n" +

               "             8,  8,   101b,   101f,   10        //  8,  8,    4:27,    4:31, flow-target\n" +

               "            12, 12,   141b,   1420,   10        // 12, 12,    5:27,    5:32, flow-target\n" +

               "             4, 12,    c19,   1420,   10        //  4, 12,    3:25,    5:32, flow-target\n" +

               "            16, 17,   1819,   1820,    8        // 16, 17,    6:25,    6:32, flow-controller\n" +

               "            18, 20,   1819,   1820,  100        // 18, 20,    6:25,    6:32, branch-false\n" +

               "            21, 21,   1c1b,   1c20,   10        // 21, 21,    7:27,    7:32, flow-target\n" +

               "            25, 25,   201b,   201f,   10        // 25, 25,    8:27,    8:31, flow-target\n" +

               "            16, 25,   1819,   201f,   10        // 16, 25,    6:25,    8:31, flow-target\n" +

               "             0, 26,    c09,   2020,    1        //  0, 26,    3:09,    8:32, statement\n" +

               "             0, 26,    824,   2406,    2        //  0, 26,    2:36,    9:06, block\n");

        doTest("    private boolean convert(int i) {\n" +

               "        return i >= 0 ? switch (i) {\n" +

               "            case 0 -> true;\n" +

               "            default -> false;\n" +

               "        } : switch (i) {\n" +

               "            case -1 -> false;\n" +

               "            default -> true;\n" +

               "        };\n" +

               "    }\n",

               "CharacterRangeTable:\n" +

               "             0,  0,    c10,    c16,    8        //  0,  0,    3:16,    3:22, flow-controller\n" +

               "             1,  3,    c10,    c16,  100        //  1,  3,    3:16,    3:22, branch-false\n" +

               "             4,  4,    c21,    c22,    8        //  4,  4,    3:33,    3:34, flow-controller\n" +

               "            24, 27,   1017,   101c,   11        // 24, 27,    4:23,    4:28, statement, flow-target\n" +

               "            28, 31,   1418,   141e,   11        // 28, 31,    5:24,    5:30, statement, flow-target\n" +

               "             4, 31,    c19,   180a,   10        //  4, 31,    3:25,    6:10, flow-target\n" +

               "            35, 35,   1815,   1816,    8        // 35, 35,    6:21,    6:22, flow-controller\n" +

               "            56, 59,   1c18,   1c1e,   11        // 56, 59,    7:24,    7:30, statement, flow-target\n" +

               "            60, 63,   2018,   201d,   11        // 60, 63,    8:24,    8:29, statement, flow-target\n" +

               "            35, 63,   180d,   240a,   10        // 35, 63,    6:13,    9:10, flow-target\n" +

               "             0, 64,    c09,   240b,    1        //  0, 64,    3:09,    9:11, statement\n" +

               "             0, 64,    824,   2806,    2        //  0, 64,    2:36,   10:06, block\n");

    }

    private void doTest(String code, String expected) throws Exception {

        Path base = Paths.get(".");

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

        tb.createDirectories(classes);

        tb.cleanDirectory(classes);

        new JavacTask(tb)

                .outdir(classes)

                .sources("public class Test {\n" +

                         code +

                         "}\n")

                .run()

                .writeAll();

        String out = new JavapTask(tb)

                .options("-private",

                         "-verbose",

                         "-s")

                .classpath(classes.toString())

                .classes("Test")

                .run()

                .writeAll()

                .getOutputLines(OutputKind.DIRECT)

                .stream()

                .collect(Collectors.joining("\n"));

        String crt = cutCRT(out);

        if (!expected.trim().equals(crt.trim())) {

            throw new AssertionError("Expected CharacterRangeTable not found, found: " + crt);

        }

    }

    private static String cutCRT(String from) {

        int start = from.indexOf("CharacterRangeTable:", from.indexOf("convert(int);"));

        int end = from.indexOf("StackMapTable:");

        return from.substring(start, end);

    }

}