Fixed issue #223.

4 files changed, 227 insertions, 0 deletions

diff --git a/spec/inputs/lists.yue b/spec/inputs/lists.yue
index f754cc1..dd951a5 100644
--- a/spec/inputs/lists.yue
+++ b/spec/inputs/lists.yue
@@ -267,4 +267,14 @@ do
         f = -> globalTB[#][#] = 1
         f1 = -> globalTB[#][# - 1]
 
+do
+        tbA[] = ...tbB
+        a, tb[], b[], c = 1, ...x, 3, 4
+
+        data =
+                a: {1,2,3}
+                b: {4,5,6}
+
+        flat = [...v for k,v in pairs data]
+
 nil
diff --git a/spec/outputs/codes_from_doc.lua b/spec/outputs/codes_from_doc.lua
index 3971bb7..00b3b8b 100644
--- a/spec/outputs/codes_from_doc.lua
+++ b/spec/outputs/codes_from_doc.lua
@@ -147,6 +147,21 @@ print((function()
 end)())
 local tab = { }
 tab[#tab + 1] = "Value"
+local tbA = {
+        1,
+        2,
+        3
+}
+local tbB = {
+        4,
+        5,
+        6
+}
+local _len_0 = #tbA + 1
+for _index_0 = 1, #tbB do
+        local _elm_0 = tbB[_index_0]
+        tbA[_len_0], _len_0 = _elm_0, _len_0 + 1
+end
 local parts = {
         "shoulders",
         "knees"
@@ -1080,6 +1095,33 @@ for _index_0 = 1, #_list_0 do
         _len_0 = _len_0 + 1
 end
 doubled = _accum_0
+local data = {
+        a = {
+                1,
+                2,
+                3
+        },
+        b = {
+                4,
+                5,
+                6
+        }
+}
+local flat
+local _accum_0 = { }
+local _len_0 = 1
+for k, v in pairs(data) do
+        do
+                local _obj_0 = _accum_0[_len_0]
+                local _len_1 = #_obj_0 + 1
+                for _index_0 = 1, #v do
+                        local _elm_0 = v[_index_0]
+                        _obj_0[_len_1], _len_1 = _elm_0, _len_1 + 1
+                end
+        end
+        _len_0 = _len_0 + 1
+end
+flat = _accum_0
 local x_coords = {
         4,
         5,
@@ -2588,6 +2630,21 @@ print((function()
 end)())
 local tab = { }
 tab[#tab + 1] = "Value"
+local tbA = {
+        1,
+        2,
+        3
+}
+local tbB = {
+        4,
+        5,
+        6
+}
+local _len_0 = #tbA + 1
+for _index_0 = 1, #tbB do
+        local _elm_0 = tbB[_index_0]
+        tbA[_len_0], _len_0 = _elm_0, _len_0 + 1
+end
 local parts = {
         "shoulders",
         "knees"
@@ -3551,6 +3608,33 @@ for _index_0 = 1, #_list_0 do
         _len_0 = _len_0 + 1
 end
 doubled = _accum_0
+local data = {
+        a = {
+                1,
+                2,
+                3
+        },
+        b = {
+                4,
+                5,
+                6
+        }
+}
+local flat
+local _accum_0 = { }
+local _len_0 = 1
+for k, v in pairs(data) do
+        do
+                local _obj_0 = _accum_0[_len_0]
+                local _len_1 = #_obj_0 + 1
+                for _index_0 = 1, #v do
+                        local _elm_0 = v[_index_0]
+                        _obj_0[_len_1], _len_1 = _elm_0, _len_1 + 1
+                end
+        end
+        _len_0 = _len_0 + 1
+end
+flat = _accum_0
 local x_coords = {
         4,
         5,
diff --git a/spec/outputs/codes_from_doc_zh.lua b/spec/outputs/codes_from_doc_zh.lua
index 403ad3e..c382bca 100644
--- a/spec/outputs/codes_from_doc_zh.lua
+++ b/spec/outputs/codes_from_doc_zh.lua
@@ -147,6 +147,21 @@ print((function()
 end)())
 local tab = { }
 tab[#tab + 1] = "Value"
+local tbA = {
+        1,
+        2,
+        3
+}
+local tbB = {
+        4,
+        5,
+        6
+}
+local _len_0 = #tbA + 1
+for _index_0 = 1, #tbB do
+        local _elm_0 = tbB[_index_0]
+        tbA[_len_0], _len_0 = _elm_0, _len_0 + 1
+end
 local parts = {
         "shoulders",
         "knees"
@@ -1074,6 +1089,33 @@ for _index_0 = 1, #_list_0 do
         _len_0 = _len_0 + 1
 end
 doubled = _accum_0
+local data = {
+        a = {
+                1,
+                2,
+                3
+        },
+        b = {
+                4,
+                5,
+                6
+        }
+}
+local flat
+local _accum_0 = { }
+local _len_0 = 1
+for k, v in pairs(data) do
+        do
+                local _obj_0 = _accum_0[_len_0]
+                local _len_1 = #_obj_0 + 1
+                for _index_0 = 1, #v do
+                        local _elm_0 = v[_index_0]
+                        _obj_0[_len_1], _len_1 = _elm_0, _len_1 + 1
+                end
+        end
+        _len_0 = _len_0 + 1
+end
+flat = _accum_0
 local x_coords = {
         4,
         5,
@@ -2582,6 +2624,21 @@ print((function()
 end)())
 local tab = { }
 tab[#tab + 1] = "Value"
+local tbA = {
+        1,
+        2,
+        3
+}
+local tbB = {
+        4,
+        5,
+        6
+}
+local _len_0 = #tbA + 1
+for _index_0 = 1, #tbB do
+        local _elm_0 = tbB[_index_0]
+        tbA[_len_0], _len_0 = _elm_0, _len_0 + 1
+end
 local parts = {
         "shoulders",
         "knees"
@@ -3539,6 +3596,33 @@ for _index_0 = 1, #_list_0 do
         _len_0 = _len_0 + 1
 end
 doubled = _accum_0
+local data = {
+        a = {
+                1,
+                2,
+                3
+        },
+        b = {
+                4,
+                5,
+                6
+        }
+}
+local flat
+local _accum_0 = { }
+local _len_0 = 1
+for k, v in pairs(data) do
+        do
+                local _obj_0 = _accum_0[_len_0]
+                local _len_1 = #_obj_0 + 1
+                for _index_0 = 1, #v do
+                        local _elm_0 = v[_index_0]
+                        _obj_0[_len_1], _len_1 = _elm_0, _len_1 + 1
+                end
+        end
+        _len_0 = _len_0 + 1
+end
+flat = _accum_0
 local x_coords = {
         4,
         5,
diff --git a/spec/outputs/lists.lua b/spec/outputs/lists.lua
index 2dd19e5..6115700 100644
--- a/spec/outputs/lists.lua
+++ b/spec/outputs/lists.lua
@@ -857,4 +857,53 @@ do
                 end
         end
 end
+do
+        do
+                local _obj_0 = tbA
+                local _len_0 = #_obj_0 + 1
+                local _list_0 = tbB
+                for _index_0 = 1, #_list_0 do
+                        local _elm_0 = _list_0[_index_0]
+                        _obj_0[_len_0], _len_0 = _elm_0, _len_0 + 1
+                end
+        end
+        a = 1
+        do
+                local _obj_0 = tb
+                local _len_0 = #_obj_0 + 1
+                for _index_0 = 1, #x do
+                        local _elm_0 = x[_index_0]
+                        _obj_0[_len_0], _len_0 = _elm_0, _len_0 + 1
+                end
+        end
+        b[#b + 1] = 3
+        c = 4
+        local data = {
+                a = {
+                        1,
+                        2,
+                        3
+                },
+                b = {
+                        4,
+                        5,
+                        6
+                }
+        }
+        local flat
+        local _accum_0 = { }
+        local _len_0 = 1
+        for k, v in pairs(data) do
+                do
+                        local _obj_0 = _accum_0[_len_0]
+                        local _len_1 = #_obj_0 + 1
+                        for _index_0 = 1, #v do
+                                local _elm_0 = v[_index_0]
+                                _obj_0[_len_1], _len_1 = _elm_0, _len_1 + 1
+                        end
+                end
+                _len_0 = _len_0 + 1
+        end
+        flat = _accum_0
+end
 return nil