修复单行注释插入到各种地方的格式化bug

CppCXY · CppCXY · commit 27741e48d940 · 2021-12-20T11:32:54.000+08:00
diff --git a/CodeService/src/LuaFormatter.cpp b/CodeService/src/LuaFormatter.cpp
@@ -1306,12 +1306,6 @@ std::shared_ptr<FormatElement> LuaFormatter::FormatTableField(std::shared_ptr<Lu
 				env->Add<TextElement>(child, TextDefineType::TableFieldNameDefine);
 				break;
 			}
-		case LuaAstNodeType::Comment:
-			{
-				env->Add<TextElement>(child);
-				env->Add<KeepElement>(1);
-				break;
-			}
 		default:
 			{
 				DefaultHandle(child, env);
@@ -1325,6 +1319,10 @@ void LuaFormatter::DefaultHandle(std::shared_ptr<LuaAstNode> node, std::shared_p
 {
 	auto childEnv = FormatNode(node);
 	envElement->AddChild(childEnv);
+	if (node->GetType() == LuaAstNodeType::Comment)
+	{
+		envElement->Add<KeepElement>(1);
+	}
 }
 
 std::shared_ptr<FormatElement> LuaFormatter::FormatAlignStatement(LuaAstNode::ChildIterator& it,

Original file line number	Diff line number	Diff line change
`@@ -1306,12 +1306,6 @@ std::shared_ptr<FormatElement> LuaFormatter::FormatTableField(std::shared_ptr<Lu`
`1306`	`1306`	`env->Add<TextElement>(child, TextDefineType::TableFieldNameDefine);`
`1307`	`1307`	`break;`
`1308`	`1308`	`}`
`1309`		`- case LuaAstNodeType::Comment:`
`1310`		`- {`
`1311`		`- env->Add<TextElement>(child);`
`1312`		`- env->Add<KeepElement>(1);`
`1313`		`- break;`
`1314`		`- }`
`1315`	`1309`	`default:`
`1316`	`1310`	`{`
`1317`	`1311`	`DefaultHandle(child, env);`
`@@ -1325,6 +1319,10 @@ void LuaFormatter::DefaultHandle(std::shared_ptr<LuaAstNode> node, std::shared_p`
`1325`	`1319`	`{`
`1326`	`1320`	`auto childEnv = FormatNode(node);`
`1327`	`1321`	`envElement->AddChild(childEnv);`
	`1322`	`+ if (node->GetType() == LuaAstNodeType::Comment)`
	`1323`	`+ {`
	`1324`	`+ envElement->Add<KeepElement>(1);`
	`1325`	`+ }`
`1328`	`1326`	`}`
`1329`	`1327`
`1330`	`1328`	`std::shared_ptr<FormatElement> LuaFormatter::FormatAlignStatement(LuaAstNode::ChildIterator& it,`