From 1b43ecc9a453e0cc7b0a481690375ca35457d54a Mon Sep 17 00:00:00 2001
From: Li Jin <dragon-fly@qq.com>
Date: Mon, 18 Dec 2023 10:28:35 +0800
Subject: fix a format case.

---
 src/yuescript/yue_ast.cpp      | 18 +++++++++++++++---
 src/yuescript/yue_compiler.cpp |  2 +-
 2 files changed, 16 insertions(+), 4 deletions(-)

(limited to 'src')

diff --git a/src/yuescript/yue_ast.cpp b/src/yuescript/yue_ast.cpp
index 69695a0..5dfacc7 100644
--- a/src/yuescript/yue_ast.cpp
+++ b/src/yuescript/yue_ast.cpp
@@ -844,9 +844,21 @@ static bool isInBlockExp(ast_node* node, bool last = false) {
 		}
 		auto value = static_cast<Value_t*>(unaryExp->expos.back());
 		if (auto simpleValue = value->item.as<SimpleValue_t>()) {
-			if (!ast_is<TableLit_t, ConstValue_t, Num_t, VarArg_t,
-					TblComprehension_t, Comprehension_t>(simpleValue->value)) {
-				return true;
+			switch (simpleValue->value->get_id()) {
+				case id<TableLit_t>():
+				case id<ConstValue_t>():
+				case id<Num_t>():
+				case id<VarArg_t>():
+				case id<TblComprehension_t>():
+				case id<Comprehension_t>():
+					return false;
+				case id<FunLit_t>():
+					if (!last) {
+						return true;
+					}
+					return false;
+				default:
+					return true;
 			}
 		} else if (auto chainValue = value->item.as<ChainValue_t>()) {
 			if (ast_is<InvokeArgs_t>(chainValue->items.back())) {
diff --git a/src/yuescript/yue_compiler.cpp b/src/yuescript/yue_compiler.cpp
index 044ee91..2283d85 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.21.3"sv;
+const std::string_view version = "0.21.4"sv;
 const std::string_view extension = "yue"sv;
 
 class CompileError : public std::logic_error {
-- 
cgit v1.2.3-55-g6feb