make vararg assignment work with line decorator.

2 files changed, 67 insertions, 0 deletions

diff --git a/spec/inputs/vararg.yue b/spec/inputs/vararg.yue
index c7b8ebf..466a4a6 100644
--- a/spec/inputs/vararg.yue
+++ b/spec/inputs/vararg.yue
@@ -52,5 +52,27 @@ join = (...) ->
         f_colon f!\func
         f_colon f(...)\func
 
+        _ = ->
+                list = {1, 2, 3, 4, 5}
+                fn = (ok) ->
+                  ok, table.unpack list
+                ok, ... = fn true
+                print ok, ...
+
+                fn_many_args = ->
+                  10, nil, 20, nil, 30
+
+                ... = fn_many_args!
+                print select "#", ...
+                print ...
+
+        do
+                ... = 1, 2 if true
+                print ...
+
+        do
+                a, ... = 1, 2 unless true
+                print a, select '#', ...
+
         nil
 
diff --git a/spec/outputs/vararg.lua b/spec/outputs/vararg.lua
index 56e2011..6cfaccb 100644
--- a/spec/outputs/vararg.lua
+++ b/spec/outputs/vararg.lua
@@ -220,5 +220,50 @@ join = function(...)
                         return _fn_0(_base_0, ...)
                 end
         end)(...))
+        local _
+        _ = function()
+                local list = {
+                        1,
+                        2,
+                        3,
+                        4,
+                        5
+                }
+                local fn
+                fn = function(ok)
+                        return ok, table.unpack(list)
+                end
+                return (function(_arg_0, ...)
+                        local ok = _arg_0
+                        print(ok, ...)
+                        local fn_many_args
+                        fn_many_args = function()
+                                return 10, nil, 20, nil, 30
+                        end
+                        return (function(...)
+                                print(select("#", ...))
+                                return print(...)
+                        end)(fn_many_args())
+                end)(fn(true))
+        end
+        do
+                (function(...)
+                        return print(...)
+                end)((function()
+                        if true then
+                                return 1, 2
+                        end
+                end)())
+        end
+        do
+                (function(_arg_0, ...)
+                        local a = _arg_0
+                        return print(a, select('#', ...))
+                end)((function()
+                        if not true then
+                                return 1, 2
+                        end
+                end)())
+        end
         return nil
 end