fix wrong codes generation for in-expression.v0.17.8

2 files changed, 69 insertions, 14 deletions

diff --git a/spec/inputs/in_expression.yue b/spec/inputs/in_expression.yue
index 4080016..6e923e1 100644
--- a/spec/inputs/in_expression.yue
+++ b/spec/inputs/in_expression.yue
@@ -39,7 +39,14 @@ do
         not_exist = item not in list
         check item in list
         check item in {1, 2, 3}
-        check item in {[1]: 1, [2]: 2, [3]: 3}
+        check item(...) in {[1]: 1, [2]: 2, [3]: 3}
+
+do
+        check -> x in tb
+        check -> x not in tb
+        local x, tb
+        check -> return x in tb
+        check -> x not in tb
 
 nil
 
diff --git a/spec/outputs/in_expression.lua b/spec/outputs/in_expression.lua
index ccc5fd0..ddba69a 100644
--- a/spec/outputs/in_expression.lua
+++ b/spec/outputs/in_expression.lua
@@ -66,15 +66,19 @@ do
 end
 do
         local item = get()
-        local exist = (function()
+        local exist
+        do
                 local _check_0 = list
+                local _find_0 = false
                 for _index_0 = 1, #_check_0 do
-                        if _check_0[_index_0] == item then
-                                return true
+                        local _item_0 = _check_0[_index_0]
+                        if _item_0 == item then
+                                _find_0 = true
+                                break
                         end
                 end
-                return false
-        end)()
+                exist = _find_0
+        end
         check((function()
                 local _check_0 = list
                 for _index_0 = 1, #_check_0 do
@@ -92,14 +96,18 @@ do
                 2,
                 3
         }
-        local not_exist = not (#list > 0 and (function()
+        local not_exist
+        do
+                local _find_0 = false
                 for _index_0 = 1, #list do
-                        if list[_index_0] == item then
-                                return true
+                        local _item_0 = list[_index_0]
+                        if _item_0 == item then
+                                _find_0 = true
+                                break
                         end
                 end
-                return false
-        end)())
+                not_exist = not _find_0
+        end
         check((#list > 0 and (function()
                 for _index_0 = 1, #list do
                         if list[_index_0] == item then
@@ -109,18 +117,58 @@ do
                 return false
         end)()))
         check((1 == item or 2 == item or 3 == item))
-        check((function()
+        check((function(...)
                 local _check_0 = {
                         [1] = 1,
                         [2] = 2,
                         [3] = 3
                 }
+                local _val_0 = item(...)
                 for _index_0 = 1, #_check_0 do
-                        if _check_0[_index_0] == item then
+                        if _check_0[_index_0] == _val_0 then
                                 return true
                         end
                 end
                 return false
-        end)())
+        end)(...))
+end
+do
+        check(function()
+                local _check_0 = tb
+                local _val_0 = x
+                for _index_0 = 1, #_check_0 do
+                        if _check_0[_index_0] == _val_0 then
+                                return true
+                        end
+                end
+                return false
+        end)
+        check(function()
+                local _check_0 = tb
+                local _val_0 = x
+                for _index_0 = 1, #_check_0 do
+                        if _check_0[_index_0] == _val_0 then
+                                return false
+                        end
+                end
+                return true
+        end)
+        local x, tb
+        check(function()
+                for _index_0 = 1, #tb do
+                        if tb[_index_0] == x then
+                                return true
+                        end
+                end
+                return false
+        end)
+        check(function()
+                for _index_0 = 1, #tb do
+                        if tb[_index_0] == x then
+                                return false
+                        end
+                end
+                return true
+        end)
 end
 return nil