FIX #27

Author: CppCXY

CodeService/src/LuaFormatter.cpp

@@ -1076,12 +1076,13 @@ std::shared_ptr<FormatElement> LuaFormatter::FormatIfStatement(std::shared_ptr<L
 				}
 
 				env->AddChild(FormatExpression(child, expression));
-				env->Add<KeepBlankElement>(1);
+				env->Add<KeepElement>(1);
 				break;
 			}
 		default:
 			{
 				DefaultHandle(child, env);
+				env->Add<KeepElement>(1);
 				break;
 			}
 		}

Test/test_script/format_text/wait_format/if.lua

@@ -23,9 +23,7 @@ if a then return a --end
 
 end
 
-
-
-
-
-
+if a --
+then
+end
 

Test/test_script/format_text/wait_format_should_be/if.lua

@@ -22,3 +22,7 @@ if a then return a --end
 
 
 end
+
+if a --
+then
+end