fix a const list destructuring case.

6 files changed, 29 insertions, 4 deletions

diff --git a/spec/inputs/attrib.yue b/spec/inputs/attrib.yue
index fbff75a..36aca99 100644
--- a/spec/inputs/attrib.yue
+++ b/spec/inputs/attrib.yue
@@ -14,6 +14,9 @@ do
         const a, {b, c}, {d} = f!
 
 do
+        const [a, b] = [2, 3]
+
+do
         close v = if flag
                 func!
         else
diff --git a/spec/outputs/5.1/attrib.lua b/spec/outputs/5.1/attrib.lua
index f782b45..70e7e24 100644
--- a/spec/outputs/5.1/attrib.lua
+++ b/spec/outputs/5.1/attrib.lua
@@ -41,6 +41,16 @@ do
         end
 end
 do
+        local a, b
+        do
+                local _obj_0 = {
+                        2,
+                        3
+                }
+                a, b = _obj_0[1], _obj_0[2]
+        end
+end
+do
         local v = (function()
                 if flag then
                         return func()
diff --git a/spec/outputs/attrib.lua b/spec/outputs/attrib.lua
index 919b7e0..181555f 100644
--- a/spec/outputs/attrib.lua
+++ b/spec/outputs/attrib.lua
@@ -30,6 +30,16 @@ do
         end
 end
 do
+        local a, b
+        do
+                local _obj_0 = {
+                        2,
+                        3
+                }
+                a, b = _obj_0[1], _obj_0[2]
+        end
+end
+do
         local v <close> = (function()
                 if flag then
                         return func()
diff --git a/src/yuescript/yue_ast.h b/src/yuescript/yue_ast.h
index 6b2b7f8..6d90add 100644
--- a/src/yuescript/yue_ast.h
+++ b/src/yuescript/yue_ast.h
@@ -80,6 +80,7 @@ class Macro_t;
 class In_t;
 class NormalDef_t;
 class SpreadListExp_t;
+class Comprehension_t;
 } // namespace yue
 
 AST_LEAF(Num)
@@ -171,7 +172,7 @@ AST_END(CloseAttrib, "close"sv)
 AST_NODE(LocalAttrib)
         ast_sel<true, ConstAttrib_t, CloseAttrib_t> attrib;
         ast_ptr<true, Seperator_t> sep;
-        ast_sel_list<true, Variable_t, SimpleTable_t, TableLit_t> leftList;
+        ast_sel_list<true, Variable_t, SimpleTable_t, TableLit_t, Comprehension_t> leftList;
         ast_ptr<true, Assign_t> assign;
         AST_MEMBER(LocalAttrib, &attrib, &sep, &leftList, &assign)
 AST_END(LocalAttrib, "local_attrib"sv)
diff --git a/src/yuescript/yue_compiler.cpp b/src/yuescript/yue_compiler.cpp
index d43f79b..f163bdf 100644
--- a/src/yuescript/yue_compiler.cpp
+++ b/src/yuescript/yue_compiler.cpp
@@ -75,7 +75,7 @@ static std::unordered_set<std::string> Metamethods = {
         "close"s // Lua 5.4
 };
 
-const std::string_view version = "0.21.5"sv;
+const std::string_view version = "0.21.6"sv;
 const std::string_view extension = "yue"sv;
 
 class CompileError : public std::logic_error {
@@ -9376,7 +9376,8 @@ private:
                                         case id<SimpleTable_t>():
                                                 value->item.set(item);
                                                 break;
-                                        case id<TableLit_t>(): {
+                                        case id<TableLit_t>():
+                                        case id<Comprehension_t>(): {
                                                 auto simpleValue = item->new_ptr<SimpleValue_t>();
                                                 simpleValue->value.set(item);
                                                 value->item.set(simpleValue);
diff --git a/src/yuescript/yue_parser.cpp b/src/yuescript/yue_parser.cpp
index 3557534..4e6105b 100644
--- a/src/yuescript/yue_parser.cpp
+++ b/src/yuescript/yue_parser.cpp
@@ -286,7 +286,7 @@ YueParser::YueParser() {
 
         ConstAttrib = key("const");
         CloseAttrib = key("close");
-        local_const_item = Variable | SimpleTable | TableLit;
+        local_const_item = Variable | SimpleTable | TableLit | Comprehension;
         LocalAttrib = (
                 ConstAttrib >> Seperator >> space >> local_const_item >> *(space >> ',' >> space >> local_const_item) |
                 CloseAttrib >> Seperator >> space >> Variable >> *(space >> ',' >> space >> Variable)