add option table_separator_style

Author: CppCXY

CodeFormatCore/include/CodeFormatCore/Config/LuaStyleEnum.h

@@ -31,7 +31,8 @@ enum class ContinuousAlign {
 enum class TableSeparatorStyle {
     None,
     Comma,
-    Semicolon
+    Semicolon,
+    OnlyKVColon
 };
 
 enum class TrailingTableSeparator {

CodeFormatCore/include/CodeFormatCore/Format/Analyzer/FormatStrategy.h

@@ -29,7 +29,8 @@ enum class TokenStrategy {
 
 enum class TokenAddStrategy {
     None,
-    TableAddSep,
+    TableAddColon,
+    TableAddComma
 };
 
 enum class IndentStrategy {

CodeFormatCore/src/Config/LuaStyle.cpp

@@ -53,6 +53,8 @@ void LuaStyle::ParseFromMap(std::map<std::string, std::string, std::less<>> &con
             table_separator_style = TableSeparatorStyle::Comma;
         } else if (style == "semicolon") {
             table_separator_style = TableSeparatorStyle::Semicolon;
+        } else if (style == "only_kv_colon") {
+            table_separator_style = TableSeparatorStyle::OnlyKVColon;
         }
     }
 

CodeFormatCore/src/Format/Analyzer/TokenAnalyzer.cpp

@@ -94,7 +94,20 @@ void TokenAnalyzer::TableFieldAddSep(FormatState &f, LuaSyntaxNode n, const LuaS
         if (lastToken.GetTokenKind(t) == TK_SHORT_COMMENT || lastToken.GetTokenKind(t) == TK_LONG_COMMENT) {
             lastToken = lastToken.GetPrevToken(t);
         }
-        MarkAdd(lastToken, t, TokenAddStrategy::TableAddSep);
+        switch (f.GetStyle().table_separator_style) {
+            case TableSeparatorStyle::Semicolon: {
+                return MarkAdd(lastToken, t, TokenAddStrategy::TableAddColon);
+            }
+            case TableSeparatorStyle::OnlyKVColon: {
+                if (n.GetChildToken('=', t).IsToken(t)) {
+                    return MarkAdd(lastToken, t, TokenAddStrategy::TableAddColon);
+                }
+                // fallthrough
+            }
+            default: {
+                return MarkAdd(lastToken, t, TokenAddStrategy::TableAddComma);
+            }
+        }
     }
 }
 
@@ -111,6 +124,20 @@ void TokenAnalyzer::AnalyzeTableField(FormatState &f, LuaSyntaxNode &syntaxNode,
         if (semicolon.IsToken(t)) {
             Mark(semicolon, t, TokenStrategy::TableSepComma);
         }
+    } else if (f.GetStyle().table_separator_style == TableSeparatorStyle::OnlyKVColon) {
+        if (syntaxNode.GetChildToken('=', t).IsToken(t)) {
+            auto sep = syntaxNode.GetChildSyntaxNode(LuaSyntaxNodeKind::TableFieldSep, t);
+            auto semicolon = sep.GetChildToken(',', t);
+            if (semicolon.IsToken(t)) {
+                Mark(semicolon, t, TokenStrategy::TableSepSemicolon);
+            }
+        } else {
+            auto sep = syntaxNode.GetChildSyntaxNode(LuaSyntaxNodeKind::TableFieldSep, t);
+            auto semicolon = sep.GetChildToken(';', t);
+            if (semicolon.IsToken(t)) {
+                Mark(semicolon, t, TokenStrategy::TableSepComma);
+            }
+        }
     }
 
     if (f.GetStyle().trailing_table_separator != TrailingTableSeparator::Keep) {

CodeFormatCore/src/Format/FormatBuilder.cpp

@@ -145,12 +145,12 @@ void FormatBuilder::DoResolve(LuaSyntaxNode &syntaxNode, const LuaSyntaxTree &t,
         }
 
         switch (resolve.GetTokenAddStrategy()) {
-            case TokenAddStrategy::TableAddSep: {
-                if (_state.GetStyle().table_separator_style == TableSeparatorStyle::Semicolon) {
-                    WriteChar(';');
-                } else {
-                    WriteChar(',');
-                }
+            case TokenAddStrategy::TableAddComma: {
+                WriteChar(',');
+                break;
+            }
+            case TokenAddStrategy::TableAddColon: {
+                WriteChar(';');
                 break;
             }
             default: {