11#include " CodeService/Format/Analyzer/TokenAnalyzer.h"
2- #include " LuaParser/Lexer/LuaTokenTypeDetail.h"
3- #include " CodeService/Format/FormatState.h"
42#include " CodeService/Format/Analyzer/SpaceAnalyzer.h"
3+ #include " CodeService/Format/FormatState.h"
4+ #include " LuaParser/Lexer/LuaTokenTypeDetail.h"
55
66TokenAnalyzer::TokenAnalyzer () {
7-
87}
98
109void TokenAnalyzer::Analyze (FormatState &f, const LuaSyntaxTree &t) {
@@ -58,7 +57,7 @@ void TokenAnalyzer::Analyze(FormatState &f, const LuaSyntaxTree &t) {
5857 }
5958}
6059
61- void TokenAnalyzer::Query (FormatState &f, LuaSyntaxNode & syntaxNode, const LuaSyntaxTree &t, FormatResolve &resolve) {
60+ void TokenAnalyzer::Query (FormatState &f, LuaSyntaxNode syntaxNode, const LuaSyntaxTree &t, FormatResolve &resolve) {
6261 if (syntaxNode.IsToken (t)) {
6362 auto it = _tokenStrategies.find (syntaxNode.GetIndex ());
6463 if (it != _tokenStrategies.end ()) {
@@ -73,7 +72,7 @@ void TokenAnalyzer::Mark(LuaSyntaxNode &n, const LuaSyntaxTree &t, TokenStrategy
7372
7473bool TokenAnalyzer::IsRemove (LuaSyntaxNode &n, const LuaSyntaxTree &t) const {
7574 auto it = _tokenStrategies.find (n.GetIndex ());
76- if (it == _tokenStrategies.end ()){
75+ if (it == _tokenStrategies.end ()) {
7776 return false ;
7877 }
7978 return it->second == TokenStrategy::Remove;
@@ -83,8 +82,7 @@ void TokenAnalyzer::TableFieldAddSep(FormatState &f, LuaSyntaxNode &n, const Lua
8382 auto sep = n.GetChildSyntaxNode (LuaSyntaxNodeKind::TableFieldSep, t);
8483 if (!sep.IsNode (t)) {
8584 auto lastToken = n.GetLastToken (t);
86- if (lastToken.GetTokenKind (t) == TK_SHORT_COMMENT
87- || lastToken.GetTokenKind (t) == TK_LONG_COMMENT) {
85+ if (lastToken.GetTokenKind (t) == TK_SHORT_COMMENT || lastToken.GetTokenKind (t) == TK_LONG_COMMENT) {
8886 lastToken = lastToken.GetPrevToken (t);
8987 }
9088 Mark (lastToken, t, TokenStrategy::TableAddSep);
@@ -108,8 +106,7 @@ void TokenAnalyzer::AnalyzeTableField(FormatState &f, LuaSyntaxNode &syntaxNode,
108106
109107 if (f.GetStyle ().trailing_table_separator != TrailingTableSeparator::Keep) {
110108 auto nextToken = syntaxNode.GetNextToken (t);
111- while (nextToken.GetTokenKind (t) == TK_SHORT_COMMENT
112- || nextToken.GetTokenKind (t) == TK_LONG_COMMENT) {
109+ while (nextToken.GetTokenKind (t) == TK_SHORT_COMMENT || nextToken.GetTokenKind (t) == TK_LONG_COMMENT) {
113110 nextToken = nextToken.GetNextToken (t);
114111 }
115112 // the last table field
@@ -147,9 +144,7 @@ void TokenAnalyzer::AnalyzeTableField(FormatState &f, LuaSyntaxNode &syntaxNode,
147144bool IsSingleTableOrStringArg (LuaSyntaxNode &syntaxNode, const LuaSyntaxTree &t) {
148145 auto children = syntaxNode.GetChildren (t);
149146 for (auto child: children) {
150- if (child.GetTokenKind (t) == TK_STRING
151- || child.GetTokenKind (t) == TK_LONG_STRING
152- || child.GetSyntaxKind (t) == LuaSyntaxNodeKind::TableExpression) {
147+ if (child.GetTokenKind (t) == TK_STRING || child.GetTokenKind (t) == TK_LONG_STRING || child.GetSyntaxKind (t) == LuaSyntaxNodeKind::TableExpression) {
153148 return true ;
154149 } else if (
155150 child.GetSyntaxKind (t) == LuaSyntaxNodeKind::ExpressionList) {
@@ -170,9 +165,7 @@ bool IsSingleTableOrStringArg(LuaSyntaxNode &syntaxNode, const LuaSyntaxTree &t)
170165LuaSyntaxNode GetSingleArgStringOrTable (LuaSyntaxNode &syntaxNode, const LuaSyntaxTree &t) {
171166 auto children = syntaxNode.GetChildren (t);
172167 for (auto child: children) {
173- if (child.GetTokenKind (t) == TK_STRING
174- || child.GetTokenKind (t) == TK_LONG_STRING
175- || child.GetSyntaxKind (t) == LuaSyntaxNodeKind::TableExpression) {
168+ if (child.GetTokenKind (t) == TK_STRING || child.GetTokenKind (t) == TK_LONG_STRING || child.GetSyntaxKind (t) == LuaSyntaxNodeKind::TableExpression) {
176169 return syntaxNode;
177170 } else if (child.GetSyntaxKind (t) == LuaSyntaxNodeKind::ExpressionList) {
178171 auto exprs = child.GetChildSyntaxNodes (LuaSyntaxMultiKind::Expression, t);
@@ -182,8 +175,7 @@ LuaSyntaxNode GetSingleArgStringOrTable(LuaSyntaxNode &syntaxNode, const LuaSynt
182175 return expr;
183176 } else if (expr.GetSyntaxKind (t) == LuaSyntaxNodeKind::StringLiteralExpression) {
184177 auto firstToken = expr.GetFirstToken (t);
185- if (firstToken.GetTokenKind (t) == TK_STRING
186- || firstToken.GetTokenKind (t) == TK_LONG_STRING) {
178+ if (firstToken.GetTokenKind (t) == TK_STRING || firstToken.GetTokenKind (t) == TK_LONG_STRING) {
187179 return firstToken;
188180 }
189181 }
@@ -198,7 +190,7 @@ void TokenAnalyzer::AnalyzeCallExpression(FormatState &f, LuaSyntaxNode &syntaxN
198190 switch (f.GetStyle ().call_arg_parentheses ) {
199191 case CallArgParentheses::Remove: {
200192 auto lbrace = syntaxNode.GetChildToken (' (' , t);
201- if (lbrace.IsToken (t)){
193+ if (lbrace.IsToken (t)) {
202194 Mark (lbrace, t, TokenStrategy::Remove);
203195 auto rbrace = syntaxNode.GetChildToken (' )' , t);
204196 Mark (rbrace, t, TokenStrategy::Remove);
@@ -208,10 +200,9 @@ void TokenAnalyzer::AnalyzeCallExpression(FormatState &f, LuaSyntaxNode &syntaxN
208200 }
209201 case CallArgParentheses::RemoveStringOnly: {
210202 auto node = GetSingleArgStringOrTable (syntaxNode, t);
211- if (node.GetTokenKind (t) == TK_STRING
212- || node.GetTokenKind (t) == TK_LONG_STRING) {
203+ if (node.GetTokenKind (t) == TK_STRING || node.GetTokenKind (t) == TK_LONG_STRING) {
213204 auto lbrace = syntaxNode.GetChildToken (' (' , t);
214- if (lbrace.IsToken (t)) {
205+ if (lbrace.IsToken (t)) {
215206 Mark (lbrace, t, TokenStrategy::Remove);
216207 auto rbrace = syntaxNode.GetChildToken (' )' , t);
217208 Mark (rbrace, t, TokenStrategy::Remove);
@@ -224,7 +215,7 @@ void TokenAnalyzer::AnalyzeCallExpression(FormatState &f, LuaSyntaxNode &syntaxN
224215 auto node = GetSingleArgStringOrTable (syntaxNode, t);
225216 if (node.GetSyntaxKind (t) == LuaSyntaxNodeKind::TableExpression) {
226217 auto lbrace = syntaxNode.GetChildToken (' (' , t);
227- if (lbrace.IsToken (t)) {
218+ if (lbrace.IsToken (t)) {
228219 Mark (lbrace, t, TokenStrategy::Remove);
229220 auto rbrace = syntaxNode.GetChildToken (' )' , t);
230221 Mark (rbrace, t, TokenStrategy::Remove);
@@ -251,8 +242,7 @@ void TokenAnalyzer::AnalyzeCallExpression(FormatState &f, LuaSyntaxNode &syntaxN
251242 }
252243 case FunctionSingleArgSpace::OnlyString: {
253244 auto firstToken = syntaxNode.GetFirstToken (t);
254- if (firstToken.GetTokenKind (t) == TK_STRING
255- || firstToken.GetTokenKind (t) == TK_LONG_STRING) {
245+ if (firstToken.GetTokenKind (t) == TK_STRING || firstToken.GetTokenKind (t) == TK_LONG_STRING) {
256246 spaceAnalyzer->SpaceLeft (syntaxNode, t, 1 );
257247 } else {
258248 spaceAnalyzer->SpaceLeft (syntaxNode, t, 0 );
@@ -275,5 +265,3 @@ void TokenAnalyzer::AnalyzeCallExpression(FormatState &f, LuaSyntaxNode &syntaxN
275265 }
276266 }
277267}
278-
279-
0 commit comments