/*

 * Copyright (c) 2013, 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     8011738

 * @author  sogoel

 * @summary Code translation test for Lambda expressions, method references

 * @modules jdk.jdeps/com.sun.tools.classfile

 * @run main ByteCodeTest

 */

import com.sun.tools.classfile.Attribute;

import com.sun.tools.classfile.BootstrapMethods_attribute;

import com.sun.tools.classfile.ClassFile;

import com.sun.tools.classfile.ConstantPool;

import com.sun.tools.classfile.ConstantPoolException;

import com.sun.tools.classfile.ConstantPool.*;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileWriter;

import java.io.IOException;

import java.io.PrintWriter;

import java.util.ArrayList;

import java.util.Collections;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

public class ByteCodeTest {

    static boolean IS_DEBUG = false;

    public static void main(String[] args) {

        File classFile = null;

        int err = 0;

        boolean verifyResult = false;

        for(TestCases tc : TestCases.values()) {

            classFile = getCompiledFile(tc.name(), tc.srcCode);

            if(classFile == null) { // either testFile or classFile was not created

               err++;

            } else {

                verifyResult = verifyClassFileAttributes(classFile, tc);

                if(!verifyResult)

                    System.out.println("Bootstrap class file attributes did not match for " + tc.name());

            }

        }

        if(err > 0)

            throw new RuntimeException("Found " + err + " found");

        else

            System.out.println("Test passed");

    }

    private static boolean verifyClassFileAttributes(File classFile, TestCases tc) {

        ClassFile c = null;

        try {

            c = ClassFile.read(classFile);

        } catch (IOException | ConstantPoolException e) {

            e.printStackTrace();

        }

        ConstantPoolVisitor cpv = new ConstantPoolVisitor(c, c.constant_pool.size());

        Map<Integer, String> hm = cpv.getBSMMap();

        List<String> expectedValList = tc.getExpectedArgValues();

        expectedValList.add(tc.bsmSpecifier.specifier);

        if(!(hm.values().containsAll(new HashSet<String>(expectedValList)))) {

            System.out.println("Values do not match");

            return false;

        }

        return true;

    }

    private static File getCompiledFile(String fname, String srcString) {

        File testFile = null, classFile = null;

        boolean isTestFileCreated = true;

        try {

            testFile = writeTestFile(fname+".java", srcString);

        } catch(IOException ioe) {

            isTestFileCreated = false;

            System.err.println("fail to write" + ioe);

        }

        if(isTestFileCreated) {

            try {

                classFile = compile(testFile);

            } catch (Error err) {

                System.err.println("fail compile. Source:\n" + srcString);

                throw err;

            }

        }

        return classFile;

    }

    static File writeTestFile(String fname, String source) throws IOException {

        File f = new File(fname);

          PrintWriter out = new PrintWriter(new BufferedWriter(new FileWriter(f)));

          out.println(source);

          out.close();

          return f;

    }

    static File compile(File f) {

        int rc = com.sun.tools.javac.Main.compile(new String[] {

                "-g", f.getPath() });

        if (rc != 0)

                throw new Error("compilation failed. rc=" + rc);

            String path = f.getPath();

            return new File(path.substring(0, path.length() - 5) + ".class");

    }

    static void debugln(String str) {

        if(IS_DEBUG)

            System.out.println(str);

    }

    enum BSMSpecifier {

        SPECIFIER1("REF_invokeStatic java/lang/invoke/LambdaMetafactory metaFactory " +

                "(Ljava/lang/invoke/MethodHandles$Lookup;Ljava/lang/String;Ljava/lang/invoke/MethodType;" +

                "Ljava/lang/invoke/MethodHandle;Ljava/lang/invoke/MethodHandle;Ljava/lang/invoke/MethodType;)" +

                "Ljava/lang/invoke/CallSite;"),

        SPECIFIER2("REF_invokeStatic java/lang/invoke/LambdaMetafactory altMetaFactory " +

                        "(Ljava/lang/invoke/MethodHandles$Lookup;Ljava/lang/String;Ljava/lang/invoke/MethodType;" +

                        "[Ljava/lang/Object;)Ljava/lang/invoke/CallSite;");

        String specifier;

        private BSMSpecifier(String specifier) {

            this.specifier = specifier;

        }

    }

    enum TestCases {

        // Single line lambda expression

