right comment binder

Author: tangzx

gen/com/tang/intellij/lua/parser/LuaParser.java

@@ -106,7 +106,7 @@
         DOUBLE_DIV = "//"
   ]
 
-  hooks("localDef|assignStat|tableField") = [leftBinder="MY_LEFT_COMMENT_BINDER"]
+  hooks("localDef|assignStat|tableField") = [leftBinder="MY_LEFT_COMMENT_BINDER" rightBinder="MY_RIGHT_COMMENT_BINDER"]
   hooks("globalFuncDef|localFuncDef|classMethodDef") = [leftBinder="MY_LEFT_COMMENT_BINDER"]
   hooks("fieldList") = [leftBinder="GREEDY_LEFT_BINDER" rightBinder="GREEDY_RIGHT_BINDER"]
   extends(".*Expr") = expr

src/main/java/com/tang/intellij/lua/lua.bnf

@@ -27,6 +27,7 @@
 
 import static com.tang.intellij.lua.psi.LuaTypes.*;
 
+@SuppressWarnings("unused")
 public class LuaParserUtil extends GeneratedParserUtilBase {
 
     public static WhitespacesAndCommentsBinder MY_LEFT_COMMENT_BINDER = (list, b, tokenTextGetter) -> {
@@ -39,6 +40,21 @@ public class LuaParserUtil extends GeneratedParserUtilBase {
         return list.size();
     };
 
+    public static WhitespacesAndCommentsBinder MY_RIGHT_COMMENT_BINDER = (list, b, tokenTextGetter) -> {
+        for (int i = 0; i < list.size(); i++) {
+            IElementType type = list.get(i);
+            if (type == DOC_COMMENT) {
+                return i + 1;
+            } else {
+                String sequence = String.valueOf(tokenTextGetter.get(i));
+                if (sequence.contains("\n")) {
+                    break;
+                }
+            }
+        }
+        return 0;
+    };
+
     public static boolean repeat(PsiBuilder builder_, int level_, Parser parser, int times) {
         PsiBuilder.Marker marker = builder_.mark();
         boolean r = true;