blob: 753eada8f1b09a2118a63737dddbd5aecf121604 [file] [log] [blame]
Daniel Jasperf7935112012-12-03 18:12:45 +00001//===--- UnwrappedLineParser.cpp - Format C++ code ------------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9///
10/// \file
11/// \brief This file contains the implementation of the UnwrappedLineParser,
12/// which turns a stream of tokens into UnwrappedLines.
13///
Daniel Jasperf7935112012-12-03 18:12:45 +000014//===----------------------------------------------------------------------===//
15
Chandler Carruth4b417452013-01-19 08:09:44 +000016#include "UnwrappedLineParser.h"
Benjamin Kramer33335df2015-03-01 21:36:40 +000017#include "llvm/ADT/STLExtras.h"
Manuel Klimekab3dc002013-01-16 12:31:12 +000018#include "llvm/Support/Debug.h"
Benjamin Kramer53f5e892015-03-23 18:05:43 +000019#include "llvm/Support/raw_ostream.h"
Manuel Klimekab3dc002013-01-16 12:31:12 +000020
Chandler Carruth10346662014-04-22 03:17:02 +000021#define DEBUG_TYPE "format-parser"
22
Daniel Jasperf7935112012-12-03 18:12:45 +000023namespace clang {
24namespace format {
25
Manuel Klimek15dfe7a2013-05-28 11:55:06 +000026class FormatTokenSource {
27public:
Angel Garcia Gomez637d1e62015-10-20 13:23:58 +000028 virtual ~FormatTokenSource() {}
Manuel Klimek15dfe7a2013-05-28 11:55:06 +000029 virtual FormatToken *getNextToken() = 0;
30
31 virtual unsigned getPosition() = 0;
32 virtual FormatToken *setPosition(unsigned Position) = 0;
33};
34
Craig Topper69665e12013-07-01 04:21:54 +000035namespace {
36
Manuel Klimek0a3a3c92013-01-23 09:32:48 +000037class ScopedDeclarationState {
38public:
39 ScopedDeclarationState(UnwrappedLine &Line, std::vector<bool> &Stack,
40 bool MustBeDeclaration)
41 : Line(Line), Stack(Stack) {
Manuel Klimek0a3a3c92013-01-23 09:32:48 +000042 Line.MustBeDeclaration = MustBeDeclaration;
Manuel Klimek39080572013-01-23 11:03:04 +000043 Stack.push_back(MustBeDeclaration);
Manuel Klimek0a3a3c92013-01-23 09:32:48 +000044 }
45 ~ScopedDeclarationState() {
Manuel Klimek0a3a3c92013-01-23 09:32:48 +000046 Stack.pop_back();
Manuel Klimekc1237a82013-01-23 14:08:21 +000047 if (!Stack.empty())
48 Line.MustBeDeclaration = Stack.back();
49 else
50 Line.MustBeDeclaration = true;
Manuel Klimek0a3a3c92013-01-23 09:32:48 +000051 }
Daniel Jasper393564f2013-05-31 14:56:29 +000052
Manuel Klimek0a3a3c92013-01-23 09:32:48 +000053private:
54 UnwrappedLine &Line;
55 std::vector<bool> &Stack;
56};
57
Krasimir Georgieva1c30932017-05-19 10:34:57 +000058static bool isLineComment(const FormatToken &FormatTok) {
59 return FormatTok.is(tok::comment) &&
60 FormatTok.TokenText.startswith("//");
61}
62
Krasimir Georgievea222a72017-05-22 10:07:56 +000063// Checks if \p FormatTok is a line comment that continues the line comment
64// \p Previous. The original column of \p MinColumnToken is used to determine
65// whether \p FormatTok is indented enough to the right to continue \p Previous.
66static bool continuesLineComment(const FormatToken &FormatTok,
67 const FormatToken *Previous,
68 const FormatToken *MinColumnToken) {
69 if (!Previous || !MinColumnToken)
70 return false;
71 unsigned MinContinueColumn =
72 MinColumnToken->OriginalColumn + (isLineComment(*MinColumnToken) ? 0 : 1);
73 return isLineComment(FormatTok) && FormatTok.NewlinesBefore == 1 &&
74 isLineComment(*Previous) &&
75 FormatTok.OriginalColumn >= MinContinueColumn;
76}
77
Manuel Klimek1abf7892013-01-04 23:34:14 +000078class ScopedMacroState : public FormatTokenSource {
79public:
80 ScopedMacroState(UnwrappedLine &Line, FormatTokenSource *&TokenSource,
Manuel Klimek20e0af62015-05-06 11:56:29 +000081 FormatToken *&ResetToken)
Manuel Klimek1abf7892013-01-04 23:34:14 +000082 : Line(Line), TokenSource(TokenSource), ResetToken(ResetToken),
Manuel Klimek1a18c402013-04-12 14:13:36 +000083 PreviousLineLevel(Line.Level), PreviousTokenSource(TokenSource),
Krasimir Georgieva1c30932017-05-19 10:34:57 +000084 Token(nullptr), PreviousToken(nullptr) {
Manuel Klimek1abf7892013-01-04 23:34:14 +000085 TokenSource = this;
Manuel Klimekef2cfb12013-01-05 22:14:16 +000086 Line.Level = 0;
Manuel Klimek1abf7892013-01-04 23:34:14 +000087 Line.InPPDirective = true;
88 }
89
Alexander Kornienko34eb2072015-04-11 02:00:23 +000090 ~ScopedMacroState() override {
Manuel Klimek1abf7892013-01-04 23:34:14 +000091 TokenSource = PreviousTokenSource;
92 ResetToken = Token;
93 Line.InPPDirective = false;
Manuel Klimekef2cfb12013-01-05 22:14:16 +000094 Line.Level = PreviousLineLevel;
Manuel Klimek1abf7892013-01-04 23:34:14 +000095 }
96
Craig Topperfb6b25b2014-03-15 04:29:04 +000097 FormatToken *getNextToken() override {
Manuel Klimek78725712013-01-07 10:03:37 +000098 // The \c UnwrappedLineParser guards against this by never calling
99 // \c getNextToken() after it has encountered the first eof token.
100 assert(!eof());
Krasimir Georgieva1c30932017-05-19 10:34:57 +0000101 PreviousToken = Token;
Manuel Klimek1abf7892013-01-04 23:34:14 +0000102 Token = PreviousTokenSource->getNextToken();
103 if (eof())
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000104 return getFakeEOF();
Manuel Klimek1abf7892013-01-04 23:34:14 +0000105 return Token;
106 }
107
Craig Topperfb6b25b2014-03-15 04:29:04 +0000108 unsigned getPosition() override { return PreviousTokenSource->getPosition(); }
Manuel Klimekab419912013-05-23 09:41:43 +0000109
Craig Topperfb6b25b2014-03-15 04:29:04 +0000110 FormatToken *setPosition(unsigned Position) override {
Krasimir Georgieva1c30932017-05-19 10:34:57 +0000111 PreviousToken = nullptr;
Manuel Klimekab419912013-05-23 09:41:43 +0000112 Token = PreviousTokenSource->setPosition(Position);
113 return Token;
114 }
115
Manuel Klimek1abf7892013-01-04 23:34:14 +0000116private:
Krasimir Georgieva1c30932017-05-19 10:34:57 +0000117 bool eof() {
118 return Token && Token->HasUnescapedNewline &&
Krasimir Georgievea222a72017-05-22 10:07:56 +0000119 !continuesLineComment(*Token, PreviousToken,
120 /*MinColumnToken=*/PreviousToken);
Krasimir Georgieva1c30932017-05-19 10:34:57 +0000121 }
Manuel Klimek1abf7892013-01-04 23:34:14 +0000122
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000123 FormatToken *getFakeEOF() {
124 static bool EOFInitialized = false;
125 static FormatToken FormatTok;
126 if (!EOFInitialized) {
127 FormatTok.Tok.startToken();
128 FormatTok.Tok.setKind(tok::eof);
129 EOFInitialized = true;
130 }
131 return &FormatTok;
Manuel Klimek1abf7892013-01-04 23:34:14 +0000132 }
133
134 UnwrappedLine &Line;
135 FormatTokenSource *&TokenSource;
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000136 FormatToken *&ResetToken;
Manuel Klimekef2cfb12013-01-05 22:14:16 +0000137 unsigned PreviousLineLevel;
Manuel Klimek1abf7892013-01-04 23:34:14 +0000138 FormatTokenSource *PreviousTokenSource;
139
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000140 FormatToken *Token;
Krasimir Georgieva1c30932017-05-19 10:34:57 +0000141 FormatToken *PreviousToken;
Manuel Klimek1abf7892013-01-04 23:34:14 +0000142};
143
Craig Topper69665e12013-07-01 04:21:54 +0000144} // end anonymous namespace
145
Manuel Klimek8e07a1b2013-01-10 11:52:21 +0000146class ScopedLineState {
147public:
Manuel Klimekd3b92fa2013-01-18 14:04:34 +0000148 ScopedLineState(UnwrappedLineParser &Parser,
149 bool SwitchToPreprocessorLines = false)
David Blaikieefb6eb22014-08-09 20:02:07 +0000150 : Parser(Parser), OriginalLines(Parser.CurrentLines) {
Manuel Klimekd3b92fa2013-01-18 14:04:34 +0000151 if (SwitchToPreprocessorLines)
152 Parser.CurrentLines = &Parser.PreprocessorDirectives;
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +0000153 else if (!Parser.Line->Tokens.empty())
154 Parser.CurrentLines = &Parser.Line->Tokens.back().Children;
David Blaikieefb6eb22014-08-09 20:02:07 +0000155 PreBlockLine = std::move(Parser.Line);
156 Parser.Line = llvm::make_unique<UnwrappedLine>();
Daniel Jasperdaffc0d2013-01-16 09:10:19 +0000157 Parser.Line->Level = PreBlockLine->Level;
158 Parser.Line->InPPDirective = PreBlockLine->InPPDirective;
Manuel Klimek8e07a1b2013-01-10 11:52:21 +0000159 }
160
161 ~ScopedLineState() {
Daniel Jasperdaffc0d2013-01-16 09:10:19 +0000162 if (!Parser.Line->Tokens.empty()) {
Manuel Klimek8e07a1b2013-01-10 11:52:21 +0000163 Parser.addUnwrappedLine();
164 }
Daniel Jasperdaffc0d2013-01-16 09:10:19 +0000165 assert(Parser.Line->Tokens.empty());
David Blaikieefb6eb22014-08-09 20:02:07 +0000166 Parser.Line = std::move(PreBlockLine);
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +0000167 if (Parser.CurrentLines == &Parser.PreprocessorDirectives)
168 Parser.MustBreakBeforeNextToken = true;
169 Parser.CurrentLines = OriginalLines;
Manuel Klimek8e07a1b2013-01-10 11:52:21 +0000170 }
171
172private:
173 UnwrappedLineParser &Parser;
174
David Blaikieefb6eb22014-08-09 20:02:07 +0000175 std::unique_ptr<UnwrappedLine> PreBlockLine;
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +0000176 SmallVectorImpl<UnwrappedLine> *OriginalLines;
Manuel Klimek8e07a1b2013-01-10 11:52:21 +0000177};
178
Alexander Kornienko3a33f022013-12-12 09:49:52 +0000179class CompoundStatementIndenter {
180public:
181 CompoundStatementIndenter(UnwrappedLineParser *Parser,
182 const FormatStyle &Style, unsigned &LineLevel)
183 : LineLevel(LineLevel), OldLineLevel(LineLevel) {
Daniel Jasperc1bc38e2015-09-29 14:57:55 +0000184 if (Style.BraceWrapping.AfterControlStatement)
Alexander Kornienko3a33f022013-12-12 09:49:52 +0000185 Parser->addUnwrappedLine();
Daniel Jasperc1bc38e2015-09-29 14:57:55 +0000186 if (Style.BraceWrapping.IndentBraces)
Alexander Kornienko3a33f022013-12-12 09:49:52 +0000187 ++LineLevel;
Alexander Kornienko3a33f022013-12-12 09:49:52 +0000188 }
Daniel Jasperb05a81d2014-05-09 13:11:16 +0000189 ~CompoundStatementIndenter() { LineLevel = OldLineLevel; }
Alexander Kornienko3a33f022013-12-12 09:49:52 +0000190
191private:
192 unsigned &LineLevel;
193 unsigned OldLineLevel;
194};
195
Craig Topper69665e12013-07-01 04:21:54 +0000196namespace {
197
Manuel Klimekab419912013-05-23 09:41:43 +0000198class IndexedTokenSource : public FormatTokenSource {
199public:
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000200 IndexedTokenSource(ArrayRef<FormatToken *> Tokens)
Manuel Klimekab419912013-05-23 09:41:43 +0000201 : Tokens(Tokens), Position(-1) {}
202
Craig Topperfb6b25b2014-03-15 04:29:04 +0000203 FormatToken *getNextToken() override {
Manuel Klimekab419912013-05-23 09:41:43 +0000204 ++Position;
205 return Tokens[Position];
206 }
207
Craig Topperfb6b25b2014-03-15 04:29:04 +0000208 unsigned getPosition() override {
Manuel Klimekab419912013-05-23 09:41:43 +0000209 assert(Position >= 0);
210 return Position;
211 }
212
Craig Topperfb6b25b2014-03-15 04:29:04 +0000213 FormatToken *setPosition(unsigned P) override {
Manuel Klimekab419912013-05-23 09:41:43 +0000214 Position = P;
215 return Tokens[Position];
216 }
217
Manuel Klimek71814b42013-10-11 21:25:45 +0000218 void reset() { Position = -1; }
219
Manuel Klimekab419912013-05-23 09:41:43 +0000220private:
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000221 ArrayRef<FormatToken *> Tokens;
Manuel Klimekab419912013-05-23 09:41:43 +0000222 int Position;
223};
224
Craig Topper69665e12013-07-01 04:21:54 +0000225} // end anonymous namespace
226
Daniel Jasperd2ae41a2013-05-15 08:14:19 +0000227UnwrappedLineParser::UnwrappedLineParser(const FormatStyle &Style,
Daniel Jasperd0ec0d62014-11-04 12:41:02 +0000228 const AdditionalKeywords &Keywords,
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000229 ArrayRef<FormatToken *> Tokens,
Daniel Jasperd2ae41a2013-05-15 08:14:19 +0000230 UnwrappedLineConsumer &Callback)
Daniel Jasperb05a81d2014-05-09 13:11:16 +0000231 : Line(new UnwrappedLine), MustBreakBeforeNextToken(false),
Krasimir Georgiev00c5c722017-02-02 15:32:19 +0000232 CurrentLines(&Lines), Style(Style), Keywords(Keywords),
233 CommentPragmasRegex(Style.CommentPragmas), Tokens(nullptr),
Manuel Klimek20e0af62015-05-06 11:56:29 +0000234 Callback(Callback), AllTokens(Tokens), PPBranchLevel(-1) {}
Manuel Klimek71814b42013-10-11 21:25:45 +0000235
236void UnwrappedLineParser::reset() {
237 PPBranchLevel = -1;
238 Line.reset(new UnwrappedLine);
239 CommentsBeforeNextToken.clear();
Craig Topper2145bc02014-05-09 08:15:10 +0000240 FormatTok = nullptr;
Manuel Klimek71814b42013-10-11 21:25:45 +0000241 MustBreakBeforeNextToken = false;
242 PreprocessorDirectives.clear();
243 CurrentLines = &Lines;
244 DeclarationScopeStack.clear();
Manuel Klimek71814b42013-10-11 21:25:45 +0000245 PPStack.clear();
246}
Daniel Jasperf7935112012-12-03 18:12:45 +0000247
Manuel Klimek20e0af62015-05-06 11:56:29 +0000248void UnwrappedLineParser::parse() {
Manuel Klimekab419912013-05-23 09:41:43 +0000249 IndexedTokenSource TokenSource(AllTokens);
Manuel Klimek71814b42013-10-11 21:25:45 +0000250 do {
251 DEBUG(llvm::dbgs() << "----\n");
252 reset();
253 Tokens = &TokenSource;
254 TokenSource.reset();
Daniel Jaspera79064a2013-03-01 18:11:39 +0000255
Manuel Klimek71814b42013-10-11 21:25:45 +0000256 readToken();
257 parseFile();
258 // Create line with eof token.
259 pushToken(FormatTok);
260 addUnwrappedLine();
261
262 for (SmallVectorImpl<UnwrappedLine>::iterator I = Lines.begin(),
263 E = Lines.end();
264 I != E; ++I) {
265 Callback.consumeUnwrappedLine(*I);
266 }
267 Callback.finishRun();
268 Lines.clear();
269 while (!PPLevelBranchIndex.empty() &&
Daniel Jasper53bd1672013-10-12 13:32:56 +0000270 PPLevelBranchIndex.back() + 1 >= PPLevelBranchCount.back()) {
Manuel Klimek71814b42013-10-11 21:25:45 +0000271 PPLevelBranchIndex.resize(PPLevelBranchIndex.size() - 1);
272 PPLevelBranchCount.resize(PPLevelBranchCount.size() - 1);
273 }
274 if (!PPLevelBranchIndex.empty()) {
275 ++PPLevelBranchIndex.back();
276 assert(PPLevelBranchIndex.size() == PPLevelBranchCount.size());
277 assert(PPLevelBranchIndex.back() <= PPLevelBranchCount.back());
278 }
279 } while (!PPLevelBranchIndex.empty());
Manuel Klimek1abf7892013-01-04 23:34:14 +0000280}
281
Manuel Klimek1a18c402013-04-12 14:13:36 +0000282void UnwrappedLineParser::parseFile() {
Daniel Jasper9326f912015-05-05 08:40:32 +0000283 // The top-level context in a file always has declarations, except for pre-
284 // processor directives and JavaScript files.
285 bool MustBeDeclaration =
286 !Line->InPPDirective && Style.Language != FormatStyle::LK_JavaScript;
287 ScopedDeclarationState DeclarationState(*Line, DeclarationScopeStack,
288 MustBeDeclaration);
Krasimir Georgiev26b144c2017-07-03 15:05:14 +0000289 if (Style.Language == FormatStyle::LK_TextProto)
290 parseBracedList();
291 else
292 parseLevel(/*HasOpeningBrace=*/false);
Manuel Klimek1abf7892013-01-04 23:34:14 +0000293 // Make sure to format the remaining tokens.
Manuel Klimekf92f7bc2013-01-22 16:31:55 +0000294 flushComments(true);
Manuel Klimek1abf7892013-01-04 23:34:14 +0000295 addUnwrappedLine();
Daniel Jasperf7935112012-12-03 18:12:45 +0000296}
297
Manuel Klimek1a18c402013-04-12 14:13:36 +0000298void UnwrappedLineParser::parseLevel(bool HasOpeningBrace) {
Daniel Jasper516d7972013-07-25 11:31:57 +0000299 bool SwitchLabelEncountered = false;
Daniel Jasperf7935112012-12-03 18:12:45 +0000300 do {
Birunthan Mohanathasb001a0b2015-07-03 17:25:16 +0000301 tok::TokenKind kind = FormatTok->Tok.getKind();
302 if (FormatTok->Type == TT_MacroBlockBegin) {
303 kind = tok::l_brace;
304 } else if (FormatTok->Type == TT_MacroBlockEnd) {
305 kind = tok::r_brace;
306 }
307
308 switch (kind) {
Daniel Jasperf7935112012-12-03 18:12:45 +0000309 case tok::comment:
Daniel Jaspere25509f2012-12-17 11:29:41 +0000310 nextToken();
311 addUnwrappedLine();
Daniel Jasperf7935112012-12-03 18:12:45 +0000312 break;
313 case tok::l_brace:
Manuel Klimek0a3a3c92013-01-23 09:32:48 +0000314 // FIXME: Add parameter whether this can happen - if this happens, we must
315 // be in a non-declaration context.
Daniel Jasperb86e2722015-08-24 13:23:37 +0000316 if (!FormatTok->is(TT_MacroBlockBegin) && tryToParseBracedList())
317 continue;
Nico Weber9096fc02013-06-26 00:30:14 +0000318 parseBlock(/*MustBeDeclaration=*/false);
Daniel Jasperf7935112012-12-03 18:12:45 +0000319 addUnwrappedLine();
320 break;
321 case tok::r_brace:
Manuel Klimek1a18c402013-04-12 14:13:36 +0000322 if (HasOpeningBrace)
323 return;
Manuel Klimek1a18c402013-04-12 14:13:36 +0000324 nextToken();
325 addUnwrappedLine();
Manuel Klimek1058d982013-01-06 20:07:31 +0000326 break;
Daniel Jasper516d7972013-07-25 11:31:57 +0000327 case tok::kw_default:
328 case tok::kw_case:
Daniel Jasper72407622013-09-02 08:26:29 +0000329 if (!SwitchLabelEncountered &&
330 (Style.IndentCaseLabels || (Line->InPPDirective && Line->Level == 1)))
331 ++Line->Level;
Daniel Jasper516d7972013-07-25 11:31:57 +0000332 SwitchLabelEncountered = true;
333 parseStructuralElement();
334 break;
Daniel Jasperf7935112012-12-03 18:12:45 +0000335 default:
Manuel Klimek6b9eeba2013-01-07 14:56:16 +0000336 parseStructuralElement();
Daniel Jasperf7935112012-12-03 18:12:45 +0000337 break;
338 }
339 } while (!eof());
340}
341
Daniel Jasperadba2aa2015-05-18 12:52:00 +0000342void UnwrappedLineParser::calculateBraceTypes(bool ExpectClassBody) {
Manuel Klimekab419912013-05-23 09:41:43 +0000343 // We'll parse forward through the tokens until we hit
344 // a closing brace or eof - note that getNextToken() will
345 // parse macros, so this will magically work inside macro
346 // definitions, too.
347 unsigned StoredPosition = Tokens->getPosition();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000348 FormatToken *Tok = FormatTok;
Daniel Jasperb9a49902016-01-09 15:56:28 +0000349 const FormatToken *PrevTok = getPreviousToken();
Manuel Klimekab419912013-05-23 09:41:43 +0000350 // Keep a stack of positions of lbrace tokens. We will
351 // update information about whether an lbrace starts a
352 // braced init list or a different block during the loop.
Daniel Jasperb1f74a82013-07-09 09:06:29 +0000353 SmallVector<FormatToken *, 8> LBraceStack;
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000354 assert(Tok->Tok.is(tok::l_brace));
Manuel Klimekab419912013-05-23 09:41:43 +0000355 do {
Daniel Jaspereb65e912015-12-21 18:31:15 +0000356 // Get next non-comment token.
Daniel Jasper7f5d53e2013-07-01 09:15:46 +0000357 FormatToken *NextTok;
Daniel Jasperca7bd722013-07-01 16:43:38 +0000358 unsigned ReadTokens = 0;
Daniel Jasper7f5d53e2013-07-01 09:15:46 +0000359 do {
360 NextTok = Tokens->getNextToken();
Daniel Jasperca7bd722013-07-01 16:43:38 +0000361 ++ReadTokens;
Daniel Jasper7f5d53e2013-07-01 09:15:46 +0000362 } while (NextTok->is(tok::comment));
363
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000364 switch (Tok->Tok.getKind()) {
Manuel Klimekab419912013-05-23 09:41:43 +0000365 case tok::l_brace:
Martin Probst95ed8e72017-05-31 09:29:40 +0000366 if (Style.Language == FormatStyle::LK_JavaScript && PrevTok) {
367 if (PrevTok->is(tok::colon))
368 // A colon indicates this code is in a type, or a braced list
369 // following a label in an object literal ({a: {b: 1}}). The code
370 // below could be confused by semicolons between the individual
371 // members in a type member list, which would normally trigger
372 // BK_Block. In both cases, this must be parsed as an inline braced
373 // init.
374 Tok->BlockKind = BK_BracedInit;
375 else if (PrevTok->is(tok::r_paren))
376 // `) { }` can only occur in function or method declarations in JS.
377 Tok->BlockKind = BK_Block;
378 } else {
Daniel Jasperb9a49902016-01-09 15:56:28 +0000379 Tok->BlockKind = BK_Unknown;
Martin Probst95ed8e72017-05-31 09:29:40 +0000380 }
Daniel Jasperb1f74a82013-07-09 09:06:29 +0000381 LBraceStack.push_back(Tok);
Manuel Klimekab419912013-05-23 09:41:43 +0000382 break;
383 case tok::r_brace:
Daniel Jasperb9a49902016-01-09 15:56:28 +0000384 if (LBraceStack.empty())
385 break;
386 if (LBraceStack.back()->BlockKind == BK_Unknown) {
387 bool ProbablyBracedList = false;
388 if (Style.Language == FormatStyle::LK_Proto) {
389 ProbablyBracedList = NextTok->isOneOf(tok::comma, tok::r_square);
390 } else {
391 // Using OriginalColumn to distinguish between ObjC methods and
392 // binary operators is a bit hacky.
393 bool NextIsObjCMethod = NextTok->isOneOf(tok::plus, tok::minus) &&
394 NextTok->OriginalColumn == 0;
Daniel Jasper91b032a2014-05-22 12:46:38 +0000395
Daniel Jasperb9a49902016-01-09 15:56:28 +0000396 // If there is a comma, semicolon or right paren after the closing
397 // brace, we assume this is a braced initializer list. Note that
398 // regardless how we mark inner braces here, we will overwrite the
399 // BlockKind later if we parse a braced list (where all blocks
400 // inside are by default braced lists), or when we explicitly detect
401 // blocks (for example while parsing lambdas).
Martin Probst95ed8e72017-05-31 09:29:40 +0000402 // FIXME: Some of these do not apply to JS, e.g. "} {" can never be a
403 // braced list in JS.
Daniel Jasperb9a49902016-01-09 15:56:28 +0000404 ProbablyBracedList =
Daniel Jasperacffeb82016-03-05 18:34:26 +0000405 (Style.Language == FormatStyle::LK_JavaScript &&
Martin Probste1e12a72016-08-19 14:35:01 +0000406 NextTok->isOneOf(Keywords.kw_of, Keywords.kw_in,
407 Keywords.kw_as)) ||
Martin Probstb7fb2672017-05-10 13:53:29 +0000408 (Style.isCpp() && NextTok->is(tok::l_paren)) ||
Daniel Jasperb9a49902016-01-09 15:56:28 +0000409 NextTok->isOneOf(tok::comma, tok::period, tok::colon,
410 tok::r_paren, tok::r_square, tok::l_brace,
Martin Probstb7fb2672017-05-10 13:53:29 +0000411 tok::l_square, tok::ellipsis) ||
Daniel Jaspere4ada022016-12-13 10:05:03 +0000412 (NextTok->is(tok::identifier) &&
413 !PrevTok->isOneOf(tok::semi, tok::r_brace, tok::l_brace)) ||
Daniel Jasperb9a49902016-01-09 15:56:28 +0000414 (NextTok->is(tok::semi) &&
415 (!ExpectClassBody || LBraceStack.size() != 1)) ||
416 (NextTok->isBinaryOperator() && !NextIsObjCMethod);
Manuel Klimekab419912013-05-23 09:41:43 +0000417 }
Daniel Jasperb9a49902016-01-09 15:56:28 +0000418 if (ProbablyBracedList) {
419 Tok->BlockKind = BK_BracedInit;
420 LBraceStack.back()->BlockKind = BK_BracedInit;
421 } else {
422 Tok->BlockKind = BK_Block;
423 LBraceStack.back()->BlockKind = BK_Block;
424 }
Manuel Klimekab419912013-05-23 09:41:43 +0000425 }
Daniel Jasperb9a49902016-01-09 15:56:28 +0000426 LBraceStack.pop_back();
Manuel Klimekab419912013-05-23 09:41:43 +0000427 break;
Daniel Jasperac7e34e2014-03-13 10:11:17 +0000428 case tok::at:
Manuel Klimekab419912013-05-23 09:41:43 +0000429 case tok::semi:
430 case tok::kw_if:
431 case tok::kw_while:
432 case tok::kw_for:
433 case tok::kw_switch:
434 case tok::kw_try:
Nico Weberfac23712015-02-04 15:26:27 +0000435 case tok::kw___try:
Daniel Jasperb9a49902016-01-09 15:56:28 +0000436 if (!LBraceStack.empty() && LBraceStack.back()->BlockKind == BK_Unknown)
Daniel Jasperb1f74a82013-07-09 09:06:29 +0000437 LBraceStack.back()->BlockKind = BK_Block;
Manuel Klimekab419912013-05-23 09:41:43 +0000438 break;
439 default:
440 break;
441 }
Daniel Jasperb9a49902016-01-09 15:56:28 +0000442 PrevTok = Tok;
Manuel Klimekab419912013-05-23 09:41:43 +0000443 Tok = NextTok;
Manuel Klimekbab25fd2013-09-04 08:20:47 +0000444 } while (Tok->Tok.isNot(tok::eof) && !LBraceStack.empty());
Daniel Jasperb9a49902016-01-09 15:56:28 +0000445
Manuel Klimekab419912013-05-23 09:41:43 +0000446 // Assume other blocks for all unclosed opening braces.
447 for (unsigned i = 0, e = LBraceStack.size(); i != e; ++i) {
Daniel Jasperb1f74a82013-07-09 09:06:29 +0000448 if (LBraceStack[i]->BlockKind == BK_Unknown)
449 LBraceStack[i]->BlockKind = BK_Block;
Manuel Klimekab419912013-05-23 09:41:43 +0000450 }
Manuel Klimekbab25fd2013-09-04 08:20:47 +0000451
Manuel Klimekab419912013-05-23 09:41:43 +0000452 FormatTok = Tokens->setPosition(StoredPosition);
453}
454
Francois Ferranda98a95c2017-07-28 07:56:14 +0000455template <class T>
456static inline void hash_combine(std::size_t &seed, const T &v) {
457 std::hash<T> hasher;
458 seed ^= hasher(v) + 0x9e3779b9 + (seed << 6) + (seed >> 2);
459}
460
461size_t UnwrappedLineParser::computePPHash() const {
462 size_t h = 0;
463 for (const auto &i : PPStack) {
464 hash_combine(h, size_t(i.Kind));
465 hash_combine(h, i.Line);
466 }
467 return h;
468}
469
Manuel Klimekb212f3b2013-10-12 22:46:56 +0000470void UnwrappedLineParser::parseBlock(bool MustBeDeclaration, bool AddLevel,
471 bool MunchSemi) {
Birunthan Mohanathasb001a0b2015-07-03 17:25:16 +0000472 assert(FormatTok->isOneOf(tok::l_brace, TT_MacroBlockBegin) &&
473 "'{' or macro block token expected");
474 const bool MacroBlock = FormatTok->is(TT_MacroBlockBegin);
Daniel Jaspereb65e912015-12-21 18:31:15 +0000475 FormatTok->BlockKind = BK_Block;
Birunthan Mohanathasb001a0b2015-07-03 17:25:16 +0000476
Francois Ferranda98a95c2017-07-28 07:56:14 +0000477 size_t PPStartHash = computePPHash();
478
Daniel Jasper516d7972013-07-25 11:31:57 +0000479 unsigned InitialLevel = Line->Level;
Krasimir Georgiev3e051052017-07-24 14:51:59 +0000480 nextToken(/*LevelDifference=*/AddLevel ? 1 : 0);
Daniel Jasperf7935112012-12-03 18:12:45 +0000481
Birunthan Mohanathasb001a0b2015-07-03 17:25:16 +0000482 if (MacroBlock && FormatTok->is(tok::l_paren))
483 parseParens();
484
Francois Ferranda98a95c2017-07-28 07:56:14 +0000485 size_t NbPreprocessorDirectives =
486 CurrentLines == &Lines ? PreprocessorDirectives.size() : 0;
Manuel Klimeka4fe1c12013-01-21 16:42:44 +0000487 addUnwrappedLine();
Francois Ferranda98a95c2017-07-28 07:56:14 +0000488 size_t OpeningLineIndex =
489 CurrentLines->empty()
490 ? (UnwrappedLine::kInvalidIndex)
491 : (CurrentLines->size() - 1 - NbPreprocessorDirectives);
Daniel Jasperf7935112012-12-03 18:12:45 +0000492
Manuel Klimek0a3a3c92013-01-23 09:32:48 +0000493 ScopedDeclarationState DeclarationState(*Line, DeclarationScopeStack,
494 MustBeDeclaration);
Daniel Jasper65ee3472013-07-31 23:16:02 +0000495 if (AddLevel)
496 ++Line->Level;
Nico Weber9096fc02013-06-26 00:30:14 +0000497 parseLevel(/*HasOpeningBrace=*/true);
Alexander Kornienko578fdd82012-12-06 18:03:27 +0000498
Marianne Mailhot-Sarrasin03137c62016-04-14 14:56:49 +0000499 if (eof())
500 return;
501
Birunthan Mohanathasb001a0b2015-07-03 17:25:16 +0000502 if (MacroBlock ? !FormatTok->is(TT_MacroBlockEnd)
503 : !FormatTok->is(tok::r_brace)) {
Daniel Jasper516d7972013-07-25 11:31:57 +0000504 Line->Level = InitialLevel;
Daniel Jaspereb65e912015-12-21 18:31:15 +0000505 FormatTok->BlockKind = BK_Block;
Manuel Klimek1a18c402013-04-12 14:13:36 +0000506 return;
Manuel Klimekf92f7bc2013-01-22 16:31:55 +0000507 }
Alexander Kornienko0ea8e102012-12-04 15:40:36 +0000508
Francois Ferranda98a95c2017-07-28 07:56:14 +0000509 size_t PPEndHash = computePPHash();
510
Krasimir Georgiev3e051052017-07-24 14:51:59 +0000511 // Munch the closing brace.
512 nextToken(/*LevelDifference=*/AddLevel ? -1 : 0);
Birunthan Mohanathasb001a0b2015-07-03 17:25:16 +0000513
514 if (MacroBlock && FormatTok->is(tok::l_paren))
515 parseParens();
516
Manuel Klimekb212f3b2013-10-12 22:46:56 +0000517 if (MunchSemi && FormatTok->Tok.is(tok::semi))
518 nextToken();
Krasimir Georgiev3e051052017-07-24 14:51:59 +0000519 Line->Level = InitialLevel;
Francois Ferranda98a95c2017-07-28 07:56:14 +0000520
521 if (PPStartHash == PPEndHash) {
522 Line->MatchingOpeningBlockLineIndex = OpeningLineIndex;
523 if (OpeningLineIndex != UnwrappedLine::kInvalidIndex) {
524 // Update the opening line to add the forward reference as well
525 (*CurrentLines)[OpeningLineIndex].MatchingOpeningBlockLineIndex =
526 CurrentLines->size() - 1;
527 }
Francois Ferrande56a8292017-06-14 12:29:47 +0000528 }
Daniel Jasperf7935112012-12-03 18:12:45 +0000529}
530
Daniel Jasper02c7bca2015-03-30 09:56:50 +0000531static bool isGoogScope(const UnwrappedLine &Line) {
Daniel Jasper616de8642014-11-23 16:46:28 +0000532 // FIXME: Closure-library specific stuff should not be hard-coded but be
533 // configurable.
Daniel Jasper4a39c842014-05-06 13:54:10 +0000534 if (Line.Tokens.size() < 4)
535 return false;
536 auto I = Line.Tokens.begin();
537 if (I->Tok->TokenText != "goog")
538 return false;
539 ++I;
540 if (I->Tok->isNot(tok::period))
541 return false;
542 ++I;
543 if (I->Tok->TokenText != "scope")
544 return false;
545 ++I;
546 return I->Tok->is(tok::l_paren);
547}
548
Martin Probst101ec892017-05-09 20:04:09 +0000549static bool isIIFE(const UnwrappedLine &Line,
550 const AdditionalKeywords &Keywords) {
551 // Look for the start of an immediately invoked anonymous function.
552 // https://en.wikipedia.org/wiki/Immediately-invoked_function_expression
553 // This is commonly done in JavaScript to create a new, anonymous scope.
554 // Example: (function() { ... })()
555 if (Line.Tokens.size() < 3)
556 return false;
557 auto I = Line.Tokens.begin();
558 if (I->Tok->isNot(tok::l_paren))
559 return false;
560 ++I;
561 if (I->Tok->isNot(Keywords.kw_function))
562 return false;
563 ++I;
564 return I->Tok->is(tok::l_paren);
565}
566
Roman Kashitsyna043ced2014-08-11 12:18:01 +0000567static bool ShouldBreakBeforeBrace(const FormatStyle &Style,
568 const FormatToken &InitialToken) {
Daniel Jasperc1bc38e2015-09-29 14:57:55 +0000569 if (InitialToken.is(tok::kw_namespace))
570 return Style.BraceWrapping.AfterNamespace;
571 if (InitialToken.is(tok::kw_class))
572 return Style.BraceWrapping.AfterClass;
573 if (InitialToken.is(tok::kw_union))
574 return Style.BraceWrapping.AfterUnion;
575 if (InitialToken.is(tok::kw_struct))
576 return Style.BraceWrapping.AfterStruct;
577 return false;
Roman Kashitsyna043ced2014-08-11 12:18:01 +0000578}
579
Manuel Klimek516e0542013-09-04 13:25:30 +0000580void UnwrappedLineParser::parseChildBlock() {
581 FormatTok->BlockKind = BK_Block;
582 nextToken();
583 {
Martin Probst101ec892017-05-09 20:04:09 +0000584 bool SkipIndent =
585 (Style.Language == FormatStyle::LK_JavaScript &&
586 (isGoogScope(*Line) || isIIFE(*Line, Keywords)));
Manuel Klimek516e0542013-09-04 13:25:30 +0000587 ScopedLineState LineState(*this);
588 ScopedDeclarationState DeclarationState(*Line, DeclarationScopeStack,
589 /*MustBeDeclaration=*/false);
Martin Probst101ec892017-05-09 20:04:09 +0000590 Line->Level += SkipIndent ? 0 : 1;
Manuel Klimek516e0542013-09-04 13:25:30 +0000591 parseLevel(/*HasOpeningBrace=*/true);
Daniel Jasper02c7bca2015-03-30 09:56:50 +0000592 flushComments(isOnNewLine(*FormatTok));
Martin Probst101ec892017-05-09 20:04:09 +0000593 Line->Level -= SkipIndent ? 0 : 1;
Manuel Klimek516e0542013-09-04 13:25:30 +0000594 }
595 nextToken();
596}
597
Daniel Jasperf7935112012-12-03 18:12:45 +0000598void UnwrappedLineParser::parsePPDirective() {
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000599 assert(FormatTok->Tok.is(tok::hash) && "'#' expected");
Manuel Klimek20e0af62015-05-06 11:56:29 +0000600 ScopedMacroState MacroState(*Line, Tokens, FormatTok);
Manuel Klimeka71e5d82013-01-02 16:30:12 +0000601 nextToken();
602
Craig Topper2145bc02014-05-09 08:15:10 +0000603 if (!FormatTok->Tok.getIdentifierInfo()) {
Manuel Klimek591b5802013-01-31 15:58:48 +0000604 parsePPUnknown();
Manuel Klimeka71e5d82013-01-02 16:30:12 +0000605 return;
Daniel Jasperf7935112012-12-03 18:12:45 +0000606 }
Manuel Klimeka71e5d82013-01-02 16:30:12 +0000607
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000608 switch (FormatTok->Tok.getIdentifierInfo()->getPPKeywordID()) {
Manuel Klimek1abf7892013-01-04 23:34:14 +0000609 case tok::pp_define:
610 parsePPDefine();
Alexander Kornienkof2e02122013-05-24 18:24:24 +0000611 return;
612 case tok::pp_if:
Manuel Klimek71814b42013-10-11 21:25:45 +0000613 parsePPIf(/*IfDef=*/false);
Alexander Kornienkof2e02122013-05-24 18:24:24 +0000614 break;
615 case tok::pp_ifdef:
616 case tok::pp_ifndef:
Manuel Klimek71814b42013-10-11 21:25:45 +0000617 parsePPIf(/*IfDef=*/true);
Alexander Kornienkof2e02122013-05-24 18:24:24 +0000618 break;
619 case tok::pp_else:
620 parsePPElse();
621 break;
622 case tok::pp_elif:
623 parsePPElIf();
624 break;
625 case tok::pp_endif:
626 parsePPEndIf();
Manuel Klimek1abf7892013-01-04 23:34:14 +0000627 break;
628 default:
629 parsePPUnknown();
630 break;
631 }
632}
633
Manuel Klimek68b03042014-04-14 09:14:11 +0000634void UnwrappedLineParser::conditionalCompilationCondition(bool Unreachable) {
Francois Ferranda98a95c2017-07-28 07:56:14 +0000635 size_t Line = CurrentLines->size();
636 if (CurrentLines == &PreprocessorDirectives)
637 Line += Lines.size();
638
639 if (Unreachable ||
640 (!PPStack.empty() && PPStack.back().Kind == PP_Unreachable))
641 PPStack.push_back({PP_Unreachable, Line});
Alexander Kornienkof2e02122013-05-24 18:24:24 +0000642 else
Francois Ferranda98a95c2017-07-28 07:56:14 +0000643 PPStack.push_back({PP_Conditional, Line});
Alexander Kornienkof2e02122013-05-24 18:24:24 +0000644}
645
Manuel Klimek68b03042014-04-14 09:14:11 +0000646void UnwrappedLineParser::conditionalCompilationStart(bool Unreachable) {
Manuel Klimek71814b42013-10-11 21:25:45 +0000647 ++PPBranchLevel;
648 assert(PPBranchLevel >= 0 && PPBranchLevel <= (int)PPLevelBranchIndex.size());
649 if (PPBranchLevel == (int)PPLevelBranchIndex.size()) {
650 PPLevelBranchIndex.push_back(0);
651 PPLevelBranchCount.push_back(0);
652 }
653 PPChainBranchIndex.push(0);
Manuel Klimek68b03042014-04-14 09:14:11 +0000654 bool Skip = PPLevelBranchIndex[PPBranchLevel] > 0;
655 conditionalCompilationCondition(Unreachable || Skip);
Alexander Kornienkof2e02122013-05-24 18:24:24 +0000656}
657
Manuel Klimek68b03042014-04-14 09:14:11 +0000658void UnwrappedLineParser::conditionalCompilationAlternative() {
Alexander Kornienkof2e02122013-05-24 18:24:24 +0000659 if (!PPStack.empty())
660 PPStack.pop_back();
Manuel Klimek71814b42013-10-11 21:25:45 +0000661 assert(PPBranchLevel < (int)PPLevelBranchIndex.size());
662 if (!PPChainBranchIndex.empty())
663 ++PPChainBranchIndex.top();
Manuel Klimek68b03042014-04-14 09:14:11 +0000664 conditionalCompilationCondition(
665 PPBranchLevel >= 0 && !PPChainBranchIndex.empty() &&
666 PPLevelBranchIndex[PPBranchLevel] != PPChainBranchIndex.top());
Alexander Kornienkof2e02122013-05-24 18:24:24 +0000667}
668
Manuel Klimek68b03042014-04-14 09:14:11 +0000669void UnwrappedLineParser::conditionalCompilationEnd() {
Manuel Klimek71814b42013-10-11 21:25:45 +0000670 assert(PPBranchLevel < (int)PPLevelBranchIndex.size());
671 if (PPBranchLevel >= 0 && !PPChainBranchIndex.empty()) {
672 if (PPChainBranchIndex.top() + 1 > PPLevelBranchCount[PPBranchLevel]) {
Manuel Klimek71814b42013-10-11 21:25:45 +0000673 PPLevelBranchCount[PPBranchLevel] = PPChainBranchIndex.top() + 1;
674 }
675 }
Manuel Klimek14bd9172014-01-29 08:49:02 +0000676 // Guard against #endif's without #if.
677 if (PPBranchLevel > 0)
678 --PPBranchLevel;
Manuel Klimek71814b42013-10-11 21:25:45 +0000679 if (!PPChainBranchIndex.empty())
680 PPChainBranchIndex.pop();
Alexander Kornienkof2e02122013-05-24 18:24:24 +0000681 if (!PPStack.empty())
682 PPStack.pop_back();
Manuel Klimek68b03042014-04-14 09:14:11 +0000683}
684
685void UnwrappedLineParser::parsePPIf(bool IfDef) {
Daniel Jasper62703eb2017-03-01 11:10:11 +0000686 bool IfNDef = FormatTok->is(tok::pp_ifndef);
Manuel Klimek68b03042014-04-14 09:14:11 +0000687 nextToken();
Daniel Jaspereab6cd42017-03-01 10:47:52 +0000688 bool Unreachable = false;
689 if (!IfDef && (FormatTok->is(tok::kw_false) || FormatTok->TokenText == "0"))
690 Unreachable = true;
Daniel Jasper62703eb2017-03-01 11:10:11 +0000691 if (IfDef && !IfNDef && FormatTok->TokenText == "SWIG")
Daniel Jaspereab6cd42017-03-01 10:47:52 +0000692 Unreachable = true;
693 conditionalCompilationStart(Unreachable);
Manuel Klimek68b03042014-04-14 09:14:11 +0000694 parsePPUnknown();
695}
696
697void UnwrappedLineParser::parsePPElse() {
698 conditionalCompilationAlternative();
699 parsePPUnknown();
700}
701
702void UnwrappedLineParser::parsePPElIf() { parsePPElse(); }
703
704void UnwrappedLineParser::parsePPEndIf() {
705 conditionalCompilationEnd();
Alexander Kornienkof2e02122013-05-24 18:24:24 +0000706 parsePPUnknown();
707}
708
Manuel Klimek1abf7892013-01-04 23:34:14 +0000709void UnwrappedLineParser::parsePPDefine() {
710 nextToken();
711
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000712 if (FormatTok->Tok.getKind() != tok::identifier) {
Manuel Klimek1abf7892013-01-04 23:34:14 +0000713 parsePPUnknown();
714 return;
715 }
716 nextToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000717 if (FormatTok->Tok.getKind() == tok::l_paren &&
718 FormatTok->WhitespaceRange.getBegin() ==
719 FormatTok->WhitespaceRange.getEnd()) {
Manuel Klimek1abf7892013-01-04 23:34:14 +0000720 parseParens();
721 }
722 addUnwrappedLine();
Manuel Klimek52b15152013-01-09 15:25:02 +0000723 Line->Level = 1;
Manuel Klimek1b896292013-01-07 09:34:28 +0000724
725 // Errors during a preprocessor directive can only affect the layout of the
726 // preprocessor directive, and thus we ignore them. An alternative approach
727 // would be to use the same approach we use on the file level (no
728 // re-indentation if there was a structural error) within the macro
729 // definition.
Manuel Klimek1abf7892013-01-04 23:34:14 +0000730 parseFile();
731}
732
733void UnwrappedLineParser::parsePPUnknown() {
Manuel Klimeka71e5d82013-01-02 16:30:12 +0000734 do {
Manuel Klimeka71e5d82013-01-02 16:30:12 +0000735 nextToken();
736 } while (!eof());
737 addUnwrappedLine();
Daniel Jasperf7935112012-12-03 18:12:45 +0000738}
739
Alexander Kornienkoa04e5e22013-04-09 16:15:19 +0000740// Here we blacklist certain tokens that are not usually the first token in an
741// unwrapped line. This is used in attempt to distinguish macro calls without
742// trailing semicolons from other constructs split to several lines.
Benjamin Kramer8407df72015-03-09 16:47:52 +0000743static bool tokenCanStartNewLine(const clang::Token &Tok) {
Alexander Kornienkoa04e5e22013-04-09 16:15:19 +0000744 // Semicolon can be a null-statement, l_square can be a start of a macro or
745 // a C++11 attribute, but this doesn't seem to be common.
746 return Tok.isNot(tok::semi) && Tok.isNot(tok::l_brace) &&
747 Tok.isNot(tok::l_square) &&
748 // Tokens that can only be used as binary operators and a part of
749 // overloaded operator names.
750 Tok.isNot(tok::period) && Tok.isNot(tok::periodstar) &&
751 Tok.isNot(tok::arrow) && Tok.isNot(tok::arrowstar) &&
752 Tok.isNot(tok::less) && Tok.isNot(tok::greater) &&
753 Tok.isNot(tok::slash) && Tok.isNot(tok::percent) &&
754 Tok.isNot(tok::lessless) && Tok.isNot(tok::greatergreater) &&
755 Tok.isNot(tok::equal) && Tok.isNot(tok::plusequal) &&
756 Tok.isNot(tok::minusequal) && Tok.isNot(tok::starequal) &&
757 Tok.isNot(tok::slashequal) && Tok.isNot(tok::percentequal) &&
758 Tok.isNot(tok::ampequal) && Tok.isNot(tok::pipeequal) &&
759 Tok.isNot(tok::caretequal) && Tok.isNot(tok::greatergreaterequal) &&
760 Tok.isNot(tok::lesslessequal) &&
761 // Colon is used in labels, base class lists, initializer lists,
762 // range-based for loops, ternary operator, but should never be the
763 // first token in an unwrapped line.
Daniel Jasper5ebb2f32014-05-21 13:08:17 +0000764 Tok.isNot(tok::colon) &&
765 // 'noexcept' is a trailing annotation.
766 Tok.isNot(tok::kw_noexcept);
Alexander Kornienkoa04e5e22013-04-09 16:15:19 +0000767}
768
Martin Probst533965c2016-04-19 18:19:06 +0000769static bool mustBeJSIdent(const AdditionalKeywords &Keywords,
770 const FormatToken *FormatTok) {
Daniel Jasper1dcbbcfc2016-03-14 19:21:36 +0000771 // FIXME: This returns true for C/C++ keywords like 'struct'.
772 return FormatTok->is(tok::identifier) &&
773 (FormatTok->Tok.getIdentifierInfo() == nullptr ||
Martin Probst3dbbefa2016-11-10 16:21:02 +0000774 !FormatTok->isOneOf(
775 Keywords.kw_in, Keywords.kw_of, Keywords.kw_as, Keywords.kw_async,
776 Keywords.kw_await, Keywords.kw_yield, Keywords.kw_finally,
777 Keywords.kw_function, Keywords.kw_import, Keywords.kw_is,
778 Keywords.kw_let, Keywords.kw_var, tok::kw_const,
779 Keywords.kw_abstract, Keywords.kw_extends, Keywords.kw_implements,
780 Keywords.kw_instanceof, Keywords.kw_interface,
Martin Probst93008f02017-07-18 14:00:19 +0000781 Keywords.kw_throws, Keywords.kw_from));
Daniel Jasper1dcbbcfc2016-03-14 19:21:36 +0000782}
783
Martin Probst533965c2016-04-19 18:19:06 +0000784static bool mustBeJSIdentOrValue(const AdditionalKeywords &Keywords,
785 const FormatToken *FormatTok) {
Martin Probstb9316ff2016-09-18 17:21:52 +0000786 return FormatTok->Tok.isLiteral() ||
787 FormatTok->isOneOf(tok::kw_true, tok::kw_false) ||
788 mustBeJSIdent(Keywords, FormatTok);
Martin Probst533965c2016-04-19 18:19:06 +0000789}
790
Daniel Jasper1dcbbcfc2016-03-14 19:21:36 +0000791// isJSDeclOrStmt returns true if |FormatTok| starts a declaration or statement
792// when encountered after a value (see mustBeJSIdentOrValue).
793static bool isJSDeclOrStmt(const AdditionalKeywords &Keywords,
794 const FormatToken *FormatTok) {
795 return FormatTok->isOneOf(
Martin Probst5f8445b2016-04-24 22:05:09 +0000796 tok::kw_return, Keywords.kw_yield,
Daniel Jasper1dcbbcfc2016-03-14 19:21:36 +0000797 // conditionals
798 tok::kw_if, tok::kw_else,
799 // loops
800 tok::kw_for, tok::kw_while, tok::kw_do, tok::kw_continue, tok::kw_break,
801 // switch/case
802 tok::kw_switch, tok::kw_case,
803 // exceptions
804 tok::kw_throw, tok::kw_try, tok::kw_catch, Keywords.kw_finally,
805 // declaration
806 tok::kw_const, tok::kw_class, Keywords.kw_var, Keywords.kw_let,
Martin Probst5f8445b2016-04-24 22:05:09 +0000807 Keywords.kw_async, Keywords.kw_function,
808 // import/export
809 Keywords.kw_import, tok::kw_export);
Daniel Jasper1dcbbcfc2016-03-14 19:21:36 +0000810}
811
812// readTokenWithJavaScriptASI reads the next token and terminates the current
813// line if JavaScript Automatic Semicolon Insertion must
814// happen between the current token and the next token.
815//
816// This method is conservative - it cannot cover all edge cases of JavaScript,
817// but only aims to correctly handle certain well known cases. It *must not*
818// return true in speculative cases.
819void UnwrappedLineParser::readTokenWithJavaScriptASI() {
820 FormatToken *Previous = FormatTok;
821 readToken();
822 FormatToken *Next = FormatTok;
823
824 bool IsOnSameLine =
825 CommentsBeforeNextToken.empty()
826 ? Next->NewlinesBefore == 0
827 : CommentsBeforeNextToken.front()->NewlinesBefore == 0;
828 if (IsOnSameLine)
829 return;
830
831 bool PreviousMustBeValue = mustBeJSIdentOrValue(Keywords, Previous);
Martin Probst717f6dc2016-10-21 05:11:38 +0000832 bool PreviousStartsTemplateExpr =
833 Previous->is(TT_TemplateString) && Previous->TokenText.endswith("${");
Martin Probstbbffeac2016-04-11 07:35:57 +0000834 if (PreviousMustBeValue && Line && Line->Tokens.size() > 1) {
835 // If the token before the previous one is an '@', the previous token is an
836 // annotation and can precede another identifier/value.
Benjamin Kramer5ffc24e2016-04-11 12:19:19 +0000837 const FormatToken *PrePrevious = std::prev(Line->Tokens.end(), 2)->Tok;
Martin Probstbbffeac2016-04-11 07:35:57 +0000838 if (PrePrevious->is(tok::at))
839 return;
840 }
Daniel Jasper1dcbbcfc2016-03-14 19:21:36 +0000841 if (Next->is(tok::exclaim) && PreviousMustBeValue)
Martin Probstd40bca42017-01-09 08:56:36 +0000842 return addUnwrappedLine();
Daniel Jasper1dcbbcfc2016-03-14 19:21:36 +0000843 bool NextMustBeValue = mustBeJSIdentOrValue(Keywords, Next);
Martin Probst717f6dc2016-10-21 05:11:38 +0000844 bool NextEndsTemplateExpr =
845 Next->is(TT_TemplateString) && Next->TokenText.startswith("}");
846 if (NextMustBeValue && !NextEndsTemplateExpr && !PreviousStartsTemplateExpr &&
847 (PreviousMustBeValue ||
848 Previous->isOneOf(tok::r_square, tok::r_paren, tok::plusplus,
849 tok::minusminus)))
Martin Probstd40bca42017-01-09 08:56:36 +0000850 return addUnwrappedLine();
Martin Probste6b5b342017-01-16 09:52:40 +0000851 if (PreviousMustBeValue && isJSDeclOrStmt(Keywords, Next))
Martin Probstd40bca42017-01-09 08:56:36 +0000852 return addUnwrappedLine();
Daniel Jasper1dcbbcfc2016-03-14 19:21:36 +0000853}
854
Manuel Klimek6b9eeba2013-01-07 14:56:16 +0000855void UnwrappedLineParser::parseStructuralElement() {
Daniel Jasper498f5582015-12-25 08:53:31 +0000856 assert(!FormatTok->is(tok::l_brace));
857 if (Style.Language == FormatStyle::LK_TableGen &&
858 FormatTok->is(tok::pp_include)) {
859 nextToken();
860 if (FormatTok->is(tok::string_literal))
861 nextToken();
862 addUnwrappedLine();
863 return;
864 }
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000865 switch (FormatTok->Tok.getKind()) {
Nico Weber04e9f1a2013-01-07 19:05:19 +0000866 case tok::at:
867 nextToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000868 if (FormatTok->Tok.is(tok::l_brace)) {
Krasimir Georgiev26b144c2017-07-03 15:05:14 +0000869 nextToken();
Nico Weber372d8dc2013-02-10 20:35:35 +0000870 parseBracedList();
871 break;
872 }
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000873 switch (FormatTok->Tok.getObjCKeywordID()) {
Nico Weber04e9f1a2013-01-07 19:05:19 +0000874 case tok::objc_public:
875 case tok::objc_protected:
876 case tok::objc_package:
877 case tok::objc_private:
878 return parseAccessSpecifier();
Nico Weber7eecf4b2013-01-09 20:25:35 +0000879 case tok::objc_interface:
Nico Weber2ce0ac52013-01-09 23:25:37 +0000880 case tok::objc_implementation:
881 return parseObjCInterfaceOrImplementation();
Nico Weber8696a8d2013-01-09 21:15:03 +0000882 case tok::objc_protocol:
883 return parseObjCProtocol();
Nico Weberd8ffe752013-01-09 21:42:32 +0000884 case tok::objc_end:
885 return; // Handled by the caller.
Nico Weber51306d22013-01-10 00:25:19 +0000886 case tok::objc_optional:
887 case tok::objc_required:
888 nextToken();
889 addUnwrappedLine();
890 return;
Nico Weber45c48122015-06-28 01:06:16 +0000891 case tok::objc_autoreleasepool:
892 nextToken();
893 if (FormatTok->Tok.is(tok::l_brace)) {
Daniel Jasperc1bc38e2015-09-29 14:57:55 +0000894 if (Style.BraceWrapping.AfterObjCDeclaration)
Nico Weber45c48122015-06-28 01:06:16 +0000895 addUnwrappedLine();
896 parseBlock(/*MustBeDeclaration=*/false);
897 }
898 addUnwrappedLine();
899 return;
Nico Weber33381f52015-02-07 01:57:32 +0000900 case tok::objc_try:
901 // This branch isn't strictly necessary (the kw_try case below would
902 // do this too after the tok::at is parsed above). But be explicit.
903 parseTryCatch();
904 return;
Nico Weber04e9f1a2013-01-07 19:05:19 +0000905 default:
906 break;
907 }
908 break;
Daniel Jasper8f463652014-08-26 23:15:12 +0000909 case tok::kw_asm:
Daniel Jasper8f463652014-08-26 23:15:12 +0000910 nextToken();
911 if (FormatTok->is(tok::l_brace)) {
Daniel Jasperc6366072015-05-10 08:42:04 +0000912 FormatTok->Type = TT_InlineASMBrace;
Daniel Jasper2337f282015-01-12 10:14:56 +0000913 nextToken();
Daniel Jasper4429f142014-08-27 17:16:46 +0000914 while (FormatTok && FormatTok->isNot(tok::eof)) {
Daniel Jasper8f463652014-08-26 23:15:12 +0000915 if (FormatTok->is(tok::r_brace)) {
Daniel Jasperc6366072015-05-10 08:42:04 +0000916 FormatTok->Type = TT_InlineASMBrace;
Daniel Jasper8f463652014-08-26 23:15:12 +0000917 nextToken();
Daniel Jasper790d4f92015-05-11 11:59:46 +0000918 addUnwrappedLine();
Daniel Jasper8f463652014-08-26 23:15:12 +0000919 break;
920 }
Daniel Jasper2337f282015-01-12 10:14:56 +0000921 FormatTok->Finalized = true;
Daniel Jasper8f463652014-08-26 23:15:12 +0000922 nextToken();
923 }
924 }
925 break;
Alexander Kornienko578fdd82012-12-06 18:03:27 +0000926 case tok::kw_namespace:
927 parseNamespace();
928 return;
Dmitri Gribenko58d64e22012-12-30 21:27:25 +0000929 case tok::kw_inline:
930 nextToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000931 if (FormatTok->Tok.is(tok::kw_namespace)) {
Dmitri Gribenko58d64e22012-12-30 21:27:25 +0000932 parseNamespace();
933 return;
934 }
935 break;
Alexander Kornienkob7076a22012-12-04 14:46:19 +0000936 case tok::kw_public:
937 case tok::kw_protected:
938 case tok::kw_private:
Daniel Jasper83709082015-02-18 17:14:05 +0000939 if (Style.Language == FormatStyle::LK_Java ||
940 Style.Language == FormatStyle::LK_JavaScript)
Daniel Jasperc58c70e2014-09-15 11:21:46 +0000941 nextToken();
942 else
943 parseAccessSpecifier();
Daniel Jasperf7935112012-12-03 18:12:45 +0000944 return;
Alexander Kornienkob7076a22012-12-04 14:46:19 +0000945 case tok::kw_if:
946 parseIfThenElse();
Daniel Jasperf7935112012-12-03 18:12:45 +0000947 return;
Alexander Kornienko37d6c942012-12-05 15:06:06 +0000948 case tok::kw_for:
949 case tok::kw_while:
950 parseForOrWhileLoop();
951 return;
Alexander Kornienkob7076a22012-12-04 14:46:19 +0000952 case tok::kw_do:
953 parseDoWhile();
954 return;
955 case tok::kw_switch:
956 parseSwitch();
957 return;
958 case tok::kw_default:
959 nextToken();
960 parseLabel();
961 return;
962 case tok::kw_case:
963 parseCaseLabel();
964 return;
Daniel Jasper04a71a42014-05-08 11:58:24 +0000965 case tok::kw_try:
Nico Weberfac23712015-02-04 15:26:27 +0000966 case tok::kw___try:
Daniel Jasper04a71a42014-05-08 11:58:24 +0000967 parseTryCatch();
968 return;
Manuel Klimekae610d12013-01-21 14:32:05 +0000969 case tok::kw_extern:
970 nextToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000971 if (FormatTok->Tok.is(tok::string_literal)) {
Manuel Klimekae610d12013-01-21 14:32:05 +0000972 nextToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +0000973 if (FormatTok->Tok.is(tok::l_brace)) {
Daniel Jasper65ee3472013-07-31 23:16:02 +0000974 parseBlock(/*MustBeDeclaration=*/true, /*AddLevel=*/false);
Manuel Klimekae610d12013-01-21 14:32:05 +0000975 addUnwrappedLine();
976 return;
977 }
978 }
Daniel Jaspere1e43192014-04-01 12:55:11 +0000979 break;
Daniel Jasperfca735c2015-02-19 16:14:18 +0000980 case tok::kw_export:
981 if (Style.Language == FormatStyle::LK_JavaScript) {
982 parseJavaScriptEs6ImportExport();
983 return;
984 }
985 break;
Daniel Jaspere1e43192014-04-01 12:55:11 +0000986 case tok::identifier:
Daniel Jasper66cb8c52015-05-04 09:22:29 +0000987 if (FormatTok->is(TT_ForEachMacro)) {
Daniel Jaspere1e43192014-04-01 12:55:11 +0000988 parseForOrWhileLoop();
989 return;
990 }
Birunthan Mohanathasb001a0b2015-07-03 17:25:16 +0000991 if (FormatTok->is(TT_MacroBlockBegin)) {
992 parseBlock(/*MustBeDeclaration=*/false, /*AddLevel=*/true,
993 /*MunchSemi=*/false);
994 return;
995 }
Daniel Jasper3d5a7d62016-06-20 18:20:38 +0000996 if (FormatTok->is(Keywords.kw_import)) {
997 if (Style.Language == FormatStyle::LK_JavaScript) {
998 parseJavaScriptEs6ImportExport();
999 return;
1000 }
1001 if (Style.Language == FormatStyle::LK_Proto) {
1002 nextToken();
Daniel Jasper8b61d142016-06-20 20:39:53 +00001003 if (FormatTok->is(tok::kw_public))
1004 nextToken();
Daniel Jasper3d5a7d62016-06-20 18:20:38 +00001005 if (!FormatTok->is(tok::string_literal))
1006 return;
1007 nextToken();
1008 if (FormatTok->is(tok::semi))
1009 nextToken();
1010 addUnwrappedLine();
1011 return;
1012 }
Daniel Jasper354aa512015-02-19 16:07:32 +00001013 }
Daniel Jasper1dbc2102017-03-31 13:30:24 +00001014 if (Style.isCpp() &&
Daniel Jasper72b33572017-03-31 12:04:37 +00001015 FormatTok->isOneOf(Keywords.kw_signals, Keywords.kw_qsignals,
Daniel Jaspera00de632015-12-01 12:05:04 +00001016 Keywords.kw_slots, Keywords.kw_qslots)) {
Daniel Jasperde0d1f32015-04-24 07:50:34 +00001017 nextToken();
1018 if (FormatTok->is(tok::colon)) {
1019 nextToken();
1020 addUnwrappedLine();
Daniel Jasper31343832016-07-27 10:13:24 +00001021 return;
Daniel Jasperde0d1f32015-04-24 07:50:34 +00001022 }
Daniel Jasper53395402015-04-07 15:04:40 +00001023 }
Manuel Klimekae610d12013-01-21 14:32:05 +00001024 // In all other cases, parse the declaration.
1025 break;
Alexander Kornienkob7076a22012-12-04 14:46:19 +00001026 default:
1027 break;
Daniel Jasperf7935112012-12-03 18:12:45 +00001028 }
Daniel Jasperf7935112012-12-03 18:12:45 +00001029 do {
Daniel Jaspera7900ad2016-05-08 18:12:22 +00001030 const FormatToken *Previous = getPreviousToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001031 switch (FormatTok->Tok.getKind()) {
Nico Weber372d8dc2013-02-10 20:35:35 +00001032 case tok::at:
1033 nextToken();
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001034 if (FormatTok->Tok.is(tok::l_brace)) {
1035 nextToken();
Nico Weber372d8dc2013-02-10 20:35:35 +00001036 parseBracedList();
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001037 }
Nico Weber372d8dc2013-02-10 20:35:35 +00001038 break;
Alexander Kornienkob7076a22012-12-04 14:46:19 +00001039 case tok::kw_enum:
Daniel Jaspera7900ad2016-05-08 18:12:22 +00001040 // Ignore if this is part of "template <enum ...".
1041 if (Previous && Previous->is(tok::less)) {
1042 nextToken();
1043 break;
1044 }
1045
Daniel Jasper90cf3802015-06-17 09:44:02 +00001046 // parseEnum falls through and does not yet add an unwrapped line as an
1047 // enum definition can start a structural element.
Daniel Jasper6f5a1932015-12-29 08:54:23 +00001048 if (!parseEnum())
1049 break;
Daniel Jasperc6dd2732015-07-16 14:25:43 +00001050 // This only applies for C++.
Daniel Jasper1dbc2102017-03-31 13:30:24 +00001051 if (!Style.isCpp()) {
Daniel Jasper90cf3802015-06-17 09:44:02 +00001052 addUnwrappedLine();
1053 return;
1054 }
Manuel Klimek2cec0192013-01-21 19:17:52 +00001055 break;
Daniel Jaspera88f80a2014-01-30 14:38:37 +00001056 case tok::kw_typedef:
1057 nextToken();
Daniel Jasper31f6c542014-12-05 10:42:21 +00001058 if (FormatTok->isOneOf(Keywords.kw_NS_ENUM, Keywords.kw_NS_OPTIONS,
1059 Keywords.kw_CF_ENUM, Keywords.kw_CF_OPTIONS))
Daniel Jaspera88f80a2014-01-30 14:38:37 +00001060 parseEnum();
1061 break;
Alexander Kornienko1231e062013-01-16 11:43:46 +00001062 case tok::kw_struct:
1063 case tok::kw_union:
Manuel Klimek28cacc72013-01-07 18:10:23 +00001064 case tok::kw_class:
Daniel Jasper910807d2015-06-12 04:52:02 +00001065 // parseRecord falls through and does not yet add an unwrapped line as a
1066 // record declaration or definition can start a structural element.
Manuel Klimeke01bab52013-01-15 13:38:33 +00001067 parseRecord();
Daniel Jasper910807d2015-06-12 04:52:02 +00001068 // This does not apply for Java and JavaScript.
1069 if (Style.Language == FormatStyle::LK_Java ||
1070 Style.Language == FormatStyle::LK_JavaScript) {
Daniel Jasperd5ec65b2016-01-08 07:06:07 +00001071 if (FormatTok->is(tok::semi))
1072 nextToken();
Daniel Jasper910807d2015-06-12 04:52:02 +00001073 addUnwrappedLine();
1074 return;
1075 }
Manuel Klimeke01bab52013-01-15 13:38:33 +00001076 break;
Daniel Jaspere5d74862014-11-26 08:17:08 +00001077 case tok::period:
1078 nextToken();
1079 // In Java, classes have an implicit static member "class".
1080 if (Style.Language == FormatStyle::LK_Java && FormatTok &&
1081 FormatTok->is(tok::kw_class))
1082 nextToken();
Daniel Jasperba52fcb2015-09-28 14:29:45 +00001083 if (Style.Language == FormatStyle::LK_JavaScript && FormatTok &&
1084 FormatTok->Tok.getIdentifierInfo())
1085 // JavaScript only has pseudo keywords, all keywords are allowed to
1086 // appear in "IdentifierName" positions. See http://es5.github.io/#x7.6
1087 nextToken();
Daniel Jaspere5d74862014-11-26 08:17:08 +00001088 break;
Daniel Jasperf7935112012-12-03 18:12:45 +00001089 case tok::semi:
1090 nextToken();
1091 addUnwrappedLine();
1092 return;
Alexander Kornienko1231e062013-01-16 11:43:46 +00001093 case tok::r_brace:
1094 addUnwrappedLine();
1095 return;
Daniel Jasperf7935112012-12-03 18:12:45 +00001096 case tok::l_paren:
1097 parseParens();
1098 break;
Daniel Jasper5af04a42015-10-07 03:43:10 +00001099 case tok::kw_operator:
1100 nextToken();
1101 if (FormatTok->isBinaryOperator())
1102 nextToken();
1103 break;
Manuel Klimek516e0542013-09-04 13:25:30 +00001104 case tok::caret:
1105 nextToken();
Daniel Jasper395193c2014-03-28 07:48:59 +00001106 if (FormatTok->Tok.isAnyIdentifier() ||
1107 FormatTok->isSimpleTypeSpecifier())
1108 nextToken();
1109 if (FormatTok->is(tok::l_paren))
1110 parseParens();
1111 if (FormatTok->is(tok::l_brace))
Manuel Klimek516e0542013-09-04 13:25:30 +00001112 parseChildBlock();
Manuel Klimek516e0542013-09-04 13:25:30 +00001113 break;
Daniel Jasperf7935112012-12-03 18:12:45 +00001114 case tok::l_brace:
Manuel Klimekab419912013-05-23 09:41:43 +00001115 if (!tryToParseBracedList()) {
1116 // A block outside of parentheses must be the last part of a
1117 // structural element.
1118 // FIXME: Figure out cases where this is not true, and add projections
1119 // for them (the one we know is missing are lambdas).
Daniel Jasperc1bc38e2015-09-29 14:57:55 +00001120 if (Style.BraceWrapping.AfterFunction)
Manuel Klimekab419912013-05-23 09:41:43 +00001121 addUnwrappedLine();
Alexander Kornienko3cfa9732013-11-20 16:33:05 +00001122 FormatTok->Type = TT_FunctionLBrace;
Nico Weber9096fc02013-06-26 00:30:14 +00001123 parseBlock(/*MustBeDeclaration=*/false);
Manuel Klimeka8eb9142013-05-13 12:51:40 +00001124 addUnwrappedLine();
Manuel Klimekab419912013-05-23 09:41:43 +00001125 return;
1126 }
1127 // Otherwise this was a braced init list, and the structural
1128 // element continues.
1129 break;
Daniel Jasper04a71a42014-05-08 11:58:24 +00001130 case tok::kw_try:
1131 // We arrive here when parsing function-try blocks.
1132 parseTryCatch();
1133 return;
Daniel Jasper40e19212013-05-29 13:16:10 +00001134 case tok::identifier: {
Birunthan Mohanathasb001a0b2015-07-03 17:25:16 +00001135 if (FormatTok->is(TT_MacroBlockEnd)) {
1136 addUnwrappedLine();
1137 return;
1138 }
1139
Martin Probst973ff792017-04-27 13:07:24 +00001140 // Function declarations (as opposed to function expressions) are parsed
1141 // on their own unwrapped line by continuing this loop. Function
1142 // expressions (functions that are not on their own line) must not create
1143 // a new unwrapped line, so they are special cased below.
1144 size_t TokenCount = Line->Tokens.size();
Daniel Jasper9326f912015-05-05 08:40:32 +00001145 if (Style.Language == FormatStyle::LK_JavaScript &&
Martin Probst973ff792017-04-27 13:07:24 +00001146 FormatTok->is(Keywords.kw_function) &&
1147 (TokenCount > 1 || (TokenCount == 1 && !Line->Tokens.front().Tok->is(
1148 Keywords.kw_async)))) {
Daniel Jasper069e5f42014-05-20 11:14:57 +00001149 tryToParseJSFunction();
1150 break;
1151 }
Daniel Jasper9326f912015-05-05 08:40:32 +00001152 if ((Style.Language == FormatStyle::LK_JavaScript ||
1153 Style.Language == FormatStyle::LK_Java) &&
1154 FormatTok->is(Keywords.kw_interface)) {
Martin Probst1e8261e2016-04-19 18:18:59 +00001155 if (Style.Language == FormatStyle::LK_JavaScript) {
1156 // In JavaScript/TypeScript, "interface" can be used as a standalone
1157 // identifier, e.g. in `var interface = 1;`. If "interface" is
1158 // followed by another identifier, it is very like to be an actual
1159 // interface declaration.
1160 unsigned StoredPosition = Tokens->getPosition();
1161 FormatToken *Next = Tokens->getNextToken();
1162 FormatTok = Tokens->setPosition(StoredPosition);
Martin Probst533965c2016-04-19 18:19:06 +00001163 if (Next && !mustBeJSIdent(Keywords, Next)) {
Martin Probst1e8261e2016-04-19 18:18:59 +00001164 nextToken();
1165 break;
1166 }
1167 }
Daniel Jasper9326f912015-05-05 08:40:32 +00001168 parseRecord();
Daniel Jasper259188b2015-06-12 04:56:34 +00001169 addUnwrappedLine();
Daniel Jasper5c235c02015-07-06 14:26:04 +00001170 return;
Daniel Jasper9326f912015-05-05 08:40:32 +00001171 }
1172
Daniel Jasper1dcbbcfc2016-03-14 19:21:36 +00001173 // See if the following token should start a new unwrapped line.
Daniel Jasper9326f912015-05-05 08:40:32 +00001174 StringRef Text = FormatTok->TokenText;
Daniel Jasperf7935112012-12-03 18:12:45 +00001175 nextToken();
Daniel Jasper83709082015-02-18 17:14:05 +00001176 if (Line->Tokens.size() == 1 &&
1177 // JS doesn't have macros, and within classes colons indicate fields,
1178 // not labels.
Daniel Jasper676e5162015-04-07 14:36:33 +00001179 Style.Language != FormatStyle::LK_JavaScript) {
1180 if (FormatTok->Tok.is(tok::colon) && !Line->MustBeDeclaration) {
Daniel Jasper40609472016-04-06 15:02:46 +00001181 Line->Tokens.begin()->Tok->MustBreakBefore = true;
Alexander Kornienkode644272013-04-08 22:16:06 +00001182 parseLabel();
1183 return;
1184 }
Daniel Jasper680b09b2014-11-05 10:48:04 +00001185 // Recognize function-like macro usages without trailing semicolon as
Daniel Jasper83709082015-02-18 17:14:05 +00001186 // well as free-standing macros like Q_OBJECT.
Daniel Jasper680b09b2014-11-05 10:48:04 +00001187 bool FunctionLike = FormatTok->is(tok::l_paren);
1188 if (FunctionLike)
Alexander Kornienkode644272013-04-08 22:16:06 +00001189 parseParens();
Daniel Jaspere60cba12015-05-13 11:35:53 +00001190
1191 bool FollowedByNewline =
1192 CommentsBeforeNextToken.empty()
1193 ? FormatTok->NewlinesBefore > 0
1194 : CommentsBeforeNextToken.front()->NewlinesBefore > 0;
1195
Daniel Jaspere6fcf7d2015-06-17 13:08:06 +00001196 if (FollowedByNewline && (Text.size() >= 5 || FunctionLike) &&
Daniel Jasper680b09b2014-11-05 10:48:04 +00001197 tokenCanStartNewLine(FormatTok->Tok) && Text == Text.upper()) {
Daniel Jasper40e19212013-05-29 13:16:10 +00001198 addUnwrappedLine();
Daniel Jasper41a0f782013-05-29 14:09:17 +00001199 return;
Alexander Kornienkode644272013-04-08 22:16:06 +00001200 }
Daniel Jasperf7935112012-12-03 18:12:45 +00001201 }
1202 break;
Daniel Jasper40e19212013-05-29 13:16:10 +00001203 }
Daniel Jaspere25509f2012-12-17 11:29:41 +00001204 case tok::equal:
Manuel Klimek79e06082015-05-21 12:23:34 +00001205 // Fat arrows (=>) have tok::TokenKind tok::equal but TokenType
1206 // TT_JsFatArrow. The always start an expression or a child block if
1207 // followed by a curly.
1208 if (FormatTok->is(TT_JsFatArrow)) {
1209 nextToken();
Daniel Jasperbe520bd2015-05-31 08:51:54 +00001210 if (FormatTok->is(tok::l_brace))
Manuel Klimek79e06082015-05-21 12:23:34 +00001211 parseChildBlock();
Manuel Klimek79e06082015-05-21 12:23:34 +00001212 break;
1213 }
1214
Daniel Jaspere25509f2012-12-17 11:29:41 +00001215 nextToken();
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001216 if (FormatTok->Tok.is(tok::l_brace)) {
1217 nextToken();
Manuel Klimek8e07a1b2013-01-10 11:52:21 +00001218 parseBracedList();
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001219 } else if (Style.Language == FormatStyle::LK_Proto &&
1220 FormatTok->Tok.is(tok::less)) {
1221 nextToken();
Krasimir Georgiev0b41fcb2017-06-27 13:58:41 +00001222 parseBracedList(/*ContinueOnSemicolons=*/false,
1223 /*ClosingBraceKind=*/tok::greater);
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001224 }
Daniel Jaspere25509f2012-12-17 11:29:41 +00001225 break;
Manuel Klimekffdeb592013-09-03 15:10:01 +00001226 case tok::l_square:
Daniel Jasperb88b25f2013-12-23 07:29:06 +00001227 parseSquare();
Manuel Klimekffdeb592013-09-03 15:10:01 +00001228 break;
Daniel Jasper6acf5132015-03-12 14:44:29 +00001229 case tok::kw_new:
1230 parseNew();
1231 break;
Alexander Kornienkob7076a22012-12-04 14:46:19 +00001232 default:
1233 nextToken();
1234 break;
Daniel Jasperf7935112012-12-03 18:12:45 +00001235 }
1236 } while (!eof());
1237}
1238
Daniel Jasperb88b25f2013-12-23 07:29:06 +00001239bool UnwrappedLineParser::tryToParseLambda() {
Daniel Jasper1dbc2102017-03-31 13:30:24 +00001240 if (!Style.isCpp()) {
Daniel Jasper1feab0f2015-06-02 15:31:37 +00001241 nextToken();
1242 return false;
1243 }
Daniel Jasperb9a49902016-01-09 15:56:28 +00001244 const FormatToken* Previous = getPreviousToken();
1245 if (Previous &&
1246 (Previous->isOneOf(tok::identifier, tok::kw_operator, tok::kw_new,
1247 tok::kw_delete) ||
1248 Previous->closesScope() || Previous->isSimpleTypeSpecifier())) {
Daniel Jasperbf02b2c12013-09-05 11:49:39 +00001249 nextToken();
Daniel Jasperb88b25f2013-12-23 07:29:06 +00001250 return false;
Daniel Jasperbf02b2c12013-09-05 11:49:39 +00001251 }
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +00001252 assert(FormatTok->is(tok::l_square));
1253 FormatToken &LSquare = *FormatTok;
Daniel Jasper9a8d48b2013-09-05 10:04:31 +00001254 if (!tryToParseLambdaIntroducer())
Daniel Jasperb88b25f2013-12-23 07:29:06 +00001255 return false;
Manuel Klimekffdeb592013-09-03 15:10:01 +00001256
Alexander Kornienkoc2ee9cf2014-03-13 13:59:48 +00001257 while (FormatTok->isNot(tok::l_brace)) {
Daniel Jaspercb51cf42014-01-16 09:11:55 +00001258 if (FormatTok->isSimpleTypeSpecifier()) {
1259 nextToken();
1260 continue;
1261 }
Manuel Klimekffdeb592013-09-03 15:10:01 +00001262 switch (FormatTok->Tok.getKind()) {
Daniel Jasper9a8d48b2013-09-05 10:04:31 +00001263 case tok::l_brace:
1264 break;
1265 case tok::l_paren:
1266 parseParens();
1267 break;
Daniel Jasperbcb55ee2014-11-21 14:08:38 +00001268 case tok::amp:
1269 case tok::star:
1270 case tok::kw_const:
Daniel Jasper3431b752014-12-08 13:22:37 +00001271 case tok::comma:
Daniel Jaspercb51cf42014-01-16 09:11:55 +00001272 case tok::less:
1273 case tok::greater:
Daniel Jasper9a8d48b2013-09-05 10:04:31 +00001274 case tok::identifier:
Daniel Jasper5eaa0092015-08-13 13:37:08 +00001275 case tok::numeric_constant:
Daniel Jasper1067ab02014-02-11 10:16:55 +00001276 case tok::coloncolon:
Daniel Jasper9a8d48b2013-09-05 10:04:31 +00001277 case tok::kw_mutable:
Daniel Jasper81a20782014-03-10 10:02:02 +00001278 nextToken();
1279 break;
Daniel Jaspercb51cf42014-01-16 09:11:55 +00001280 case tok::arrow:
Daniel Jasper6f2b88a2015-06-05 13:18:09 +00001281 FormatTok->Type = TT_LambdaArrow;
Daniel Jasper9a8d48b2013-09-05 10:04:31 +00001282 nextToken();
1283 break;
1284 default:
Daniel Jasperb88b25f2013-12-23 07:29:06 +00001285 return true;
Manuel Klimekffdeb592013-09-03 15:10:01 +00001286 }
1287 }
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +00001288 LSquare.Type = TT_LambdaLSquare;
Manuel Klimek516e0542013-09-04 13:25:30 +00001289 parseChildBlock();
Daniel Jasperb88b25f2013-12-23 07:29:06 +00001290 return true;
Manuel Klimekffdeb592013-09-03 15:10:01 +00001291}
1292
1293bool UnwrappedLineParser::tryToParseLambdaIntroducer() {
1294 nextToken();
1295 if (FormatTok->is(tok::equal)) {
1296 nextToken();
Daniel Jasper9a8d48b2013-09-05 10:04:31 +00001297 if (FormatTok->is(tok::r_square)) {
1298 nextToken();
1299 return true;
1300 }
1301 if (FormatTok->isNot(tok::comma))
1302 return false;
Manuel Klimekffdeb592013-09-03 15:10:01 +00001303 nextToken();
1304 } else if (FormatTok->is(tok::amp)) {
1305 nextToken();
Daniel Jasper9a8d48b2013-09-05 10:04:31 +00001306 if (FormatTok->is(tok::r_square)) {
1307 nextToken();
1308 return true;
1309 }
Manuel Klimekffdeb592013-09-03 15:10:01 +00001310 if (!FormatTok->isOneOf(tok::comma, tok::identifier)) {
1311 return false;
1312 }
Daniel Jasper9a8d48b2013-09-05 10:04:31 +00001313 if (FormatTok->is(tok::comma))
1314 nextToken();
Manuel Klimekffdeb592013-09-03 15:10:01 +00001315 } else if (FormatTok->is(tok::r_square)) {
1316 nextToken();
1317 return true;
1318 }
1319 do {
Daniel Jasper9a8d48b2013-09-05 10:04:31 +00001320 if (FormatTok->is(tok::amp))
1321 nextToken();
1322 if (!FormatTok->isOneOf(tok::identifier, tok::kw_this))
1323 return false;
Manuel Klimekffdeb592013-09-03 15:10:01 +00001324 nextToken();
Daniel Jasperda18fd82014-06-10 06:39:03 +00001325 if (FormatTok->is(tok::ellipsis))
1326 nextToken();
Manuel Klimekffdeb592013-09-03 15:10:01 +00001327 if (FormatTok->is(tok::comma)) {
1328 nextToken();
1329 } else if (FormatTok->is(tok::r_square)) {
1330 nextToken();
1331 return true;
1332 } else {
1333 return false;
1334 }
1335 } while (!eof());
1336 return false;
1337}
1338
Daniel Jasperc03e16a2014-05-08 09:25:39 +00001339void UnwrappedLineParser::tryToParseJSFunction() {
Martin Probst409697e2016-05-29 14:41:07 +00001340 assert(FormatTok->is(Keywords.kw_function) ||
1341 FormatTok->startsSequence(Keywords.kw_async, Keywords.kw_function));
Martin Probst5f8445b2016-04-24 22:05:09 +00001342 if (FormatTok->is(Keywords.kw_async))
1343 nextToken();
1344 // Consume "function".
Daniel Jasperc03e16a2014-05-08 09:25:39 +00001345 nextToken();
Daniel Jasper5217a8b2014-06-13 07:02:04 +00001346
Daniel Jasper71e50af2016-11-01 06:22:59 +00001347 // Consume * (generator function). Treat it like C++'s overloaded operators.
1348 if (FormatTok->is(tok::star)) {
1349 FormatTok->Type = TT_OverloadedOperator;
Martin Probst5f8445b2016-04-24 22:05:09 +00001350 nextToken();
Daniel Jasper71e50af2016-11-01 06:22:59 +00001351 }
Martin Probst5f8445b2016-04-24 22:05:09 +00001352
Daniel Jasper5217a8b2014-06-13 07:02:04 +00001353 // Consume function name.
1354 if (FormatTok->is(tok::identifier))
Daniel Jasperfca735c2015-02-19 16:14:18 +00001355 nextToken();
Daniel Jasper5217a8b2014-06-13 07:02:04 +00001356
Daniel Jasperc03e16a2014-05-08 09:25:39 +00001357 if (FormatTok->isNot(tok::l_paren))
1358 return;
Manuel Klimek79e06082015-05-21 12:23:34 +00001359
1360 // Parse formal parameter list.
Daniel Jasperbe520bd2015-05-31 08:51:54 +00001361 parseParens();
Manuel Klimek79e06082015-05-21 12:23:34 +00001362
1363 if (FormatTok->is(tok::colon)) {
1364 // Parse a type definition.
1365 nextToken();
1366
1367 // Eat the type declaration. For braced inline object types, balance braces,
1368 // otherwise just parse until finding an l_brace for the function body.
Daniel Jasperbe520bd2015-05-31 08:51:54 +00001369 if (FormatTok->is(tok::l_brace))
1370 tryToParseBracedList();
1371 else
Martin Probstaf16c502017-01-04 13:36:43 +00001372 while (!FormatTok->isOneOf(tok::l_brace, tok::semi) && !eof())
Manuel Klimek79e06082015-05-21 12:23:34 +00001373 nextToken();
Manuel Klimek79e06082015-05-21 12:23:34 +00001374 }
1375
Martin Probstaf16c502017-01-04 13:36:43 +00001376 if (FormatTok->is(tok::semi))
1377 return;
1378
Manuel Klimek79e06082015-05-21 12:23:34 +00001379 parseChildBlock();
1380}
1381
Daniel Jasper3c883d12015-05-18 14:49:19 +00001382bool UnwrappedLineParser::tryToParseBracedList() {
Daniel Jasperb1f74a82013-07-09 09:06:29 +00001383 if (FormatTok->BlockKind == BK_Unknown)
Daniel Jasper3c883d12015-05-18 14:49:19 +00001384 calculateBraceTypes();
Daniel Jasperb1f74a82013-07-09 09:06:29 +00001385 assert(FormatTok->BlockKind != BK_Unknown);
1386 if (FormatTok->BlockKind == BK_Block)
Manuel Klimekab419912013-05-23 09:41:43 +00001387 return false;
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001388 nextToken();
Manuel Klimekab419912013-05-23 09:41:43 +00001389 parseBracedList();
1390 return true;
1391}
1392
Krasimir Georgievff747be2017-06-27 13:43:07 +00001393bool UnwrappedLineParser::parseBracedList(bool ContinueOnSemicolons,
1394 tok::TokenKind ClosingBraceKind) {
Daniel Jasper015ed022013-09-13 09:20:45 +00001395 bool HasError = false;
Manuel Klimek8e07a1b2013-01-10 11:52:21 +00001396
Manuel Klimeka3ff45e2013-04-10 09:52:05 +00001397 // FIXME: Once we have an expression parser in the UnwrappedLineParser,
1398 // replace this by using parseAssigmentExpression() inside.
Manuel Klimek8e07a1b2013-01-10 11:52:21 +00001399 do {
Manuel Klimek79e06082015-05-21 12:23:34 +00001400 if (Style.Language == FormatStyle::LK_JavaScript) {
Martin Probst409697e2016-05-29 14:41:07 +00001401 if (FormatTok->is(Keywords.kw_function) ||
1402 FormatTok->startsSequence(Keywords.kw_async, Keywords.kw_function)) {
Manuel Klimek79e06082015-05-21 12:23:34 +00001403 tryToParseJSFunction();
1404 continue;
Daniel Jasperbe520bd2015-05-31 08:51:54 +00001405 }
1406 if (FormatTok->is(TT_JsFatArrow)) {
Manuel Klimek79e06082015-05-21 12:23:34 +00001407 nextToken();
1408 // Fat arrows can be followed by simple expressions or by child blocks
1409 // in curly braces.
Daniel Jaspere6fcf7d2015-06-17 13:08:06 +00001410 if (FormatTok->is(tok::l_brace)) {
Manuel Klimek79e06082015-05-21 12:23:34 +00001411 parseChildBlock();
1412 continue;
1413 }
1414 }
Martin Probst8e3eba02017-02-07 16:33:13 +00001415 if (FormatTok->is(tok::l_brace)) {
1416 // Could be a method inside of a braced list `{a() { return 1; }}`.
1417 if (tryToParseBracedList())
1418 continue;
1419 parseChildBlock();
1420 }
Daniel Jasperc03e16a2014-05-08 09:25:39 +00001421 }
Krasimir Georgievff747be2017-06-27 13:43:07 +00001422 if (FormatTok->Tok.getKind() == ClosingBraceKind) {
1423 nextToken();
1424 return !HasError;
1425 }
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001426 switch (FormatTok->Tok.getKind()) {
Manuel Klimek516e0542013-09-04 13:25:30 +00001427 case tok::caret:
1428 nextToken();
1429 if (FormatTok->is(tok::l_brace)) {
1430 parseChildBlock();
1431 }
1432 break;
1433 case tok::l_square:
1434 tryToParseLambda();
1435 break;
Daniel Jaspera87af7a2015-06-30 11:32:22 +00001436 case tok::l_paren:
1437 parseParens();
Daniel Jasperf46dec82015-03-31 14:34:15 +00001438 // JavaScript can just have free standing methods and getters/setters in
1439 // object literals. Detect them by a "{" following ")".
1440 if (Style.Language == FormatStyle::LK_JavaScript) {
Daniel Jasperf46dec82015-03-31 14:34:15 +00001441 if (FormatTok->is(tok::l_brace))
1442 parseChildBlock();
1443 break;
1444 }
Daniel Jasperf46dec82015-03-31 14:34:15 +00001445 break;
Martin Probst8e3eba02017-02-07 16:33:13 +00001446 case tok::l_brace:
1447 // Assume there are no blocks inside a braced init list apart
1448 // from the ones we explicitly parse out (like lambdas).
1449 FormatTok->BlockKind = BK_BracedInit;
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001450 nextToken();
Martin Probst8e3eba02017-02-07 16:33:13 +00001451 parseBracedList();
1452 break;
Manuel Klimeka3ff45e2013-04-10 09:52:05 +00001453 case tok::semi:
Daniel Jasperb9a49902016-01-09 15:56:28 +00001454 // JavaScript (or more precisely TypeScript) can have semicolons in braced
1455 // lists (in so-called TypeMemberLists). Thus, the semicolon cannot be
1456 // used for error recovery if we have otherwise determined that this is
1457 // a braced list.
1458 if (Style.Language == FormatStyle::LK_JavaScript) {
1459 nextToken();
1460 break;
1461 }
Daniel Jasper015ed022013-09-13 09:20:45 +00001462 HasError = true;
1463 if (!ContinueOnSemicolons)
1464 return !HasError;
1465 nextToken();
1466 break;
Manuel Klimeka3ff45e2013-04-10 09:52:05 +00001467 case tok::comma:
1468 nextToken();
Manuel Klimeka3ff45e2013-04-10 09:52:05 +00001469 break;
Manuel Klimek8e07a1b2013-01-10 11:52:21 +00001470 default:
1471 nextToken();
1472 break;
1473 }
1474 } while (!eof());
Daniel Jasper015ed022013-09-13 09:20:45 +00001475 return false;
Manuel Klimek8e07a1b2013-01-10 11:52:21 +00001476}
1477
Daniel Jasperf7935112012-12-03 18:12:45 +00001478void UnwrappedLineParser::parseParens() {
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001479 assert(FormatTok->Tok.is(tok::l_paren) && "'(' expected.");
Daniel Jasperf7935112012-12-03 18:12:45 +00001480 nextToken();
1481 do {
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001482 switch (FormatTok->Tok.getKind()) {
Daniel Jasperf7935112012-12-03 18:12:45 +00001483 case tok::l_paren:
1484 parseParens();
Daniel Jasper5f1fa852015-01-04 20:40:51 +00001485 if (Style.Language == FormatStyle::LK_Java && FormatTok->is(tok::l_brace))
1486 parseChildBlock();
Daniel Jasperf7935112012-12-03 18:12:45 +00001487 break;
1488 case tok::r_paren:
1489 nextToken();
1490 return;
Daniel Jasper393564f2013-05-31 14:56:29 +00001491 case tok::r_brace:
1492 // A "}" inside parenthesis is an error if there wasn't a matching "{".
1493 return;
Daniel Jasper9a8d48b2013-09-05 10:04:31 +00001494 case tok::l_square:
1495 tryToParseLambda();
1496 break;
Daniel Jasper5f1fa852015-01-04 20:40:51 +00001497 case tok::l_brace:
Daniel Jasperadba2aa2015-05-18 12:52:00 +00001498 if (!tryToParseBracedList())
Manuel Klimekf017dc02013-09-04 13:34:14 +00001499 parseChildBlock();
Manuel Klimek8e07a1b2013-01-10 11:52:21 +00001500 break;
Nico Weber372d8dc2013-02-10 20:35:35 +00001501 case tok::at:
1502 nextToken();
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001503 if (FormatTok->Tok.is(tok::l_brace)) {
1504 nextToken();
Nico Weber372d8dc2013-02-10 20:35:35 +00001505 parseBracedList();
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001506 }
Nico Weber372d8dc2013-02-10 20:35:35 +00001507 break;
Martin Probst1027fb82017-02-07 14:05:30 +00001508 case tok::kw_class:
1509 if (Style.Language == FormatStyle::LK_JavaScript)
1510 parseRecord(/*ParseAsExpr=*/true);
1511 else
1512 nextToken();
1513 break;
Daniel Jasper3f69ba12014-09-05 08:42:27 +00001514 case tok::identifier:
1515 if (Style.Language == FormatStyle::LK_JavaScript &&
Martin Probst409697e2016-05-29 14:41:07 +00001516 (FormatTok->is(Keywords.kw_function) ||
1517 FormatTok->startsSequence(Keywords.kw_async, Keywords.kw_function)))
Daniel Jasper3f69ba12014-09-05 08:42:27 +00001518 tryToParseJSFunction();
1519 else
1520 nextToken();
1521 break;
Daniel Jasperf7935112012-12-03 18:12:45 +00001522 default:
1523 nextToken();
1524 break;
1525 }
1526 } while (!eof());
1527}
1528
Daniel Jasperb88b25f2013-12-23 07:29:06 +00001529void UnwrappedLineParser::parseSquare() {
1530 assert(FormatTok->Tok.is(tok::l_square) && "'[' expected.");
1531 if (tryToParseLambda())
1532 return;
1533 do {
Daniel Jasperb88b25f2013-12-23 07:29:06 +00001534 switch (FormatTok->Tok.getKind()) {
1535 case tok::l_paren:
1536 parseParens();
1537 break;
1538 case tok::r_square:
1539 nextToken();
1540 return;
1541 case tok::r_brace:
1542 // A "}" inside parenthesis is an error if there wasn't a matching "{".
1543 return;
1544 case tok::l_square:
1545 parseSquare();
1546 break;
1547 case tok::l_brace: {
Daniel Jasperadba2aa2015-05-18 12:52:00 +00001548 if (!tryToParseBracedList())
Daniel Jasperb88b25f2013-12-23 07:29:06 +00001549 parseChildBlock();
Daniel Jasperb88b25f2013-12-23 07:29:06 +00001550 break;
1551 }
1552 case tok::at:
1553 nextToken();
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001554 if (FormatTok->Tok.is(tok::l_brace)) {
1555 nextToken();
Daniel Jasperb88b25f2013-12-23 07:29:06 +00001556 parseBracedList();
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001557 }
Daniel Jasperb88b25f2013-12-23 07:29:06 +00001558 break;
1559 default:
1560 nextToken();
1561 break;
1562 }
1563 } while (!eof());
1564}
1565
Daniel Jasperf7935112012-12-03 18:12:45 +00001566void UnwrappedLineParser::parseIfThenElse() {
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001567 assert(FormatTok->Tok.is(tok::kw_if) && "'if' expected");
Daniel Jasperf7935112012-12-03 18:12:45 +00001568 nextToken();
Daniel Jasper6a7d5a72017-06-19 07:40:49 +00001569 if (FormatTok->Tok.is(tok::kw_constexpr))
1570 nextToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001571 if (FormatTok->Tok.is(tok::l_paren))
Manuel Klimekadededf2013-01-11 18:28:36 +00001572 parseParens();
Daniel Jasperf7935112012-12-03 18:12:45 +00001573 bool NeedsUnwrappedLine = false;
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001574 if (FormatTok->Tok.is(tok::l_brace)) {
Alexander Kornienko3a33f022013-12-12 09:49:52 +00001575 CompoundStatementIndenter Indenter(this, Style, Line->Level);
Nico Weber9096fc02013-06-26 00:30:14 +00001576 parseBlock(/*MustBeDeclaration=*/false);
Daniel Jasperc1bc38e2015-09-29 14:57:55 +00001577 if (Style.BraceWrapping.BeforeElse)
Manuel Klimekd3ed59a2013-08-02 21:31:59 +00001578 addUnwrappedLine();
Daniel Jasperc1bc38e2015-09-29 14:57:55 +00001579 else
Manuel Klimekd3ed59a2013-08-02 21:31:59 +00001580 NeedsUnwrappedLine = true;
Daniel Jasperf7935112012-12-03 18:12:45 +00001581 } else {
1582 addUnwrappedLine();
Manuel Klimek52b15152013-01-09 15:25:02 +00001583 ++Line->Level;
Manuel Klimek6b9eeba2013-01-07 14:56:16 +00001584 parseStructuralElement();
Manuel Klimek52b15152013-01-09 15:25:02 +00001585 --Line->Level;
Daniel Jasperf7935112012-12-03 18:12:45 +00001586 }
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001587 if (FormatTok->Tok.is(tok::kw_else)) {
Daniel Jasperf7935112012-12-03 18:12:45 +00001588 nextToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001589 if (FormatTok->Tok.is(tok::l_brace)) {
Alexander Kornienko3a33f022013-12-12 09:49:52 +00001590 CompoundStatementIndenter Indenter(this, Style, Line->Level);
Nico Weber9096fc02013-06-26 00:30:14 +00001591 parseBlock(/*MustBeDeclaration=*/false);
Daniel Jasperf7935112012-12-03 18:12:45 +00001592 addUnwrappedLine();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001593 } else if (FormatTok->Tok.is(tok::kw_if)) {
Daniel Jasperf7935112012-12-03 18:12:45 +00001594 parseIfThenElse();
1595 } else {
1596 addUnwrappedLine();
Manuel Klimek52b15152013-01-09 15:25:02 +00001597 ++Line->Level;
Manuel Klimek6b9eeba2013-01-07 14:56:16 +00001598 parseStructuralElement();
Daniel Jasper451544a2016-05-19 06:30:48 +00001599 if (FormatTok->is(tok::eof))
1600 addUnwrappedLine();
Manuel Klimek52b15152013-01-09 15:25:02 +00001601 --Line->Level;
Daniel Jasperf7935112012-12-03 18:12:45 +00001602 }
1603 } else if (NeedsUnwrappedLine) {
1604 addUnwrappedLine();
1605 }
1606}
1607
Daniel Jasper04a71a42014-05-08 11:58:24 +00001608void UnwrappedLineParser::parseTryCatch() {
Nico Weberfac23712015-02-04 15:26:27 +00001609 assert(FormatTok->isOneOf(tok::kw_try, tok::kw___try) && "'try' expected");
Daniel Jasper04a71a42014-05-08 11:58:24 +00001610 nextToken();
1611 bool NeedsUnwrappedLine = false;
1612 if (FormatTok->is(tok::colon)) {
1613 // We are in a function try block, what comes is an initializer list.
1614 nextToken();
1615 while (FormatTok->is(tok::identifier)) {
1616 nextToken();
1617 if (FormatTok->is(tok::l_paren))
1618 parseParens();
Daniel Jasper04a71a42014-05-08 11:58:24 +00001619 if (FormatTok->is(tok::comma))
1620 nextToken();
1621 }
1622 }
Daniel Jaspere189d462015-01-14 10:48:41 +00001623 // Parse try with resource.
1624 if (Style.Language == FormatStyle::LK_Java && FormatTok->is(tok::l_paren)) {
1625 parseParens();
1626 }
Daniel Jasper04a71a42014-05-08 11:58:24 +00001627 if (FormatTok->is(tok::l_brace)) {
1628 CompoundStatementIndenter Indenter(this, Style, Line->Level);
1629 parseBlock(/*MustBeDeclaration=*/false);
Daniel Jasperc1bc38e2015-09-29 14:57:55 +00001630 if (Style.BraceWrapping.BeforeCatch) {
Daniel Jasper04a71a42014-05-08 11:58:24 +00001631 addUnwrappedLine();
1632 } else {
1633 NeedsUnwrappedLine = true;
1634 }
1635 } else if (!FormatTok->is(tok::kw_catch)) {
1636 // The C++ standard requires a compound-statement after a try.
1637 // If there's none, we try to assume there's a structuralElement
1638 // and try to continue.
Daniel Jasper04a71a42014-05-08 11:58:24 +00001639 addUnwrappedLine();
1640 ++Line->Level;
1641 parseStructuralElement();
1642 --Line->Level;
1643 }
Nico Weber33381f52015-02-07 01:57:32 +00001644 while (1) {
1645 if (FormatTok->is(tok::at))
1646 nextToken();
1647 if (!(FormatTok->isOneOf(tok::kw_catch, Keywords.kw___except,
1648 tok::kw___finally) ||
1649 ((Style.Language == FormatStyle::LK_Java ||
1650 Style.Language == FormatStyle::LK_JavaScript) &&
1651 FormatTok->is(Keywords.kw_finally)) ||
1652 (FormatTok->Tok.isObjCAtKeyword(tok::objc_catch) ||
1653 FormatTok->Tok.isObjCAtKeyword(tok::objc_finally))))
1654 break;
Daniel Jasper04a71a42014-05-08 11:58:24 +00001655 nextToken();
1656 while (FormatTok->isNot(tok::l_brace)) {
1657 if (FormatTok->is(tok::l_paren)) {
1658 parseParens();
1659 continue;
1660 }
Daniel Jasper2bd7a642015-01-19 10:50:51 +00001661 if (FormatTok->isOneOf(tok::semi, tok::r_brace, tok::eof))
Daniel Jasper04a71a42014-05-08 11:58:24 +00001662 return;
1663 nextToken();
1664 }
1665 NeedsUnwrappedLine = false;
1666 CompoundStatementIndenter Indenter(this, Style, Line->Level);
1667 parseBlock(/*MustBeDeclaration=*/false);
Daniel Jasperc1bc38e2015-09-29 14:57:55 +00001668 if (Style.BraceWrapping.BeforeCatch)
Daniel Jasper04a71a42014-05-08 11:58:24 +00001669 addUnwrappedLine();
Daniel Jasperc1bc38e2015-09-29 14:57:55 +00001670 else
Daniel Jasper04a71a42014-05-08 11:58:24 +00001671 NeedsUnwrappedLine = true;
Daniel Jasper04a71a42014-05-08 11:58:24 +00001672 }
Daniel Jasperc1bc38e2015-09-29 14:57:55 +00001673 if (NeedsUnwrappedLine)
Daniel Jasper04a71a42014-05-08 11:58:24 +00001674 addUnwrappedLine();
Daniel Jasper04a71a42014-05-08 11:58:24 +00001675}
1676
Alexander Kornienko578fdd82012-12-06 18:03:27 +00001677void UnwrappedLineParser::parseNamespace() {
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001678 assert(FormatTok->Tok.is(tok::kw_namespace) && "'namespace' expected");
Roman Kashitsyna043ced2014-08-11 12:18:01 +00001679
1680 const FormatToken &InitialToken = *FormatTok;
Alexander Kornienko578fdd82012-12-06 18:03:27 +00001681 nextToken();
Saleem Abdulrasool328085f2015-10-30 05:07:56 +00001682 while (FormatTok->isOneOf(tok::identifier, tok::coloncolon))
Alexander Kornienko578fdd82012-12-06 18:03:27 +00001683 nextToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001684 if (FormatTok->Tok.is(tok::l_brace)) {
Roman Kashitsyna043ced2014-08-11 12:18:01 +00001685 if (ShouldBreakBeforeBrace(Style, InitialToken))
Manuel Klimeka8eb9142013-05-13 12:51:40 +00001686 addUnwrappedLine();
1687
Daniel Jasper65ee3472013-07-31 23:16:02 +00001688 bool AddLevel = Style.NamespaceIndentation == FormatStyle::NI_All ||
1689 (Style.NamespaceIndentation == FormatStyle::NI_Inner &&
1690 DeclarationScopeStack.size() > 1);
1691 parseBlock(/*MustBeDeclaration=*/true, AddLevel);
Manuel Klimek046b9302013-02-06 16:08:09 +00001692 // Munch the semicolon after a namespace. This is more common than one would
1693 // think. Puttin the semicolon into its own line is very ugly.
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001694 if (FormatTok->Tok.is(tok::semi))
Manuel Klimek046b9302013-02-06 16:08:09 +00001695 nextToken();
Alexander Kornienko578fdd82012-12-06 18:03:27 +00001696 addUnwrappedLine();
1697 }
1698 // FIXME: Add error handling.
1699}
1700
Daniel Jasper6acf5132015-03-12 14:44:29 +00001701void UnwrappedLineParser::parseNew() {
1702 assert(FormatTok->is(tok::kw_new) && "'new' expected");
1703 nextToken();
1704 if (Style.Language != FormatStyle::LK_Java)
1705 return;
1706
1707 // In Java, we can parse everything up to the parens, which aren't optional.
1708 do {
1709 // There should not be a ;, { or } before the new's open paren.
1710 if (FormatTok->isOneOf(tok::semi, tok::l_brace, tok::r_brace))
1711 return;
1712
1713 // Consume the parens.
1714 if (FormatTok->is(tok::l_paren)) {
1715 parseParens();
1716
1717 // If there is a class body of an anonymous class, consume that as child.
1718 if (FormatTok->is(tok::l_brace))
1719 parseChildBlock();
1720 return;
1721 }
1722 nextToken();
1723 } while (!eof());
1724}
1725
Alexander Kornienko37d6c942012-12-05 15:06:06 +00001726void UnwrappedLineParser::parseForOrWhileLoop() {
Daniel Jasper66cb8c52015-05-04 09:22:29 +00001727 assert(FormatTok->isOneOf(tok::kw_for, tok::kw_while, TT_ForEachMacro) &&
Daniel Jaspere1e43192014-04-01 12:55:11 +00001728 "'for', 'while' or foreach macro expected");
Alexander Kornienko37d6c942012-12-05 15:06:06 +00001729 nextToken();
Martin Probsta050f412017-05-18 21:19:29 +00001730 // JS' for await ( ...
Martin Probstbd49e322017-05-15 19:33:20 +00001731 if (Style.Language == FormatStyle::LK_JavaScript &&
Martin Probsta050f412017-05-18 21:19:29 +00001732 FormatTok->is(Keywords.kw_await))
Martin Probstbd49e322017-05-15 19:33:20 +00001733 nextToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001734 if (FormatTok->Tok.is(tok::l_paren))
Manuel Klimek9fa8d552013-01-11 19:23:05 +00001735 parseParens();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001736 if (FormatTok->Tok.is(tok::l_brace)) {
Alexander Kornienko3a33f022013-12-12 09:49:52 +00001737 CompoundStatementIndenter Indenter(this, Style, Line->Level);
Nico Weber9096fc02013-06-26 00:30:14 +00001738 parseBlock(/*MustBeDeclaration=*/false);
Alexander Kornienko37d6c942012-12-05 15:06:06 +00001739 addUnwrappedLine();
1740 } else {
1741 addUnwrappedLine();
Manuel Klimek52b15152013-01-09 15:25:02 +00001742 ++Line->Level;
Manuel Klimek6b9eeba2013-01-07 14:56:16 +00001743 parseStructuralElement();
Manuel Klimek52b15152013-01-09 15:25:02 +00001744 --Line->Level;
Alexander Kornienko37d6c942012-12-05 15:06:06 +00001745 }
1746}
1747
Daniel Jasperf7935112012-12-03 18:12:45 +00001748void UnwrappedLineParser::parseDoWhile() {
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001749 assert(FormatTok->Tok.is(tok::kw_do) && "'do' expected");
Daniel Jasperf7935112012-12-03 18:12:45 +00001750 nextToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001751 if (FormatTok->Tok.is(tok::l_brace)) {
Alexander Kornienko3a33f022013-12-12 09:49:52 +00001752 CompoundStatementIndenter Indenter(this, Style, Line->Level);
Nico Weber9096fc02013-06-26 00:30:14 +00001753 parseBlock(/*MustBeDeclaration=*/false);
Daniel Jasperc1bc38e2015-09-29 14:57:55 +00001754 if (Style.BraceWrapping.IndentBraces)
Alexander Kornienko3a33f022013-12-12 09:49:52 +00001755 addUnwrappedLine();
Daniel Jasperf7935112012-12-03 18:12:45 +00001756 } else {
1757 addUnwrappedLine();
Manuel Klimek52b15152013-01-09 15:25:02 +00001758 ++Line->Level;
Manuel Klimek6b9eeba2013-01-07 14:56:16 +00001759 parseStructuralElement();
Manuel Klimek52b15152013-01-09 15:25:02 +00001760 --Line->Level;
Daniel Jasperf7935112012-12-03 18:12:45 +00001761 }
1762
Alexander Kornienko0ea8e102012-12-04 15:40:36 +00001763 // FIXME: Add error handling.
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001764 if (!FormatTok->Tok.is(tok::kw_while)) {
Alexander Kornienko0ea8e102012-12-04 15:40:36 +00001765 addUnwrappedLine();
1766 return;
1767 }
1768
Daniel Jasperf7935112012-12-03 18:12:45 +00001769 nextToken();
Manuel Klimek6b9eeba2013-01-07 14:56:16 +00001770 parseStructuralElement();
Daniel Jasperf7935112012-12-03 18:12:45 +00001771}
1772
1773void UnwrappedLineParser::parseLabel() {
Daniel Jasperf7935112012-12-03 18:12:45 +00001774 nextToken();
Manuel Klimek52b15152013-01-09 15:25:02 +00001775 unsigned OldLineLevel = Line->Level;
Daniel Jaspera1275122013-03-20 10:23:53 +00001776 if (Line->Level > 1 || (!Line->InPPDirective && Line->Level > 0))
Manuel Klimek52b15152013-01-09 15:25:02 +00001777 --Line->Level;
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001778 if (CommentsBeforeNextToken.empty() && FormatTok->Tok.is(tok::l_brace)) {
Alexander Kornienko3a33f022013-12-12 09:49:52 +00001779 CompoundStatementIndenter Indenter(this, Style, Line->Level);
Nico Weber9096fc02013-06-26 00:30:14 +00001780 parseBlock(/*MustBeDeclaration=*/false);
Manuel Klimekd3ed59a2013-08-02 21:31:59 +00001781 if (FormatTok->Tok.is(tok::kw_break)) {
Daniel Jasperc1bc38e2015-09-29 14:57:55 +00001782 if (Style.BraceWrapping.AfterControlStatement)
Manuel Klimekd3ed59a2013-08-02 21:31:59 +00001783 addUnwrappedLine();
1784 parseStructuralElement();
1785 }
Alexander Kornienko3a33f022013-12-12 09:49:52 +00001786 addUnwrappedLine();
1787 } else {
Daniel Jasper1fe0d5c2015-05-06 15:19:47 +00001788 if (FormatTok->is(tok::semi))
1789 nextToken();
Alexander Kornienko3a33f022013-12-12 09:49:52 +00001790 addUnwrappedLine();
Daniel Jasperf7935112012-12-03 18:12:45 +00001791 }
Manuel Klimek52b15152013-01-09 15:25:02 +00001792 Line->Level = OldLineLevel;
Daniel Jasper2cce7b72016-04-06 16:41:39 +00001793 if (FormatTok->isNot(tok::l_brace)) {
Daniel Jasper40609472016-04-06 15:02:46 +00001794 parseStructuralElement();
Daniel Jasper2cce7b72016-04-06 16:41:39 +00001795 addUnwrappedLine();
1796 }
Daniel Jasperf7935112012-12-03 18:12:45 +00001797}
1798
1799void UnwrappedLineParser::parseCaseLabel() {
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001800 assert(FormatTok->Tok.is(tok::kw_case) && "'case' expected");
Daniel Jasperf7935112012-12-03 18:12:45 +00001801 // FIXME: fix handling of complex expressions here.
1802 do {
1803 nextToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001804 } while (!eof() && !FormatTok->Tok.is(tok::colon));
Daniel Jasperf7935112012-12-03 18:12:45 +00001805 parseLabel();
1806}
1807
1808void UnwrappedLineParser::parseSwitch() {
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001809 assert(FormatTok->Tok.is(tok::kw_switch) && "'switch' expected");
Daniel Jasperf7935112012-12-03 18:12:45 +00001810 nextToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001811 if (FormatTok->Tok.is(tok::l_paren))
Manuel Klimek9fa8d552013-01-11 19:23:05 +00001812 parseParens();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001813 if (FormatTok->Tok.is(tok::l_brace)) {
Alexander Kornienko3a33f022013-12-12 09:49:52 +00001814 CompoundStatementIndenter Indenter(this, Style, Line->Level);
Daniel Jasper65ee3472013-07-31 23:16:02 +00001815 parseBlock(/*MustBeDeclaration=*/false);
Daniel Jasperf7935112012-12-03 18:12:45 +00001816 addUnwrappedLine();
1817 } else {
1818 addUnwrappedLine();
Daniel Jasper516d7972013-07-25 11:31:57 +00001819 ++Line->Level;
Manuel Klimek6b9eeba2013-01-07 14:56:16 +00001820 parseStructuralElement();
Daniel Jasper516d7972013-07-25 11:31:57 +00001821 --Line->Level;
Daniel Jasperf7935112012-12-03 18:12:45 +00001822 }
1823}
1824
1825void UnwrappedLineParser::parseAccessSpecifier() {
1826 nextToken();
Daniel Jasper84c47a12013-11-23 17:53:41 +00001827 // Understand Qt's slots.
Daniel Jasper53395402015-04-07 15:04:40 +00001828 if (FormatTok->isOneOf(Keywords.kw_slots, Keywords.kw_qslots))
Daniel Jasper84c47a12013-11-23 17:53:41 +00001829 nextToken();
Alexander Kornienko2ca766f2012-12-10 16:34:48 +00001830 // Otherwise, we don't know what it is, and we'd better keep the next token.
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00001831 if (FormatTok->Tok.is(tok::colon))
Alexander Kornienko2ca766f2012-12-10 16:34:48 +00001832 nextToken();
Daniel Jasperf7935112012-12-03 18:12:45 +00001833 addUnwrappedLine();
1834}
1835
Daniel Jasper6f5a1932015-12-29 08:54:23 +00001836bool UnwrappedLineParser::parseEnum() {
Daniel Jasper6be0f552014-11-13 15:56:28 +00001837 // Won't be 'enum' for NS_ENUMs.
1838 if (FormatTok->Tok.is(tok::kw_enum))
Daniel Jasperccb68b42014-11-19 22:38:18 +00001839 nextToken();
Daniel Jasper6be0f552014-11-13 15:56:28 +00001840
Daniel Jasper6f5a1932015-12-29 08:54:23 +00001841 // In TypeScript, "enum" can also be used as property name, e.g. in interface
1842 // declarations. An "enum" keyword followed by a colon would be a syntax
1843 // error and thus assume it is just an identifier.
Daniel Jasper87379302016-02-03 05:33:44 +00001844 if (Style.Language == FormatStyle::LK_JavaScript &&
1845 FormatTok->isOneOf(tok::colon, tok::question))
Daniel Jasper6f5a1932015-12-29 08:54:23 +00001846 return false;
1847
Daniel Jasper2b41a822013-08-20 12:42:50 +00001848 // Eat up enum class ...
Daniel Jasperb05a81d2014-05-09 13:11:16 +00001849 if (FormatTok->Tok.is(tok::kw_class) || FormatTok->Tok.is(tok::kw_struct))
1850 nextToken();
Daniel Jasperb5a0b852015-06-19 08:17:32 +00001851
Daniel Jasper786a5502013-09-06 21:32:35 +00001852 while (FormatTok->Tok.getIdentifierInfo() ||
Daniel Jasperccb68b42014-11-19 22:38:18 +00001853 FormatTok->isOneOf(tok::colon, tok::coloncolon, tok::less,
1854 tok::greater, tok::comma, tok::question)) {
Manuel Klimek2cec0192013-01-21 19:17:52 +00001855 nextToken();
1856 // We can have macros or attributes in between 'enum' and the enum name.
Daniel Jasperccb68b42014-11-19 22:38:18 +00001857 if (FormatTok->is(tok::l_paren))
Alexander Kornienkob7076a22012-12-04 14:46:19 +00001858 parseParens();
Daniel Jasperb5a0b852015-06-19 08:17:32 +00001859 if (FormatTok->is(tok::identifier)) {
Manuel Klimek2cec0192013-01-21 19:17:52 +00001860 nextToken();
Daniel Jasperb5a0b852015-06-19 08:17:32 +00001861 // If there are two identifiers in a row, this is likely an elaborate
1862 // return type. In Java, this can be "implements", etc.
Daniel Jasper1dbc2102017-03-31 13:30:24 +00001863 if (Style.isCpp() && FormatTok->is(tok::identifier))
Daniel Jasper6f5a1932015-12-29 08:54:23 +00001864 return false;
Daniel Jasperb5a0b852015-06-19 08:17:32 +00001865 }
Manuel Klimek2cec0192013-01-21 19:17:52 +00001866 }
Daniel Jasper6be0f552014-11-13 15:56:28 +00001867
1868 // Just a declaration or something is wrong.
Daniel Jasperccb68b42014-11-19 22:38:18 +00001869 if (FormatTok->isNot(tok::l_brace))
Daniel Jasper6f5a1932015-12-29 08:54:23 +00001870 return true;
Daniel Jasper6be0f552014-11-13 15:56:28 +00001871 FormatTok->BlockKind = BK_Block;
1872
1873 if (Style.Language == FormatStyle::LK_Java) {
1874 // Java enums are different.
1875 parseJavaEnumBody();
Daniel Jasper6f5a1932015-12-29 08:54:23 +00001876 return true;
1877 }
1878 if (Style.Language == FormatStyle::LK_Proto) {
Daniel Jasperc6dd2732015-07-16 14:25:43 +00001879 parseBlock(/*MustBeDeclaration=*/true);
Daniel Jasper6f5a1932015-12-29 08:54:23 +00001880 return true;
Manuel Klimek2cec0192013-01-21 19:17:52 +00001881 }
Daniel Jasper6be0f552014-11-13 15:56:28 +00001882
1883 // Parse enum body.
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001884 nextToken();
Daniel Jasper6be0f552014-11-13 15:56:28 +00001885 bool HasError = !parseBracedList(/*ContinueOnSemicolons=*/true);
1886 if (HasError) {
1887 if (FormatTok->is(tok::semi))
1888 nextToken();
1889 addUnwrappedLine();
1890 }
Daniel Jasper6f5a1932015-12-29 08:54:23 +00001891 return true;
Daniel Jasper6be0f552014-11-13 15:56:28 +00001892
Daniel Jasper90cf3802015-06-17 09:44:02 +00001893 // There is no addUnwrappedLine() here so that we fall through to parsing a
1894 // structural element afterwards. Thus, in "enum A {} n, m;",
Manuel Klimek2cec0192013-01-21 19:17:52 +00001895 // "} n, m;" will end up in one unwrapped line.
Daniel Jasper6be0f552014-11-13 15:56:28 +00001896}
1897
1898void UnwrappedLineParser::parseJavaEnumBody() {
1899 // Determine whether the enum is simple, i.e. does not have a semicolon or
1900 // constants with class bodies. Simple enums can be formatted like braced
1901 // lists, contracted to a single line, etc.
1902 unsigned StoredPosition = Tokens->getPosition();
1903 bool IsSimple = true;
1904 FormatToken *Tok = Tokens->getNextToken();
1905 while (Tok) {
1906 if (Tok->is(tok::r_brace))
1907 break;
1908 if (Tok->isOneOf(tok::l_brace, tok::semi)) {
1909 IsSimple = false;
1910 break;
1911 }
1912 // FIXME: This will also mark enums with braces in the arguments to enum
1913 // constants as "not simple". This is probably fine in practice, though.
1914 Tok = Tokens->getNextToken();
1915 }
1916 FormatTok = Tokens->setPosition(StoredPosition);
1917
1918 if (IsSimple) {
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001919 nextToken();
Daniel Jasper6be0f552014-11-13 15:56:28 +00001920 parseBracedList();
Daniel Jasperdf2ff002014-11-02 22:31:39 +00001921 addUnwrappedLine();
Daniel Jasper6be0f552014-11-13 15:56:28 +00001922 return;
1923 }
1924
1925 // Parse the body of a more complex enum.
1926 // First add a line for everything up to the "{".
1927 nextToken();
1928 addUnwrappedLine();
1929 ++Line->Level;
1930
1931 // Parse the enum constants.
1932 while (FormatTok) {
1933 if (FormatTok->is(tok::l_brace)) {
1934 // Parse the constant's class body.
1935 parseBlock(/*MustBeDeclaration=*/true, /*AddLevel=*/true,
1936 /*MunchSemi=*/false);
1937 } else if (FormatTok->is(tok::l_paren)) {
1938 parseParens();
1939 } else if (FormatTok->is(tok::comma)) {
1940 nextToken();
1941 addUnwrappedLine();
1942 } else if (FormatTok->is(tok::semi)) {
1943 nextToken();
1944 addUnwrappedLine();
1945 break;
1946 } else if (FormatTok->is(tok::r_brace)) {
1947 addUnwrappedLine();
1948 break;
1949 } else {
1950 nextToken();
1951 }
1952 }
1953
1954 // Parse the class body after the enum's ";" if any.
1955 parseLevel(/*HasOpeningBrace=*/true);
1956 nextToken();
1957 --Line->Level;
1958 addUnwrappedLine();
Daniel Jasperf7935112012-12-03 18:12:45 +00001959}
1960
Martin Probst1027fb82017-02-07 14:05:30 +00001961void UnwrappedLineParser::parseRecord(bool ParseAsExpr) {
Roman Kashitsyna043ced2014-08-11 12:18:01 +00001962 const FormatToken &InitialToken = *FormatTok;
Manuel Klimek28cacc72013-01-07 18:10:23 +00001963 nextToken();
Daniel Jasper04785d02015-05-06 14:03:02 +00001964
Daniel Jasper04785d02015-05-06 14:03:02 +00001965 // The actual identifier can be a nested name specifier, and in macros
1966 // it is often token-pasted.
1967 while (FormatTok->isOneOf(tok::identifier, tok::coloncolon, tok::hashhash,
1968 tok::kw___attribute, tok::kw___declspec,
1969 tok::kw_alignas) ||
1970 ((Style.Language == FormatStyle::LK_Java ||
1971 Style.Language == FormatStyle::LK_JavaScript) &&
1972 FormatTok->isOneOf(tok::period, tok::comma))) {
Martin Probstcb870c52017-08-01 15:46:10 +00001973 if (Style.Language == FormatStyle::LK_JavaScript &&
1974 FormatTok->isOneOf(Keywords.kw_extends, Keywords.kw_implements)) {
1975 // JavaScript/TypeScript supports inline object types in
1976 // extends/implements positions:
1977 // class Foo implements {bar: number} { }
1978 nextToken();
1979 if (FormatTok->is(tok::l_brace)) {
1980 tryToParseBracedList();
1981 continue;
1982 }
1983 }
Daniel Jasper04785d02015-05-06 14:03:02 +00001984 bool IsNonMacroIdentifier =
1985 FormatTok->is(tok::identifier) &&
1986 FormatTok->TokenText != FormatTok->TokenText.upper();
Manuel Klimeke01bab52013-01-15 13:38:33 +00001987 nextToken();
1988 // We can have macros or attributes in between 'class' and the class name.
Daniel Jasper04785d02015-05-06 14:03:02 +00001989 if (!IsNonMacroIdentifier && FormatTok->Tok.is(tok::l_paren))
Manuel Klimeke01bab52013-01-15 13:38:33 +00001990 parseParens();
Daniel Jasper04785d02015-05-06 14:03:02 +00001991 }
Manuel Klimeke01bab52013-01-15 13:38:33 +00001992
Daniel Jasper04785d02015-05-06 14:03:02 +00001993 // Note that parsing away template declarations here leads to incorrectly
1994 // accepting function declarations as record declarations.
1995 // In general, we cannot solve this problem. Consider:
1996 // class A<int> B() {}
1997 // which can be a function definition or a class definition when B() is a
1998 // macro. If we find enough real-world cases where this is a problem, we
1999 // can parse for the 'template' keyword in the beginning of the statement,
2000 // and thus rule out the record production in case there is no template
2001 // (this would still leave us with an ambiguity between template function
2002 // and class declarations).
Daniel Jasperadba2aa2015-05-18 12:52:00 +00002003 if (FormatTok->isOneOf(tok::colon, tok::less)) {
2004 while (!eof()) {
Daniel Jasper3c883d12015-05-18 14:49:19 +00002005 if (FormatTok->is(tok::l_brace)) {
2006 calculateBraceTypes(/*ExpectClassBody=*/true);
2007 if (!tryToParseBracedList())
2008 break;
2009 }
Daniel Jasper04785d02015-05-06 14:03:02 +00002010 if (FormatTok->Tok.is(tok::semi))
2011 return;
2012 nextToken();
Manuel Klimeke01bab52013-01-15 13:38:33 +00002013 }
2014 }
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00002015 if (FormatTok->Tok.is(tok::l_brace)) {
Martin Probst1027fb82017-02-07 14:05:30 +00002016 if (ParseAsExpr) {
2017 parseChildBlock();
2018 } else {
2019 if (ShouldBreakBeforeBrace(Style, InitialToken))
2020 addUnwrappedLine();
Manuel Klimeka8eb9142013-05-13 12:51:40 +00002021
Martin Probst1027fb82017-02-07 14:05:30 +00002022 parseBlock(/*MustBeDeclaration=*/true, /*AddLevel=*/true,
2023 /*MunchSemi=*/false);
2024 }
Manuel Klimeka8eb9142013-05-13 12:51:40 +00002025 }
Daniel Jasper90cf3802015-06-17 09:44:02 +00002026 // There is no addUnwrappedLine() here so that we fall through to parsing a
2027 // structural element afterwards. Thus, in "class A {} n, m;",
2028 // "} n, m;" will end up in one unwrapped line.
Manuel Klimek28cacc72013-01-07 18:10:23 +00002029}
2030
Nico Weber8696a8d2013-01-09 21:15:03 +00002031void UnwrappedLineParser::parseObjCProtocolList() {
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00002032 assert(FormatTok->Tok.is(tok::less) && "'<' expected.");
Nico Weber8696a8d2013-01-09 21:15:03 +00002033 do
2034 nextToken();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00002035 while (!eof() && FormatTok->Tok.isNot(tok::greater));
Nico Weber8696a8d2013-01-09 21:15:03 +00002036 nextToken(); // Skip '>'.
2037}
2038
2039void UnwrappedLineParser::parseObjCUntilAtEnd() {
2040 do {
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00002041 if (FormatTok->Tok.isObjCAtKeyword(tok::objc_end)) {
Nico Weber8696a8d2013-01-09 21:15:03 +00002042 nextToken();
2043 addUnwrappedLine();
2044 break;
2045 }
Daniel Jaspera15da302013-08-28 08:04:23 +00002046 if (FormatTok->is(tok::l_brace)) {
2047 parseBlock(/*MustBeDeclaration=*/false);
2048 // In ObjC interfaces, nothing should be following the "}".
2049 addUnwrappedLine();
Benjamin Kramere21cb742014-01-08 15:59:42 +00002050 } else if (FormatTok->is(tok::r_brace)) {
2051 // Ignore stray "}". parseStructuralElement doesn't consume them.
2052 nextToken();
2053 addUnwrappedLine();
Daniel Jaspera15da302013-08-28 08:04:23 +00002054 } else {
2055 parseStructuralElement();
2056 }
Nico Weber8696a8d2013-01-09 21:15:03 +00002057 } while (!eof());
2058}
2059
Nico Weber2ce0ac52013-01-09 23:25:37 +00002060void UnwrappedLineParser::parseObjCInterfaceOrImplementation() {
Nico Weber7eecf4b2013-01-09 20:25:35 +00002061 nextToken();
Daniel Jasperd1ae3582013-03-20 12:37:50 +00002062 nextToken(); // interface name
Nico Weber7eecf4b2013-01-09 20:25:35 +00002063
2064 // @interface can be followed by either a base class, or a category.
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00002065 if (FormatTok->Tok.is(tok::colon)) {
Nico Weber7eecf4b2013-01-09 20:25:35 +00002066 nextToken();
Daniel Jasperd1ae3582013-03-20 12:37:50 +00002067 nextToken(); // base class name
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00002068 } else if (FormatTok->Tok.is(tok::l_paren))
Nico Weber7eecf4b2013-01-09 20:25:35 +00002069 // Skip category, if present.
2070 parseParens();
2071
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00002072 if (FormatTok->Tok.is(tok::less))
Nico Weber8696a8d2013-01-09 21:15:03 +00002073 parseObjCProtocolList();
Nico Weber7eecf4b2013-01-09 20:25:35 +00002074
Dinesh Dwivediea3aca82014-05-02 17:01:46 +00002075 if (FormatTok->Tok.is(tok::l_brace)) {
Daniel Jasperc1bc38e2015-09-29 14:57:55 +00002076 if (Style.BraceWrapping.AfterObjCDeclaration)
Dinesh Dwivediea3aca82014-05-02 17:01:46 +00002077 addUnwrappedLine();
Nico Weber9096fc02013-06-26 00:30:14 +00002078 parseBlock(/*MustBeDeclaration=*/true);
Dinesh Dwivediea3aca82014-05-02 17:01:46 +00002079 }
Nico Weber7eecf4b2013-01-09 20:25:35 +00002080
2081 // With instance variables, this puts '}' on its own line. Without instance
2082 // variables, this ends the @interface line.
2083 addUnwrappedLine();
2084
Nico Weber8696a8d2013-01-09 21:15:03 +00002085 parseObjCUntilAtEnd();
2086}
Nico Weber7eecf4b2013-01-09 20:25:35 +00002087
Nico Weber8696a8d2013-01-09 21:15:03 +00002088void UnwrappedLineParser::parseObjCProtocol() {
2089 nextToken();
Daniel Jasperd1ae3582013-03-20 12:37:50 +00002090 nextToken(); // protocol name
Nico Weber8696a8d2013-01-09 21:15:03 +00002091
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00002092 if (FormatTok->Tok.is(tok::less))
Nico Weber8696a8d2013-01-09 21:15:03 +00002093 parseObjCProtocolList();
2094
2095 // Check for protocol declaration.
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00002096 if (FormatTok->Tok.is(tok::semi)) {
Nico Weber8696a8d2013-01-09 21:15:03 +00002097 nextToken();
2098 return addUnwrappedLine();
2099 }
2100
2101 addUnwrappedLine();
2102 parseObjCUntilAtEnd();
Nico Weber7eecf4b2013-01-09 20:25:35 +00002103}
2104
Daniel Jasperfca735c2015-02-19 16:14:18 +00002105void UnwrappedLineParser::parseJavaScriptEs6ImportExport() {
Martin Probst053f1aa2016-04-19 14:55:37 +00002106 bool IsImport = FormatTok->is(Keywords.kw_import);
2107 assert(IsImport || FormatTok->is(tok::kw_export));
Daniel Jasper354aa512015-02-19 16:07:32 +00002108 nextToken();
Daniel Jasperfca735c2015-02-19 16:14:18 +00002109
Daniel Jasperec05fc72015-05-11 09:14:50 +00002110 // Consume the "default" in "export default class/function".
Daniel Jasper668c7bb2015-05-11 09:03:10 +00002111 if (FormatTok->is(tok::kw_default))
2112 nextToken();
Daniel Jasperec05fc72015-05-11 09:14:50 +00002113
Martin Probst5f8445b2016-04-24 22:05:09 +00002114 // Consume "async function", "function" and "default function", so that these
2115 // get parsed as free-standing JS functions, i.e. do not require a trailing
2116 // semicolon.
2117 if (FormatTok->is(Keywords.kw_async))
2118 nextToken();
Daniel Jasper668c7bb2015-05-11 09:03:10 +00002119 if (FormatTok->is(Keywords.kw_function)) {
2120 nextToken();
2121 return;
2122 }
2123
Martin Probst053f1aa2016-04-19 14:55:37 +00002124 // For imports, `export *`, `export {...}`, consume the rest of the line up
2125 // to the terminating `;`. For everything else, just return and continue
2126 // parsing the structural element, i.e. the declaration or expression for
2127 // `export default`.
2128 if (!IsImport && !FormatTok->isOneOf(tok::l_brace, tok::star) &&
2129 !FormatTok->isStringLiteral())
2130 return;
Daniel Jasperfca735c2015-02-19 16:14:18 +00002131
Martin Probstd40bca42017-01-09 08:56:36 +00002132 while (!eof()) {
2133 if (FormatTok->is(tok::semi))
2134 return;
2135 if (Line->Tokens.size() == 0) {
2136 // Common issue: Automatic Semicolon Insertion wrapped the line, so the
2137 // import statement should terminate.
2138 return;
2139 }
Daniel Jasperefc1a832016-01-07 08:53:35 +00002140 if (FormatTok->is(tok::l_brace)) {
2141 FormatTok->BlockKind = BK_Block;
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00002142 nextToken();
Daniel Jasperefc1a832016-01-07 08:53:35 +00002143 parseBracedList();
2144 } else {
2145 nextToken();
2146 }
Daniel Jasper354aa512015-02-19 16:07:32 +00002147 }
2148}
2149
Daniel Jasper3b203a62013-09-05 16:05:56 +00002150LLVM_ATTRIBUTE_UNUSED static void printDebugInfo(const UnwrappedLine &Line,
2151 StringRef Prefix = "") {
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +00002152 llvm::dbgs() << Prefix << "Line(" << Line.Level << ")"
2153 << (Line.InPPDirective ? " MACRO" : "") << ": ";
2154 for (std::list<UnwrappedLineNode>::const_iterator I = Line.Tokens.begin(),
2155 E = Line.Tokens.end();
2156 I != E; ++I) {
Krasimir Georgiev91834222017-01-25 13:58:58 +00002157 llvm::dbgs() << I->Tok->Tok.getName() << "["
2158 << "T=" << I->Tok->Type
2159 << ", OC=" << I->Tok->OriginalColumn << "] ";
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +00002160 }
2161 for (std::list<UnwrappedLineNode>::const_iterator I = Line.Tokens.begin(),
2162 E = Line.Tokens.end();
2163 I != E; ++I) {
2164 const UnwrappedLineNode &Node = *I;
2165 for (SmallVectorImpl<UnwrappedLine>::const_iterator
2166 I = Node.Children.begin(),
2167 E = Node.Children.end();
2168 I != E; ++I) {
2169 printDebugInfo(*I, "\nChild: ");
2170 }
2171 }
2172 llvm::dbgs() << "\n";
2173}
2174
Daniel Jasperf7935112012-12-03 18:12:45 +00002175void UnwrappedLineParser::addUnwrappedLine() {
Daniel Jasperdaffc0d2013-01-16 09:10:19 +00002176 if (Line->Tokens.empty())
Daniel Jasper7c85fde2013-01-08 14:56:18 +00002177 return;
Manuel Klimekab3dc002013-01-16 12:31:12 +00002178 DEBUG({
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +00002179 if (CurrentLines == &Lines)
2180 printDebugInfo(*Line);
Manuel Klimekab3dc002013-01-16 12:31:12 +00002181 });
Benjamin Kramerc7551a42015-05-31 11:18:05 +00002182 CurrentLines->push_back(std::move(*Line));
Daniel Jasperdaffc0d2013-01-16 09:10:19 +00002183 Line->Tokens.clear();
Krasimir Georgiev85c37042017-03-01 16:38:08 +00002184 Line->MatchingOpeningBlockLineIndex = UnwrappedLine::kInvalidIndex;
Manuel Klimekd3b92fa2013-01-18 14:04:34 +00002185 if (CurrentLines == &Lines && !PreprocessorDirectives.empty()) {
Benjamin Kramerc7551a42015-05-31 11:18:05 +00002186 CurrentLines->append(
2187 std::make_move_iterator(PreprocessorDirectives.begin()),
2188 std::make_move_iterator(PreprocessorDirectives.end()));
Manuel Klimekd3b92fa2013-01-18 14:04:34 +00002189 PreprocessorDirectives.clear();
2190 }
Daniel Jasperf7935112012-12-03 18:12:45 +00002191}
2192
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00002193bool UnwrappedLineParser::eof() const { return FormatTok->Tok.is(tok::eof); }
Daniel Jasperf7935112012-12-03 18:12:45 +00002194
Daniel Jasperb05a81d2014-05-09 13:11:16 +00002195bool UnwrappedLineParser::isOnNewLine(const FormatToken &FormatTok) {
Manuel Klimek1fcbe672014-04-11 12:27:47 +00002196 return (Line->InPPDirective || FormatTok.HasUnescapedNewline) &&
2197 FormatTok.NewlinesBefore > 0;
2198}
2199
Krasimir Georgiev91834222017-01-25 13:58:58 +00002200// Checks if \p FormatTok is a line comment that continues the line comment
2201// section on \p Line.
Krasimir Georgievea222a72017-05-22 10:07:56 +00002202static bool continuesLineCommentSection(const FormatToken &FormatTok,
2203 const UnwrappedLine &Line,
2204 llvm::Regex &CommentPragmasRegex) {
Krasimir Georgiev91834222017-01-25 13:58:58 +00002205 if (Line.Tokens.empty())
2206 return false;
Krasimir Georgiev84321612017-01-30 19:18:55 +00002207
Krasimir Georgiev00c5c722017-02-02 15:32:19 +00002208 StringRef IndentContent = FormatTok.TokenText;
2209 if (FormatTok.TokenText.startswith("//") ||
2210 FormatTok.TokenText.startswith("/*"))
2211 IndentContent = FormatTok.TokenText.substr(2);
2212 if (CommentPragmasRegex.match(IndentContent))
2213 return false;
2214
Krasimir Georgiev91834222017-01-25 13:58:58 +00002215 // If Line starts with a line comment, then FormatTok continues the comment
Krasimir Georgiev84321612017-01-30 19:18:55 +00002216 // section if its original column is greater or equal to the original start
Krasimir Georgiev91834222017-01-25 13:58:58 +00002217 // column of the line.
2218 //
Krasimir Georgiev84321612017-01-30 19:18:55 +00002219 // Define the min column token of a line as follows: if a line ends in '{' or
2220 // contains a '{' followed by a line comment, then the min column token is
2221 // that '{'. Otherwise, the min column token of the line is the first token of
2222 // the line.
2223 //
2224 // If Line starts with a token other than a line comment, then FormatTok
2225 // continues the comment section if its original column is greater than the
2226 // original start column of the min column token of the line.
Krasimir Georgiev91834222017-01-25 13:58:58 +00002227 //
2228 // For example, the second line comment continues the first in these cases:
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002229 //
Krasimir Georgiev91834222017-01-25 13:58:58 +00002230 // // first line
2231 // // second line
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002232 //
Krasimir Georgiev91834222017-01-25 13:58:58 +00002233 // and:
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002234 //
Krasimir Georgiev91834222017-01-25 13:58:58 +00002235 // // first line
2236 // // second line
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002237 //
Krasimir Georgiev91834222017-01-25 13:58:58 +00002238 // and:
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002239 //
Krasimir Georgiev91834222017-01-25 13:58:58 +00002240 // int i; // first line
2241 // // second line
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002242 //
Krasimir Georgiev84321612017-01-30 19:18:55 +00002243 // and:
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002244 //
Krasimir Georgiev84321612017-01-30 19:18:55 +00002245 // do { // first line
2246 // // second line
2247 // int i;
2248 // } while (true);
Krasimir Georgiev91834222017-01-25 13:58:58 +00002249 //
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002250 // and:
2251 //
2252 // enum {
2253 // a, // first line
2254 // // second line
2255 // b
2256 // };
2257 //
Krasimir Georgiev91834222017-01-25 13:58:58 +00002258 // The second line comment doesn't continue the first in these cases:
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002259 //
Krasimir Georgiev91834222017-01-25 13:58:58 +00002260 // // first line
2261 // // second line
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002262 //
Krasimir Georgiev91834222017-01-25 13:58:58 +00002263 // and:
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002264 //
Krasimir Georgiev91834222017-01-25 13:58:58 +00002265 // int i; // first line
2266 // // second line
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002267 //
Krasimir Georgiev84321612017-01-30 19:18:55 +00002268 // and:
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002269 //
Krasimir Georgiev84321612017-01-30 19:18:55 +00002270 // do { // first line
2271 // // second line
2272 // int i;
2273 // } while (true);
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002274 //
2275 // and:
2276 //
2277 // enum {
2278 // a, // first line
2279 // // second line
2280 // };
Krasimir Georgiev84321612017-01-30 19:18:55 +00002281 const FormatToken *MinColumnToken = Line.Tokens.front().Tok;
2282
2283 // Scan for '{//'. If found, use the column of '{' as a min column for line
2284 // comment section continuation.
2285 const FormatToken *PreviousToken = nullptr;
Krasimir Georgievd86c25d2017-03-10 13:09:29 +00002286 for (const UnwrappedLineNode &Node : Line.Tokens) {
Krasimir Georgiev84321612017-01-30 19:18:55 +00002287 if (PreviousToken && PreviousToken->is(tok::l_brace) &&
2288 isLineComment(*Node.Tok)) {
2289 MinColumnToken = PreviousToken;
2290 break;
2291 }
2292 PreviousToken = Node.Tok;
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002293
2294 // Grab the last newline preceding a token in this unwrapped line.
2295 if (Node.Tok->NewlinesBefore > 0) {
2296 MinColumnToken = Node.Tok;
2297 }
Krasimir Georgiev84321612017-01-30 19:18:55 +00002298 }
2299 if (PreviousToken && PreviousToken->is(tok::l_brace)) {
2300 MinColumnToken = PreviousToken;
2301 }
2302
Krasimir Georgievea222a72017-05-22 10:07:56 +00002303 return continuesLineComment(FormatTok, /*Previous=*/Line.Tokens.back().Tok,
2304 MinColumnToken);
Krasimir Georgiev91834222017-01-25 13:58:58 +00002305}
2306
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002307void UnwrappedLineParser::flushComments(bool NewlineBeforeNext) {
2308 bool JustComments = Line->Tokens.empty();
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00002309 for (SmallVectorImpl<FormatToken *>::const_iterator
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002310 I = CommentsBeforeNextToken.begin(),
2311 E = CommentsBeforeNextToken.end();
2312 I != E; ++I) {
Krasimir Georgiev91834222017-01-25 13:58:58 +00002313 // Line comments that belong to the same line comment section are put on the
2314 // same line since later we might want to reflow content between them.
Krasimir Georgiev753625b2017-01-31 13:32:38 +00002315 // Additional fine-grained breaking of line comment sections is controlled
2316 // by the class BreakableLineCommentSection in case it is desirable to keep
2317 // several line comment sections in the same unwrapped line.
2318 //
2319 // FIXME: Consider putting separate line comment sections as children to the
2320 // unwrapped line instead.
Krasimir Georgiev00c5c722017-02-02 15:32:19 +00002321 (*I)->ContinuesLineCommentSection =
Krasimir Georgievea222a72017-05-22 10:07:56 +00002322 continuesLineCommentSection(**I, *Line, CommentPragmasRegex);
Krasimir Georgievb6ccd382017-02-02 14:36:50 +00002323 if (isOnNewLine(**I) && JustComments && !(*I)->ContinuesLineCommentSection)
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002324 addUnwrappedLine();
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002325 pushToken(*I);
2326 }
Daniel Jaspere60cba12015-05-13 11:35:53 +00002327 if (NewlineBeforeNext && JustComments)
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002328 addUnwrappedLine();
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002329 CommentsBeforeNextToken.clear();
2330}
2331
Krasimir Georgiev3e051052017-07-24 14:51:59 +00002332void UnwrappedLineParser::nextToken(int LevelDifference) {
Daniel Jasperf7935112012-12-03 18:12:45 +00002333 if (eof())
2334 return;
Manuel Klimek1fcbe672014-04-11 12:27:47 +00002335 flushComments(isOnNewLine(*FormatTok));
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002336 pushToken(FormatTok);
Daniel Jasper1dcbbcfc2016-03-14 19:21:36 +00002337 if (Style.Language != FormatStyle::LK_JavaScript)
Krasimir Georgiev3e051052017-07-24 14:51:59 +00002338 readToken(LevelDifference);
Daniel Jasper1dcbbcfc2016-03-14 19:21:36 +00002339 else
2340 readTokenWithJavaScriptASI();
Manuel Klimek1abf7892013-01-04 23:34:14 +00002341}
2342
Daniel Jasperb9a49902016-01-09 15:56:28 +00002343const FormatToken *UnwrappedLineParser::getPreviousToken() {
2344 // FIXME: This is a dirty way to access the previous token. Find a better
2345 // solution.
2346 if (!Line || Line->Tokens.empty())
2347 return nullptr;
2348 return Line->Tokens.back().Tok;
2349}
2350
Krasimir Georgievf62f9582017-02-08 10:30:44 +00002351void UnwrappedLineParser::distributeComments(
2352 const SmallVectorImpl<FormatToken *> &Comments,
2353 const FormatToken *NextTok) {
2354 // Whether or not a line comment token continues a line is controlled by
Krasimir Georgievea222a72017-05-22 10:07:56 +00002355 // the method continuesLineCommentSection, with the following caveat:
Krasimir Georgievf62f9582017-02-08 10:30:44 +00002356 //
2357 // Define a trail of Comments to be a nonempty proper postfix of Comments such
2358 // that each comment line from the trail is aligned with the next token, if
2359 // the next token exists. If a trail exists, the beginning of the maximal
2360 // trail is marked as a start of a new comment section.
2361 //
2362 // For example in this code:
2363 //
2364 // int a; // line about a
2365 // // line 1 about b
2366 // // line 2 about b
2367 // int b;
2368 //
2369 // the two lines about b form a maximal trail, so there are two sections, the
2370 // first one consisting of the single comment "// line about a" and the
2371 // second one consisting of the next two comments.
2372 if (Comments.empty())
2373 return;
2374 bool ShouldPushCommentsInCurrentLine = true;
2375 bool HasTrailAlignedWithNextToken = false;
2376 unsigned StartOfTrailAlignedWithNextToken = 0;
2377 if (NextTok) {
2378 // We are skipping the first element intentionally.
2379 for (unsigned i = Comments.size() - 1; i > 0; --i) {
2380 if (Comments[i]->OriginalColumn == NextTok->OriginalColumn) {
2381 HasTrailAlignedWithNextToken = true;
2382 StartOfTrailAlignedWithNextToken = i;
2383 }
2384 }
2385 }
2386 for (unsigned i = 0, e = Comments.size(); i < e; ++i) {
2387 FormatToken *FormatTok = Comments[i];
2388 if (HasTrailAlignedWithNextToken &&
2389 i == StartOfTrailAlignedWithNextToken) {
2390 FormatTok->ContinuesLineCommentSection = false;
2391 } else {
2392 FormatTok->ContinuesLineCommentSection =
Krasimir Georgievea222a72017-05-22 10:07:56 +00002393 continuesLineCommentSection(*FormatTok, *Line, CommentPragmasRegex);
Krasimir Georgievf62f9582017-02-08 10:30:44 +00002394 }
2395 if (!FormatTok->ContinuesLineCommentSection &&
2396 (isOnNewLine(*FormatTok) || FormatTok->IsFirst)) {
2397 ShouldPushCommentsInCurrentLine = false;
2398 }
2399 if (ShouldPushCommentsInCurrentLine) {
2400 pushToken(FormatTok);
2401 } else {
2402 CommentsBeforeNextToken.push_back(FormatTok);
2403 }
2404 }
2405}
2406
Krasimir Georgiev3e051052017-07-24 14:51:59 +00002407void UnwrappedLineParser::readToken(int LevelDifference) {
Krasimir Georgievf62f9582017-02-08 10:30:44 +00002408 SmallVector<FormatToken *, 1> Comments;
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002409 do {
2410 FormatTok = Tokens->getNextToken();
Alexander Kornienkoc2ee9cf2014-03-13 13:59:48 +00002411 assert(FormatTok);
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00002412 while (!Line->InPPDirective && FormatTok->Tok.is(tok::hash) &&
2413 (FormatTok->HasUnescapedNewline || FormatTok->IsFirst)) {
Krasimir Georgievf62f9582017-02-08 10:30:44 +00002414 distributeComments(Comments, FormatTok);
2415 Comments.clear();
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002416 // If there is an unfinished unwrapped line, we flush the preprocessor
2417 // directives only after that unwrapped line was finished later.
Daniel Jasper29d39d52015-02-08 09:34:49 +00002418 bool SwitchToPreprocessorLines = !Line->Tokens.empty();
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002419 ScopedLineState BlockState(*this, SwitchToPreprocessorLines);
Krasimir Georgiev3e051052017-07-24 14:51:59 +00002420 assert((LevelDifference >= 0 ||
2421 static_cast<unsigned>(-LevelDifference) <= Line->Level) &&
2422 "LevelDifference makes Line->Level negative");
2423 Line->Level += LevelDifference;
Alexander Kornienkob1be9d62013-04-03 12:38:53 +00002424 // Comments stored before the preprocessor directive need to be output
2425 // before the preprocessor directive, at the same level as the
2426 // preprocessor directive, as we consider them to apply to the directive.
Manuel Klimek1fcbe672014-04-11 12:27:47 +00002427 flushComments(isOnNewLine(*FormatTok));
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002428 parsePPDirective();
2429 }
Manuel Klimek68b03042014-04-14 09:14:11 +00002430 while (FormatTok->Type == TT_ConflictStart ||
2431 FormatTok->Type == TT_ConflictEnd ||
2432 FormatTok->Type == TT_ConflictAlternative) {
2433 if (FormatTok->Type == TT_ConflictStart) {
2434 conditionalCompilationStart(/*Unreachable=*/false);
2435 } else if (FormatTok->Type == TT_ConflictAlternative) {
2436 conditionalCompilationAlternative();
Daniel Jasperb05a81d2014-05-09 13:11:16 +00002437 } else if (FormatTok->Type == TT_ConflictEnd) {
Manuel Klimek68b03042014-04-14 09:14:11 +00002438 conditionalCompilationEnd();
2439 }
2440 FormatTok = Tokens->getNextToken();
2441 FormatTok->MustBreakBefore = true;
2442 }
Alexander Kornienkof2e02122013-05-24 18:24:24 +00002443
Francois Ferranda98a95c2017-07-28 07:56:14 +00002444 if (!PPStack.empty() && (PPStack.back().Kind == PP_Unreachable) &&
Alexander Kornienkof2e02122013-05-24 18:24:24 +00002445 !Line->InPPDirective) {
2446 continue;
2447 }
2448
Krasimir Georgievf62f9582017-02-08 10:30:44 +00002449 if (!FormatTok->Tok.is(tok::comment)) {
2450 distributeComments(Comments, FormatTok);
2451 Comments.clear();
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002452 return;
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002453 }
Krasimir Georgievf62f9582017-02-08 10:30:44 +00002454
2455 Comments.push_back(FormatTok);
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002456 } while (!eof());
Krasimir Georgievf62f9582017-02-08 10:30:44 +00002457
2458 distributeComments(Comments, nullptr);
2459 Comments.clear();
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002460}
2461
Manuel Klimek15dfe7a2013-05-28 11:55:06 +00002462void UnwrappedLineParser::pushToken(FormatToken *Tok) {
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +00002463 Line->Tokens.push_back(UnwrappedLineNode(Tok));
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002464 if (MustBreakBeforeNextToken) {
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +00002465 Line->Tokens.back().Tok->MustBreakBefore = true;
Manuel Klimekf92f7bc2013-01-22 16:31:55 +00002466 MustBreakBeforeNextToken = false;
Manuel Klimek1abf7892013-01-04 23:34:14 +00002467 }
Daniel Jasperf7935112012-12-03 18:12:45 +00002468}
2469
Daniel Jasper8d1832e2013-01-07 13:26:07 +00002470} // end namespace format
2471} // end namespace clang