        TC1("class TC1 {\n" +

            "    public static void main(String[] args) {\n" +

            "        Object o = (Runnable) () -> { System.out.println(\"hi\");};\n" +

            "    }\n"+

            "}", BSMSpecifier.SPECIFIER1) {

            @Override

            List<String> getExpectedArgValues() {

                List<String> valList = new ArrayList<>();

                valList.add("REF_invokeInterface java/lang/Runnable run ()V");

                valList.add("REF_invokeStatic TC1 lambda$0 ()V");

                valList.add("()V");

                return valList;

            }

        },

        // Lambda expression in a for loop

        TC2("import java.util.*;\n" +

            "public class TC2 {\n" +

            "    void TC2_test() {\n" +

            "        List<String> list = new ArrayList<>();\n" +

            "        list.add(\"A\");\n" +

            "        list.add(\"B\");\n" +

            "        list.stream().forEach( s -> { System.out.println(s); } );\n" +

            "    }\n" +

            "    public static void main(String[] args) {\n" +

            "        new TC2().TC2_test();\n" +

            "    }\n" +

            "}", BSMSpecifier.SPECIFIER1) {

            @Override

            List<String> getExpectedArgValues() {

                List<String> valList = new ArrayList<>();

                valList.add("REF_invokeInterface java/util/function/Consumer accept (Ljava/lang/Object;)V");

                valList.add("REF_invokeStatic TC2 lambda$0 (Ljava/lang/String;)V");

                valList.add("(Ljava/lang/String;)V");

                return valList;

            }

        },

        // Lambda initializer

        TC3("class TC3 {\n" +

            "    interface SAM {\n" +

            "        void m(int i);\n" +

            "    }\n" +

            "    SAM lambda_03 = (int pos) -> { };\n" +

            "}", BSMSpecifier.SPECIFIER1) {

            @Override

            List<String> getExpectedArgValues() {

                List<String> valList = new ArrayList<>();

                valList.add("REF_invokeInterface TC3$SAM m (I)V");

                valList.add("REF_invokeStatic TC3 lambda$0 (I)V");

                valList.add("(I)V");

                return valList;

            }

        },

        // Array initializer

        TC4("class TC4 {\n" +

            "    interface Block<T> {\n" +

            "        void m(T t);\n" +

            "     }\n" +

            "     void test1() {\n" +

            "         Block<?>[] arr1 =  { t -> { }, t -> { } };\n" +

            "     }\n" +

            "}", BSMSpecifier.SPECIFIER1) {

            @Override

            List<String> getExpectedArgValues() {

                List<String> valList = new ArrayList<>();

                valList.add("REF_invokeInterface TC4$Block m (Ljava/lang/Object;)V");

                valList.add("REF_invokeStatic TC4 lambda$0 (Ljava/lang/Object;)V");

                valList.add("(Ljava/lang/Object;)V");

                valList.add("REF_invokeStatic TC4 lambda$1 (Ljava/lang/Object;)V");

                return valList;

            }

        },

        //Lambda expression as a method arg

        TC5("class TC5 {\n"+

            "    interface MapFun<T,R> {  R m( T n); }\n" +

            "    void meth( MapFun<String,Integer> mf ) {\n" +

            "        assert( mf.m(\"four\") == 4);\n" +

            "    }\n"+

            "    void test(Integer i) {\n" +

            "        meth(s -> { Integer len = s.length(); return len; } );\n" +

            "    }\n"+

            "}", BSMSpecifier.SPECIFIER1) {

            @Override

            List<String> getExpectedArgValues() {

                List<String> valList = new ArrayList<>();

                valList.add("REF_invokeInterface TC5$MapFun m (Ljava/lang/Object;)Ljava/lang/Object;");

                valList.add("REF_invokeStatic TC5 lambda$0 (Ljava/lang/String;)Ljava/lang/Integer;");

                valList.add("(Ljava/lang/String;)Ljava/lang/Integer;");

                return valList;

            }

        },

        //Inner class of Lambda expression

