Fix reserve comment parsing for table block comment lines

1 files changed, 4 insertions, 3 deletions

diff --git a/src/yuescript/yue_parser.cpp b/src/yuescript/yue_parser.cpp
index e445945..21f19a3 100644
--- a/src/yuescript/yue_parser.cpp
+++ b/src/yuescript/yue_parser.cpp
@@ -909,11 +909,12 @@ YueParser::YueParser() {
                         white >> '}'
                 );
 
-        table_block_inner = Seperator >> key_value_line >> *(+space_break >> key_value_line);
-        TableBlock = +space_break >> advance_match >> ensure(table_block_inner, pop_indent);
+        table_block_inner = Seperator >> key_value_line >> *((+(plain_space >> line_break) >> key_value_line) | (+space_break >> not_(expr("--")) >> key_value_line));
+        TableBlock = ((+(plain_space >> line_break)) | (+space_break >> not_(expr("--")))) >> advance_match >> ensure(table_block_inner, pop_indent);
         TableBlockIndent = ('*' | '-' >> space_one) >> Seperator >> disable_arg_table_block_rule(
                 space >> key_value_list >> -(space >> ',') >>
-                -(+space_break >> advance_match >> space >> ensure(key_value_list >> -(space >> ',') >> *(+space_break >> key_value_line), pop_indent)));
+                -((plain_space >> line_break >> advance_match >> space >> ensure(key_value_list >> -(space >> ',') >> *((+(plain_space >> line_break) >> key_value_line) | (+space_break >> not_(expr("--")) >> key_value_line)), pop_indent)) |
+                  (+space_break >> advance_match >> space >> ensure(key_value_list >> -(space >> ',') >> *(+space_break >> key_value_line), pop_indent))));
 
         ClassMemberList = Seperator >> key_value >> *(space >> ',' >> space >> key_value);
         class_line = check_indent_match >> space >> (ClassMemberList | Statement) >> -(space >> ',');