fix table matching syntax in switch statement with list table.v0.20.4

3 files changed, 51 insertions, 3 deletions

diff --git a/spec/inputs/switch.yue b/spec/inputs/switch.yue
index cb1eb31..49d47f3 100644
--- a/spec/inputs/switch.yue
+++ b/spec/inputs/switch.yue
@@ -159,5 +159,11 @@ do
                 when {x: <>: mt}
                         print mt
 
+do
+        switch tb
+                when [ [item,],]
+                        print item
+                when [a = 1, b = "abc"]
+                        print a, b
 nil
 
diff --git a/spec/outputs/switch.lua b/spec/outputs/switch.lua
index 1ace1e3..bbdf794 100644
--- a/spec/outputs/switch.lua
+++ b/spec/outputs/switch.lua
@@ -398,4 +398,39 @@ do
                 end
         end
 end
+do
+        do
+                local _exp_0 = tb
+                local _type_0 = type(_exp_0)
+                local _tab_0 = "table" == _type_0 or "userdata" == _type_0
+                local _match_0 = false
+                if _tab_0 then
+                        local item
+                        do
+                                local _obj_0 = _exp_0[1]
+                                local _type_1 = type(_obj_0)
+                                if "table" == _type_1 or "userdata" == _type_1 then
+                                        item = _obj_0[1]
+                                end
+                        end
+                        if item ~= nil then
+                                _match_0 = true
+                                print(item)
+                        end
+                end
+                if not _match_0 then
+                        if _tab_0 then
+                                local a = _exp_0[1]
+                                local b = _exp_0[2]
+                                if a == nil then
+                                        a = 1
+                                end
+                                if b == nil then
+                                        b = "abc"
+                                end
+                                print(a, b)
+                        end
+                end
+        end
+end
 return nil
diff --git a/src/yuescript/yue_compiler.cpp b/src/yuescript/yue_compiler.cpp
index 648dab2..1733967 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.20.3"sv;
+const std::string_view version = "0.20.4"sv;
 const std::string_view extension = "yue"sv;
 
 class CompileError : public std::logic_error {
@@ -9155,8 +9155,15 @@ private:
                 for (auto branch_ : branches) {
                         auto branch = static_cast<SwitchCase_t*>(branch_);
                         auto valueList = static_cast<SwitchList_t*>(branch->condition.get());
-                        if (auto value = singleValueFrom(valueList);
-                                value && (value->item.is<SimpleTable_t>() || value->get_by_path<SimpleValue_t, TableLit_t>())) {
+                        bool tableMatching = false;
+                        if (auto value = singleValueFrom(valueList)) {
+                                if (value->item.is<SimpleTable_t>()) {
+                                        tableMatching = true;
+                                } else if (auto sVal = value->item.as<SimpleValue_t>()){
+                                        tableMatching = ast_is<TableLit_t, Comprehension_t>(sVal->value);
+                                }
+                        }
+                        if (tableMatching) {
                                 if (!firstBranch) {
                                         temp.push_back(indent() + "else"s + nll(branch));
                                         pushScope();