        TC6("class TC6 {\n" +

            "    interface MapFun<T, R> {  R m( T n); }\n" +

            "    MapFun<Class<?>,String> cs;\n" +

            "    void test() {\n" +

            "        cs = c -> {\n" +

            "                 class innerClass   {\n" +

            "                    Class<?> icc;\n" +

            "                    innerClass(Class<?> _c) { icc = _c; }\n" +

            "                    String getString() { return icc.toString(); }\n" +

            "                  }\n" +

            "             return new innerClass(c).getString();\n"+

            "             };\n" +

            "    }\n" +

            "}\n", BSMSpecifier.SPECIFIER1) {

            @Override

            List<String> getExpectedArgValues() {

                List<String> valList = new ArrayList<>();

                valList.add("REF_invokeInterface TC6$MapFun m (Ljava/lang/Object;)Ljava/lang/Object;");

                valList.add("REF_invokeSpecial TC6 lambda$0 (Ljava/lang/Class;)Ljava/lang/String;");

                valList.add("(Ljava/lang/Class;)Ljava/lang/String;");

                return valList;

            }

        },

        // Method reference

        TC7("class TC7 {\n" +

            "    static interface SAM {\n" +

            "       void m(Integer i);\n" +

            "    }\n" +

            "    void m(Integer i) {}\n" +

            "       SAM s = this::m;\n" +

            "}\n", BSMSpecifier.SPECIFIER1) {

            @Override

            List<String> getExpectedArgValues() {

                List<String> valList = new ArrayList<>();

                valList.add("REF_invokeInterface TC7$SAM m (Ljava/lang/Integer;)V");

                valList.add("REF_invokeVirtual TC7 m (Ljava/lang/Integer;)V");

                valList.add("(Ljava/lang/Integer;)V");

                return valList;

            }

        },

        // Constructor reference

        TC8("public class TC8 {\n" +

            "    static interface A {Fee<String> m();}\n" +

            "    static class Fee<T> {\n" +

            "        private T t;\n" +

            "        public Fee() {}\n" +

            "    }\n" +

            "    public static void main(String[] args) {\n" +

            "        A a = Fee<String>::new; \n" +

            "    }\n" +

            "}\n", BSMSpecifier.SPECIFIER1) {

            @Override

            List<String> getExpectedArgValues() {

                List<String> valList = new ArrayList<>();

                valList.add("REF_invokeInterface TC8$A m ()LTC8$Fee;");

                valList.add("REF_newInvokeSpecial TC8$Fee <init> ()V");

                valList.add("()LTC8$Fee;");

                return valList;

            }

        },

        // Recursive lambda expression

        TC9("class TC9 {\n" +

            "    interface Recursive<T, R> { T apply(R n); };\n" +

            "    Recursive<Integer,Integer> factorial;\n" +

            "    void test(Integer j) {\n" +

            "        factorial = i -> { return i == 0 ? 1 : i * factorial.apply( i - 1 ); };\n" +

            "    }\n" +

            "}\n", BSMSpecifier.SPECIFIER1) {

            @Override

            List<String> getExpectedArgValues() {

                List<String> valList = new ArrayList<>();

                valList.add("REF_invokeInterface TC9$Recursive apply (Ljava/lang/Object;)Ljava/lang/Object;");

                valList.add("REF_invokeSpecial TC9 lambda$0 (Ljava/lang/Integer;)Ljava/lang/Integer;");

                valList.add("(Ljava/lang/Integer;)Ljava/lang/Integer;");

                return valList;

            }

        },

        //Serializable Lambda

        TC10("import java.io.Serializable;\n" +

              "class TC10 {\n" +

              "    interface Foo { int m(); }\n" +

              "    public static void main(String[] args) {\n" +

              "        Foo f1 = (Foo & Serializable)() -> 3;\n" +

              "    }\n" +

              "}\n", BSMSpecifier.SPECIFIER2) {

            @Override

            List<String> getExpectedArgValues() {

                List<String> valList = new ArrayList<>();

                valList.add("REF_invokeStatic java/lang/invoke/LambdaMetafactory altMetaFactory (Ljava/lang/invoke/MethodHandles$Lookup;Ljava/lang/String;Ljava/lang/invoke/MethodType;[Ljava/lang/Object;)Ljava/lang/invoke/CallSite;");

                valList.add("REF_invokeInterface TC10$Foo m ()I");

                valList.add("REF_invokeStatic TC10 lambda$main$3231c38a$0 ()I");

                valList.add("()I");

                valList.add("1");

                return valList;

            }

        };

        String srcCode;

        BSMSpecifier bsmSpecifier;

        TestCases(String src, BSMSpecifier bsmSpecifier) {

            this.srcCode = src;

            // By default, all test cases will have bootstrap method specifier as Lambda.MetaFactory

            // For serializable lambda test cases, bootstrap method specifier changed to altMetaFactory

            this.bsmSpecifier = bsmSpecifier;

        }

