fix missing checks and issues related to destrucuring in if branches.

5 files changed, 101 insertions, 7 deletions

diff --git a/spec/outputs/assign.lua b/spec/outputs/assign.lua
index 628f763..c7f2e20 100644
--- a/spec/outputs/assign.lua
+++ b/spec/outputs/assign.lua
@@ -7,14 +7,14 @@ _ = function()
                 local a, b, c
                 a, b, c, d = 1, 2, 3, 4
                 local hello
-                hello[232], (5 + 5)[121], hello, x[99] = 100, 200, 300
+                hello[232], (5 + 5)[121], hello, x[99] = 100, 200, 300, 400
                 joop = 12
         end
 end
 local joop = 2345
 local a, b
 if hello then
-        a, b = "hello"
+        a, b = "hello", nil
 else
         a, b = "nothing", "yeah"
 end
@@ -22,7 +22,7 @@ if hello then
         if yeah then
                 a, b = "one", "two"
         else
-                a, b = "mmhh"
+                a, b = "mmhh", nil
         end
 else
         print("the other")
@@ -42,3 +42,90 @@ do
                 end)())
         end
 end
+return _(function()
+        setmetatable(a, (function()
+                print(123)
+                return { }
+        end)())
+        do
+                local a, b
+                if "a" == c then
+                        do
+                                local _obj_0 = tb
+                                a, b = _obj_0[1], _obj_0[2]
+                        end
+                elseif "b" == c then
+                        do
+                                local _obj_0 = f()
+                                a, b = _obj_0[1], _obj_0[2]
+                        end
+                end
+        end
+        if x then
+                a[#a + 1] = 1
+        else
+                a[#a + 1] = 2
+        end
+        do
+                local a
+                if x then
+                        do
+                                local _exp_0 = y
+                                if 1 == _exp_0 then
+                                        do
+                                                local _obj_0, _obj_1 = f()
+                                                b[#b + 1] = _obj_1
+                                                a = _obj_0
+                                        end
+                                end
+                        end
+                end
+        end
+        do
+                local a
+                local b
+                if x then
+                        do
+                                local _obj_0, _obj_1 = f()
+                                a = _obj_0
+                                b = _obj_1[1]
+                        end
+                else
+                        a = 123
+                        b = tb[1]
+                end
+        end
+        do
+                local a, c
+                if x then
+                        do
+                                local _exp_0 = y
+                                if 1 == _exp_0 then
+                                        do
+                                                local _obj_0, _obj_1, _obj_2, _obj_3 = f()
+                                                b[#b + 1] = _obj_1
+                                                a, c, getmetatable(d).__add = _obj_0, _obj_2, _obj_3
+                                        end
+                                end
+                        end
+                elseif x2 then
+                        if y2 then
+                                do
+                                        local _obj_0, _obj_1, _obj_2, _obj_3 = f1()
+                                        b[#b + 1] = _obj_1
+                                        a, c, getmetatable(d).__add = _obj_0, _obj_2, _obj_3
+                                end
+                        end
+                else
+                        print("hello")
+                        do
+                                print(123)
+                                do
+                                        local _obj_0, _obj_1, _obj_2 = f2()
+                                        b[#b + 1] = _obj_0
+                                        a, c, getmetatable(d).__add = 1, _obj_1, _obj_2
+                                end
+                        end
+                end
+        end
+end)
diff --git a/spec/outputs/cond.lua b/spec/outputs/cond.lua
index 178bf9b..7764439 100644
--- a/spec/outputs/cond.lua
+++ b/spec/outputs/cond.lua
@@ -251,7 +251,7 @@ end
 local a = 12
 local c, b
 if something then
-        a, c, b = "cool"
+        a, c, b = "cool", nil, nil
 end
 local j
 if 1 then
diff --git a/spec/outputs/global.lua b/spec/outputs/global.lua
index ef73823..930ecf6 100644
--- a/spec/outputs/global.lua
+++ b/spec/outputs/global.lua
@@ -1,5 +1,5 @@
 do
-        a, b, c = 223, 343
+        a, b, c = 223, 343, nil
         cool = "dad"
 end
 do
@@ -29,7 +29,7 @@ do
 end
 do
         local d
-        a, b, c, d = "hello"
+        a, b, c, d = "hello", nil, nil, nil
 end
 do
         local What
diff --git a/spec/outputs/metatable.lua b/spec/outputs/metatable.lua
index 2d72e1d..6f5ceed 100644
--- a/spec/outputs/metatable.lua
+++ b/spec/outputs/metatable.lua
@@ -66,7 +66,7 @@ setmetatable(a.b, { })
 x.abc = 123
 setmetatable(func(), mt)
 setmetatable(b.c, mt)
-a, d, e = 1, "abc"
+a, d, e = 1, "abc", nil
 local is_same = getmetatable(a).__index == getmetatable(a).__index
 setmetatable(a, {
         __index = tb
diff --git a/spec/outputs/using.lua b/spec/outputs/using.lua
index 6d0d888..ff23a30 100644
--- a/spec/outputs/using.lua
+++ b/spec/outputs/using.lua
@@ -21,3 +21,10 @@ end
 _ = function()
         local hello = hello or 2
 end
+do
+        local a = { };
+        _(function()
+                local x, y = a[1], a[2]
+        end)
+end
+return nil