1 //===--- UnwrappedLineParser.cpp - Format C++ code ------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 /// 9 /// \file 10 /// This file contains the implementation of the UnwrappedLineParser, 11 /// which turns a stream of tokens into UnwrappedLines. 12 /// 13 //===----------------------------------------------------------------------===// 14 15 #include "UnwrappedLineParser.h" 16 #include "FormatToken.h" 17 #include "llvm/ADT/STLExtras.h" 18 #include "llvm/Support/Debug.h" 19 #include "llvm/Support/raw_ostream.h" 20 21 #include <algorithm> 22 23 #define DEBUG_TYPE "format-parser" 24 25 namespace clang { 26 namespace format { 27 28 class FormatTokenSource { 29 public: 30 virtual ~FormatTokenSource() {} 31 virtual FormatToken *getNextToken() = 0; 32 33 virtual unsigned getPosition() = 0; 34 virtual FormatToken *setPosition(unsigned Position) = 0; 35 }; 36 37 namespace { 38 39 class ScopedDeclarationState { 40 public: 41 ScopedDeclarationState(UnwrappedLine &Line, std::vector<bool> &Stack, 42 bool MustBeDeclaration) 43 : Line(Line), Stack(Stack) { 44 Line.MustBeDeclaration = MustBeDeclaration; 45 Stack.push_back(MustBeDeclaration); 46 } 47 ~ScopedDeclarationState() { 48 Stack.pop_back(); 49 if (!Stack.empty()) 50 Line.MustBeDeclaration = Stack.back(); 51 else 52 Line.MustBeDeclaration = true; 53 } 54 55 private: 56 UnwrappedLine &Line; 57 std::vector<bool> &Stack; 58 }; 59 60 static bool isLineComment(const FormatToken &FormatTok) { 61 return FormatTok.is(tok::comment) && !FormatTok.TokenText.startswith("/*"); 62 } 63 64 // Checks if \p FormatTok is a line comment that continues the line comment 65 // \p Previous. The original column of \p MinColumnToken is used to determine 66 // whether \p FormatTok is indented enough to the right to continue \p Previous. 67 static bool continuesLineComment(const FormatToken &FormatTok, 68 const FormatToken *Previous, 69 const FormatToken *MinColumnToken) { 70 if (!Previous || !MinColumnToken) 71 return false; 72 unsigned MinContinueColumn = 73 MinColumnToken->OriginalColumn + (isLineComment(*MinColumnToken) ? 0 : 1); 74 return isLineComment(FormatTok) && FormatTok.NewlinesBefore == 1 && 75 isLineComment(*Previous) && 76 FormatTok.OriginalColumn >= MinContinueColumn; 77 } 78 79 class ScopedMacroState : public FormatTokenSource { 80 public: 81 ScopedMacroState(UnwrappedLine &Line, FormatTokenSource *&TokenSource, 82 FormatToken *&ResetToken) 83 : Line(Line), TokenSource(TokenSource), ResetToken(ResetToken), 84 PreviousLineLevel(Line.Level), PreviousTokenSource(TokenSource), 85 Token(nullptr), PreviousToken(nullptr) { 86 FakeEOF.Tok.startToken(); 87 FakeEOF.Tok.setKind(tok::eof); 88 TokenSource = this; 89 Line.Level = 0; 90 Line.InPPDirective = true; 91 } 92 93 ~ScopedMacroState() override { 94 TokenSource = PreviousTokenSource; 95 ResetToken = Token; 96 Line.InPPDirective = false; 97 Line.Level = PreviousLineLevel; 98 } 99 100 FormatToken *getNextToken() override { 101 // The \c UnwrappedLineParser guards against this by never calling 102 // \c getNextToken() after it has encountered the first eof token. 103 assert(!eof()); 104 PreviousToken = Token; 105 Token = PreviousTokenSource->getNextToken(); 106 if (eof()) 107 return &FakeEOF; 108 return Token; 109 } 110 111 unsigned getPosition() override { return PreviousTokenSource->getPosition(); } 112 113 FormatToken *setPosition(unsigned Position) override { 114 PreviousToken = nullptr; 115 Token = PreviousTokenSource->setPosition(Position); 116 return Token; 117 } 118 119 private: 120 bool eof() { 121 return Token && Token->HasUnescapedNewline && 122 !continuesLineComment(*Token, PreviousToken, 123 /*MinColumnToken=*/PreviousToken); 124 } 125 126 FormatToken FakeEOF; 127 UnwrappedLine &Line; 128 FormatTokenSource *&TokenSource; 129 FormatToken *&ResetToken; 130 unsigned PreviousLineLevel; 131 FormatTokenSource *PreviousTokenSource; 132 133 FormatToken *Token; 134 FormatToken *PreviousToken; 135 }; 136 137 } // end anonymous namespace 138 139 class ScopedLineState { 140 public: 141 ScopedLineState(UnwrappedLineParser &Parser, 142 bool SwitchToPreprocessorLines = false) 143 : Parser(Parser), OriginalLines(Parser.CurrentLines) { 144 if (SwitchToPreprocessorLines) 145 Parser.CurrentLines = &Parser.PreprocessorDirectives; 146 else if (!Parser.Line->Tokens.empty()) 147 Parser.CurrentLines = &Parser.Line->Tokens.back().Children; 148 PreBlockLine = std::move(Parser.Line); 149 Parser.Line = std::make_unique<UnwrappedLine>(); 150 Parser.Line->Level = PreBlockLine->Level; 151 Parser.Line->InPPDirective = PreBlockLine->InPPDirective; 152 } 153 154 ~ScopedLineState() { 155 if (!Parser.Line->Tokens.empty()) { 156 Parser.addUnwrappedLine(); 157 } 158 assert(Parser.Line->Tokens.empty()); 159 Parser.Line = std::move(PreBlockLine); 160 if (Parser.CurrentLines == &Parser.PreprocessorDirectives) 161 Parser.MustBreakBeforeNextToken = true; 162 Parser.CurrentLines = OriginalLines; 163 } 164 165 private: 166 UnwrappedLineParser &Parser; 167 168 std::unique_ptr<UnwrappedLine> PreBlockLine; 169 SmallVectorImpl<UnwrappedLine> *OriginalLines; 170 }; 171 172 class CompoundStatementIndenter { 173 public: 174 CompoundStatementIndenter(UnwrappedLineParser *Parser, 175 const FormatStyle &Style, unsigned &LineLevel) 176 : CompoundStatementIndenter(Parser, LineLevel, 177 Style.BraceWrapping.AfterControlStatement, 178 Style.BraceWrapping.IndentBraces) {} 179 CompoundStatementIndenter(UnwrappedLineParser *Parser, unsigned &LineLevel, 180 bool WrapBrace, bool IndentBrace) 181 : LineLevel(LineLevel), OldLineLevel(LineLevel) { 182 if (WrapBrace) 183 Parser->addUnwrappedLine(); 184 if (IndentBrace) 185 ++LineLevel; 186 } 187 ~CompoundStatementIndenter() { LineLevel = OldLineLevel; } 188 189 private: 190 unsigned &LineLevel; 191 unsigned OldLineLevel; 192 }; 193 194 namespace { 195 196 class IndexedTokenSource : public FormatTokenSource { 197 public: 198 IndexedTokenSource(ArrayRef<FormatToken *> Tokens) 199 : Tokens(Tokens), Position(-1) {} 200 201 FormatToken *getNextToken() override { 202 ++Position; 203 return Tokens[Position]; 204 } 205 206 unsigned getPosition() override { 207 assert(Position >= 0); 208 return Position; 209 } 210 211 FormatToken *setPosition(unsigned P) override { 212 Position = P; 213 return Tokens[Position]; 214 } 215 216 void reset() { Position = -1; } 217 218 private: 219 ArrayRef<FormatToken *> Tokens; 220 int Position; 221 }; 222 223 } // end anonymous namespace 224 225 UnwrappedLineParser::UnwrappedLineParser(const FormatStyle &Style, 226 const AdditionalKeywords &Keywords, 227 unsigned FirstStartColumn, 228 ArrayRef<FormatToken *> Tokens, 229 UnwrappedLineConsumer &Callback) 230 : Line(new UnwrappedLine), MustBreakBeforeNextToken(false), 231 CurrentLines(&Lines), Style(Style), Keywords(Keywords), 232 CommentPragmasRegex(Style.CommentPragmas), Tokens(nullptr), 233 Callback(Callback), AllTokens(Tokens), PPBranchLevel(-1), 234 IncludeGuard(Style.IndentPPDirectives == FormatStyle::PPDIS_None 235 ? IG_Rejected 236 : IG_Inited), 237 IncludeGuardToken(nullptr), FirstStartColumn(FirstStartColumn) {} 238 239 void UnwrappedLineParser::reset() { 240 PPBranchLevel = -1; 241 IncludeGuard = Style.IndentPPDirectives == FormatStyle::PPDIS_None 242 ? IG_Rejected 243 : IG_Inited; 244 IncludeGuardToken = nullptr; 245 Line.reset(new UnwrappedLine); 246 CommentsBeforeNextToken.clear(); 247 FormatTok = nullptr; 248 MustBreakBeforeNextToken = false; 249 PreprocessorDirectives.clear(); 250 CurrentLines = &Lines; 251 DeclarationScopeStack.clear(); 252 PPStack.clear(); 253 Line->FirstStartColumn = FirstStartColumn; 254 } 255 256 void UnwrappedLineParser::parse() { 257 IndexedTokenSource TokenSource(AllTokens); 258 Line->FirstStartColumn = FirstStartColumn; 259 do { 260 LLVM_DEBUG(llvm::dbgs() << "----\n"); 261 reset(); 262 Tokens = &TokenSource; 263 TokenSource.reset(); 264 265 readToken(); 266 parseFile(); 267 268 // If we found an include guard then all preprocessor directives (other than 269 // the guard) are over-indented by one. 270 if (IncludeGuard == IG_Found) 271 for (auto &Line : Lines) 272 if (Line.InPPDirective && Line.Level > 0) 273 --Line.Level; 274 275 // Create line with eof token. 276 pushToken(FormatTok); 277 addUnwrappedLine(); 278 279 for (SmallVectorImpl<UnwrappedLine>::iterator I = Lines.begin(), 280 E = Lines.end(); 281 I != E; ++I) { 282 Callback.consumeUnwrappedLine(*I); 283 } 284 Callback.finishRun(); 285 Lines.clear(); 286 while (!PPLevelBranchIndex.empty() && 287 PPLevelBranchIndex.back() + 1 >= PPLevelBranchCount.back()) { 288 PPLevelBranchIndex.resize(PPLevelBranchIndex.size() - 1); 289 PPLevelBranchCount.resize(PPLevelBranchCount.size() - 1); 290 } 291 if (!PPLevelBranchIndex.empty()) { 292 ++PPLevelBranchIndex.back(); 293 assert(PPLevelBranchIndex.size() == PPLevelBranchCount.size()); 294 assert(PPLevelBranchIndex.back() <= PPLevelBranchCount.back()); 295 } 296 } while (!PPLevelBranchIndex.empty()); 297 } 298 299 void UnwrappedLineParser::parseFile() { 300 // The top-level context in a file always has declarations, except for pre- 301 // processor directives and JavaScript files. 302 bool MustBeDeclaration = 303 !Line->InPPDirective && Style.Language != FormatStyle::LK_JavaScript; 304 ScopedDeclarationState DeclarationState(*Line, DeclarationScopeStack, 305 MustBeDeclaration); 306 if (Style.Language == FormatStyle::LK_TextProto) 307 parseBracedList(); 308 else 309 parseLevel(/*HasOpeningBrace=*/false); 310 // Make sure to format the remaining tokens. 311 // 312 // LK_TextProto is special since its top-level is parsed as the body of a 313 // braced list, which does not necessarily have natural line separators such 314 // as a semicolon. Comments after the last entry that have been determined to 315 // not belong to that line, as in: 316 // key: value 317 // // endfile comment 318 // do not have a chance to be put on a line of their own until this point. 319 // Here we add this newline before end-of-file comments. 320 if (Style.Language == FormatStyle::LK_TextProto && 321 !CommentsBeforeNextToken.empty()) 322 addUnwrappedLine(); 323 flushComments(true); 324 addUnwrappedLine(); 325 } 326 327 void UnwrappedLineParser::parseCSharpGenericTypeConstraint() { 328 do { 329 switch (FormatTok->Tok.getKind()) { 330 case tok::l_brace: 331 return; 332 default: 333 if (FormatTok->is(Keywords.kw_where)) { 334 addUnwrappedLine(); 335 nextToken(); 336 parseCSharpGenericTypeConstraint(); 337 break; 338 } 339 nextToken(); 340 break; 341 } 342 } while (!eof()); 343 } 344 345 void UnwrappedLineParser::parseCSharpAttribute() { 346 int UnpairedSquareBrackets = 1; 347 do { 348 switch (FormatTok->Tok.getKind()) { 349 case tok::r_square: 350 nextToken(); 351 --UnpairedSquareBrackets; 352 if (UnpairedSquareBrackets == 0) { 353 addUnwrappedLine(); 354 return; 355 } 356 break; 357 case tok::l_square: 358 ++UnpairedSquareBrackets; 359 nextToken(); 360 break; 361 default: 362 nextToken(); 363 break; 364 } 365 } while (!eof()); 366 } 367 368 void UnwrappedLineParser::parseLevel(bool HasOpeningBrace) { 369 bool SwitchLabelEncountered = false; 370 do { 371 tok::TokenKind kind = FormatTok->Tok.getKind(); 372 if (FormatTok->Type == TT_MacroBlockBegin) { 373 kind = tok::l_brace; 374 } else if (FormatTok->Type == TT_MacroBlockEnd) { 375 kind = tok::r_brace; 376 } 377 378 switch (kind) { 379 case tok::comment: 380 nextToken(); 381 addUnwrappedLine(); 382 break; 383 case tok::l_brace: 384 // FIXME: Add parameter whether this can happen - if this happens, we must 385 // be in a non-declaration context. 386 if (!FormatTok->is(TT_MacroBlockBegin) && tryToParseBracedList()) 387 continue; 388 parseBlock(/*MustBeDeclaration=*/false); 389 addUnwrappedLine(); 390 break; 391 case tok::r_brace: 392 if (HasOpeningBrace) 393 return; 394 nextToken(); 395 addUnwrappedLine(); 396 break; 397 case tok::kw_default: { 398 unsigned StoredPosition = Tokens->getPosition(); 399 FormatToken *Next; 400 do { 401 Next = Tokens->getNextToken(); 402 } while (Next && Next->is(tok::comment)); 403 FormatTok = Tokens->setPosition(StoredPosition); 404 if (Next && Next->isNot(tok::colon)) { 405 // default not followed by ':' is not a case label; treat it like 406 // an identifier. 407 parseStructuralElement(); 408 break; 409 } 410 // Else, if it is 'default:', fall through to the case handling. 411 LLVM_FALLTHROUGH; 412 } 413 case tok::kw_case: 414 if (Style.Language == FormatStyle::LK_JavaScript && 415 Line->MustBeDeclaration) { 416 // A 'case: string' style field declaration. 417 parseStructuralElement(); 418 break; 419 } 420 if (!SwitchLabelEncountered && 421 (Style.IndentCaseLabels || (Line->InPPDirective && Line->Level == 1))) 422 ++Line->Level; 423 SwitchLabelEncountered = true; 424 parseStructuralElement(); 425 break; 426 case tok::l_square: 427 if (Style.isCSharp()) { 428 nextToken(); 429 parseCSharpAttribute(); 430 break; 431 } 432 LLVM_FALLTHROUGH; 433 default: 434 parseStructuralElement(); 435 break; 436 } 437 } while (!eof()); 438 } 439 440 void UnwrappedLineParser::calculateBraceTypes(bool ExpectClassBody) { 441 // We'll parse forward through the tokens until we hit 442 // a closing brace or eof - note that getNextToken() will 443 // parse macros, so this will magically work inside macro 444 // definitions, too. 445 unsigned StoredPosition = Tokens->getPosition(); 446 FormatToken *Tok = FormatTok; 447 const FormatToken *PrevTok = Tok->Previous; 448 // Keep a stack of positions of lbrace tokens. We will 449 // update information about whether an lbrace starts a 450 // braced init list or a different block during the loop. 451 SmallVector<FormatToken *, 8> LBraceStack; 452 assert(Tok->Tok.is(tok::l_brace)); 453 do { 454 // Get next non-comment token. 455 FormatToken *NextTok; 456 unsigned ReadTokens = 0; 457 do { 458 NextTok = Tokens->getNextToken(); 459 ++ReadTokens; 460 } while (NextTok->is(tok::comment)); 461 462 switch (Tok->Tok.getKind()) { 463 case tok::l_brace: 464 if (Style.Language == FormatStyle::LK_JavaScript && PrevTok) { 465 if (PrevTok->isOneOf(tok::colon, tok::less)) 466 // A ':' indicates this code is in a type, or a braced list 467 // following a label in an object literal ({a: {b: 1}}). 468 // A '<' could be an object used in a comparison, but that is nonsense 469 // code (can never return true), so more likely it is a generic type 470 // argument (`X<{a: string; b: number}>`). 471 // The code below could be confused by semicolons between the 472 // individual members in a type member list, which would normally 473 // trigger BK_Block. In both cases, this must be parsed as an inline 474 // braced init. 475 Tok->BlockKind = BK_BracedInit; 476 else if (PrevTok->is(tok::r_paren)) 477 // `) { }` can only occur in function or method declarations in JS. 478 Tok->BlockKind = BK_Block; 479 } else { 480 Tok->BlockKind = BK_Unknown; 481 } 482 LBraceStack.push_back(Tok); 483 break; 484 case tok::r_brace: 485 if (LBraceStack.empty()) 486 break; 487 if (LBraceStack.back()->BlockKind == BK_Unknown) { 488 bool ProbablyBracedList = false; 489 if (Style.Language == FormatStyle::LK_Proto) { 490 ProbablyBracedList = NextTok->isOneOf(tok::comma, tok::r_square); 491 } else { 492 // Using OriginalColumn to distinguish between ObjC methods and 493 // binary operators is a bit hacky. 494 bool NextIsObjCMethod = NextTok->isOneOf(tok::plus, tok::minus) && 495 NextTok->OriginalColumn == 0; 496 497 // If there is a comma, semicolon or right paren after the closing 498 // brace, we assume this is a braced initializer list. Note that 499 // regardless how we mark inner braces here, we will overwrite the 500 // BlockKind later if we parse a braced list (where all blocks 501 // inside are by default braced lists), or when we explicitly detect 502 // blocks (for example while parsing lambdas). 503 // FIXME: Some of these do not apply to JS, e.g. "} {" can never be a 504 // braced list in JS. 505 ProbablyBracedList = 506 (Style.Language == FormatStyle::LK_JavaScript && 507 NextTok->isOneOf(Keywords.kw_of, Keywords.kw_in, 508 Keywords.kw_as)) || 509 (Style.isCpp() && NextTok->is(tok::l_paren)) || 510 NextTok->isOneOf(tok::comma, tok::period, tok::colon, 511 tok::r_paren, tok::r_square, tok::l_brace, 512 tok::ellipsis) || 513 (NextTok->is(tok::identifier) && 514 !PrevTok->isOneOf(tok::semi, tok::r_brace, tok::l_brace)) || 515 (NextTok->is(tok::semi) && 516 (!ExpectClassBody || LBraceStack.size() != 1)) || 517 (NextTok->isBinaryOperator() && !NextIsObjCMethod); 518 if (!Style.isCSharp() && NextTok->is(tok::l_square)) { 519 // We can have an array subscript after a braced init 520 // list, but C++11 attributes are expected after blocks. 521 NextTok = Tokens->getNextToken(); 522 ++ReadTokens; 523 ProbablyBracedList = NextTok->isNot(tok::l_square); 524 } 525 } 526 if (ProbablyBracedList) { 527 Tok->BlockKind = BK_BracedInit; 528 LBraceStack.back()->BlockKind = BK_BracedInit; 529 } else { 530 Tok->BlockKind = BK_Block; 531 LBraceStack.back()->BlockKind = BK_Block; 532 } 533 } 534 LBraceStack.pop_back(); 535 break; 536 case tok::identifier: 537 if (!Tok->is(TT_StatementMacro)) 538 break; 539 LLVM_FALLTHROUGH; 540 case tok::at: 541 case tok::semi: 542 case tok::kw_if: 543 case tok::kw_while: 544 case tok::kw_for: 545 case tok::kw_switch: 546 case tok::kw_try: 547 case tok::kw___try: 548 if (!LBraceStack.empty() && LBraceStack.back()->BlockKind == BK_Unknown) 549 LBraceStack.back()->BlockKind = BK_Block; 550 break; 551 default: 552 break; 553 } 554 PrevTok = Tok; 555 Tok = NextTok; 556 } while (Tok->Tok.isNot(tok::eof) && !LBraceStack.empty()); 557 558 // Assume other blocks for all unclosed opening braces. 559 for (unsigned i = 0, e = LBraceStack.size(); i != e; ++i) { 560 if (LBraceStack[i]->BlockKind == BK_Unknown) 561 LBraceStack[i]->BlockKind = BK_Block; 562 } 563 564 FormatTok = Tokens->setPosition(StoredPosition); 565 } 566 567 template <class T> 568 static inline void hash_combine(std::size_t &seed, const T &v) { 569 std::hash<T> hasher; 570 seed ^= hasher(v) + 0x9e3779b9 + (seed << 6) + (seed >> 2); 571 } 572 573 size_t UnwrappedLineParser::computePPHash() const { 574 size_t h = 0; 575 for (const auto &i : PPStack) { 576 hash_combine(h, size_t(i.Kind)); 577 hash_combine(h, i.Line); 578 } 579 return h; 580 } 581 582 void UnwrappedLineParser::parseBlock(bool MustBeDeclaration, bool AddLevel, 583 bool MunchSemi) { 584 assert(FormatTok->isOneOf(tok::l_brace, TT_MacroBlockBegin) && 585 "'{' or macro block token expected"); 586 const bool MacroBlock = FormatTok->is(TT_MacroBlockBegin); 587 FormatTok->BlockKind = BK_Block; 588 589 size_t PPStartHash = computePPHash(); 590 591 unsigned InitialLevel = Line->Level; 592 nextToken(/*LevelDifference=*/AddLevel ? 1 : 0); 593 594 if (MacroBlock && FormatTok->is(tok::l_paren)) 595 parseParens(); 596 597 size_t NbPreprocessorDirectives = 598 CurrentLines == &Lines ? PreprocessorDirectives.size() : 0; 599 addUnwrappedLine(); 600 size_t OpeningLineIndex = 601 CurrentLines->empty() 602 ? (UnwrappedLine::kInvalidIndex) 603 : (CurrentLines->size() - 1 - NbPreprocessorDirectives); 604 605 ScopedDeclarationState DeclarationState(*Line, DeclarationScopeStack, 606 MustBeDeclaration); 607 if (AddLevel) 608 ++Line->Level; 609 parseLevel(/*HasOpeningBrace=*/true); 610 611 if (eof()) 612 return; 613 614 if (MacroBlock ? !FormatTok->is(TT_MacroBlockEnd) 615 : !FormatTok->is(tok::r_brace)) { 616 Line->Level = InitialLevel; 617 FormatTok->BlockKind = BK_Block; 618 return; 619 } 620 621 size_t PPEndHash = computePPHash(); 622 623 // Munch the closing brace. 624 nextToken(/*LevelDifference=*/AddLevel ? -1 : 0); 625 626 if (MacroBlock && FormatTok->is(tok::l_paren)) 627 parseParens(); 628 629 if (MunchSemi && FormatTok->Tok.is(tok::semi)) 630 nextToken(); 631 Line->Level = InitialLevel; 632 633 if (PPStartHash == PPEndHash) { 634 Line->MatchingOpeningBlockLineIndex = OpeningLineIndex; 635 if (OpeningLineIndex != UnwrappedLine::kInvalidIndex) { 636 // Update the opening line to add the forward reference as well 637 (*CurrentLines)[OpeningLineIndex].MatchingClosingBlockLineIndex = 638 CurrentLines->size() - 1; 639 } 640 } 641 } 642 643 static bool isGoogScope(const UnwrappedLine &Line) { 644 // FIXME: Closure-library specific stuff should not be hard-coded but be 645 // configurable. 646 if (Line.Tokens.size() < 4) 647 return false; 648 auto I = Line.Tokens.begin(); 649 if (I->Tok->TokenText != "goog") 650 return false; 651 ++I; 652 if (I->Tok->isNot(tok::period)) 653 return false; 654 ++I; 655 if (I->Tok->TokenText != "scope") 656 return false; 657 ++I; 658 return I->Tok->is(tok::l_paren); 659 } 660 661 static bool isIIFE(const UnwrappedLine &Line, 662 const AdditionalKeywords &Keywords) { 663 // Look for the start of an immediately invoked anonymous function. 664 // https://en.wikipedia.org/wiki/Immediately-invoked_function_expression 665 // This is commonly done in JavaScript to create a new, anonymous scope. 666 // Example: (function() { ... })() 667 if (Line.Tokens.size() < 3) 668 return false; 669 auto I = Line.Tokens.begin(); 670 if (I->Tok->isNot(tok::l_paren)) 671 return false; 672 ++I; 673 if (I->Tok->isNot(Keywords.kw_function)) 674 return false; 675 ++I; 676 return I->Tok->is(tok::l_paren); 677 } 678 679 static bool ShouldBreakBeforeBrace(const FormatStyle &Style, 680 const FormatToken &InitialToken) { 681 if (InitialToken.isOneOf(tok::kw_namespace, TT_NamespaceMacro)) 682 return Style.BraceWrapping.AfterNamespace; 683 if (InitialToken.is(tok::kw_class)) 684 return Style.BraceWrapping.AfterClass; 685 if (InitialToken.is(tok::kw_union)) 686 return Style.BraceWrapping.AfterUnion; 687 if (InitialToken.is(tok::kw_struct)) 688 return Style.BraceWrapping.AfterStruct; 689 return false; 690 } 691 692 void UnwrappedLineParser::parseChildBlock() { 693 FormatTok->BlockKind = BK_Block; 694 nextToken(); 695 { 696 bool SkipIndent = (Style.Language == FormatStyle::LK_JavaScript && 697 (isGoogScope(*Line) || isIIFE(*Line, Keywords))); 698 ScopedLineState LineState(*this); 699 ScopedDeclarationState DeclarationState(*Line, DeclarationScopeStack, 700 /*MustBeDeclaration=*/false); 701 Line->Level += SkipIndent ? 0 : 1; 702 parseLevel(/*HasOpeningBrace=*/true); 703 flushComments(isOnNewLine(*FormatTok)); 704 Line->Level -= SkipIndent ? 0 : 1; 705 } 706 nextToken(); 707 } 708 709 void UnwrappedLineParser::parsePPDirective() { 710 assert(FormatTok->Tok.is(tok::hash) && "'#' expected"); 711 ScopedMacroState MacroState(*Line, Tokens, FormatTok); 712 713 nextToken(); 714 715 if (!FormatTok->Tok.getIdentifierInfo()) { 716 parsePPUnknown(); 717 return; 718 } 719 720 switch (FormatTok->Tok.getIdentifierInfo()->getPPKeywordID()) { 721 case tok::pp_define: 722 parsePPDefine(); 723 return; 724 case tok::pp_if: 725 parsePPIf(/*IfDef=*/false); 726 break; 727 case tok::pp_ifdef: 728 case tok::pp_ifndef: 729 parsePPIf(/*IfDef=*/true); 730 break; 731 case tok::pp_else: 732 parsePPElse(); 733 break; 734 case tok::pp_elif: 735 parsePPElIf(); 736 break; 737 case tok::pp_endif: 738 parsePPEndIf(); 739 break; 740 default: 741 parsePPUnknown(); 742 break; 743 } 744 } 745 746 void UnwrappedLineParser::conditionalCompilationCondition(bool Unreachable) { 747 size_t Line = CurrentLines->size(); 748 if (CurrentLines == &PreprocessorDirectives) 749 Line += Lines.size(); 750 751 if (Unreachable || 752 (!PPStack.empty() && PPStack.back().Kind == PP_Unreachable)) 753 PPStack.push_back({PP_Unreachable, Line}); 754 else 755 PPStack.push_back({PP_Conditional, Line}); 756 } 757 758 void UnwrappedLineParser::conditionalCompilationStart(bool Unreachable) { 759 ++PPBranchLevel; 760 assert(PPBranchLevel >= 0 && PPBranchLevel <= (int)PPLevelBranchIndex.size()); 761 if (PPBranchLevel == (int)PPLevelBranchIndex.size()) { 762 PPLevelBranchIndex.push_back(0); 763 PPLevelBranchCount.push_back(0); 764 } 765 PPChainBranchIndex.push(0); 766 bool Skip = PPLevelBranchIndex[PPBranchLevel] > 0; 767 conditionalCompilationCondition(Unreachable || Skip); 768 } 769 770 void UnwrappedLineParser::conditionalCompilationAlternative() { 771 if (!PPStack.empty()) 772 PPStack.pop_back(); 773 assert(PPBranchLevel < (int)PPLevelBranchIndex.size()); 774 if (!PPChainBranchIndex.empty()) 775 ++PPChainBranchIndex.top(); 776 conditionalCompilationCondition( 777 PPBranchLevel >= 0 && !PPChainBranchIndex.empty() && 778 PPLevelBranchIndex[PPBranchLevel] != PPChainBranchIndex.top()); 779 } 780 781 void UnwrappedLineParser::conditionalCompilationEnd() { 782 assert(PPBranchLevel < (int)PPLevelBranchIndex.size()); 783 if (PPBranchLevel >= 0 && !PPChainBranchIndex.empty()) { 784 if (PPChainBranchIndex.top() + 1 > PPLevelBranchCount[PPBranchLevel]) { 785 PPLevelBranchCount[PPBranchLevel] = PPChainBranchIndex.top() + 1; 786 } 787 } 788 // Guard against #endif's without #if. 789 if (PPBranchLevel > -1) 790 --PPBranchLevel; 791 if (!PPChainBranchIndex.empty()) 792 PPChainBranchIndex.pop(); 793 if (!PPStack.empty()) 794 PPStack.pop_back(); 795 } 796 797 void UnwrappedLineParser::parsePPIf(bool IfDef) { 798 bool IfNDef = FormatTok->is(tok::pp_ifndef); 799 nextToken(); 800 bool Unreachable = false; 801 if (!IfDef && (FormatTok->is(tok::kw_false) || FormatTok->TokenText == "0")) 802 Unreachable = true; 803 if (IfDef && !IfNDef && FormatTok->TokenText == "SWIG") 804 Unreachable = true; 805 conditionalCompilationStart(Unreachable); 806 FormatToken *IfCondition = FormatTok; 807 // If there's a #ifndef on the first line, and the only lines before it are 808 // comments, it could be an include guard. 809 bool MaybeIncludeGuard = IfNDef; 810 if (IncludeGuard == IG_Inited && MaybeIncludeGuard) 811 for (auto &Line : Lines) { 812 if (!Line.Tokens.front().Tok->is(tok::comment)) { 813 MaybeIncludeGuard = false; 814 IncludeGuard = IG_Rejected; 815 break; 816 } 817 } 818 --PPBranchLevel; 819 parsePPUnknown(); 820 ++PPBranchLevel; 821 if (IncludeGuard == IG_Inited && MaybeIncludeGuard) { 822 IncludeGuard = IG_IfNdefed; 823 IncludeGuardToken = IfCondition; 824 } 825 } 826 827 void UnwrappedLineParser::parsePPElse() { 828 // If a potential include guard has an #else, it's not an include guard. 829 if (IncludeGuard == IG_Defined && PPBranchLevel == 0) 830 IncludeGuard = IG_Rejected; 831 conditionalCompilationAlternative(); 832 if (PPBranchLevel > -1) 833 --PPBranchLevel; 834 parsePPUnknown(); 835 ++PPBranchLevel; 836 } 837 838 void UnwrappedLineParser::parsePPElIf() { parsePPElse(); } 839 840 void UnwrappedLineParser::parsePPEndIf() { 841 conditionalCompilationEnd(); 842 parsePPUnknown(); 843 // If the #endif of a potential include guard is the last thing in the file, 844 // then we found an include guard. 845 unsigned TokenPosition = Tokens->getPosition(); 846 FormatToken *PeekNext = AllTokens[TokenPosition]; 847 if (IncludeGuard == IG_Defined && PPBranchLevel == -1 && 848 PeekNext->is(tok::eof) && 849 Style.IndentPPDirectives != FormatStyle::PPDIS_None) 850 IncludeGuard = IG_Found; 851 } 852 853 void UnwrappedLineParser::parsePPDefine() { 854 nextToken(); 855 856 if (!FormatTok->Tok.getIdentifierInfo()) { 857 IncludeGuard = IG_Rejected; 858 IncludeGuardToken = nullptr; 859 parsePPUnknown(); 860 return; 861 } 862 863 if (IncludeGuard == IG_IfNdefed && 864 IncludeGuardToken->TokenText == FormatTok->TokenText) { 865 IncludeGuard = IG_Defined; 866 IncludeGuardToken = nullptr; 867 for (auto &Line : Lines) { 868 if (!Line.Tokens.front().Tok->isOneOf(tok::comment, tok::hash)) { 869 IncludeGuard = IG_Rejected; 870 break; 871 } 872 } 873 } 874 875 nextToken(); 876 if (FormatTok->Tok.getKind() == tok::l_paren && 877 FormatTok->WhitespaceRange.getBegin() == 878 FormatTok->WhitespaceRange.getEnd()) { 879 parseParens(); 880 } 881 if (Style.IndentPPDirectives != FormatStyle::PPDIS_None) 882 Line->Level += PPBranchLevel + 1; 883 addUnwrappedLine(); 884 ++Line->Level; 885 886 // Errors during a preprocessor directive can only affect the layout of the 887 // preprocessor directive, and thus we ignore them. An alternative approach 888 // would be to use the same approach we use on the file level (no 889 // re-indentation if there was a structural error) within the macro 890 // definition. 891 parseFile(); 892 } 893 894 void UnwrappedLineParser::parsePPUnknown() { 895 do { 896 nextToken(); 897 } while (!eof()); 898 if (Style.IndentPPDirectives != FormatStyle::PPDIS_None) 899 Line->Level += PPBranchLevel + 1; 900 addUnwrappedLine(); 901 } 902 903 // Here we blacklist certain tokens that are not usually the first token in an 904 // unwrapped line. This is used in attempt to distinguish macro calls without 905 // trailing semicolons from other constructs split to several lines. 906 static bool tokenCanStartNewLine(const clang::Token &Tok) { 907 // Semicolon can be a null-statement, l_square can be a start of a macro or 908 // a C++11 attribute, but this doesn't seem to be common. 909 return Tok.isNot(tok::semi) && Tok.isNot(tok::l_brace) && 910 Tok.isNot(tok::l_square) && 911 // Tokens that can only be used as binary operators and a part of 912 // overloaded operator names. 913 Tok.isNot(tok::period) && Tok.isNot(tok::periodstar) && 914 Tok.isNot(tok::arrow) && Tok.isNot(tok::arrowstar) && 915 Tok.isNot(tok::less) && Tok.isNot(tok::greater) && 916 Tok.isNot(tok::slash) && Tok.isNot(tok::percent) && 917 Tok.isNot(tok::lessless) && Tok.isNot(tok::greatergreater) && 918 Tok.isNot(tok::equal) && Tok.isNot(tok::plusequal) && 919 Tok.isNot(tok::minusequal) && Tok.isNot(tok::starequal) && 920 Tok.isNot(tok::slashequal) && Tok.isNot(tok::percentequal) && 921 Tok.isNot(tok::ampequal) && Tok.isNot(tok::pipeequal) && 922 Tok.isNot(tok::caretequal) && Tok.isNot(tok::greatergreaterequal) && 923 Tok.isNot(tok::lesslessequal) && 924 // Colon is used in labels, base class lists, initializer lists, 925 // range-based for loops, ternary operator, but should never be the 926 // first token in an unwrapped line. 927 Tok.isNot(tok::colon) && 928 // 'noexcept' is a trailing annotation. 929 Tok.isNot(tok::kw_noexcept); 930 } 931 932 static bool mustBeJSIdent(const AdditionalKeywords &Keywords, 933 const FormatToken *FormatTok) { 934 // FIXME: This returns true for C/C++ keywords like 'struct'. 935 return FormatTok->is(tok::identifier) && 936 (FormatTok->Tok.getIdentifierInfo() == nullptr || 937 !FormatTok->isOneOf( 938 Keywords.kw_in, Keywords.kw_of, Keywords.kw_as, Keywords.kw_async, 939 Keywords.kw_await, Keywords.kw_yield, Keywords.kw_finally, 940 Keywords.kw_function, Keywords.kw_import, Keywords.kw_is, 941 Keywords.kw_let, Keywords.kw_var, tok::kw_const, 942 Keywords.kw_abstract, Keywords.kw_extends, Keywords.kw_implements, 943 Keywords.kw_instanceof, Keywords.kw_interface, Keywords.kw_throws, 944 Keywords.kw_from)); 945 } 946 947 static bool mustBeJSIdentOrValue(const AdditionalKeywords &Keywords, 948 const FormatToken *FormatTok) { 949 return FormatTok->Tok.isLiteral() || 950 FormatTok->isOneOf(tok::kw_true, tok::kw_false) || 951 mustBeJSIdent(Keywords, FormatTok); 952 } 953 954 // isJSDeclOrStmt returns true if |FormatTok| starts a declaration or statement 955 // when encountered after a value (see mustBeJSIdentOrValue). 956 static bool isJSDeclOrStmt(const AdditionalKeywords &Keywords, 957 const FormatToken *FormatTok) { 958 return FormatTok->isOneOf( 959 tok::kw_return, Keywords.kw_yield, 960 // conditionals 961 tok::kw_if, tok::kw_else, 962 // loops 963 tok::kw_for, tok::kw_while, tok::kw_do, tok::kw_continue, tok::kw_break, 964 // switch/case 965 tok::kw_switch, tok::kw_case, 966 // exceptions 967 tok::kw_throw, tok::kw_try, tok::kw_catch, Keywords.kw_finally, 968 // declaration 969 tok::kw_const, tok::kw_class, Keywords.kw_var, Keywords.kw_let, 970 Keywords.kw_async, Keywords.kw_function, 971 // import/export 972 Keywords.kw_import, tok::kw_export); 973 } 974 975 // readTokenWithJavaScriptASI reads the next token and terminates the current 976 // line if JavaScript Automatic Semicolon Insertion must 977 // happen between the current token and the next token. 978 // 979 // This method is conservative - it cannot cover all edge cases of JavaScript, 980 // but only aims to correctly handle certain well known cases. It *must not* 981 // return true in speculative cases. 982 void UnwrappedLineParser::readTokenWithJavaScriptASI() { 983 FormatToken *Previous = FormatTok; 984 readToken(); 985 FormatToken *Next = FormatTok; 986 987 bool IsOnSameLine = 988 CommentsBeforeNextToken.empty() 989 ? Next->NewlinesBefore == 0 990 : CommentsBeforeNextToken.front()->NewlinesBefore == 0; 991 if (IsOnSameLine) 992 return; 993 994 bool PreviousMustBeValue = mustBeJSIdentOrValue(Keywords, Previous); 995 bool PreviousStartsTemplateExpr = 996 Previous->is(TT_TemplateString) && Previous->TokenText.endswith("${"); 997 if (PreviousMustBeValue || Previous->is(tok::r_paren)) { 998 // If the line contains an '@' sign, the previous token might be an 999 // annotation, which can precede another identifier/value. 1000 bool HasAt = std::find_if(Line->Tokens.begin(), Line->Tokens.end(), 1001 [](UnwrappedLineNode &LineNode) { 1002 return LineNode.Tok->is(tok::at); 1003 }) != Line->Tokens.end(); 1004 if (HasAt) 1005 return; 1006 } 1007 if (Next->is(tok::exclaim) && PreviousMustBeValue) 1008 return addUnwrappedLine(); 1009 bool NextMustBeValue = mustBeJSIdentOrValue(Keywords, Next); 1010 bool NextEndsTemplateExpr = 1011 Next->is(TT_TemplateString) && Next->TokenText.startswith("}"); 1012 if (NextMustBeValue && !NextEndsTemplateExpr && !PreviousStartsTemplateExpr && 1013 (PreviousMustBeValue || 1014 Previous->isOneOf(tok::r_square, tok::r_paren, tok::plusplus, 1015 tok::minusminus))) 1016 return addUnwrappedLine(); 1017 if ((PreviousMustBeValue || Previous->is(tok::r_paren)) && 1018 isJSDeclOrStmt(Keywords, Next)) 1019 return addUnwrappedLine(); 1020 } 1021 1022 void UnwrappedLineParser::parseStructuralElement() { 1023 assert(!FormatTok->is(tok::l_brace)); 1024 if (Style.Language == FormatStyle::LK_TableGen && 1025 FormatTok->is(tok::pp_include)) { 1026 nextToken(); 1027 if (FormatTok->is(tok::string_literal)) 1028 nextToken(); 1029 addUnwrappedLine(); 1030 return; 1031 } 1032 switch (FormatTok->Tok.getKind()) { 1033 case tok::kw_asm: 1034 nextToken(); 1035 if (FormatTok->is(tok::l_brace)) { 1036 FormatTok->Type = TT_InlineASMBrace; 1037 nextToken(); 1038 while (FormatTok && FormatTok->isNot(tok::eof)) { 1039 if (FormatTok->is(tok::r_brace)) { 1040 FormatTok->Type = TT_InlineASMBrace; 1041 nextToken(); 1042 addUnwrappedLine(); 1043 break; 1044 } 1045 FormatTok->Finalized = true; 1046 nextToken(); 1047 } 1048 } 1049 break; 1050 case tok::kw_namespace: 1051 parseNamespace(); 1052 return; 1053 case tok::kw_public: 1054 case tok::kw_protected: 1055 case tok::kw_private: 1056 if (Style.Language == FormatStyle::LK_Java || 1057 Style.Language == FormatStyle::LK_JavaScript || Style.isCSharp()) 1058 nextToken(); 1059 else 1060 parseAccessSpecifier(); 1061 return; 1062 case tok::kw_if: 1063 if (Style.Language == FormatStyle::LK_JavaScript && Line->MustBeDeclaration) 1064 // field/method declaration. 1065 break; 1066 parseIfThenElse(); 1067 return; 1068 case tok::kw_for: 1069 case tok::kw_while: 1070 if (Style.Language == FormatStyle::LK_JavaScript && Line->MustBeDeclaration) 1071 // field/method declaration. 1072 break; 1073 parseForOrWhileLoop(); 1074 return; 1075 case tok::kw_do: 1076 if (Style.Language == FormatStyle::LK_JavaScript && Line->MustBeDeclaration) 1077 // field/method declaration. 1078 break; 1079 parseDoWhile(); 1080 return; 1081 case tok::kw_switch: 1082 if (Style.Language == FormatStyle::LK_JavaScript && Line->MustBeDeclaration) 1083 // 'switch: string' field declaration. 1084 break; 1085 parseSwitch(); 1086 return; 1087 case tok::kw_default: 1088 if (Style.Language == FormatStyle::LK_JavaScript && Line->MustBeDeclaration) 1089 // 'default: string' field declaration. 1090 break; 1091 nextToken(); 1092 if (FormatTok->is(tok::colon)) { 1093 parseLabel(); 1094 return; 1095 } 1096 // e.g. "default void f() {}" in a Java interface. 1097 break; 1098 case tok::kw_case: 1099 if (Style.Language == FormatStyle::LK_JavaScript && Line->MustBeDeclaration) 1100 // 'case: string' field declaration. 1101 break; 1102 parseCaseLabel(); 1103 return; 1104 case tok::kw_try: 1105 case tok::kw___try: 1106 if (Style.Language == FormatStyle::LK_JavaScript && Line->MustBeDeclaration) 1107 // field/method declaration. 1108 break; 1109 parseTryCatch(); 1110 return; 1111 case tok::kw_extern: 1112 nextToken(); 1113 if (FormatTok->Tok.is(tok::string_literal)) { 1114 nextToken(); 1115 if (FormatTok->Tok.is(tok::l_brace)) { 1116 if (Style.BraceWrapping.AfterExternBlock) { 1117 addUnwrappedLine(); 1118 parseBlock(/*MustBeDeclaration=*/true); 1119 } else { 1120 parseBlock(/*MustBeDeclaration=*/true, /*AddLevel=*/false); 1121 } 1122 addUnwrappedLine(); 1123 return; 1124 } 1125 } 1126 break; 1127 case tok::kw_export: 1128 if (Style.Language == FormatStyle::LK_JavaScript) { 1129 parseJavaScriptEs6ImportExport(); 1130 return; 1131 } 1132 if (!Style.isCpp()) 1133 break; 1134 // Handle C++ "(inline|export) namespace". 1135 LLVM_FALLTHROUGH; 1136 case tok::kw_inline: 1137 nextToken(); 1138 if (FormatTok->Tok.is(tok::kw_namespace)) { 1139 parseNamespace(); 1140 return; 1141 } 1142 break; 1143 case tok::identifier: 1144 if (FormatTok->is(TT_ForEachMacro)) { 1145 parseForOrWhileLoop(); 1146 return; 1147 } 1148 if (FormatTok->is(TT_MacroBlockBegin)) { 1149 parseBlock(/*MustBeDeclaration=*/false, /*AddLevel=*/true, 1150 /*MunchSemi=*/false); 1151 return; 1152 } 1153 if (FormatTok->is(Keywords.kw_import)) { 1154 if (Style.Language == FormatStyle::LK_JavaScript) { 1155 parseJavaScriptEs6ImportExport(); 1156 return; 1157 } 1158 if (Style.Language == FormatStyle::LK_Proto) { 1159 nextToken(); 1160 if (FormatTok->is(tok::kw_public)) 1161 nextToken(); 1162 if (!FormatTok->is(tok::string_literal)) 1163 return; 1164 nextToken(); 1165 if (FormatTok->is(tok::semi)) 1166 nextToken(); 1167 addUnwrappedLine(); 1168 return; 1169 } 1170 } 1171 if (Style.isCpp() && 1172 FormatTok->isOneOf(Keywords.kw_signals, Keywords.kw_qsignals, 1173 Keywords.kw_slots, Keywords.kw_qslots)) { 1174 nextToken(); 1175 if (FormatTok->is(tok::colon)) { 1176 nextToken(); 1177 addUnwrappedLine(); 1178 return; 1179 } 1180 } 1181 if (Style.isCpp() && FormatTok->is(TT_StatementMacro)) { 1182 parseStatementMacro(); 1183 return; 1184 } 1185 if (Style.isCpp() && FormatTok->is(TT_NamespaceMacro)) { 1186 parseNamespace(); 1187 return; 1188 } 1189 // In all other cases, parse the declaration. 1190 break; 1191 default: 1192 break; 1193 } 1194 do { 1195 const FormatToken *Previous = FormatTok->Previous; 1196 switch (FormatTok->Tok.getKind()) { 1197 case tok::at: 1198 nextToken(); 1199 if (FormatTok->Tok.is(tok::l_brace)) { 1200 nextToken(); 1201 parseBracedList(); 1202 break; 1203 } else if (Style.Language == FormatStyle::LK_Java && 1204 FormatTok->is(Keywords.kw_interface)) { 1205 nextToken(); 1206 break; 1207 } 1208 switch (FormatTok->Tok.getObjCKeywordID()) { 1209 case tok::objc_public: 1210 case tok::objc_protected: 1211 case tok::objc_package: 1212 case tok::objc_private: 1213 return parseAccessSpecifier(); 1214 case tok::objc_interface: 1215 case tok::objc_implementation: 1216 return parseObjCInterfaceOrImplementation(); 1217 case tok::objc_protocol: 1218 if (parseObjCProtocol()) 1219 return; 1220 break; 1221 case tok::objc_end: 1222 return; // Handled by the caller. 1223 case tok::objc_optional: 1224 case tok::objc_required: 1225 nextToken(); 1226 addUnwrappedLine(); 1227 return; 1228 case tok::objc_autoreleasepool: 1229 nextToken(); 1230 if (FormatTok->Tok.is(tok::l_brace)) { 1231 if (Style.BraceWrapping.AfterControlStatement == 1232 FormatStyle::BWACS_Always) 1233 addUnwrappedLine(); 1234 parseBlock(/*MustBeDeclaration=*/false); 1235 } 1236 addUnwrappedLine(); 1237 return; 1238 case tok::objc_synchronized: 1239 nextToken(); 1240 if (FormatTok->Tok.is(tok::l_paren)) 1241 // Skip synchronization object 1242 parseParens(); 1243 if (FormatTok->Tok.is(tok::l_brace)) { 1244 if (Style.BraceWrapping.AfterControlStatement == 1245 FormatStyle::BWACS_Always) 1246 addUnwrappedLine(); 1247 parseBlock(/*MustBeDeclaration=*/false); 1248 } 1249 addUnwrappedLine(); 1250 return; 1251 case tok::objc_try: 1252 // This branch isn't strictly necessary (the kw_try case below would 1253 // do this too after the tok::at is parsed above). But be explicit. 1254 parseTryCatch(); 1255 return; 1256 default: 1257 break; 1258 } 1259 break; 1260 case tok::kw_enum: 1261 // Ignore if this is part of "template <enum ...". 1262 if (Previous && Previous->is(tok::less)) { 1263 nextToken(); 1264 break; 1265 } 1266 1267 // parseEnum falls through and does not yet add an unwrapped line as an 1268 // enum definition can start a structural element. 1269 if (!parseEnum()) 1270 break; 1271 // This only applies for C++. 1272 if (!Style.isCpp()) { 1273 addUnwrappedLine(); 1274 return; 1275 } 1276 break; 1277 case tok::kw_typedef: 1278 nextToken(); 1279 if (FormatTok->isOneOf(Keywords.kw_NS_ENUM, Keywords.kw_NS_OPTIONS, 1280 Keywords.kw_CF_ENUM, Keywords.kw_CF_OPTIONS, 1281 Keywords.kw_CF_CLOSED_ENUM, 1282 Keywords.kw_NS_CLOSED_ENUM)) 1283 parseEnum(); 1284 break; 1285 case tok::kw_struct: 1286 case tok::kw_union: 1287 case tok::kw_class: 1288 // parseRecord falls through and does not yet add an unwrapped line as a 1289 // record declaration or definition can start a structural element. 1290 parseRecord(); 1291 // This does not apply for Java, JavaScript and C#. 1292 if (Style.Language == FormatStyle::LK_Java || 1293 Style.Language == FormatStyle::LK_JavaScript || Style.isCSharp()) { 1294 if (FormatTok->is(tok::semi)) 1295 nextToken(); 1296 addUnwrappedLine(); 1297 return; 1298 } 1299 break; 1300 case tok::period: 1301 nextToken(); 1302 // In Java, classes have an implicit static member "class". 1303 if (Style.Language == FormatStyle::LK_Java && FormatTok && 1304 FormatTok->is(tok::kw_class)) 1305 nextToken(); 1306 if (Style.Language == FormatStyle::LK_JavaScript && FormatTok && 1307 FormatTok->Tok.getIdentifierInfo()) 1308 // JavaScript only has pseudo keywords, all keywords are allowed to 1309 // appear in "IdentifierName" positions. See http://es5.github.io/#x7.6 1310 nextToken(); 1311 break; 1312 case tok::semi: 1313 nextToken(); 1314 addUnwrappedLine(); 1315 return; 1316 case tok::r_brace: 1317 addUnwrappedLine(); 1318 return; 1319 case tok::l_paren: 1320 parseParens(); 1321 break; 1322 case tok::kw_operator: 1323 nextToken(); 1324 if (FormatTok->isBinaryOperator()) 1325 nextToken(); 1326 break; 1327 case tok::caret: 1328 nextToken(); 1329 if (FormatTok->Tok.isAnyIdentifier() || 1330 FormatTok->isSimpleTypeSpecifier()) 1331 nextToken(); 1332 if (FormatTok->is(tok::l_paren)) 1333 parseParens(); 1334 if (FormatTok->is(tok::l_brace)) 1335 parseChildBlock(); 1336 break; 1337 case tok::l_brace: 1338 if (!tryToParsePropertyAccessor() && !tryToParseBracedList()) { 1339 // A block outside of parentheses must be the last part of a 1340 // structural element. 1341 // FIXME: Figure out cases where this is not true, and add projections 1342 // for them (the one we know is missing are lambdas). 1343 if (Style.BraceWrapping.AfterFunction) 1344 addUnwrappedLine(); 1345 FormatTok->Type = TT_FunctionLBrace; 1346 parseBlock(/*MustBeDeclaration=*/false); 1347 addUnwrappedLine(); 1348 return; 1349 } 1350 // Otherwise this was a braced init list, and the structural 1351 // element continues. 1352 break; 1353 case tok::kw_try: 1354 if (Style.Language == FormatStyle::LK_JavaScript && 1355 Line->MustBeDeclaration) { 1356 // field/method declaration. 1357 nextToken(); 1358 break; 1359 } 1360 // We arrive here when parsing function-try blocks. 1361 if (Style.BraceWrapping.AfterFunction) 1362 addUnwrappedLine(); 1363 parseTryCatch(); 1364 return; 1365 case tok::identifier: { 1366 if (Style.isCSharp() && FormatTok->is(Keywords.kw_where) && 1367 Line->MustBeDeclaration) { 1368 addUnwrappedLine(); 1369 parseCSharpGenericTypeConstraint(); 1370 break; 1371 } 1372 if (FormatTok->is(TT_MacroBlockEnd)) { 1373 addUnwrappedLine(); 1374 return; 1375 } 1376 1377 // Function declarations (as opposed to function expressions) are parsed 1378 // on their own unwrapped line by continuing this loop. Function 1379 // expressions (functions that are not on their own line) must not create 1380 // a new unwrapped line, so they are special cased below. 1381 size_t TokenCount = Line->Tokens.size(); 1382 if (Style.Language == FormatStyle::LK_JavaScript && 1383 FormatTok->is(Keywords.kw_function) && 1384 (TokenCount > 1 || (TokenCount == 1 && !Line->Tokens.front().Tok->is( 1385 Keywords.kw_async)))) { 1386 tryToParseJSFunction(); 1387 break; 1388 } 1389 if ((Style.Language == FormatStyle::LK_JavaScript || 1390 Style.Language == FormatStyle::LK_Java) && 1391 FormatTok->is(Keywords.kw_interface)) { 1392 if (Style.Language == FormatStyle::LK_JavaScript) { 1393 // In JavaScript/TypeScript, "interface" can be used as a standalone 1394 // identifier, e.g. in `var interface = 1;`. If "interface" is 1395 // followed by another identifier, it is very like to be an actual 1396 // interface declaration. 1397 unsigned StoredPosition = Tokens->getPosition(); 1398 FormatToken *Next = Tokens->getNextToken(); 1399 FormatTok = Tokens->setPosition(StoredPosition); 1400 if (Next && !mustBeJSIdent(Keywords, Next)) { 1401 nextToken(); 1402 break; 1403 } 1404 } 1405 parseRecord(); 1406 addUnwrappedLine(); 1407 return; 1408 } 1409 1410 if (Style.isCpp() && FormatTok->is(TT_StatementMacro)) { 1411 parseStatementMacro(); 1412 return; 1413 } 1414 1415 // See if the following token should start a new unwrapped line. 1416 StringRef Text = FormatTok->TokenText; 1417 nextToken(); 1418 1419 // JS doesn't have macros, and within classes colons indicate fields, not 1420 // labels. 1421 if (Style.Language == FormatStyle::LK_JavaScript) 1422 break; 1423 1424 TokenCount = Line->Tokens.size(); 1425 if (TokenCount == 1 || 1426 (TokenCount == 2 && Line->Tokens.front().Tok->is(tok::comment))) { 1427 if (FormatTok->Tok.is(tok::colon) && !Line->MustBeDeclaration) { 1428 Line->Tokens.begin()->Tok->MustBreakBefore = true; 1429 parseLabel(!Style.IndentGotoLabels); 1430 return; 1431 } 1432 // Recognize function-like macro usages without trailing semicolon as 1433 // well as free-standing macros like Q_OBJECT. 1434 bool FunctionLike = FormatTok->is(tok::l_paren); 1435 if (FunctionLike) 1436 parseParens(); 1437 1438 bool FollowedByNewline = 1439 CommentsBeforeNextToken.empty() 1440 ? FormatTok->NewlinesBefore > 0 1441 : CommentsBeforeNextToken.front()->NewlinesBefore > 0; 1442 1443 if (FollowedByNewline && (Text.size() >= 5 || FunctionLike) && 1444 tokenCanStartNewLine(FormatTok->Tok) && Text == Text.upper()) { 1445 addUnwrappedLine(); 1446 return; 1447 } 1448 } 1449 break; 1450 } 1451 case tok::equal: 1452 // Fat arrows (=>) have tok::TokenKind tok::equal but TokenType 1453 // TT_JsFatArrow. The always start an expression or a child block if 1454 // followed by a curly. 1455 if (FormatTok->is(TT_JsFatArrow)) { 1456 nextToken(); 1457 if (FormatTok->is(tok::l_brace)) 1458 parseChildBlock(); 1459 break; 1460 } 1461 1462 nextToken(); 1463 if (FormatTok->Tok.is(tok::l_brace)) { 1464 // Block kind should probably be set to BK_BracedInit for any language. 1465 // C# needs this change to ensure that array initialisers and object 1466 // initialisers are indented the same way. 1467 if (Style.isCSharp()) 1468 FormatTok->BlockKind = BK_BracedInit; 1469 nextToken(); 1470 parseBracedList(); 1471 } else if (Style.Language == FormatStyle::LK_Proto && 1472 FormatTok->Tok.is(tok::less)) { 1473 nextToken(); 1474 parseBracedList(/*ContinueOnSemicolons=*/false, 1475 /*ClosingBraceKind=*/tok::greater); 1476 } 1477 break; 1478 case tok::l_square: 1479 parseSquare(); 1480 break; 1481 case tok::kw_new: 1482 parseNew(); 1483 break; 1484 default: 1485 nextToken(); 1486 break; 1487 } 1488 } while (!eof()); 1489 } 1490 1491 bool UnwrappedLineParser::tryToParsePropertyAccessor() { 1492 assert(FormatTok->is(tok::l_brace)); 1493 if (!Style.isCSharp()) 1494 return false; 1495 // See if it's a property accessor. 1496 if (FormatTok->Previous->isNot(tok::identifier)) 1497 return false; 1498 1499 // See if we are inside a property accessor. 1500 // 1501 // Record the current tokenPosition so that we can advance and 1502 // reset the current token. `Next` is not set yet so we need 1503 // another way to advance along the token stream. 1504 unsigned int StoredPosition = Tokens->getPosition(); 1505 FormatToken *Tok = Tokens->getNextToken(); 1506 1507 // A trivial property accessor is of the form: 1508 // { [ACCESS_SPECIFIER] [get]; [ACCESS_SPECIFIER] [set] } 1509 // Track these as they do not require line breaks to be introduced. 1510 bool HasGetOrSet = false; 1511 bool IsTrivialPropertyAccessor = true; 1512 while (!eof()) { 1513 if (Tok->isOneOf(tok::semi, tok::kw_public, tok::kw_private, 1514 tok::kw_protected, Keywords.kw_internal, Keywords.kw_get, 1515 Keywords.kw_set)) { 1516 if (Tok->isOneOf(Keywords.kw_get, Keywords.kw_set)) 1517 HasGetOrSet = true; 1518 Tok = Tokens->getNextToken(); 1519 continue; 1520 } 1521 if (Tok->isNot(tok::r_brace)) 1522 IsTrivialPropertyAccessor = false; 1523 break; 1524 } 1525 1526 if (!HasGetOrSet) { 1527 Tokens->setPosition(StoredPosition); 1528 return false; 1529 } 1530 1531 // Try to parse the property accessor: 1532 // https://docs.microsoft.com/en-us/dotnet/csharp/programming-guide/classes-and-structs/properties 1533 Tokens->setPosition(StoredPosition); 1534 nextToken(); 1535 do { 1536 switch (FormatTok->Tok.getKind()) { 1537 case tok::r_brace: 1538 nextToken(); 1539 if (FormatTok->is(tok::equal)) { 1540 while (!eof() && FormatTok->isNot(tok::semi)) 1541 nextToken(); 1542 nextToken(); 1543 } 1544 addUnwrappedLine(); 1545 return true; 1546 case tok::l_brace: 1547 ++Line->Level; 1548 parseBlock(/*MustBeDeclaration=*/true); 1549 addUnwrappedLine(); 1550 --Line->Level; 1551 break; 1552 case tok::equal: 1553 if (FormatTok->is(TT_JsFatArrow)) { 1554 ++Line->Level; 1555 do { 1556 nextToken(); 1557 } while (!eof() && FormatTok->isNot(tok::semi)); 1558 nextToken(); 1559 addUnwrappedLine(); 1560 --Line->Level; 1561 break; 1562 } 1563 nextToken(); 1564 break; 1565 default: 1566 if (FormatTok->isOneOf(Keywords.kw_get, Keywords.kw_set) && 1567 !IsTrivialPropertyAccessor) { 1568 // Non-trivial get/set needs to be on its own line. 1569 addUnwrappedLine(); 1570 } 1571 nextToken(); 1572 } 1573 } while (!eof()); 1574 1575 // Unreachable for well-formed code (paired '{' and '}'). 1576 return true; 1577 } 1578 1579 bool UnwrappedLineParser::tryToParseLambda() { 1580 if (!Style.isCpp()) { 1581 nextToken(); 1582 return false; 1583 } 1584 assert(FormatTok->is(tok::l_square)); 1585 FormatToken &LSquare = *FormatTok; 1586 if (!tryToParseLambdaIntroducer()) 1587 return false; 1588 1589 bool SeenArrow = false; 1590 1591 while (FormatTok->isNot(tok::l_brace)) { 1592 if (FormatTok->isSimpleTypeSpecifier()) { 1593 nextToken(); 1594 continue; 1595 } 1596 switch (FormatTok->Tok.getKind()) { 1597 case tok::l_brace: 1598 break; 1599 case tok::l_paren: 1600 parseParens(); 1601 break; 1602 case tok::amp: 1603 case tok::star: 1604 case tok::kw_const: 1605 case tok::comma: 1606 case tok::less: 1607 case tok::greater: 1608 case tok::identifier: 1609 case tok::numeric_constant: 1610 case tok::coloncolon: 1611 case tok::kw_class: 1612 case tok::kw_mutable: 1613 case tok::kw_noexcept: 1614 case tok::kw_template: 1615 case tok::kw_typename: 1616 nextToken(); 1617 break; 1618 // Specialization of a template with an integer parameter can contain 1619 // arithmetic, logical, comparison and ternary operators. 1620 // 1621 // FIXME: This also accepts sequences of operators that are not in the scope 1622 // of a template argument list. 1623 // 1624 // In a C++ lambda a template type can only occur after an arrow. We use 1625 // this as an heuristic to distinguish between Objective-C expressions 1626 // followed by an `a->b` expression, such as: 1627 // ([obj func:arg] + a->b) 1628 // Otherwise the code below would parse as a lambda. 1629 // 1630 // FIXME: This heuristic is incorrect for C++20 generic lambdas with 1631 // explicit template lists: []<bool b = true && false>(U &&u){} 1632 case tok::plus: 1633 case tok::minus: 1634 case tok::exclaim: 1635 case tok::tilde: 1636 case tok::slash: 1637 case tok::percent: 1638 case tok::lessless: 1639 case tok::pipe: 1640 case tok::pipepipe: 1641 case tok::ampamp: 1642 case tok::caret: 1643 case tok::equalequal: 1644 case tok::exclaimequal: 1645 case tok::greaterequal: 1646 case tok::lessequal: 1647 case tok::question: 1648 case tok::colon: 1649 case tok::ellipsis: 1650 case tok::kw_true: 1651 case tok::kw_false: 1652 if (SeenArrow) { 1653 nextToken(); 1654 break; 1655 } 1656 return true; 1657 case tok::arrow: 1658 // This might or might not actually be a lambda arrow (this could be an 1659 // ObjC method invocation followed by a dereferencing arrow). We might 1660 // reset this back to TT_Unknown in TokenAnnotator. 1661 FormatTok->Type = TT_LambdaArrow; 1662 SeenArrow = true; 1663 nextToken(); 1664 break; 1665 default: 1666 return true; 1667 } 1668 } 1669 FormatTok->Type = TT_LambdaLBrace; 1670 LSquare.Type = TT_LambdaLSquare; 1671 parseChildBlock(); 1672 return true; 1673 } 1674 1675 bool UnwrappedLineParser::tryToParseLambdaIntroducer() { 1676 const FormatToken *Previous = FormatTok->Previous; 1677 if (Previous && 1678 (Previous->isOneOf(tok::identifier, tok::kw_operator, tok::kw_new, 1679 tok::kw_delete, tok::l_square) || 1680 FormatTok->isCppStructuredBinding(Style) || Previous->closesScope() || 1681 Previous->isSimpleTypeSpecifier())) { 1682 nextToken(); 1683 return false; 1684 } 1685 nextToken(); 1686 if (FormatTok->is(tok::l_square)) { 1687 return false; 1688 } 1689 parseSquare(/*LambdaIntroducer=*/true); 1690 return true; 1691 } 1692 1693 void UnwrappedLineParser::tryToParseJSFunction() { 1694 assert(FormatTok->is(Keywords.kw_function) || 1695 FormatTok->startsSequence(Keywords.kw_async, Keywords.kw_function)); 1696 if (FormatTok->is(Keywords.kw_async)) 1697 nextToken(); 1698 // Consume "function". 1699 nextToken(); 1700 1701 // Consume * (generator function). Treat it like C++'s overloaded operators. 1702 if (FormatTok->is(tok::star)) { 1703 FormatTok->Type = TT_OverloadedOperator; 1704 nextToken(); 1705 } 1706 1707 // Consume function name. 1708 if (FormatTok->is(tok::identifier)) 1709 nextToken(); 1710 1711 if (FormatTok->isNot(tok::l_paren)) 1712 return; 1713 1714 // Parse formal parameter list. 1715 parseParens(); 1716 1717 if (FormatTok->is(tok::colon)) { 1718 // Parse a type definition. 1719 nextToken(); 1720 1721 // Eat the type declaration. For braced inline object types, balance braces, 1722 // otherwise just parse until finding an l_brace for the function body. 1723 if (FormatTok->is(tok::l_brace)) 1724 tryToParseBracedList(); 1725 else 1726 while (!FormatTok->isOneOf(tok::l_brace, tok::semi) && !eof()) 1727 nextToken(); 1728 } 1729 1730 if (FormatTok->is(tok::semi)) 1731 return; 1732 1733 parseChildBlock(); 1734 } 1735 1736 bool UnwrappedLineParser::tryToParseBracedList() { 1737 if (FormatTok->BlockKind == BK_Unknown) 1738 calculateBraceTypes(); 1739 assert(FormatTok->BlockKind != BK_Unknown); 1740 if (FormatTok->BlockKind == BK_Block) 1741 return false; 1742 nextToken(); 1743 parseBracedList(); 1744 return true; 1745 } 1746 1747 bool UnwrappedLineParser::parseBracedList(bool ContinueOnSemicolons, 1748 tok::TokenKind ClosingBraceKind) { 1749 bool HasError = false; 1750 1751 // FIXME: Once we have an expression parser in the UnwrappedLineParser, 1752 // replace this by using parseAssigmentExpression() inside. 1753 do { 1754 if (Style.isCSharp()) { 1755 if (FormatTok->is(TT_JsFatArrow)) { 1756 nextToken(); 1757 // Fat arrows can be followed by simple expressions or by child blocks 1758 // in curly braces. 1759 if (FormatTok->is(tok::l_brace)) { 1760 parseChildBlock(); 1761 continue; 1762 } 1763 } 1764 } 1765 if (Style.Language == FormatStyle::LK_JavaScript) { 1766 if (FormatTok->is(Keywords.kw_function) || 1767 FormatTok->startsSequence(Keywords.kw_async, Keywords.kw_function)) { 1768 tryToParseJSFunction(); 1769 continue; 1770 } 1771 if (FormatTok->is(TT_JsFatArrow)) { 1772 nextToken(); 1773 // Fat arrows can be followed by simple expressions or by child blocks 1774 // in curly braces. 1775 if (FormatTok->is(tok::l_brace)) { 1776 parseChildBlock(); 1777 continue; 1778 } 1779 } 1780 if (FormatTok->is(tok::l_brace)) { 1781 // Could be a method inside of a braced list `{a() { return 1; }}`. 1782 if (tryToParseBracedList()) 1783 continue; 1784 parseChildBlock(); 1785 } 1786 } 1787 if (FormatTok->Tok.getKind() == ClosingBraceKind) { 1788 nextToken(); 1789 return !HasError; 1790 } 1791 switch (FormatTok->Tok.getKind()) { 1792 case tok::caret: 1793 nextToken(); 1794 if (FormatTok->is(tok::l_brace)) { 1795 parseChildBlock(); 1796 } 1797 break; 1798 case tok::l_square: 1799 if (Style.isCSharp()) 1800 parseSquare(); 1801 else 1802 tryToParseLambda(); 1803 break; 1804 case tok::l_paren: 1805 parseParens(); 1806 // JavaScript can just have free standing methods and getters/setters in 1807 // object literals. Detect them by a "{" following ")". 1808 if (Style.Language == FormatStyle::LK_JavaScript) { 1809 if (FormatTok->is(tok::l_brace)) 1810 parseChildBlock(); 1811 break; 1812 } 1813 break; 1814 case tok::l_brace: 1815 // Assume there are no blocks inside a braced init list apart 1816 // from the ones we explicitly parse out (like lambdas). 1817 FormatTok->BlockKind = BK_BracedInit; 1818 nextToken(); 1819 parseBracedList(); 1820 break; 1821 case tok::less: 1822 if (Style.Language == FormatStyle::LK_Proto) { 1823 nextToken(); 1824 parseBracedList(/*ContinueOnSemicolons=*/false, 1825 /*ClosingBraceKind=*/tok::greater); 1826 } else { 1827 nextToken(); 1828 } 1829 break; 1830 case tok::semi: 1831 // JavaScript (or more precisely TypeScript) can have semicolons in braced 1832 // lists (in so-called TypeMemberLists). Thus, the semicolon cannot be 1833 // used for error recovery if we have otherwise determined that this is 1834 // a braced list. 1835 if (Style.Language == FormatStyle::LK_JavaScript) { 1836 nextToken(); 1837 break; 1838 } 1839 HasError = true; 1840 if (!ContinueOnSemicolons) 1841 return !HasError; 1842 nextToken(); 1843 break; 1844 case tok::comma: 1845 nextToken(); 1846 break; 1847 default: 1848 nextToken(); 1849 break; 1850 } 1851 } while (!eof()); 1852 return false; 1853 } 1854 1855 void UnwrappedLineParser::parseParens() { 1856 assert(FormatTok->Tok.is(tok::l_paren) && "'(' expected."); 1857 nextToken(); 1858 do { 1859 switch (FormatTok->Tok.getKind()) { 1860 case tok::l_paren: 1861 parseParens(); 1862 if (Style.Language == FormatStyle::LK_Java && FormatTok->is(tok::l_brace)) 1863 parseChildBlock(); 1864 break; 1865 case tok::r_paren: 1866 nextToken(); 1867 return; 1868 case tok::r_brace: 1869 // A "}" inside parenthesis is an error if there wasn't a matching "{". 1870 return; 1871 case tok::l_square: 1872 tryToParseLambda(); 1873 break; 1874 case tok::l_brace: 1875 if (!tryToParseBracedList()) 1876 parseChildBlock(); 1877 break; 1878 case tok::at: 1879 nextToken(); 1880 if (FormatTok->Tok.is(tok::l_brace)) { 1881 nextToken(); 1882 parseBracedList(); 1883 } 1884 break; 1885 case tok::kw_class: 1886 if (Style.Language == FormatStyle::LK_JavaScript) 1887 parseRecord(/*ParseAsExpr=*/true); 1888 else 1889 nextToken(); 1890 break; 1891 case tok::identifier: 1892 if (Style.Language == FormatStyle::LK_JavaScript && 1893 (FormatTok->is(Keywords.kw_function) || 1894 FormatTok->startsSequence(Keywords.kw_async, Keywords.kw_function))) 1895 tryToParseJSFunction(); 1896 else 1897 nextToken(); 1898 break; 1899 default: 1900 nextToken(); 1901 break; 1902 } 1903 } while (!eof()); 1904 } 1905 1906 void UnwrappedLineParser::parseSquare(bool LambdaIntroducer) { 1907 if (!LambdaIntroducer) { 1908 assert(FormatTok->Tok.is(tok::l_square) && "'[' expected."); 1909 if (tryToParseLambda()) 1910 return; 1911 } 1912 do { 1913 switch (FormatTok->Tok.getKind()) { 1914 case tok::l_paren: 1915 parseParens(); 1916 break; 1917 case tok::r_square: 1918 nextToken(); 1919 return; 1920 case tok::r_brace: 1921 // A "}" inside parenthesis is an error if there wasn't a matching "{". 1922 return; 1923 case tok::l_square: 1924 parseSquare(); 1925 break; 1926 case tok::l_brace: { 1927 if (!tryToParseBracedList()) 1928 parseChildBlock(); 1929 break; 1930 } 1931 case tok::at: 1932 nextToken(); 1933 if (FormatTok->Tok.is(tok::l_brace)) { 1934 nextToken(); 1935 parseBracedList(); 1936 } 1937 break; 1938 default: 1939 nextToken(); 1940 break; 1941 } 1942 } while (!eof()); 1943 } 1944 1945 void UnwrappedLineParser::parseIfThenElse() { 1946 assert(FormatTok->Tok.is(tok::kw_if) && "'if' expected"); 1947 nextToken(); 1948 if (FormatTok->Tok.isOneOf(tok::kw_constexpr, tok::identifier)) 1949 nextToken(); 1950 if (FormatTok->Tok.is(tok::l_paren)) 1951 parseParens(); 1952 bool NeedsUnwrappedLine = false; 1953 if (FormatTok->Tok.is(tok::l_brace)) { 1954 CompoundStatementIndenter Indenter(this, Style, Line->Level); 1955 parseBlock(/*MustBeDeclaration=*/false); 1956 if (Style.BraceWrapping.BeforeElse) 1957 addUnwrappedLine(); 1958 else 1959 NeedsUnwrappedLine = true; 1960 } else { 1961 addUnwrappedLine(); 1962 ++Line->Level; 1963 parseStructuralElement(); 1964 --Line->Level; 1965 } 1966 if (FormatTok->Tok.is(tok::kw_else)) { 1967 nextToken(); 1968 if (FormatTok->Tok.is(tok::l_brace)) { 1969 CompoundStatementIndenter Indenter(this, Style, Line->Level); 1970 parseBlock(/*MustBeDeclaration=*/false); 1971 addUnwrappedLine(); 1972 } else if (FormatTok->Tok.is(tok::kw_if)) { 1973 parseIfThenElse(); 1974 } else { 1975 addUnwrappedLine(); 1976 ++Line->Level; 1977 parseStructuralElement(); 1978 if (FormatTok->is(tok::eof)) 1979 addUnwrappedLine(); 1980 --Line->Level; 1981 } 1982 } else if (NeedsUnwrappedLine) { 1983 addUnwrappedLine(); 1984 } 1985 } 1986 1987 void UnwrappedLineParser::parseTryCatch() { 1988 assert(FormatTok->isOneOf(tok::kw_try, tok::kw___try) && "'try' expected"); 1989 nextToken(); 1990 bool NeedsUnwrappedLine = false; 1991 if (FormatTok->is(tok::colon)) { 1992 // We are in a function try block, what comes is an initializer list. 1993 nextToken(); 1994 1995 // In case identifiers were removed by clang-tidy, what might follow is 1996 // multiple commas in sequence - before the first identifier. 1997 while (FormatTok->is(tok::comma)) 1998 nextToken(); 1999 2000 while (FormatTok->is(tok::identifier)) { 2001 nextToken(); 2002 if (FormatTok->is(tok::l_paren)) 2003 parseParens(); 2004 2005 // In case identifiers were removed by clang-tidy, what might follow is 2006 // multiple commas in sequence - after the first identifier. 2007 while (FormatTok->is(tok::comma)) 2008 nextToken(); 2009 } 2010 } 2011 // Parse try with resource. 2012 if (Style.Language == FormatStyle::LK_Java && FormatTok->is(tok::l_paren)) { 2013 parseParens(); 2014 } 2015 if (FormatTok->is(tok::l_brace)) { 2016 CompoundStatementIndenter Indenter(this, Style, Line->Level); 2017 parseBlock(/*MustBeDeclaration=*/false); 2018 if (Style.BraceWrapping.BeforeCatch) { 2019 addUnwrappedLine(); 2020 } else { 2021 NeedsUnwrappedLine = true; 2022 } 2023 } else if (!FormatTok->is(tok::kw_catch)) { 2024 // The C++ standard requires a compound-statement after a try. 2025 // If there's none, we try to assume there's a structuralElement 2026 // and try to continue. 2027 addUnwrappedLine(); 2028 ++Line->Level; 2029 parseStructuralElement(); 2030 --Line->Level; 2031 } 2032 while (1) { 2033 if (FormatTok->is(tok::at)) 2034 nextToken(); 2035 if (!(FormatTok->isOneOf(tok::kw_catch, Keywords.kw___except, 2036 tok::kw___finally) || 2037 ((Style.Language == FormatStyle::LK_Java || 2038 Style.Language == FormatStyle::LK_JavaScript) && 2039 FormatTok->is(Keywords.kw_finally)) || 2040 (FormatTok->Tok.isObjCAtKeyword(tok::objc_catch) || 2041 FormatTok->Tok.isObjCAtKeyword(tok::objc_finally)))) 2042 break; 2043 nextToken(); 2044 while (FormatTok->isNot(tok::l_brace)) { 2045 if (FormatTok->is(tok::l_paren)) { 2046 parseParens(); 2047 continue; 2048 } 2049 if (FormatTok->isOneOf(tok::semi, tok::r_brace, tok::eof)) 2050 return; 2051 nextToken(); 2052 } 2053 NeedsUnwrappedLine = false; 2054 CompoundStatementIndenter Indenter(this, Style, Line->Level); 2055 parseBlock(/*MustBeDeclaration=*/false); 2056 if (Style.BraceWrapping.BeforeCatch) 2057 addUnwrappedLine(); 2058 else 2059 NeedsUnwrappedLine = true; 2060 } 2061 if (NeedsUnwrappedLine) 2062 addUnwrappedLine(); 2063 } 2064 2065 void UnwrappedLineParser::parseNamespace() { 2066 assert(FormatTok->isOneOf(tok::kw_namespace, TT_NamespaceMacro) && 2067 "'namespace' expected"); 2068 2069 const FormatToken &InitialToken = *FormatTok; 2070 nextToken(); 2071 if (InitialToken.is(TT_NamespaceMacro)) { 2072 parseParens(); 2073 } else { 2074 while (FormatTok->isOneOf(tok::identifier, tok::coloncolon, tok::kw_inline, 2075 tok::l_square)) { 2076 if (FormatTok->is(tok::l_square)) 2077 parseSquare(); 2078 else 2079 nextToken(); 2080 } 2081 } 2082 if (FormatTok->Tok.is(tok::l_brace)) { 2083 if (ShouldBreakBeforeBrace(Style, InitialToken)) 2084 addUnwrappedLine(); 2085 2086 bool AddLevel = Style.NamespaceIndentation == FormatStyle::NI_All || 2087 (Style.NamespaceIndentation == FormatStyle::NI_Inner && 2088 DeclarationScopeStack.size() > 1); 2089 parseBlock(/*MustBeDeclaration=*/true, AddLevel); 2090 // Munch the semicolon after a namespace. This is more common than one would 2091 // think. Putting the semicolon into its own line is very ugly. 2092 if (FormatTok->Tok.is(tok::semi)) 2093 nextToken(); 2094 addUnwrappedLine(); 2095 } 2096 // FIXME: Add error handling. 2097 } 2098 2099 void UnwrappedLineParser::parseNew() { 2100 assert(FormatTok->is(tok::kw_new) && "'new' expected"); 2101 nextToken(); 2102 2103 if (Style.isCSharp()) { 2104 do { 2105 if (FormatTok->is(tok::l_brace)) 2106 parseBracedList(); 2107 2108 if (FormatTok->isOneOf(tok::semi, tok::comma)) 2109 return; 2110 2111 nextToken(); 2112 } while (!eof()); 2113 } 2114 2115 if (Style.Language != FormatStyle::LK_Java) 2116 return; 2117 2118 // In Java, we can parse everything up to the parens, which aren't optional. 2119 do { 2120 // There should not be a ;, { or } before the new's open paren. 2121 if (FormatTok->isOneOf(tok::semi, tok::l_brace, tok::r_brace)) 2122 return; 2123 2124 // Consume the parens. 2125 if (FormatTok->is(tok::l_paren)) { 2126 parseParens(); 2127 2128 // If there is a class body of an anonymous class, consume that as child. 2129 if (FormatTok->is(tok::l_brace)) 2130 parseChildBlock(); 2131 return; 2132 } 2133 nextToken(); 2134 } while (!eof()); 2135 } 2136 2137 void UnwrappedLineParser::parseForOrWhileLoop() { 2138 assert(FormatTok->isOneOf(tok::kw_for, tok::kw_while, TT_ForEachMacro) && 2139 "'for', 'while' or foreach macro expected"); 2140 nextToken(); 2141 // JS' for await ( ... 2142 if (Style.Language == FormatStyle::LK_JavaScript && 2143 FormatTok->is(Keywords.kw_await)) 2144 nextToken(); 2145 if (FormatTok->Tok.is(tok::l_paren)) 2146 parseParens(); 2147 if (FormatTok->Tok.is(tok::l_brace)) { 2148 CompoundStatementIndenter Indenter(this, Style, Line->Level); 2149 parseBlock(/*MustBeDeclaration=*/false); 2150 addUnwrappedLine(); 2151 } else { 2152 addUnwrappedLine(); 2153 ++Line->Level; 2154 parseStructuralElement(); 2155 --Line->Level; 2156 } 2157 } 2158 2159 void UnwrappedLineParser::parseDoWhile() { 2160 assert(FormatTok->Tok.is(tok::kw_do) && "'do' expected"); 2161 nextToken(); 2162 if (FormatTok->Tok.is(tok::l_brace)) { 2163 CompoundStatementIndenter Indenter(this, Style, Line->Level); 2164 parseBlock(/*MustBeDeclaration=*/false); 2165 if (Style.BraceWrapping.IndentBraces) 2166 addUnwrappedLine(); 2167 } else { 2168 addUnwrappedLine(); 2169 ++Line->Level; 2170 parseStructuralElement(); 2171 --Line->Level; 2172 } 2173 2174 // FIXME: Add error handling. 2175 if (!FormatTok->Tok.is(tok::kw_while)) { 2176 addUnwrappedLine(); 2177 return; 2178 } 2179 2180 nextToken(); 2181 parseStructuralElement(); 2182 } 2183 2184 void UnwrappedLineParser::parseLabel(bool LeftAlignLabel) { 2185 nextToken(); 2186 unsigned OldLineLevel = Line->Level; 2187 if (Line->Level > 1 || (!Line->InPPDirective && Line->Level > 0)) 2188 --Line->Level; 2189 if (LeftAlignLabel) 2190 Line->Level = 0; 2191 if (!Style.IndentCaseBlocks && CommentsBeforeNextToken.empty() && 2192 FormatTok->Tok.is(tok::l_brace)) { 2193 CompoundStatementIndenter Indenter(this, Line->Level, 2194 Style.BraceWrapping.AfterCaseLabel, 2195 Style.BraceWrapping.IndentBraces); 2196 parseBlock(/*MustBeDeclaration=*/false); 2197 if (FormatTok->Tok.is(tok::kw_break)) { 2198 if (Style.BraceWrapping.AfterControlStatement == 2199 FormatStyle::BWACS_Always) 2200 addUnwrappedLine(); 2201 parseStructuralElement(); 2202 } 2203 addUnwrappedLine(); 2204 } else { 2205 if (FormatTok->is(tok::semi)) 2206 nextToken(); 2207 addUnwrappedLine(); 2208 } 2209 Line->Level = OldLineLevel; 2210 if (FormatTok->isNot(tok::l_brace)) { 2211 parseStructuralElement(); 2212 addUnwrappedLine(); 2213 } 2214 } 2215 2216 void UnwrappedLineParser::parseCaseLabel() { 2217 assert(FormatTok->Tok.is(tok::kw_case) && "'case' expected"); 2218 // FIXME: fix handling of complex expressions here. 2219 do { 2220 nextToken(); 2221 } while (!eof() && !FormatTok->Tok.is(tok::colon)); 2222 parseLabel(); 2223 } 2224 2225 void UnwrappedLineParser::parseSwitch() { 2226 assert(FormatTok->Tok.is(tok::kw_switch) && "'switch' expected"); 2227 nextToken(); 2228 if (FormatTok->Tok.is(tok::l_paren)) 2229 parseParens(); 2230 if (FormatTok->Tok.is(tok::l_brace)) { 2231 CompoundStatementIndenter Indenter(this, Style, Line->Level); 2232 parseBlock(/*MustBeDeclaration=*/false); 2233 addUnwrappedLine(); 2234 } else { 2235 addUnwrappedLine(); 2236 ++Line->Level; 2237 parseStructuralElement(); 2238 --Line->Level; 2239 } 2240 } 2241 2242 void UnwrappedLineParser::parseAccessSpecifier() { 2243 nextToken(); 2244 // Understand Qt's slots. 2245 if (FormatTok->isOneOf(Keywords.kw_slots, Keywords.kw_qslots)) 2246 nextToken(); 2247 // Otherwise, we don't know what it is, and we'd better keep the next token. 2248 if (FormatTok->Tok.is(tok::colon)) 2249 nextToken(); 2250 addUnwrappedLine(); 2251 } 2252 2253 bool UnwrappedLineParser::parseEnum() { 2254 // Won't be 'enum' for NS_ENUMs. 2255 if (FormatTok->Tok.is(tok::kw_enum)) 2256 nextToken(); 2257 2258 // In TypeScript, "enum" can also be used as property name, e.g. in interface 2259 // declarations. An "enum" keyword followed by a colon would be a syntax 2260 // error and thus assume it is just an identifier. 2261 if (Style.Language == FormatStyle::LK_JavaScript && 2262 FormatTok->isOneOf(tok::colon, tok::question)) 2263 return false; 2264 2265 // In protobuf, "enum" can be used as a field name. 2266 if (Style.Language == FormatStyle::LK_Proto && FormatTok->is(tok::equal)) 2267 return false; 2268 2269 // Eat up enum class ... 2270 if (FormatTok->Tok.is(tok::kw_class) || FormatTok->Tok.is(tok::kw_struct)) 2271 nextToken(); 2272 2273 while (FormatTok->Tok.getIdentifierInfo() || 2274 FormatTok->isOneOf(tok::colon, tok::coloncolon, tok::less, 2275 tok::greater, tok::comma, tok::question)) { 2276 nextToken(); 2277 // We can have macros or attributes in between 'enum' and the enum name. 2278 if (FormatTok->is(tok::l_paren)) 2279 parseParens(); 2280 if (FormatTok->is(tok::identifier)) { 2281 nextToken(); 2282 // If there are two identifiers in a row, this is likely an elaborate 2283 // return type. In Java, this can be "implements", etc. 2284 if (Style.isCpp() && FormatTok->is(tok::identifier)) 2285 return false; 2286 } 2287 } 2288 2289 // Just a declaration or something is wrong. 2290 if (FormatTok->isNot(tok::l_brace)) 2291 return true; 2292 FormatTok->BlockKind = BK_Block; 2293 2294 if (Style.Language == FormatStyle::LK_Java) { 2295 // Java enums are different. 2296 parseJavaEnumBody(); 2297 return true; 2298 } 2299 if (Style.Language == FormatStyle::LK_Proto) { 2300 parseBlock(/*MustBeDeclaration=*/true); 2301 return true; 2302 } 2303 2304 // Parse enum body. 2305 nextToken(); 2306 bool HasError = !parseBracedList(/*ContinueOnSemicolons=*/true); 2307 if (HasError) { 2308 if (FormatTok->is(tok::semi)) 2309 nextToken(); 2310 addUnwrappedLine(); 2311 } 2312 return true; 2313 2314 // There is no addUnwrappedLine() here so that we fall through to parsing a 2315 // structural element afterwards. Thus, in "enum A {} n, m;", 2316 // "} n, m;" will end up in one unwrapped line. 2317 } 2318 2319 void UnwrappedLineParser::parseJavaEnumBody() { 2320 // Determine whether the enum is simple, i.e. does not have a semicolon or 2321 // constants with class bodies. Simple enums can be formatted like braced 2322 // lists, contracted to a single line, etc. 2323 unsigned StoredPosition = Tokens->getPosition(); 2324 bool IsSimple = true; 2325 FormatToken *Tok = Tokens->getNextToken(); 2326 while (Tok) { 2327 if (Tok->is(tok::r_brace)) 2328 break; 2329 if (Tok->isOneOf(tok::l_brace, tok::semi)) { 2330 IsSimple = false; 2331 break; 2332 } 2333 // FIXME: This will also mark enums with braces in the arguments to enum 2334 // constants as "not simple". This is probably fine in practice, though. 2335 Tok = Tokens->getNextToken(); 2336 } 2337 FormatTok = Tokens->setPosition(StoredPosition); 2338 2339 if (IsSimple) { 2340 nextToken(); 2341 parseBracedList(); 2342 addUnwrappedLine(); 2343 return; 2344 } 2345 2346 // Parse the body of a more complex enum. 2347 // First add a line for everything up to the "{". 2348 nextToken(); 2349 addUnwrappedLine(); 2350 ++Line->Level; 2351 2352 // Parse the enum constants. 2353 while (FormatTok) { 2354 if (FormatTok->is(tok::l_brace)) { 2355 // Parse the constant's class body. 2356 parseBlock(/*MustBeDeclaration=*/true, /*AddLevel=*/true, 2357 /*MunchSemi=*/false); 2358 } else if (FormatTok->is(tok::l_paren)) { 2359 parseParens(); 2360 } else if (FormatTok->is(tok::comma)) { 2361 nextToken(); 2362 addUnwrappedLine(); 2363 } else if (FormatTok->is(tok::semi)) { 2364 nextToken(); 2365 addUnwrappedLine(); 2366 break; 2367 } else if (FormatTok->is(tok::r_brace)) { 2368 addUnwrappedLine(); 2369 break; 2370 } else { 2371 nextToken(); 2372 } 2373 } 2374 2375 // Parse the class body after the enum's ";" if any. 2376 parseLevel(/*HasOpeningBrace=*/true); 2377 nextToken(); 2378 --Line->Level; 2379 addUnwrappedLine(); 2380 } 2381 2382 void UnwrappedLineParser::parseRecord(bool ParseAsExpr) { 2383 const FormatToken &InitialToken = *FormatTok; 2384 nextToken(); 2385 2386 // The actual identifier can be a nested name specifier, and in macros 2387 // it is often token-pasted. 2388 while (FormatTok->isOneOf(tok::identifier, tok::coloncolon, tok::hashhash, 2389 tok::kw___attribute, tok::kw___declspec, 2390 tok::kw_alignas) || 2391 ((Style.Language == FormatStyle::LK_Java || 2392 Style.Language == FormatStyle::LK_JavaScript) && 2393 FormatTok->isOneOf(tok::period, tok::comma))) { 2394 if (Style.Language == FormatStyle::LK_JavaScript && 2395 FormatTok->isOneOf(Keywords.kw_extends, Keywords.kw_implements)) { 2396 // JavaScript/TypeScript supports inline object types in 2397 // extends/implements positions: 2398 // class Foo implements {bar: number} { } 2399 nextToken(); 2400 if (FormatTok->is(tok::l_brace)) { 2401 tryToParseBracedList(); 2402 continue; 2403 } 2404 } 2405 bool IsNonMacroIdentifier = 2406 FormatTok->is(tok::identifier) && 2407 FormatTok->TokenText != FormatTok->TokenText.upper(); 2408 nextToken(); 2409 // We can have macros or attributes in between 'class' and the class name. 2410 if (!IsNonMacroIdentifier && FormatTok->Tok.is(tok::l_paren)) 2411 parseParens(); 2412 } 2413 2414 // Note that parsing away template declarations here leads to incorrectly 2415 // accepting function declarations as record declarations. 2416 // In general, we cannot solve this problem. Consider: 2417 // class A<int> B() {} 2418 // which can be a function definition or a class definition when B() is a 2419 // macro. If we find enough real-world cases where this is a problem, we 2420 // can parse for the 'template' keyword in the beginning of the statement, 2421 // and thus rule out the record production in case there is no template 2422 // (this would still leave us with an ambiguity between template function 2423 // and class declarations). 2424 if (FormatTok->isOneOf(tok::colon, tok::less)) { 2425 while (!eof()) { 2426 if (FormatTok->is(tok::l_brace)) { 2427 calculateBraceTypes(/*ExpectClassBody=*/true); 2428 if (!tryToParseBracedList()) 2429 break; 2430 } 2431 if (FormatTok->Tok.is(tok::semi)) 2432 return; 2433 if (Style.isCSharp() && FormatTok->is(Keywords.kw_where)) { 2434 addUnwrappedLine(); 2435 nextToken(); 2436 parseCSharpGenericTypeConstraint(); 2437 break; 2438 } 2439 nextToken(); 2440 } 2441 } 2442 if (FormatTok->Tok.is(tok::l_brace)) { 2443 if (ParseAsExpr) { 2444 parseChildBlock(); 2445 } else { 2446 if (ShouldBreakBeforeBrace(Style, InitialToken)) 2447 addUnwrappedLine(); 2448 2449 parseBlock(/*MustBeDeclaration=*/true, /*AddLevel=*/true, 2450 /*MunchSemi=*/false); 2451 } 2452 } 2453 // There is no addUnwrappedLine() here so that we fall through to parsing a 2454 // structural element afterwards. Thus, in "class A {} n, m;", 2455 // "} n, m;" will end up in one unwrapped line. 2456 } 2457 2458 void UnwrappedLineParser::parseObjCMethod() { 2459 assert(FormatTok->Tok.isOneOf(tok::l_paren, tok::identifier) && 2460 "'(' or identifier expected."); 2461 do { 2462 if (FormatTok->Tok.is(tok::semi)) { 2463 nextToken(); 2464 addUnwrappedLine(); 2465 return; 2466 } else if (FormatTok->Tok.is(tok::l_brace)) { 2467 if (Style.BraceWrapping.AfterFunction) 2468 addUnwrappedLine(); 2469 parseBlock(/*MustBeDeclaration=*/false); 2470 addUnwrappedLine(); 2471 return; 2472 } else { 2473 nextToken(); 2474 } 2475 } while (!eof()); 2476 } 2477 2478 void UnwrappedLineParser::parseObjCProtocolList() { 2479 assert(FormatTok->Tok.is(tok::less) && "'<' expected."); 2480 do { 2481 nextToken(); 2482 // Early exit in case someone forgot a close angle. 2483 if (FormatTok->isOneOf(tok::semi, tok::l_brace) || 2484 FormatTok->Tok.isObjCAtKeyword(tok::objc_end)) 2485 return; 2486 } while (!eof() && FormatTok->Tok.isNot(tok::greater)); 2487 nextToken(); // Skip '>'. 2488 } 2489 2490 void UnwrappedLineParser::parseObjCUntilAtEnd() { 2491 do { 2492 if (FormatTok->Tok.isObjCAtKeyword(tok::objc_end)) { 2493 nextToken(); 2494 addUnwrappedLine(); 2495 break; 2496 } 2497 if (FormatTok->is(tok::l_brace)) { 2498 parseBlock(/*MustBeDeclaration=*/false); 2499 // In ObjC interfaces, nothing should be following the "}". 2500 addUnwrappedLine(); 2501 } else if (FormatTok->is(tok::r_brace)) { 2502 // Ignore stray "}". parseStructuralElement doesn't consume them. 2503 nextToken(); 2504 addUnwrappedLine(); 2505 } else if (FormatTok->isOneOf(tok::minus, tok::plus)) { 2506 nextToken(); 2507 parseObjCMethod(); 2508 } else { 2509 parseStructuralElement(); 2510 } 2511 } while (!eof()); 2512 } 2513 2514 void UnwrappedLineParser::parseObjCInterfaceOrImplementation() { 2515 assert(FormatTok->Tok.getObjCKeywordID() == tok::objc_interface || 2516 FormatTok->Tok.getObjCKeywordID() == tok::objc_implementation); 2517 nextToken(); 2518 nextToken(); // interface name 2519 2520 // @interface can be followed by a lightweight generic 2521 // specialization list, then either a base class or a category. 2522 if (FormatTok->Tok.is(tok::less)) { 2523 // Unlike protocol lists, generic parameterizations support 2524 // nested angles: 2525 // 2526 // @interface Foo<ValueType : id <NSCopying, NSSecureCoding>> : 2527 // NSObject <NSCopying, NSSecureCoding> 2528 // 2529 // so we need to count how many open angles we have left. 2530 unsigned NumOpenAngles = 1; 2531 do { 2532 nextToken(); 2533 // Early exit in case someone forgot a close angle. 2534 if (FormatTok->isOneOf(tok::semi, tok::l_brace) || 2535 FormatTok->Tok.isObjCAtKeyword(tok::objc_end)) 2536 break; 2537 if (FormatTok->Tok.is(tok::less)) 2538 ++NumOpenAngles; 2539 else if (FormatTok->Tok.is(tok::greater)) { 2540 assert(NumOpenAngles > 0 && "'>' makes NumOpenAngles negative"); 2541 --NumOpenAngles; 2542 } 2543 } while (!eof() && NumOpenAngles != 0); 2544 nextToken(); // Skip '>'. 2545 } 2546 if (FormatTok->Tok.is(tok::colon)) { 2547 nextToken(); 2548 nextToken(); // base class name 2549 } else if (FormatTok->Tok.is(tok::l_paren)) 2550 // Skip category, if present. 2551 parseParens(); 2552 2553 if (FormatTok->Tok.is(tok::less)) 2554 parseObjCProtocolList(); 2555 2556 if (FormatTok->Tok.is(tok::l_brace)) { 2557 if (Style.BraceWrapping.AfterObjCDeclaration) 2558 addUnwrappedLine(); 2559 parseBlock(/*MustBeDeclaration=*/true); 2560 } 2561 2562 // With instance variables, this puts '}' on its own line. Without instance 2563 // variables, this ends the @interface line. 2564 addUnwrappedLine(); 2565 2566 parseObjCUntilAtEnd(); 2567 } 2568 2569 // Returns true for the declaration/definition form of @protocol, 2570 // false for the expression form. 2571 bool UnwrappedLineParser::parseObjCProtocol() { 2572 assert(FormatTok->Tok.getObjCKeywordID() == tok::objc_protocol); 2573 nextToken(); 2574 2575 if (FormatTok->is(tok::l_paren)) 2576 // The expression form of @protocol, e.g. "Protocol* p = @protocol(foo);". 2577 return false; 2578 2579 // The definition/declaration form, 2580 // @protocol Foo 2581 // - (int)someMethod; 2582 // @end 2583 2584 nextToken(); // protocol name 2585 2586 if (FormatTok->Tok.is(tok::less)) 2587 parseObjCProtocolList(); 2588 2589 // Check for protocol declaration. 2590 if (FormatTok->Tok.is(tok::semi)) { 2591 nextToken(); 2592 addUnwrappedLine(); 2593 return true; 2594 } 2595 2596 addUnwrappedLine(); 2597 parseObjCUntilAtEnd(); 2598 return true; 2599 } 2600 2601 void UnwrappedLineParser::parseJavaScriptEs6ImportExport() { 2602 bool IsImport = FormatTok->is(Keywords.kw_import); 2603 assert(IsImport || FormatTok->is(tok::kw_export)); 2604 nextToken(); 2605 2606 // Consume the "default" in "export default class/function". 2607 if (FormatTok->is(tok::kw_default)) 2608 nextToken(); 2609 2610 // Consume "async function", "function" and "default function", so that these 2611 // get parsed as free-standing JS functions, i.e. do not require a trailing 2612 // semicolon. 2613 if (FormatTok->is(Keywords.kw_async)) 2614 nextToken(); 2615 if (FormatTok->is(Keywords.kw_function)) { 2616 nextToken(); 2617 return; 2618 } 2619 2620 // For imports, `export *`, `export {...}`, consume the rest of the line up 2621 // to the terminating `;`. For everything else, just return and continue 2622 // parsing the structural element, i.e. the declaration or expression for 2623 // `export default`. 2624 if (!IsImport && !FormatTok->isOneOf(tok::l_brace, tok::star) && 2625 !FormatTok->isStringLiteral()) 2626 return; 2627 2628 while (!eof()) { 2629 if (FormatTok->is(tok::semi)) 2630 return; 2631 if (Line->Tokens.empty()) { 2632 // Common issue: Automatic Semicolon Insertion wrapped the line, so the 2633 // import statement should terminate. 2634 return; 2635 } 2636 if (FormatTok->is(tok::l_brace)) { 2637 FormatTok->BlockKind = BK_Block; 2638 nextToken(); 2639 parseBracedList(); 2640 } else { 2641 nextToken(); 2642 } 2643 } 2644 } 2645 2646 void UnwrappedLineParser::parseStatementMacro() { 2647 nextToken(); 2648 if (FormatTok->is(tok::l_paren)) 2649 parseParens(); 2650 if (FormatTok->is(tok::semi)) 2651 nextToken(); 2652 addUnwrappedLine(); 2653 } 2654 2655 LLVM_ATTRIBUTE_UNUSED static void printDebugInfo(const UnwrappedLine &Line, 2656 StringRef Prefix = "") { 2657 llvm::dbgs() << Prefix << "Line(" << Line.Level 2658 << ", FSC=" << Line.FirstStartColumn << ")" 2659 << (Line.InPPDirective ? " MACRO" : "") << ": "; 2660 for (std::list<UnwrappedLineNode>::const_iterator I = Line.Tokens.begin(), 2661 E = Line.Tokens.end(); 2662 I != E; ++I) { 2663 llvm::dbgs() << I->Tok->Tok.getName() << "[" 2664 << "T=" << I->Tok->Type << ", OC=" << I->Tok->OriginalColumn 2665 << "] "; 2666 } 2667 for (std::list<UnwrappedLineNode>::const_iterator I = Line.Tokens.begin(), 2668 E = Line.Tokens.end(); 2669 I != E; ++I) { 2670 const UnwrappedLineNode &Node = *I; 2671 for (SmallVectorImpl<UnwrappedLine>::const_iterator 2672 I = Node.Children.begin(), 2673 E = Node.Children.end(); 2674 I != E; ++I) { 2675 printDebugInfo(*I, "\nChild: "); 2676 } 2677 } 2678 llvm::dbgs() << "\n"; 2679 } 2680 2681 void UnwrappedLineParser::addUnwrappedLine() { 2682 if (Line->Tokens.empty()) 2683 return; 2684 LLVM_DEBUG({ 2685 if (CurrentLines == &Lines) 2686 printDebugInfo(*Line); 2687 }); 2688 CurrentLines->push_back(std::move(*Line)); 2689 Line->Tokens.clear(); 2690 Line->MatchingOpeningBlockLineIndex = UnwrappedLine::kInvalidIndex; 2691 Line->FirstStartColumn = 0; 2692 if (CurrentLines == &Lines && !PreprocessorDirectives.empty()) { 2693 CurrentLines->append( 2694 std::make_move_iterator(PreprocessorDirectives.begin()), 2695 std::make_move_iterator(PreprocessorDirectives.end())); 2696 PreprocessorDirectives.clear(); 2697 } 2698 // Disconnect the current token from the last token on the previous line. 2699 FormatTok->Previous = nullptr; 2700 } 2701 2702 bool UnwrappedLineParser::eof() const { return FormatTok->Tok.is(tok::eof); } 2703 2704 bool UnwrappedLineParser::isOnNewLine(const FormatToken &FormatTok) { 2705 return (Line->InPPDirective || FormatTok.HasUnescapedNewline) && 2706 FormatTok.NewlinesBefore > 0; 2707 } 2708 2709 // Checks if \p FormatTok is a line comment that continues the line comment 2710 // section on \p Line. 2711 static bool 2712 continuesLineCommentSection(const FormatToken &FormatTok, 2713 const UnwrappedLine &Line, 2714 const llvm::Regex &CommentPragmasRegex) { 2715 if (Line.Tokens.empty()) 2716 return false; 2717 2718 StringRef IndentContent = FormatTok.TokenText; 2719 if (FormatTok.TokenText.startswith("//") || 2720 FormatTok.TokenText.startswith("/*")) 2721 IndentContent = FormatTok.TokenText.substr(2); 2722 if (CommentPragmasRegex.match(IndentContent)) 2723 return false; 2724 2725 // If Line starts with a line comment, then FormatTok continues the comment 2726 // section if its original column is greater or equal to the original start 2727 // column of the line. 2728 // 2729 // Define the min column token of a line as follows: if a line ends in '{' or 2730 // contains a '{' followed by a line comment, then the min column token is 2731 // that '{'. Otherwise, the min column token of the line is the first token of 2732 // the line. 2733 // 2734 // If Line starts with a token other than a line comment, then FormatTok 2735 // continues the comment section if its original column is greater than the 2736 // original start column of the min column token of the line. 2737 // 2738 // For example, the second line comment continues the first in these cases: 2739 // 2740 // // first line 2741 // // second line 2742 // 2743 // and: 2744 // 2745 // // first line 2746 // // second line 2747 // 2748 // and: 2749 // 2750 // int i; // first line 2751 // // second line 2752 // 2753 // and: 2754 // 2755 // do { // first line 2756 // // second line 2757 // int i; 2758 // } while (true); 2759 // 2760 // and: 2761 // 2762 // enum { 2763 // a, // first line 2764 // // second line 2765 // b 2766 // }; 2767 // 2768 // The second line comment doesn't continue the first in these cases: 2769 // 2770 // // first line 2771 // // second line 2772 // 2773 // and: 2774 // 2775 // int i; // first line 2776 // // second line 2777 // 2778 // and: 2779 // 2780 // do { // first line 2781 // // second line 2782 // int i; 2783 // } while (true); 2784 // 2785 // and: 2786 // 2787 // enum { 2788 // a, // first line 2789 // // second line 2790 // }; 2791 const FormatToken *MinColumnToken = Line.Tokens.front().Tok; 2792 2793 // Scan for '{//'. If found, use the column of '{' as a min column for line 2794 // comment section continuation. 2795 const FormatToken *PreviousToken = nullptr; 2796 for (const UnwrappedLineNode &Node : Line.Tokens) { 2797 if (PreviousToken && PreviousToken->is(tok::l_brace) && 2798 isLineComment(*Node.Tok)) { 2799 MinColumnToken = PreviousToken; 2800 break; 2801 } 2802 PreviousToken = Node.Tok; 2803 2804 // Grab the last newline preceding a token in this unwrapped line. 2805 if (Node.Tok->NewlinesBefore > 0) { 2806 MinColumnToken = Node.Tok; 2807 } 2808 } 2809 if (PreviousToken && PreviousToken->is(tok::l_brace)) { 2810 MinColumnToken = PreviousToken; 2811 } 2812 2813 return continuesLineComment(FormatTok, /*Previous=*/Line.Tokens.back().Tok, 2814 MinColumnToken); 2815 } 2816 2817 void UnwrappedLineParser::flushComments(bool NewlineBeforeNext) { 2818 bool JustComments = Line->Tokens.empty(); 2819 for (SmallVectorImpl<FormatToken *>::const_iterator 2820 I = CommentsBeforeNextToken.begin(), 2821 E = CommentsBeforeNextToken.end(); 2822 I != E; ++I) { 2823 // Line comments that belong to the same line comment section are put on the 2824 // same line since later we might want to reflow content between them. 2825 // Additional fine-grained breaking of line comment sections is controlled 2826 // by the class BreakableLineCommentSection in case it is desirable to keep 2827 // several line comment sections in the same unwrapped line. 2828 // 2829 // FIXME: Consider putting separate line comment sections as children to the 2830 // unwrapped line instead. 2831 (*I)->ContinuesLineCommentSection = 2832 continuesLineCommentSection(**I, *Line, CommentPragmasRegex); 2833 if (isOnNewLine(**I) && JustComments && !(*I)->ContinuesLineCommentSection) 2834 addUnwrappedLine(); 2835 pushToken(*I); 2836 } 2837 if (NewlineBeforeNext && JustComments) 2838 addUnwrappedLine(); 2839 CommentsBeforeNextToken.clear(); 2840 } 2841 2842 void UnwrappedLineParser::nextToken(int LevelDifference) { 2843 if (eof()) 2844 return; 2845 flushComments(isOnNewLine(*FormatTok)); 2846 pushToken(FormatTok); 2847 FormatToken *Previous = FormatTok; 2848 if (Style.Language != FormatStyle::LK_JavaScript) 2849 readToken(LevelDifference); 2850 else 2851 readTokenWithJavaScriptASI(); 2852 FormatTok->Previous = Previous; 2853 } 2854 2855 void UnwrappedLineParser::distributeComments( 2856 const SmallVectorImpl<FormatToken *> &Comments, 2857 const FormatToken *NextTok) { 2858 // Whether or not a line comment token continues a line is controlled by 2859 // the method continuesLineCommentSection, with the following caveat: 2860 // 2861 // Define a trail of Comments to be a nonempty proper postfix of Comments such 2862 // that each comment line from the trail is aligned with the next token, if 2863 // the next token exists. If a trail exists, the beginning of the maximal 2864 // trail is marked as a start of a new comment section. 2865 // 2866 // For example in this code: 2867 // 2868 // int a; // line about a 2869 // // line 1 about b 2870 // // line 2 about b 2871 // int b; 2872 // 2873 // the two lines about b form a maximal trail, so there are two sections, the 2874 // first one consisting of the single comment "// line about a" and the 2875 // second one consisting of the next two comments. 2876 if (Comments.empty()) 2877 return; 2878 bool ShouldPushCommentsInCurrentLine = true; 2879 bool HasTrailAlignedWithNextToken = false; 2880 unsigned StartOfTrailAlignedWithNextToken = 0; 2881 if (NextTok) { 2882 // We are skipping the first element intentionally. 2883 for (unsigned i = Comments.size() - 1; i > 0; --i) { 2884 if (Comments[i]->OriginalColumn == NextTok->OriginalColumn) { 2885 HasTrailAlignedWithNextToken = true; 2886 StartOfTrailAlignedWithNextToken = i; 2887 } 2888 } 2889 } 2890 for (unsigned i = 0, e = Comments.size(); i < e; ++i) { 2891 FormatToken *FormatTok = Comments[i]; 2892 if (HasTrailAlignedWithNextToken && i == StartOfTrailAlignedWithNextToken) { 2893 FormatTok->ContinuesLineCommentSection = false; 2894 } else { 2895 FormatTok->ContinuesLineCommentSection = 2896 continuesLineCommentSection(*FormatTok, *Line, CommentPragmasRegex); 2897 } 2898 if (!FormatTok->ContinuesLineCommentSection && 2899 (isOnNewLine(*FormatTok) || FormatTok->IsFirst)) { 2900 ShouldPushCommentsInCurrentLine = false; 2901 } 2902 if (ShouldPushCommentsInCurrentLine) { 2903 pushToken(FormatTok); 2904 } else { 2905 CommentsBeforeNextToken.push_back(FormatTok); 2906 } 2907 } 2908 } 2909 2910 void UnwrappedLineParser::readToken(int LevelDifference) { 2911 SmallVector<FormatToken *, 1> Comments; 2912 do { 2913 FormatTok = Tokens->getNextToken(); 2914 assert(FormatTok); 2915 while (!Line->InPPDirective && FormatTok->Tok.is(tok::hash) && 2916 (FormatTok->HasUnescapedNewline || FormatTok->IsFirst)) { 2917 distributeComments(Comments, FormatTok); 2918 Comments.clear(); 2919 // If there is an unfinished unwrapped line, we flush the preprocessor 2920 // directives only after that unwrapped line was finished later. 2921 bool SwitchToPreprocessorLines = !Line->Tokens.empty(); 2922 ScopedLineState BlockState(*this, SwitchToPreprocessorLines); 2923 assert((LevelDifference >= 0 || 2924 static_cast<unsigned>(-LevelDifference) <= Line->Level) && 2925 "LevelDifference makes Line->Level negative"); 2926 Line->Level += LevelDifference; 2927 // Comments stored before the preprocessor directive need to be output 2928 // before the preprocessor directive, at the same level as the 2929 // preprocessor directive, as we consider them to apply to the directive. 2930 if (Style.IndentPPDirectives == FormatStyle::PPDIS_BeforeHash && 2931 PPBranchLevel > 0) 2932 Line->Level += PPBranchLevel; 2933 flushComments(isOnNewLine(*FormatTok)); 2934 parsePPDirective(); 2935 } 2936 while (FormatTok->Type == TT_ConflictStart || 2937 FormatTok->Type == TT_ConflictEnd || 2938 FormatTok->Type == TT_ConflictAlternative) { 2939 if (FormatTok->Type == TT_ConflictStart) { 2940 conditionalCompilationStart(/*Unreachable=*/false); 2941 } else if (FormatTok->Type == TT_ConflictAlternative) { 2942 conditionalCompilationAlternative(); 2943 } else if (FormatTok->Type == TT_ConflictEnd) { 2944 conditionalCompilationEnd(); 2945 } 2946 FormatTok = Tokens->getNextToken(); 2947 FormatTok->MustBreakBefore = true; 2948 } 2949 2950 if (!PPStack.empty() && (PPStack.back().Kind == PP_Unreachable) && 2951 !Line->InPPDirective) { 2952 continue; 2953 } 2954 2955 if (!FormatTok->Tok.is(tok::comment)) { 2956 distributeComments(Comments, FormatTok); 2957 Comments.clear(); 2958 return; 2959 } 2960 2961 Comments.push_back(FormatTok); 2962 } while (!eof()); 2963 2964 distributeComments(Comments, nullptr); 2965 Comments.clear(); 2966 } 2967 2968 void UnwrappedLineParser::pushToken(FormatToken *Tok) { 2969 Line->Tokens.push_back(UnwrappedLineNode(Tok)); 2970 if (MustBreakBeforeNextToken) { 2971 Line->Tokens.back().Tok->MustBreakBefore = true; 2972 MustBreakBeforeNextToken = false; 2973 } 2974 } 2975 2976 } // end namespace format 2977 } // end namespace clang 2978