        List<String> getExpectedArgValues() {

            return null;

        }

        void setSrcCode(String src) {

            srcCode = src;

        }

    }

    static class ConstantPoolVisitor implements ConstantPool.Visitor<String, Integer> {

        final List<String> slist;

        final ClassFile cf;

        final ConstantPool cfpool;

        final Map<Integer, String> bsmMap;

        public ConstantPoolVisitor(ClassFile cf, int size) {

            slist = new ArrayList<>(size);

            for (int i = 0 ; i < size; i++) {

                slist.add(null);

            }

            this.cf = cf;

            this.cfpool = cf.constant_pool;

            bsmMap = readBSM();

        }

        public Map<Integer, String> getBSMMap() {

            return Collections.unmodifiableMap(bsmMap);

        }

        public String visit(CPInfo c, int index) {

            return c.accept(this, index);

        }

        private Map<Integer, String> readBSM() {

            BootstrapMethods_attribute bsmAttr =

                    (BootstrapMethods_attribute) cf.getAttribute(Attribute.BootstrapMethods);

            if (bsmAttr != null) {

                Map<Integer, String> out =

                        new HashMap<>(bsmAttr.bootstrap_method_specifiers.length);

                for (BootstrapMethods_attribute.BootstrapMethodSpecifier bsms :

                        bsmAttr.bootstrap_method_specifiers) {

                    int index = bsms.bootstrap_method_ref;

                    try {

                        String value = slist.get(index);

                        if (value == null) {

                            value = visit(cfpool.get(index), index);

                            debugln("[SG]: index " + index);

                            debugln("[SG]: value " + value);

                            slist.set(index, value);

                            out.put(index, value);

                        }

                        for (int idx : bsms.bootstrap_arguments) {

                            value = slist.get(idx);

                            if (value == null) {

                                value = visit(cfpool.get(idx), idx);

                                debugln("[SG]: idx " + idx);

                                debugln("[SG]: value " + value);

                                slist.set(idx, value);

                                out.put(idx, value);

                            }

                        }

                    } catch (InvalidIndex ex) {

                        ex.printStackTrace();

                    }

                }

                return out;

            }

            return new HashMap<>(0);

        }

        @Override

        public String visitClass(CONSTANT_Class_info c, Integer p) {

            String value = slist.get(p);

            if (value == null) {

                try {

                    value = visit(cfpool.get(c.name_index), c.name_index);

                    slist.set(p, value);

                } catch (ConstantPoolException ex) {

                    ex.printStackTrace();

                }

            }

            return value;

        }

        @Override

        public String visitDouble(CONSTANT_Double_info c, Integer p) {

            String value = slist.get(p);

            if (value == null) {

                value = Double.toString(c.value);

                slist.set(p, value);

            }

            return value;

        }

        @Override

        public String visitFieldref(CONSTANT_Fieldref_info c, Integer p) {

        String value = slist.get(p);

            if (value == null) {

                try {

                    value = visit(cfpool.get(c.class_index), c.class_index);

                    value = value.concat(" " + visit(cfpool.get(c.name_and_type_index),

                                         c.name_and_type_index));

                    slist.set(p, value);

                } catch (ConstantPoolException ex) {

                    ex.printStackTrace();

                }

            }

            return value;

        }

        @Override

        public String visitFloat(CONSTANT_Float_info c, Integer p) {

            String value = slist.get(p);

            if (value == null) {

                value = Float.toString(c.value);

                slist.set(p, value);

            }

            return value;

        }

        @Override

        public String visitInteger(CONSTANT_Integer_info cnstnt, Integer p) {

            String value = slist.get(p);

            if (value == null) {

                value = Integer.toString(cnstnt.value);

                slist.set(p, value);

            }

            return value;

        }

        @Override

        public String visitInterfaceMethodref(CONSTANT_InterfaceMethodref_info c,

                                              Integer p) {

            String value = slist.get(p);

            if (value == null) {

                try {

                    value = visit(cfpool.get(c.class_index), c.class_index);

                    value = value.concat(" " +

                                         visit(cfpool.get(c.name_and_type_index),

                                         c.name_and_type_index));

                    slist.set(p, value);

                } catch (ConstantPoolException ex) {

                    ex.printStackTrace();

                }

            }

            return value;

        }

        @Override

        public String visitInvokeDynamic(CONSTANT_InvokeDynamic_info c, Integer p) {