1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902 |
- //===--- TokenAnnotator.cpp - Format C++ code -----------------------------===//
- //
- // The LLVM Compiler Infrastructure
- //
- // This file is distributed under the University of Illinois Open Source
- // License. See LICENSE.TXT for details.
- //
- //===----------------------------------------------------------------------===//
- ///
- /// \file
- /// \brief This file implements a token annotator, i.e. creates
- /// \c AnnotatedTokens out of \c FormatTokens with required extra information.
- ///
- //===----------------------------------------------------------------------===//
- #include "TokenAnnotator.h"
- #include "clang/Basic/SourceManager.h"
- #include "llvm/ADT/SmallPtrSet.h"
- #include "llvm/Support/Debug.h"
- #define DEBUG_TYPE "format-token-annotator"
- namespace clang {
- namespace format {
- namespace {
- /// \brief A parser that gathers additional information about tokens.
- ///
- /// The \c TokenAnnotator tries to match parenthesis and square brakets and
- /// store a parenthesis levels. It also tries to resolve matching "<" and ">"
- /// into template parameter lists.
- class AnnotatingParser {
- public:
- AnnotatingParser(const FormatStyle &Style, AnnotatedLine &Line,
- const AdditionalKeywords &Keywords)
- : Style(Style), Line(Line), CurrentToken(Line.First), AutoFound(false),
- Keywords(Keywords) {
- Contexts.push_back(Context(tok::unknown, 1, /*IsExpression=*/false));
- resetTokenMetadata(CurrentToken);
- }
- private:
- bool parseAngle() {
- if (!CurrentToken || !CurrentToken->Previous)
- return false;
- if (NonTemplateLess.count(CurrentToken->Previous))
- return false;
- const FormatToken &Previous = *CurrentToken->Previous; // The '<'.
- if (Previous.Previous) {
- if (Previous.Previous->Tok.isLiteral())
- return false;
- if (Previous.Previous->is(tok::r_paren) && Contexts.size() > 1 &&
- (!Previous.Previous->MatchingParen ||
- !Previous.Previous->MatchingParen->is(TT_OverloadedOperatorLParen)))
- return false;
- }
- FormatToken *Left = CurrentToken->Previous;
- Left->ParentBracket = Contexts.back().ContextKind;
- ScopedContextCreator ContextCreator(*this, tok::less, 12);
- // If this angle is in the context of an expression, we need to be more
- // hesitant to detect it as opening template parameters.
- bool InExprContext = Contexts.back().IsExpression;
- Contexts.back().IsExpression = false;
- // If there's a template keyword before the opening angle bracket, this is a
- // template parameter, not an argument.
- Contexts.back().InTemplateArgument =
- Left->Previous && Left->Previous->Tok.isNot(tok::kw_template);
- if (Style.Language == FormatStyle::LK_Java &&
- CurrentToken->is(tok::question))
- next();
- while (CurrentToken) {
- if (CurrentToken->is(tok::greater)) {
- Left->MatchingParen = CurrentToken;
- CurrentToken->MatchingParen = Left;
- CurrentToken->Type = TT_TemplateCloser;
- next();
- return true;
- }
- if (CurrentToken->is(tok::question) &&
- Style.Language == FormatStyle::LK_Java) {
- next();
- continue;
- }
- if (CurrentToken->isOneOf(tok::r_paren, tok::r_square, tok::r_brace) ||
- (CurrentToken->isOneOf(tok::colon, tok::question) && InExprContext &&
- Style.Language != FormatStyle::LK_Proto &&
- Style.Language != FormatStyle::LK_TextProto))
- return false;
- // If a && or || is found and interpreted as a binary operator, this set
- // of angles is likely part of something like "a < b && c > d". If the
- // angles are inside an expression, the ||/&& might also be a binary
- // operator that was misinterpreted because we are parsing template
- // parameters.
- // FIXME: This is getting out of hand, write a decent parser.
- if (CurrentToken->Previous->isOneOf(tok::pipepipe, tok::ampamp) &&
- CurrentToken->Previous->is(TT_BinaryOperator) &&
- Contexts[Contexts.size() - 2].IsExpression &&
- !Line.startsWith(tok::kw_template))
- return false;
- updateParameterCount(Left, CurrentToken);
- if (Style.Language == FormatStyle::LK_Proto) {
- if (FormatToken *Previous = CurrentToken->getPreviousNonComment()) {
- if (CurrentToken->is(tok::colon) ||
- (CurrentToken->isOneOf(tok::l_brace, tok::less) &&
- Previous->isNot(tok::colon)))
- Previous->Type = TT_SelectorName;
- }
- }
- if (!consumeToken())
- return false;
- }
- return false;
- }
- bool parseParens(bool LookForDecls = false) {
- if (!CurrentToken)
- return false;
- FormatToken *Left = CurrentToken->Previous;
- Left->ParentBracket = Contexts.back().ContextKind;
- ScopedContextCreator ContextCreator(*this, tok::l_paren, 1);
- // FIXME: This is a bit of a hack. Do better.
- Contexts.back().ColonIsForRangeExpr =
- Contexts.size() == 2 && Contexts[0].ColonIsForRangeExpr;
- bool StartsObjCMethodExpr = false;
- if (CurrentToken->is(tok::caret)) {
- // (^ can start a block type.
- Left->Type = TT_ObjCBlockLParen;
- } else if (FormatToken *MaybeSel = Left->Previous) {
- // @selector( starts a selector.
- if (MaybeSel->isObjCAtKeyword(tok::objc_selector) && MaybeSel->Previous &&
- MaybeSel->Previous->is(tok::at)) {
- StartsObjCMethodExpr = true;
- }
- }
- if (Left->is(TT_OverloadedOperatorLParen)) {
- Contexts.back().IsExpression = false;
- } else if (Style.Language == FormatStyle::LK_JavaScript &&
- (Line.startsWith(Keywords.kw_type, tok::identifier) ||
- Line.startsWith(tok::kw_export, Keywords.kw_type,
- tok::identifier))) {
- // type X = (...);
- // export type X = (...);
- Contexts.back().IsExpression = false;
- } else if (Left->Previous &&
- (Left->Previous->isOneOf(tok::kw_static_assert, tok::kw_decltype,
- tok::kw_if, tok::kw_while, tok::l_paren,
- tok::comma) ||
- Left->Previous->endsSequence(tok::kw_constexpr, tok::kw_if) ||
- Left->Previous->is(TT_BinaryOperator))) {
- // static_assert, if and while usually contain expressions.
- Contexts.back().IsExpression = true;
- } else if (Style.Language == FormatStyle::LK_JavaScript && Left->Previous &&
- (Left->Previous->is(Keywords.kw_function) ||
- (Left->Previous->endsSequence(tok::identifier,
- Keywords.kw_function)))) {
- // function(...) or function f(...)
- Contexts.back().IsExpression = false;
- } else if (Style.Language == FormatStyle::LK_JavaScript && Left->Previous &&
- Left->Previous->is(TT_JsTypeColon)) {
- // let x: (SomeType);
- Contexts.back().IsExpression = false;
- } else if (Left->Previous && Left->Previous->is(tok::r_square) &&
- Left->Previous->MatchingParen &&
- Left->Previous->MatchingParen->is(TT_LambdaLSquare)) {
- // This is a parameter list of a lambda expression.
- Contexts.back().IsExpression = false;
- } else if (Line.InPPDirective &&
- (!Left->Previous || !Left->Previous->is(tok::identifier))) {
- Contexts.back().IsExpression = true;
- } else if (Contexts[Contexts.size() - 2].CaretFound) {
- // This is the parameter list of an ObjC block.
- Contexts.back().IsExpression = false;
- } else if (Left->Previous && Left->Previous->is(tok::kw___attribute)) {
- Left->Type = TT_AttributeParen;
- } else if (Left->Previous && Left->Previous->is(TT_ForEachMacro)) {
- // The first argument to a foreach macro is a declaration.
- Contexts.back().IsForEachMacro = true;
- Contexts.back().IsExpression = false;
- } else if (Left->Previous && Left->Previous->MatchingParen &&
- Left->Previous->MatchingParen->is(TT_ObjCBlockLParen)) {
- Contexts.back().IsExpression = false;
- } else if (!Line.MustBeDeclaration && !Line.InPPDirective) {
- bool IsForOrCatch =
- Left->Previous && Left->Previous->isOneOf(tok::kw_for, tok::kw_catch);
- Contexts.back().IsExpression = !IsForOrCatch;
- }
- if (StartsObjCMethodExpr) {
- Contexts.back().ColonIsObjCMethodExpr = true;
- Left->Type = TT_ObjCMethodExpr;
- }
- bool MightBeFunctionType = !Contexts[Contexts.size() - 2].IsExpression;
- bool ProbablyFunctionType = CurrentToken->isOneOf(tok::star, tok::amp);
- bool HasMultipleLines = false;
- bool HasMultipleParametersOnALine = false;
- bool MightBeObjCForRangeLoop =
- Left->Previous && Left->Previous->is(tok::kw_for);
- while (CurrentToken) {
- // LookForDecls is set when "if (" has been seen. Check for
- // 'identifier' '*' 'identifier' followed by not '=' -- this
- // '*' has to be a binary operator but determineStarAmpUsage() will
- // categorize it as an unary operator, so set the right type here.
- if (LookForDecls && CurrentToken->Next) {
- FormatToken *Prev = CurrentToken->getPreviousNonComment();
- if (Prev) {
- FormatToken *PrevPrev = Prev->getPreviousNonComment();
- FormatToken *Next = CurrentToken->Next;
- if (PrevPrev && PrevPrev->is(tok::identifier) &&
- Prev->isOneOf(tok::star, tok::amp, tok::ampamp) &&
- CurrentToken->is(tok::identifier) && Next->isNot(tok::equal)) {
- Prev->Type = TT_BinaryOperator;
- LookForDecls = false;
- }
- }
- }
- if (CurrentToken->Previous->is(TT_PointerOrReference) &&
- CurrentToken->Previous->Previous->isOneOf(tok::l_paren,
- tok::coloncolon))
- ProbablyFunctionType = true;
- if (CurrentToken->is(tok::comma))
- MightBeFunctionType = false;
- if (CurrentToken->Previous->is(TT_BinaryOperator))
- Contexts.back().IsExpression = true;
- if (CurrentToken->is(tok::r_paren)) {
- if (MightBeFunctionType && ProbablyFunctionType && CurrentToken->Next &&
- (CurrentToken->Next->is(tok::l_paren) ||
- (CurrentToken->Next->is(tok::l_square) && Line.MustBeDeclaration)))
- Left->Type = TT_FunctionTypeLParen;
- Left->MatchingParen = CurrentToken;
- CurrentToken->MatchingParen = Left;
- if (CurrentToken->Next && CurrentToken->Next->is(tok::l_brace) &&
- Left->Previous && Left->Previous->is(tok::l_paren)) {
- // Detect the case where macros are used to generate lambdas or
- // function bodies, e.g.:
- // auto my_lambda = MARCO((Type *type, int i) { .. body .. });
- for (FormatToken *Tok = Left; Tok != CurrentToken; Tok = Tok->Next) {
- if (Tok->is(TT_BinaryOperator) &&
- Tok->isOneOf(tok::star, tok::amp, tok::ampamp))
- Tok->Type = TT_PointerOrReference;
- }
- }
- if (StartsObjCMethodExpr) {
- CurrentToken->Type = TT_ObjCMethodExpr;
- if (Contexts.back().FirstObjCSelectorName) {
- Contexts.back().FirstObjCSelectorName->LongestObjCSelectorName =
- Contexts.back().LongestObjCSelectorName;
- }
- }
- if (Left->is(TT_AttributeParen))
- CurrentToken->Type = TT_AttributeParen;
- if (Left->Previous && Left->Previous->is(TT_JavaAnnotation))
- CurrentToken->Type = TT_JavaAnnotation;
- if (Left->Previous && Left->Previous->is(TT_LeadingJavaAnnotation))
- CurrentToken->Type = TT_LeadingJavaAnnotation;
- if (!HasMultipleLines)
- Left->PackingKind = PPK_Inconclusive;
- else if (HasMultipleParametersOnALine)
- Left->PackingKind = PPK_BinPacked;
- else
- Left->PackingKind = PPK_OnePerLine;
- next();
- return true;
- }
- if (CurrentToken->isOneOf(tok::r_square, tok::r_brace))
- return false;
- if (CurrentToken->is(tok::l_brace))
- Left->Type = TT_Unknown; // Not TT_ObjCBlockLParen
- if (CurrentToken->is(tok::comma) && CurrentToken->Next &&
- !CurrentToken->Next->HasUnescapedNewline &&
- !CurrentToken->Next->isTrailingComment())
- HasMultipleParametersOnALine = true;
- if ((CurrentToken->Previous->isOneOf(tok::kw_const, tok::kw_auto) ||
- CurrentToken->Previous->isSimpleTypeSpecifier()) &&
- !CurrentToken->is(tok::l_brace))
- Contexts.back().IsExpression = false;
- if (CurrentToken->isOneOf(tok::semi, tok::colon))
- MightBeObjCForRangeLoop = false;
- if (MightBeObjCForRangeLoop && CurrentToken->is(Keywords.kw_in))
- CurrentToken->Type = TT_ObjCForIn;
- // When we discover a 'new', we set CanBeExpression to 'false' in order to
- // parse the type correctly. Reset that after a comma.
- if (CurrentToken->is(tok::comma))
- Contexts.back().CanBeExpression = true;
- FormatToken *Tok = CurrentToken;
- if (!consumeToken())
- return false;
- updateParameterCount(Left, Tok);
- if (CurrentToken && CurrentToken->HasUnescapedNewline)
- HasMultipleLines = true;
- }
- return false;
- }
- bool parseSquare() {
- if (!CurrentToken)
- return false;
- // A '[' could be an index subscript (after an identifier or after
- // ')' or ']'), it could be the start of an Objective-C method
- // expression, or it could the start of an Objective-C array literal.
- FormatToken *Left = CurrentToken->Previous;
- Left->ParentBracket = Contexts.back().ContextKind;
- FormatToken *Parent = Left->getPreviousNonComment();
- // Cases where '>' is followed by '['.
- // In C++, this can happen either in array of templates (foo<int>[10])
- // or when array is a nested template type (unique_ptr<type1<type2>[]>).
- bool CppArrayTemplates =
- Style.isCpp() && Parent && Parent->is(TT_TemplateCloser) &&
- (Contexts.back().CanBeExpression || Contexts.back().IsExpression ||
- Contexts.back().InTemplateArgument);
- bool StartsObjCMethodExpr =
- !CppArrayTemplates && Style.isCpp() &&
- Contexts.back().CanBeExpression && Left->isNot(TT_LambdaLSquare) &&
- CurrentToken->isNot(tok::l_brace) &&
- (!Parent ||
- Parent->isOneOf(tok::colon, tok::l_square, tok::l_paren,
- tok::kw_return, tok::kw_throw) ||
- Parent->isUnaryOperator() ||
- Parent->isOneOf(TT_ObjCForIn, TT_CastRParen) ||
- getBinOpPrecedence(Parent->Tok.getKind(), true, true) > prec::Unknown);
- bool ColonFound = false;
- unsigned BindingIncrease = 1;
- if (Left->isCppStructuredBinding(Style)) {
- Left->Type = TT_StructuredBindingLSquare;
- } else if (Left->is(TT_Unknown)) {
- if (StartsObjCMethodExpr) {
- Left->Type = TT_ObjCMethodExpr;
- } else if (Style.Language == FormatStyle::LK_JavaScript && Parent &&
- Contexts.back().ContextKind == tok::l_brace &&
- Parent->isOneOf(tok::l_brace, tok::comma)) {
- Left->Type = TT_JsComputedPropertyName;
- } else if (Style.isCpp() && Contexts.back().ContextKind == tok::l_brace &&
- Parent && Parent->isOneOf(tok::l_brace, tok::comma)) {
- Left->Type = TT_DesignatedInitializerLSquare;
- } else if (CurrentToken->is(tok::r_square) && Parent &&
- Parent->is(TT_TemplateCloser)) {
- Left->Type = TT_ArraySubscriptLSquare;
- } else if (Style.Language == FormatStyle::LK_Proto ||
- (!CppArrayTemplates && Parent &&
- Parent->isOneOf(TT_BinaryOperator, TT_TemplateCloser, tok::at,
- tok::comma, tok::l_paren, tok::l_square,
- tok::question, tok::colon, tok::kw_return,
- // Should only be relevant to JavaScript:
- tok::kw_default))) {
- Left->Type = TT_ArrayInitializerLSquare;
- } else {
- BindingIncrease = 10;
- Left->Type = TT_ArraySubscriptLSquare;
- }
- }
- ScopedContextCreator ContextCreator(*this, tok::l_square, BindingIncrease);
- Contexts.back().IsExpression = true;
- if (Style.Language == FormatStyle::LK_JavaScript && Parent &&
- Parent->is(TT_JsTypeColon))
- Contexts.back().IsExpression = false;
- Contexts.back().ColonIsObjCMethodExpr = StartsObjCMethodExpr;
- while (CurrentToken) {
- if (CurrentToken->is(tok::r_square)) {
- if (CurrentToken->Next && CurrentToken->Next->is(tok::l_paren) &&
- Left->is(TT_ObjCMethodExpr)) {
- // An ObjC method call is rarely followed by an open parenthesis.
- // FIXME: Do we incorrectly label ":" with this?
- StartsObjCMethodExpr = false;
- Left->Type = TT_Unknown;
- }
- if (StartsObjCMethodExpr && CurrentToken->Previous != Left) {
- CurrentToken->Type = TT_ObjCMethodExpr;
- // determineStarAmpUsage() thinks that '*' '[' is allocating an
- // array of pointers, but if '[' starts a selector then '*' is a
- // binary operator.
- if (Parent && Parent->is(TT_PointerOrReference))
- Parent->Type = TT_BinaryOperator;
- }
- Left->MatchingParen = CurrentToken;
- CurrentToken->MatchingParen = Left;
- if (Contexts.back().FirstObjCSelectorName) {
- Contexts.back().FirstObjCSelectorName->LongestObjCSelectorName =
- Contexts.back().LongestObjCSelectorName;
- if (Left->BlockParameterCount > 1)
- Contexts.back().FirstObjCSelectorName->LongestObjCSelectorName = 0;
- }
- next();
- return true;
- }
- if (CurrentToken->isOneOf(tok::r_paren, tok::r_brace))
- return false;
- if (CurrentToken->is(tok::colon)) {
- if (Left->isOneOf(TT_ArraySubscriptLSquare,
- TT_DesignatedInitializerLSquare)) {
- Left->Type = TT_ObjCMethodExpr;
- StartsObjCMethodExpr = true;
- Contexts.back().ColonIsObjCMethodExpr = true;
- if (Parent && Parent->is(tok::r_paren))
- Parent->Type = TT_CastRParen;
- }
- ColonFound = true;
- }
- if (CurrentToken->is(tok::comma) && Left->is(TT_ObjCMethodExpr) &&
- !ColonFound)
- Left->Type = TT_ArrayInitializerLSquare;
- FormatToken *Tok = CurrentToken;
- if (!consumeToken())
- return false;
- updateParameterCount(Left, Tok);
- }
- return false;
- }
- bool parseBrace() {
- if (CurrentToken) {
- FormatToken *Left = CurrentToken->Previous;
- Left->ParentBracket = Contexts.back().ContextKind;
- if (Contexts.back().CaretFound)
- Left->Type = TT_ObjCBlockLBrace;
- Contexts.back().CaretFound = false;
- ScopedContextCreator ContextCreator(*this, tok::l_brace, 1);
- Contexts.back().ColonIsDictLiteral = true;
- if (Left->BlockKind == BK_BracedInit)
- Contexts.back().IsExpression = true;
- if (Style.Language == FormatStyle::LK_JavaScript && Left->Previous &&
- Left->Previous->is(TT_JsTypeColon))
- Contexts.back().IsExpression = false;
- while (CurrentToken) {
- if (CurrentToken->is(tok::r_brace)) {
- Left->MatchingParen = CurrentToken;
- CurrentToken->MatchingParen = Left;
- next();
- return true;
- }
- if (CurrentToken->isOneOf(tok::r_paren, tok::r_square))
- return false;
- updateParameterCount(Left, CurrentToken);
- if (CurrentToken->isOneOf(tok::colon, tok::l_brace, tok::less)) {
- FormatToken *Previous = CurrentToken->getPreviousNonComment();
- if (Previous->is(TT_JsTypeOptionalQuestion))
- Previous = Previous->getPreviousNonComment();
- if (((CurrentToken->is(tok::colon) &&
- (!Contexts.back().ColonIsDictLiteral || !Style.isCpp())) ||
- Style.Language == FormatStyle::LK_Proto ||
- Style.Language == FormatStyle::LK_TextProto) &&
- (Previous->Tok.getIdentifierInfo() ||
- Previous->is(tok::string_literal)))
- Previous->Type = TT_SelectorName;
- if (CurrentToken->is(tok::colon) ||
- Style.Language == FormatStyle::LK_JavaScript)
- Left->Type = TT_DictLiteral;
- }
- if (CurrentToken->is(tok::comma) &&
- Style.Language == FormatStyle::LK_JavaScript)
- Left->Type = TT_DictLiteral;
- if (!consumeToken())
- return false;
- }
- }
- return true;
- }
- void updateParameterCount(FormatToken *Left, FormatToken *Current) {
- if (Current->is(tok::l_brace) && Current->BlockKind == BK_Block)
- ++Left->BlockParameterCount;
- if (Current->is(tok::comma)) {
- ++Left->ParameterCount;
- if (!Left->Role)
- Left->Role.reset(new CommaSeparatedList(Style));
- Left->Role->CommaFound(Current);
- } else if (Left->ParameterCount == 0 && Current->isNot(tok::comment)) {
- Left->ParameterCount = 1;
- }
- }
- bool parseConditional() {
- while (CurrentToken) {
- if (CurrentToken->is(tok::colon)) {
- CurrentToken->Type = TT_ConditionalExpr;
- next();
- return true;
- }
- if (!consumeToken())
- return false;
- }
- return false;
- }
- bool parseTemplateDeclaration() {
- if (CurrentToken && CurrentToken->is(tok::less)) {
- CurrentToken->Type = TT_TemplateOpener;
- next();
- if (!parseAngle())
- return false;
- if (CurrentToken)
- CurrentToken->Previous->ClosesTemplateDeclaration = true;
- return true;
- }
- return false;
- }
- bool consumeToken() {
- FormatToken *Tok = CurrentToken;
- next();
- switch (Tok->Tok.getKind()) {
- case tok::plus:
- case tok::minus:
- if (!Tok->Previous && Line.MustBeDeclaration)
- Tok->Type = TT_ObjCMethodSpecifier;
- break;
- case tok::colon:
- if (!Tok->Previous)
- return false;
- // Colons from ?: are handled in parseConditional().
- if (Style.Language == FormatStyle::LK_JavaScript) {
- if (Contexts.back().ColonIsForRangeExpr || // colon in for loop
- (Contexts.size() == 1 && // switch/case labels
- !Line.First->isOneOf(tok::kw_enum, tok::kw_case)) ||
- Contexts.back().ContextKind == tok::l_paren || // function params
- Contexts.back().ContextKind == tok::l_square || // array type
- (!Contexts.back().IsExpression &&
- Contexts.back().ContextKind == tok::l_brace) || // object type
- (Contexts.size() == 1 &&
- Line.MustBeDeclaration)) { // method/property declaration
- Contexts.back().IsExpression = false;
- Tok->Type = TT_JsTypeColon;
- break;
- }
- }
- if (Contexts.back().ColonIsDictLiteral ||
- Style.Language == FormatStyle::LK_Proto ||
- Style.Language == FormatStyle::LK_TextProto) {
- Tok->Type = TT_DictLiteral;
- if (Style.Language == FormatStyle::LK_TextProto) {
- if (FormatToken *Previous = Tok->getPreviousNonComment())
- Previous->Type = TT_SelectorName;
- }
- } else if (Contexts.back().ColonIsObjCMethodExpr ||
- Line.startsWith(TT_ObjCMethodSpecifier)) {
- Tok->Type = TT_ObjCMethodExpr;
- const FormatToken *BeforePrevious = Tok->Previous->Previous;
- if (!BeforePrevious ||
- !(BeforePrevious->is(TT_CastRParen) ||
- (BeforePrevious->is(TT_ObjCMethodExpr) &&
- BeforePrevious->is(tok::colon))) ||
- BeforePrevious->is(tok::r_square) ||
- Contexts.back().LongestObjCSelectorName == 0) {
- Tok->Previous->Type = TT_SelectorName;
- if (Tok->Previous->ColumnWidth >
- Contexts.back().LongestObjCSelectorName)
- Contexts.back().LongestObjCSelectorName =
- Tok->Previous->ColumnWidth;
- if (!Contexts.back().FirstObjCSelectorName)
- Contexts.back().FirstObjCSelectorName = Tok->Previous;
- }
- } else if (Contexts.back().ColonIsForRangeExpr) {
- Tok->Type = TT_RangeBasedForLoopColon;
- } else if (CurrentToken && CurrentToken->is(tok::numeric_constant)) {
- Tok->Type = TT_BitFieldColon;
- } else if (Contexts.size() == 1 &&
- !Line.First->isOneOf(tok::kw_enum, tok::kw_case)) {
- if (Tok->getPreviousNonComment()->isOneOf(tok::r_paren,
- tok::kw_noexcept))
- Tok->Type = TT_CtorInitializerColon;
- else
- Tok->Type = TT_InheritanceColon;
- } else if (Tok->Previous->is(tok::identifier) && Tok->Next &&
- Tok->Next->isOneOf(tok::r_paren, tok::comma)) {
- // This handles a special macro in ObjC code where selectors including
- // the colon are passed as macro arguments.
- Tok->Type = TT_ObjCMethodExpr;
- } else if (Contexts.back().ContextKind == tok::l_paren) {
- Tok->Type = TT_InlineASMColon;
- }
- break;
- case tok::pipe:
- case tok::amp:
- // | and & in declarations/type expressions represent union and
- // intersection types, respectively.
- if (Style.Language == FormatStyle::LK_JavaScript &&
- !Contexts.back().IsExpression)
- Tok->Type = TT_JsTypeOperator;
- break;
- case tok::kw_if:
- case tok::kw_while:
- if (Tok->is(tok::kw_if) && CurrentToken &&
- CurrentToken->is(tok::kw_constexpr))
- next();
- if (CurrentToken && CurrentToken->is(tok::l_paren)) {
- next();
- if (!parseParens(/*LookForDecls=*/true))
- return false;
- }
- break;
- case tok::kw_for:
- if (Style.Language == FormatStyle::LK_JavaScript) {
- if (Tok->Previous && Tok->Previous->is(tok::period))
- break;
- // JS' for await ( ...
- if (CurrentToken && CurrentToken->is(Keywords.kw_await))
- next();
- }
- Contexts.back().ColonIsForRangeExpr = true;
- next();
- if (!parseParens())
- return false;
- break;
- case tok::l_paren:
- // When faced with 'operator()()', the kw_operator handler incorrectly
- // marks the first l_paren as a OverloadedOperatorLParen. Here, we make
- // the first two parens OverloadedOperators and the second l_paren an
- // OverloadedOperatorLParen.
- if (Tok->Previous && Tok->Previous->is(tok::r_paren) &&
- Tok->Previous->MatchingParen &&
- Tok->Previous->MatchingParen->is(TT_OverloadedOperatorLParen)) {
- Tok->Previous->Type = TT_OverloadedOperator;
- Tok->Previous->MatchingParen->Type = TT_OverloadedOperator;
- Tok->Type = TT_OverloadedOperatorLParen;
- }
- if (!parseParens())
- return false;
- if (Line.MustBeDeclaration && Contexts.size() == 1 &&
- !Contexts.back().IsExpression && !Line.startsWith(TT_ObjCProperty) &&
- (!Tok->Previous ||
- !Tok->Previous->isOneOf(tok::kw_decltype, tok::kw___attribute,
- TT_LeadingJavaAnnotation)))
- Line.MightBeFunctionDecl = true;
- break;
- case tok::l_square:
- if (!parseSquare())
- return false;
- break;
- case tok::l_brace:
- if (Style.Language == FormatStyle::LK_TextProto) {
- FormatToken *Previous = Tok->getPreviousNonComment();
- if (Previous && Previous->Type != TT_DictLiteral)
- Previous->Type = TT_SelectorName;
- }
- if (!parseBrace())
- return false;
- break;
- case tok::less:
- if (parseAngle()) {
- Tok->Type = TT_TemplateOpener;
- if (Style.Language == FormatStyle::LK_TextProto) {
- FormatToken *Previous = Tok->getPreviousNonComment();
- if (Previous && Previous->Type != TT_DictLiteral)
- Previous->Type = TT_SelectorName;
- }
- } else {
- Tok->Type = TT_BinaryOperator;
- NonTemplateLess.insert(Tok);
- CurrentToken = Tok;
- next();
- }
- break;
- case tok::r_paren:
- case tok::r_square:
- return false;
- case tok::r_brace:
- // Lines can start with '}'.
- if (Tok->Previous)
- return false;
- break;
- case tok::greater:
- Tok->Type = TT_BinaryOperator;
- break;
- case tok::kw_operator:
- while (CurrentToken &&
- !CurrentToken->isOneOf(tok::l_paren, tok::semi, tok::r_paren)) {
- if (CurrentToken->isOneOf(tok::star, tok::amp))
- CurrentToken->Type = TT_PointerOrReference;
- consumeToken();
- if (CurrentToken &&
- CurrentToken->Previous->isOneOf(TT_BinaryOperator, TT_UnaryOperator,
- tok::comma))
- CurrentToken->Previous->Type = TT_OverloadedOperator;
- }
- if (CurrentToken) {
- CurrentToken->Type = TT_OverloadedOperatorLParen;
- if (CurrentToken->Previous->is(TT_BinaryOperator))
- CurrentToken->Previous->Type = TT_OverloadedOperator;
- }
- break;
- case tok::question:
- if (Style.Language == FormatStyle::LK_JavaScript && Tok->Next &&
- Tok->Next->isOneOf(tok::semi, tok::comma, tok::colon, tok::r_paren,
- tok::r_brace)) {
- // Question marks before semicolons, colons, etc. indicate optional
- // types (fields, parameters), e.g.
- // function(x?: string, y?) {...}
- // class X { y?; }
- Tok->Type = TT_JsTypeOptionalQuestion;
- break;
- }
- // Declarations cannot be conditional expressions, this can only be part
- // of a type declaration.
- if (Line.MustBeDeclaration && !Contexts.back().IsExpression &&
- Style.Language == FormatStyle::LK_JavaScript)
- break;
- parseConditional();
- break;
- case tok::kw_template:
- parseTemplateDeclaration();
- break;
- case tok::comma:
- if (Contexts.back().InCtorInitializer)
- Tok->Type = TT_CtorInitializerComma;
- else if (Contexts.back().InInheritanceList)
- Tok->Type = TT_InheritanceComma;
- else if (Contexts.back().FirstStartOfName &&
- (Contexts.size() == 1 || Line.startsWith(tok::kw_for))) {
- Contexts.back().FirstStartOfName->PartOfMultiVariableDeclStmt = true;
- Line.IsMultiVariableDeclStmt = true;
- }
- if (Contexts.back().IsForEachMacro)
- Contexts.back().IsExpression = true;
- break;
- case tok::identifier:
- if (Tok->isOneOf(Keywords.kw___has_include,
- Keywords.kw___has_include_next)) {
- parseHasInclude();
- }
- break;
- default:
- break;
- }
- return true;
- }
- void parseIncludeDirective() {
- if (CurrentToken && CurrentToken->is(tok::less)) {
- next();
- while (CurrentToken) {
- // Mark tokens up to the trailing line comments as implicit string
- // literals.
- if (CurrentToken->isNot(tok::comment) &&
- !CurrentToken->TokenText.startswith("//"))
- CurrentToken->Type = TT_ImplicitStringLiteral;
- next();
- }
- }
- }
- void parseWarningOrError() {
- next();
- // We still want to format the whitespace left of the first token of the
- // warning or error.
- next();
- while (CurrentToken) {
- CurrentToken->Type = TT_ImplicitStringLiteral;
- next();
- }
- }
- void parsePragma() {
- next(); // Consume "pragma".
- if (CurrentToken &&
- CurrentToken->isOneOf(Keywords.kw_mark, Keywords.kw_option)) {
- bool IsMark = CurrentToken->is(Keywords.kw_mark);
- next(); // Consume "mark".
- next(); // Consume first token (so we fix leading whitespace).
- while (CurrentToken) {
- if (IsMark || CurrentToken->Previous->is(TT_BinaryOperator))
- CurrentToken->Type = TT_ImplicitStringLiteral;
- next();
- }
- }
- }
- void parseHasInclude() {
- if (!CurrentToken || !CurrentToken->is(tok::l_paren))
- return;
- next(); // '('
- parseIncludeDirective();
- next(); // ')'
- }
- LineType parsePreprocessorDirective() {
- bool IsFirstToken = CurrentToken->IsFirst;
- LineType Type = LT_PreprocessorDirective;
- next();
- if (!CurrentToken)
- return Type;
- if (Style.Language == FormatStyle::LK_JavaScript && IsFirstToken) {
- // JavaScript files can contain shebang lines of the form:
- // #!/usr/bin/env node
- // Treat these like C++ #include directives.
- while (CurrentToken) {
- // Tokens cannot be comments here.
- CurrentToken->Type = TT_ImplicitStringLiteral;
- next();
- }
- return LT_ImportStatement;
- }
- if (CurrentToken->Tok.is(tok::numeric_constant)) {
- CurrentToken->SpacesRequiredBefore = 1;
- return Type;
- }
- // Hashes in the middle of a line can lead to any strange token
- // sequence.
- if (!CurrentToken->Tok.getIdentifierInfo())
- return Type;
- switch (CurrentToken->Tok.getIdentifierInfo()->getPPKeywordID()) {
- case tok::pp_include:
- case tok::pp_include_next:
- case tok::pp_import:
- next();
- parseIncludeDirective();
- Type = LT_ImportStatement;
- break;
- case tok::pp_error:
- case tok::pp_warning:
- parseWarningOrError();
- break;
- case tok::pp_pragma:
- parsePragma();
- break;
- case tok::pp_if:
- case tok::pp_elif:
- Contexts.back().IsExpression = true;
- parseLine();
- break;
- default:
- break;
- }
- while (CurrentToken) {
- FormatToken *Tok = CurrentToken;
- next();
- if (Tok->is(tok::l_paren))
- parseParens();
- else if (Tok->isOneOf(Keywords.kw___has_include,
- Keywords.kw___has_include_next))
- parseHasInclude();
- }
- return Type;
- }
- public:
- LineType parseLine() {
- NonTemplateLess.clear();
- if (CurrentToken->is(tok::hash))
- return parsePreprocessorDirective();
- // Directly allow to 'import <string-literal>' to support protocol buffer
- // definitions (code.google.com/p/protobuf) or missing "#" (either way we
- // should not break the line).
- IdentifierInfo *Info = CurrentToken->Tok.getIdentifierInfo();
- if ((Style.Language == FormatStyle::LK_Java &&
- CurrentToken->is(Keywords.kw_package)) ||
- (Info && Info->getPPKeywordID() == tok::pp_import &&
- CurrentToken->Next &&
- CurrentToken->Next->isOneOf(tok::string_literal, tok::identifier,
- tok::kw_static))) {
- next();
- parseIncludeDirective();
- return LT_ImportStatement;
- }
- // If this line starts and ends in '<' and '>', respectively, it is likely
- // part of "#define <a/b.h>".
- if (CurrentToken->is(tok::less) && Line.Last->is(tok::greater)) {
- parseIncludeDirective();
- return LT_ImportStatement;
- }
- // In .proto files, top-level options are very similar to import statements
- // and should not be line-wrapped.
- if (Style.Language == FormatStyle::LK_Proto && Line.Level == 0 &&
- CurrentToken->is(Keywords.kw_option)) {
- next();
- if (CurrentToken && CurrentToken->is(tok::identifier))
- return LT_ImportStatement;
- }
- bool KeywordVirtualFound = false;
- bool ImportStatement = false;
- // import {...} from '...';
- if (Style.Language == FormatStyle::LK_JavaScript &&
- CurrentToken->is(Keywords.kw_import))
- ImportStatement = true;
- while (CurrentToken) {
- if (CurrentToken->is(tok::kw_virtual))
- KeywordVirtualFound = true;
- if (Style.Language == FormatStyle::LK_JavaScript) {
- // export {...} from '...';
- // An export followed by "from 'some string';" is a re-export from
- // another module identified by a URI and is treated as a
- // LT_ImportStatement (i.e. prevent wraps on it for long URIs).
- // Just "export {...};" or "export class ..." should not be treated as
- // an import in this sense.
- if (Line.First->is(tok::kw_export) &&
- CurrentToken->is(Keywords.kw_from) && CurrentToken->Next &&
- CurrentToken->Next->isStringLiteral())
- ImportStatement = true;
- if (isClosureImportStatement(*CurrentToken))
- ImportStatement = true;
- }
- if (!consumeToken())
- return LT_Invalid;
- }
- if (KeywordVirtualFound)
- return LT_VirtualFunctionDecl;
- if (ImportStatement)
- return LT_ImportStatement;
- if (Line.startsWith(TT_ObjCMethodSpecifier)) {
- if (Contexts.back().FirstObjCSelectorName)
- Contexts.back().FirstObjCSelectorName->LongestObjCSelectorName =
- Contexts.back().LongestObjCSelectorName;
- return LT_ObjCMethodDecl;
- }
- return LT_Other;
- }
- private:
- bool isClosureImportStatement(const FormatToken &Tok) {
- // FIXME: Closure-library specific stuff should not be hard-coded but be
- // configurable.
- return Tok.TokenText == "goog" && Tok.Next && Tok.Next->is(tok::period) &&
- Tok.Next->Next &&
- (Tok.Next->Next->TokenText == "module" ||
- Tok.Next->Next->TokenText == "provide" ||
- Tok.Next->Next->TokenText == "require" ||
- Tok.Next->Next->TokenText == "forwardDeclare") &&
- Tok.Next->Next->Next && Tok.Next->Next->Next->is(tok::l_paren);
- }
- void resetTokenMetadata(FormatToken *Token) {
- if (!Token)
- return;
- // Reset token type in case we have already looked at it and then
- // recovered from an error (e.g. failure to find the matching >).
- if (!CurrentToken->isOneOf(TT_LambdaLSquare, TT_ForEachMacro,
- TT_FunctionLBrace, TT_ImplicitStringLiteral,
- TT_InlineASMBrace, TT_JsFatArrow, TT_LambdaArrow,
- TT_OverloadedOperator, TT_RegexLiteral,
- TT_TemplateString, TT_ObjCStringLiteral))
- CurrentToken->Type = TT_Unknown;
- CurrentToken->Role.reset();
- CurrentToken->MatchingParen = nullptr;
- CurrentToken->FakeLParens.clear();
- CurrentToken->FakeRParens = 0;
- }
- void next() {
- if (CurrentToken) {
- CurrentToken->NestingLevel = Contexts.size() - 1;
- CurrentToken->BindingStrength = Contexts.back().BindingStrength;
- modifyContext(*CurrentToken);
- determineTokenType(*CurrentToken);
- CurrentToken = CurrentToken->Next;
- }
- resetTokenMetadata(CurrentToken);
- }
- /// \brief A struct to hold information valid in a specific context, e.g.
- /// a pair of parenthesis.
- struct Context {
- Context(tok::TokenKind ContextKind, unsigned BindingStrength,
- bool IsExpression)
- : ContextKind(ContextKind), BindingStrength(BindingStrength),
- IsExpression(IsExpression) {}
- tok::TokenKind ContextKind;
- unsigned BindingStrength;
- bool IsExpression;
- unsigned LongestObjCSelectorName = 0;
- bool ColonIsForRangeExpr = false;
- bool ColonIsDictLiteral = false;
- bool ColonIsObjCMethodExpr = false;
- FormatToken *FirstObjCSelectorName = nullptr;
- FormatToken *FirstStartOfName = nullptr;
- bool CanBeExpression = true;
- bool InTemplateArgument = false;
- bool InCtorInitializer = false;
- bool InInheritanceList = false;
- bool CaretFound = false;
- bool IsForEachMacro = false;
- };
- /// \brief Puts a new \c Context onto the stack \c Contexts for the lifetime
- /// of each instance.
- struct ScopedContextCreator {
- AnnotatingParser &P;
- ScopedContextCreator(AnnotatingParser &P, tok::TokenKind ContextKind,
- unsigned Increase)
- : P(P) {
- P.Contexts.push_back(Context(ContextKind,
- P.Contexts.back().BindingStrength + Increase,
- P.Contexts.back().IsExpression));
- }
- ~ScopedContextCreator() { P.Contexts.pop_back(); }
- };
- void modifyContext(const FormatToken &Current) {
- if (Current.getPrecedence() == prec::Assignment &&
- !Line.First->isOneOf(tok::kw_template, tok::kw_using, tok::kw_return) &&
- // Type aliases use `type X = ...;` in TypeScript and can be exported
- // using `export type ...`.
- !(Style.Language == FormatStyle::LK_JavaScript &&
- (Line.startsWith(Keywords.kw_type, tok::identifier) ||
- Line.startsWith(tok::kw_export, Keywords.kw_type,
- tok::identifier))) &&
- (!Current.Previous || Current.Previous->isNot(tok::kw_operator))) {
- Contexts.back().IsExpression = true;
- if (!Line.startsWith(TT_UnaryOperator)) {
- for (FormatToken *Previous = Current.Previous;
- Previous && Previous->Previous &&
- !Previous->Previous->isOneOf(tok::comma, tok::semi);
- Previous = Previous->Previous) {
- if (Previous->isOneOf(tok::r_square, tok::r_paren)) {
- Previous = Previous->MatchingParen;
- if (!Previous)
- break;
- }
- if (Previous->opensScope())
- break;
- if (Previous->isOneOf(TT_BinaryOperator, TT_UnaryOperator) &&
- Previous->isOneOf(tok::star, tok::amp, tok::ampamp) &&
- Previous->Previous && Previous->Previous->isNot(tok::equal))
- Previous->Type = TT_PointerOrReference;
- }
- }
- } else if (Current.is(tok::lessless) &&
- (!Current.Previous || !Current.Previous->is(tok::kw_operator))) {
- Contexts.back().IsExpression = true;
- } else if (Current.isOneOf(tok::kw_return, tok::kw_throw)) {
- Contexts.back().IsExpression = true;
- } else if (Current.is(TT_TrailingReturnArrow)) {
- Contexts.back().IsExpression = false;
- } else if (Current.is(TT_LambdaArrow) || Current.is(Keywords.kw_assert)) {
- Contexts.back().IsExpression = Style.Language == FormatStyle::LK_Java;
- } else if (Current.Previous &&
- Current.Previous->is(TT_CtorInitializerColon)) {
- Contexts.back().IsExpression = true;
- Contexts.back().InCtorInitializer = true;
- } else if (Current.Previous && Current.Previous->is(TT_InheritanceColon)) {
- Contexts.back().InInheritanceList = true;
- } else if (Current.isOneOf(tok::r_paren, tok::greater, tok::comma)) {
- for (FormatToken *Previous = Current.Previous;
- Previous && Previous->isOneOf(tok::star, tok::amp);
- Previous = Previous->Previous)
- Previous->Type = TT_PointerOrReference;
- if (Line.MustBeDeclaration && !Contexts.front().InCtorInitializer)
- Contexts.back().IsExpression = false;
- } else if (Current.is(tok::kw_new)) {
- Contexts.back().CanBeExpression = false;
- } else if (Current.isOneOf(tok::semi, tok::exclaim)) {
- // This should be the condition or increment in a for-loop.
- Contexts.back().IsExpression = true;
- }
- }
- void determineTokenType(FormatToken &Current) {
- if (!Current.is(TT_Unknown))
- // The token type is already known.
- return;
- if (Style.Language == FormatStyle::LK_JavaScript) {
- if (Current.is(tok::exclaim)) {
- if (Current.Previous &&
- (Current.Previous->isOneOf(tok::identifier, tok::kw_namespace,
- tok::r_paren, tok::r_square,
- tok::r_brace) ||
- Current.Previous->Tok.isLiteral())) {
- Current.Type = TT_JsNonNullAssertion;
- return;
- }
- if (Current.Next &&
- Current.Next->isOneOf(TT_BinaryOperator, Keywords.kw_as)) {
- Current.Type = TT_JsNonNullAssertion;
- return;
- }
- }
- }
- // Line.MightBeFunctionDecl can only be true after the parentheses of a
- // function declaration have been found. In this case, 'Current' is a
- // trailing token of this declaration and thus cannot be a name.
- if (Current.is(Keywords.kw_instanceof)) {
- Current.Type = TT_BinaryOperator;
- } else if (isStartOfName(Current) &&
- (!Line.MightBeFunctionDecl || Current.NestingLevel != 0)) {
- Contexts.back().FirstStartOfName = &Current;
- Current.Type = TT_StartOfName;
- } else if (Current.is(tok::semi)) {
- // Reset FirstStartOfName after finding a semicolon so that a for loop
- // with multiple increment statements is not confused with a for loop
- // having multiple variable declarations.
- Contexts.back().FirstStartOfName = nullptr;
- } else if (Current.isOneOf(tok::kw_auto, tok::kw___auto_type)) {
- AutoFound = true;
- } else if (Current.is(tok::arrow) &&
- Style.Language == FormatStyle::LK_Java) {
- Current.Type = TT_LambdaArrow;
- } else if (Current.is(tok::arrow) && AutoFound && Line.MustBeDeclaration &&
- Current.NestingLevel == 0) {
- Current.Type = TT_TrailingReturnArrow;
- } else if (Current.isOneOf(tok::star, tok::amp, tok::ampamp)) {
- Current.Type = determineStarAmpUsage(Current,
- Contexts.back().CanBeExpression &&
- Contexts.back().IsExpression,
- Contexts.back().InTemplateArgument);
- } else if (Current.isOneOf(tok::minus, tok::plus, tok::caret)) {
- Current.Type = determinePlusMinusCaretUsage(Current);
- if (Current.is(TT_UnaryOperator) && Current.is(tok::caret))
- Contexts.back().CaretFound = true;
- } else if (Current.isOneOf(tok::minusminus, tok::plusplus)) {
- Current.Type = determineIncrementUsage(Current);
- } else if (Current.isOneOf(tok::exclaim, tok::tilde)) {
- Current.Type = TT_UnaryOperator;
- } else if (Current.is(tok::question)) {
- if (Style.Language == FormatStyle::LK_JavaScript &&
- Line.MustBeDeclaration && !Contexts.back().IsExpression) {
- // In JavaScript, `interface X { foo?(): bar; }` is an optional method
- // on the interface, not a ternary expression.
- Current.Type = TT_JsTypeOptionalQuestion;
- } else {
- Current.Type = TT_ConditionalExpr;
- }
- } else if (Current.isBinaryOperator() &&
- (!Current.Previous || Current.Previous->isNot(tok::l_square))) {
- Current.Type = TT_BinaryOperator;
- } else if (Current.is(tok::comment)) {
- if (Current.TokenText.startswith("/*")) {
- if (Current.TokenText.endswith("*/"))
- Current.Type = TT_BlockComment;
- else
- // The lexer has for some reason determined a comment here. But we
- // cannot really handle it, if it isn't properly terminated.
- Current.Tok.setKind(tok::unknown);
- } else {
- Current.Type = TT_LineComment;
- }
- } else if (Current.is(tok::r_paren)) {
- if (rParenEndsCast(Current))
- Current.Type = TT_CastRParen;
- if (Current.MatchingParen && Current.Next &&
- !Current.Next->isBinaryOperator() &&
- !Current.Next->isOneOf(tok::semi, tok::colon, tok::l_brace,
- tok::comma, tok::period, tok::arrow,
- tok::coloncolon))
- if (FormatToken *AfterParen = Current.MatchingParen->Next) {
- // Make sure this isn't the return type of an Obj-C block declaration
- if (AfterParen->Tok.isNot(tok::caret)) {
- if (FormatToken *BeforeParen = Current.MatchingParen->Previous)
- if (BeforeParen->is(tok::identifier) &&
- BeforeParen->TokenText == BeforeParen->TokenText.upper() &&
- (!BeforeParen->Previous ||
- BeforeParen->Previous->ClosesTemplateDeclaration))
- Current.Type = TT_FunctionAnnotationRParen;
- }
- }
- } else if (Current.is(tok::at) && Current.Next &&
- Style.Language != FormatStyle::LK_JavaScript &&
- Style.Language != FormatStyle::LK_Java) {
- // In Java & JavaScript, "@..." is a decorator or annotation. In ObjC, it
- // marks declarations and properties that need special formatting.
- switch (Current.Next->Tok.getObjCKeywordID()) {
- case tok::objc_interface:
- case tok::objc_implementation:
- case tok::objc_protocol:
- Current.Type = TT_ObjCDecl;
- break;
- case tok::objc_property:
- Current.Type = TT_ObjCProperty;
- break;
- default:
- break;
- }
- } else if (Current.is(tok::period)) {
- FormatToken *PreviousNoComment = Current.getPreviousNonComment();
- if (PreviousNoComment &&
- PreviousNoComment->isOneOf(tok::comma, tok::l_brace))
- Current.Type = TT_DesignatedInitializerPeriod;
- else if (Style.Language == FormatStyle::LK_Java && Current.Previous &&
- Current.Previous->isOneOf(TT_JavaAnnotation,
- TT_LeadingJavaAnnotation)) {
- Current.Type = Current.Previous->Type;
- }
- } else if (Current.isOneOf(tok::identifier, tok::kw_const) &&
- Current.Previous &&
- !Current.Previous->isOneOf(tok::equal, tok::at) &&
- Line.MightBeFunctionDecl && Contexts.size() == 1) {
- // Line.MightBeFunctionDecl can only be true after the parentheses of a
- // function declaration have been found.
- Current.Type = TT_TrailingAnnotation;
- } else if ((Style.Language == FormatStyle::LK_Java ||
- Style.Language == FormatStyle::LK_JavaScript) &&
- Current.Previous) {
- if (Current.Previous->is(tok::at) &&
- Current.isNot(Keywords.kw_interface)) {
- const FormatToken &AtToken = *Current.Previous;
- const FormatToken *Previous = AtToken.getPreviousNonComment();
- if (!Previous || Previous->is(TT_LeadingJavaAnnotation))
- Current.Type = TT_LeadingJavaAnnotation;
- else
- Current.Type = TT_JavaAnnotation;
- } else if (Current.Previous->is(tok::period) &&
- Current.Previous->isOneOf(TT_JavaAnnotation,
- TT_LeadingJavaAnnotation)) {
- Current.Type = Current.Previous->Type;
- }
- }
- }
- /// \brief Take a guess at whether \p Tok starts a name of a function or
- /// variable declaration.
- ///
- /// This is a heuristic based on whether \p Tok is an identifier following
- /// something that is likely a type.
- bool isStartOfName(const FormatToken &Tok) {
- if (Tok.isNot(tok::identifier) || !Tok.Previous)
- return false;
- if (Tok.Previous->isOneOf(TT_LeadingJavaAnnotation, Keywords.kw_instanceof,
- Keywords.kw_as))
- return false;
- if (Style.Language == FormatStyle::LK_JavaScript &&
- Tok.Previous->is(Keywords.kw_in))
- return false;
- // Skip "const" as it does not have an influence on whether this is a name.
- FormatToken *PreviousNotConst = Tok.getPreviousNonComment();
- while (PreviousNotConst && PreviousNotConst->is(tok::kw_const))
- PreviousNotConst = PreviousNotConst->getPreviousNonComment();
- if (!PreviousNotConst)
- return false;
- bool IsPPKeyword = PreviousNotConst->is(tok::identifier) &&
- PreviousNotConst->Previous &&
- PreviousNotConst->Previous->is(tok::hash);
- if (PreviousNotConst->is(TT_TemplateCloser))
- return PreviousNotConst && PreviousNotConst->MatchingParen &&
- PreviousNotConst->MatchingParen->Previous &&
- PreviousNotConst->MatchingParen->Previous->isNot(tok::period) &&
- PreviousNotConst->MatchingParen->Previous->isNot(tok::kw_template);
- if (PreviousNotConst->is(tok::r_paren) && PreviousNotConst->MatchingParen &&
- PreviousNotConst->MatchingParen->Previous &&
- PreviousNotConst->MatchingParen->Previous->is(tok::kw_decltype))
- return true;
- return (!IsPPKeyword &&
- PreviousNotConst->isOneOf(tok::identifier, tok::kw_auto)) ||
- PreviousNotConst->is(TT_PointerOrReference) ||
- PreviousNotConst->isSimpleTypeSpecifier();
- }
- /// \brief Determine whether ')' is ending a cast.
- bool rParenEndsCast(const FormatToken &Tok) {
- // C-style casts are only used in C++ and Java.
- if (!Style.isCpp() && Style.Language != FormatStyle::LK_Java)
- return false;
- // Empty parens aren't casts and there are no casts at the end of the line.
- if (Tok.Previous == Tok.MatchingParen || !Tok.Next || !Tok.MatchingParen)
- return false;
- FormatToken *LeftOfParens = Tok.MatchingParen->getPreviousNonComment();
- if (LeftOfParens) {
- // If there is a closing parenthesis left of the current parentheses,
- // look past it as these might be chained casts.
- if (LeftOfParens->is(tok::r_paren)) {
- if (!LeftOfParens->MatchingParen ||
- !LeftOfParens->MatchingParen->Previous)
- return false;
- LeftOfParens = LeftOfParens->MatchingParen->Previous;
- }
- // If there is an identifier (or with a few exceptions a keyword) right
- // before the parentheses, this is unlikely to be a cast.
- if (LeftOfParens->Tok.getIdentifierInfo() &&
- !LeftOfParens->isOneOf(Keywords.kw_in, tok::kw_return, tok::kw_case,
- tok::kw_delete))
- return false;
- // Certain other tokens right before the parentheses are also signals that
- // this cannot be a cast.
- if (LeftOfParens->isOneOf(tok::at, tok::r_square, TT_OverloadedOperator,
- TT_TemplateCloser, tok::ellipsis))
- return false;
- }
- if (Tok.Next->is(tok::question))
- return false;
- // As Java has no function types, a "(" after the ")" likely means that this
- // is a cast.
- if (Style.Language == FormatStyle::LK_Java && Tok.Next->is(tok::l_paren))
- return true;
- // If a (non-string) literal follows, this is likely a cast.
- if (Tok.Next->isNot(tok::string_literal) &&
- (Tok.Next->Tok.isLiteral() ||
- Tok.Next->isOneOf(tok::kw_sizeof, tok::kw_alignof)))
- return true;
- // Heuristically try to determine whether the parentheses contain a type.
- bool ParensAreType =
- !Tok.Previous ||
- Tok.Previous->isOneOf(TT_PointerOrReference, TT_TemplateCloser) ||
- Tok.Previous->isSimpleTypeSpecifier();
- bool ParensCouldEndDecl =
- Tok.Next->isOneOf(tok::equal, tok::semi, tok::l_brace, tok::greater);
- if (ParensAreType && !ParensCouldEndDecl)
- return true;
- // At this point, we heuristically assume that there are no casts at the
- // start of the line. We assume that we have found most cases where there
- // are by the logic above, e.g. "(void)x;".
- if (!LeftOfParens)
- return false;
- // Certain token types inside the parentheses mean that this can't be a
- // cast.
- for (const FormatToken *Token = Tok.MatchingParen->Next; Token != &Tok;
- Token = Token->Next)
- if (Token->is(TT_BinaryOperator))
- return false;
- // If the following token is an identifier or 'this', this is a cast. All
- // cases where this can be something else are handled above.
- if (Tok.Next->isOneOf(tok::identifier, tok::kw_this))
- return true;
- if (!Tok.Next->Next)
- return false;
- // If the next token after the parenthesis is a unary operator, assume
- // that this is cast, unless there are unexpected tokens inside the
- // parenthesis.
- bool NextIsUnary =
- Tok.Next->isUnaryOperator() || Tok.Next->isOneOf(tok::amp, tok::star);
- if (!NextIsUnary || Tok.Next->is(tok::plus) ||
- !Tok.Next->Next->isOneOf(tok::identifier, tok::numeric_constant))
- return false;
- // Search for unexpected tokens.
- for (FormatToken *Prev = Tok.Previous; Prev != Tok.MatchingParen;
- Prev = Prev->Previous) {
- if (!Prev->isOneOf(tok::kw_const, tok::identifier, tok::coloncolon))
- return false;
- }
- return true;
- }
- /// \brief Return the type of the given token assuming it is * or &.
- TokenType determineStarAmpUsage(const FormatToken &Tok, bool IsExpression,
- bool InTemplateArgument) {
- if (Style.Language == FormatStyle::LK_JavaScript)
- return TT_BinaryOperator;
- const FormatToken *PrevToken = Tok.getPreviousNonComment();
- if (!PrevToken)
- return TT_UnaryOperator;
- const FormatToken *NextToken = Tok.getNextNonComment();
- if (!NextToken ||
- NextToken->isOneOf(tok::arrow, tok::equal, tok::kw_const) ||
- (NextToken->is(tok::l_brace) && !NextToken->getNextNonComment()))
- return TT_PointerOrReference;
- if (PrevToken->is(tok::coloncolon))
- return TT_PointerOrReference;
- if (PrevToken->isOneOf(tok::l_paren, tok::l_square, tok::l_brace,
- tok::comma, tok::semi, tok::kw_return, tok::colon,
- tok::equal, tok::kw_delete, tok::kw_sizeof,
- tok::kw_throw) ||
- PrevToken->isOneOf(TT_BinaryOperator, TT_ConditionalExpr,
- TT_UnaryOperator, TT_CastRParen))
- return TT_UnaryOperator;
- if (NextToken->is(tok::l_square) && NextToken->isNot(TT_LambdaLSquare))
- return TT_PointerOrReference;
- if (NextToken->is(tok::kw_operator) && !IsExpression)
- return TT_PointerOrReference;
- if (NextToken->isOneOf(tok::comma, tok::semi))
- return TT_PointerOrReference;
- if (PrevToken->is(tok::r_paren) && PrevToken->MatchingParen) {
- FormatToken *TokenBeforeMatchingParen =
- PrevToken->MatchingParen->getPreviousNonComment();
- if (TokenBeforeMatchingParen &&
- TokenBeforeMatchingParen->isOneOf(tok::kw_typeof, tok::kw_decltype))
- return TT_PointerOrReference;
- }
- if (PrevToken->Tok.isLiteral() ||
- PrevToken->isOneOf(tok::r_paren, tok::r_square, tok::kw_true,
- tok::kw_false, tok::r_brace) ||
- NextToken->Tok.isLiteral() ||
- NextToken->isOneOf(tok::kw_true, tok::kw_false) ||
- NextToken->isUnaryOperator() ||
- // If we know we're in a template argument, there are no named
- // declarations. Thus, having an identifier on the right-hand side
- // indicates a binary operator.
- (InTemplateArgument && NextToken->Tok.isAnyIdentifier()))
- return TT_BinaryOperator;
- // "&&(" is quite unlikely to be two successive unary "&".
- if (Tok.is(tok::ampamp) && NextToken && NextToken->is(tok::l_paren))
- return TT_BinaryOperator;
- // This catches some cases where evaluation order is used as control flow:
- // aaa && aaa->f();
- const FormatToken *NextNextToken = NextToken->getNextNonComment();
- if (NextNextToken && NextNextToken->is(tok::arrow))
- return TT_BinaryOperator;
- // It is very unlikely that we are going to find a pointer or reference type
- // definition on the RHS of an assignment.
- if (IsExpression && !Contexts.back().CaretFound)
- return TT_BinaryOperator;
- return TT_PointerOrReference;
- }
- TokenType determinePlusMinusCaretUsage(const FormatToken &Tok) {
- const FormatToken *PrevToken = Tok.getPreviousNonComment();
- if (!PrevToken)
- return TT_UnaryOperator;
- if (PrevToken->isOneOf(TT_CastRParen, TT_UnaryOperator) &&
- !PrevToken->is(tok::exclaim))
- // There aren't any trailing unary operators except for TypeScript's
- // non-null operator (!). Thus, this must be squence of leading operators.
- return TT_UnaryOperator;
- // Use heuristics to recognize unary operators.
- if (PrevToken->isOneOf(tok::equal, tok::l_paren, tok::comma, tok::l_square,
- tok::question, tok::colon, tok::kw_return,
- tok::kw_case, tok::at, tok::l_brace))
- return TT_UnaryOperator;
- // There can't be two consecutive binary operators.
- if (PrevToken->is(TT_BinaryOperator))
- return TT_UnaryOperator;
- // Fall back to marking the token as binary operator.
- return TT_BinaryOperator;
- }
- /// \brief Determine whether ++/-- are pre- or post-increments/-decrements.
- TokenType determineIncrementUsage(const FormatToken &Tok) {
- const FormatToken *PrevToken = Tok.getPreviousNonComment();
- if (!PrevToken || PrevToken->is(TT_CastRParen))
- return TT_UnaryOperator;
- if (PrevToken->isOneOf(tok::r_paren, tok::r_square, tok::identifier))
- return TT_TrailingUnaryOperator;
- return TT_UnaryOperator;
- }
- SmallVector<Context, 8> Contexts;
- const FormatStyle &Style;
- AnnotatedLine &Line;
- FormatToken *CurrentToken;
- bool AutoFound;
- const AdditionalKeywords &Keywords;
- // Set of "<" tokens that do not open a template parameter list. If parseAngle
- // determines that a specific token can't be a template opener, it will make
- // same decision irrespective of the decisions for tokens leading up to it.
- // Store this information to prevent this from causing exponential runtime.
- llvm::SmallPtrSet<FormatToken *, 16> NonTemplateLess;
- };
- static const int PrecedenceUnaryOperator = prec::PointerToMember + 1;
- static const int PrecedenceArrowAndPeriod = prec::PointerToMember + 2;
- /// \brief Parses binary expressions by inserting fake parenthesis based on
- /// operator precedence.
- class ExpressionParser {
- public:
- ExpressionParser(const FormatStyle &Style, const AdditionalKeywords &Keywords,
- AnnotatedLine &Line)
- : Style(Style), Keywords(Keywords), Current(Line.First) {}
- /// \brief Parse expressions with the given operatore precedence.
- void parse(int Precedence = 0) {
- // Skip 'return' and ObjC selector colons as they are not part of a binary
- // expression.
- while (Current && (Current->is(tok::kw_return) ||
- (Current->is(tok::colon) &&
- Current->isOneOf(TT_ObjCMethodExpr, TT_DictLiteral))))
- next();
- if (!Current || Precedence > PrecedenceArrowAndPeriod)
- return;
- // Conditional expressions need to be parsed separately for proper nesting.
- if (Precedence == prec::Conditional) {
- parseConditionalExpr();
- return;
- }
- // Parse unary operators, which all have a higher precedence than binary
- // operators.
- if (Precedence == PrecedenceUnaryOperator) {
- parseUnaryOperator();
- return;
- }
- FormatToken *Start = Current;
- FormatToken *LatestOperator = nullptr;
- unsigned OperatorIndex = 0;
- while (Current) {
- // Consume operators with higher precedence.
- parse(Precedence + 1);
- int CurrentPrecedence = getCurrentPrecedence();
- if (Current && Current->is(TT_SelectorName) &&
- Precedence == CurrentPrecedence) {
- if (LatestOperator)
- addFakeParenthesis(Start, prec::Level(Precedence));
- Start = Current;
- }
- // At the end of the line or when an operator with higher precedence is
- // found, insert fake parenthesis and return.
- if (!Current ||
- (Current->closesScope() &&
- (Current->MatchingParen || Current->is(TT_TemplateString))) ||
- (CurrentPrecedence != -1 && CurrentPrecedence < Precedence) ||
- (CurrentPrecedence == prec::Conditional &&
- Precedence == prec::Assignment && Current->is(tok::colon))) {
- break;
- }
- // Consume scopes: (), [], <> and {}
- if (Current->opensScope()) {
- // In fragment of a JavaScript template string can look like '}..${' and
- // thus close a scope and open a new one at the same time.
- while (Current && (!Current->closesScope() || Current->opensScope())) {
- next();
- parse();
- }
- next();
- } else {
- // Operator found.
- if (CurrentPrecedence == Precedence) {
- if (LatestOperator)
- LatestOperator->NextOperator = Current;
- LatestOperator = Current;
- Current->OperatorIndex = OperatorIndex;
- ++OperatorIndex;
- }
- next(/*SkipPastLeadingComments=*/Precedence > 0);
- }
- }
- if (LatestOperator && (Current || Precedence > 0)) {
- // LatestOperator->LastOperator = true;
- if (Precedence == PrecedenceArrowAndPeriod) {
- // Call expressions don't have a binary operator precedence.
- addFakeParenthesis(Start, prec::Unknown);
- } else {
- addFakeParenthesis(Start, prec::Level(Precedence));
- }
- }
- }
- private:
- /// \brief Gets the precedence (+1) of the given token for binary operators
- /// and other tokens that we treat like binary operators.
- int getCurrentPrecedence() {
- if (Current) {
- const FormatToken *NextNonComment = Current->getNextNonComment();
- if (Current->is(TT_ConditionalExpr))
- return prec::Conditional;
- if (NextNonComment && Current->is(TT_SelectorName) &&
- (NextNonComment->isOneOf(TT_DictLiteral, TT_JsTypeColon) ||
- ((Style.Language == FormatStyle::LK_Proto ||
- Style.Language == FormatStyle::LK_TextProto) &&
- NextNonComment->is(tok::less))))
- return prec::Assignment;
- if (Current->is(TT_JsComputedPropertyName))
- return prec::Assignment;
- if (Current->is(TT_LambdaArrow))
- return prec::Comma;
- if (Current->is(TT_JsFatArrow))
- return prec::Assignment;
- if (Current->isOneOf(tok::semi, TT_InlineASMColon, TT_SelectorName) ||
- (Current->is(tok::comment) && NextNonComment &&
- NextNonComment->is(TT_SelectorName)))
- return 0;
- if (Current->is(TT_RangeBasedForLoopColon))
- return prec::Comma;
- if ((Style.Language == FormatStyle::LK_Java ||
- Style.Language == FormatStyle::LK_JavaScript) &&
- Current->is(Keywords.kw_instanceof))
- return prec::Relational;
- if (Style.Language == FormatStyle::LK_JavaScript &&
- Current->isOneOf(Keywords.kw_in, Keywords.kw_as))
- return prec::Relational;
- if (Current->is(TT_BinaryOperator) || Current->is(tok::comma))
- return Current->getPrecedence();
- if (Current->isOneOf(tok::period, tok::arrow))
- return PrecedenceArrowAndPeriod;
- if ((Style.Language == FormatStyle::LK_Java ||
- Style.Language == FormatStyle::LK_JavaScript) &&
- Current->isOneOf(Keywords.kw_extends, Keywords.kw_implements,
- Keywords.kw_throws))
- return 0;
- }
- return -1;
- }
- void addFakeParenthesis(FormatToken *Start, prec::Level Precedence) {
- Start->FakeLParens.push_back(Precedence);
- if (Precedence > prec::Unknown)
- Start->StartsBinaryExpression = true;
- if (Current) {
- FormatToken *Previous = Current->Previous;
- while (Previous->is(tok::comment) && Previous->Previous)
- Previous = Previous->Previous;
- ++Previous->FakeRParens;
- if (Precedence > prec::Unknown)
- Previous->EndsBinaryExpression = true;
- }
- }
- /// \brief Parse unary operator expressions and surround them with fake
- /// parentheses if appropriate.
- void parseUnaryOperator() {
- llvm::SmallVector<FormatToken *, 2> Tokens;
- while (Current && Current->is(TT_UnaryOperator)) {
- Tokens.push_back(Current);
- next();
- }
- parse(PrecedenceArrowAndPeriod);
- for (FormatToken *Token : llvm::reverse(Tokens))
- // The actual precedence doesn't matter.
- addFakeParenthesis(Token, prec::Unknown);
- }
- void parseConditionalExpr() {
- while (Current && Current->isTrailingComment()) {
- next();
- }
- FormatToken *Start = Current;
- parse(prec::LogicalOr);
- if (!Current || !Current->is(tok::question))
- return;
- next();
- parse(prec::Assignment);
- if (!Current || Current->isNot(TT_ConditionalExpr))
- return;
- next();
- parse(prec::Assignment);
- addFakeParenthesis(Start, prec::Conditional);
- }
- void next(bool SkipPastLeadingComments = true) {
- if (Current)
- Current = Current->Next;
- while (Current &&
- (Current->NewlinesBefore == 0 || SkipPastLeadingComments) &&
- Current->isTrailingComment())
- Current = Current->Next;
- }
- const FormatStyle &Style;
- const AdditionalKeywords &Keywords;
- FormatToken *Current;
- };
- } // end anonymous namespace
- void TokenAnnotator::setCommentLineLevels(
- SmallVectorImpl<AnnotatedLine *> &Lines) {
- const AnnotatedLine *NextNonCommentLine = nullptr;
- for (SmallVectorImpl<AnnotatedLine *>::reverse_iterator I = Lines.rbegin(),
- E = Lines.rend();
- I != E; ++I) {
- bool CommentLine = true;
- for (const FormatToken *Tok = (*I)->First; Tok; Tok = Tok->Next) {
- if (!Tok->is(tok::comment)) {
- CommentLine = false;
- break;
- }
- }
- if (NextNonCommentLine && CommentLine) {
- // If the comment is currently aligned with the line immediately following
- // it, that's probably intentional and we should keep it.
- bool AlignedWithNextLine =
- NextNonCommentLine->First->NewlinesBefore <= 1 &&
- NextNonCommentLine->First->OriginalColumn ==
- (*I)->First->OriginalColumn;
- if (AlignedWithNextLine)
- (*I)->Level = NextNonCommentLine->Level;
- } else {
- NextNonCommentLine = (*I)->First->isNot(tok::r_brace) ? (*I) : nullptr;
- }
- setCommentLineLevels((*I)->Children);
- }
- }
- static unsigned maxNestingDepth(const AnnotatedLine &Line) {
- unsigned Result = 0;
- for (const auto *Tok = Line.First; Tok != nullptr; Tok = Tok->Next)
- Result = std::max(Result, Tok->NestingLevel);
- return Result;
- }
- void TokenAnnotator::annotate(AnnotatedLine &Line) {
- for (SmallVectorImpl<AnnotatedLine *>::iterator I = Line.Children.begin(),
- E = Line.Children.end();
- I != E; ++I) {
- annotate(**I);
- }
- AnnotatingParser Parser(Style, Line, Keywords);
- Line.Type = Parser.parseLine();
- // With very deep nesting, ExpressionParser uses lots of stack and the
- // formatting algorithm is very slow. We're not going to do a good job here
- // anyway - it's probably generated code being formatted by mistake.
- // Just skip the whole line.
- if (maxNestingDepth(Line) > 50)
- Line.Type = LT_Invalid;
- if (Line.Type == LT_Invalid)
- return;
- ExpressionParser ExprParser(Style, Keywords, Line);
- ExprParser.parse();
- if (Line.startsWith(TT_ObjCMethodSpecifier))
- Line.Type = LT_ObjCMethodDecl;
- else if (Line.startsWith(TT_ObjCDecl))
- Line.Type = LT_ObjCDecl;
- else if (Line.startsWith(TT_ObjCProperty))
- Line.Type = LT_ObjCProperty;
- Line.First->SpacesRequiredBefore = 1;
- Line.First->CanBreakBefore = Line.First->MustBreakBefore;
- }
- // This function heuristically determines whether 'Current' starts the name of a
- // function declaration.
- static bool isFunctionDeclarationName(const FormatToken &Current,
- const AnnotatedLine &Line) {
- auto skipOperatorName = [](const FormatToken *Next) -> const FormatToken * {
- for (; Next; Next = Next->Next) {
- if (Next->is(TT_OverloadedOperatorLParen))
- return Next;
- if (Next->is(TT_OverloadedOperator))
- continue;
- if (Next->isOneOf(tok::kw_new, tok::kw_delete)) {
- // For 'new[]' and 'delete[]'.
- if (Next->Next && Next->Next->is(tok::l_square) && Next->Next->Next &&
- Next->Next->Next->is(tok::r_square))
- Next = Next->Next->Next;
- continue;
- }
- break;
- }
- return nullptr;
- };
- // Find parentheses of parameter list.
- const FormatToken *Next = Current.Next;
- if (Current.is(tok::kw_operator)) {
- if (Current.Previous && Current.Previous->is(tok::coloncolon))
- return false;
- Next = skipOperatorName(Next);
- } else {
- if (!Current.is(TT_StartOfName) || Current.NestingLevel != 0)
- return false;
- for (; Next; Next = Next->Next) {
- if (Next->is(TT_TemplateOpener)) {
- Next = Next->MatchingParen;
- } else if (Next->is(tok::coloncolon)) {
- Next = Next->Next;
- if (!Next)
- return false;
- if (Next->is(tok::kw_operator)) {
- Next = skipOperatorName(Next->Next);
- break;
- }
- if (!Next->is(tok::identifier))
- return false;
- } else if (Next->is(tok::l_paren)) {
- break;
- } else {
- return false;
- }
- }
- }
- // Check whether parameter list can belong to a function declaration.
- if (!Next || !Next->is(tok::l_paren) || !Next->MatchingParen)
- return false;
- // If the lines ends with "{", this is likely an function definition.
- if (Line.Last->is(tok::l_brace))
- return true;
- if (Next->Next == Next->MatchingParen)
- return true; // Empty parentheses.
- // If there is an &/&& after the r_paren, this is likely a function.
- if (Next->MatchingParen->Next &&
- Next->MatchingParen->Next->is(TT_PointerOrReference))
- return true;
- for (const FormatToken *Tok = Next->Next; Tok && Tok != Next->MatchingParen;
- Tok = Tok->Next) {
- if (Tok->is(tok::l_paren) && Tok->MatchingParen) {
- Tok = Tok->MatchingParen;
- continue;
- }
- if (Tok->is(tok::kw_const) || Tok->isSimpleTypeSpecifier() ||
- Tok->isOneOf(TT_PointerOrReference, TT_StartOfName, tok::ellipsis))
- return true;
- if (Tok->isOneOf(tok::l_brace, tok::string_literal, TT_ObjCMethodExpr) ||
- Tok->Tok.isLiteral())
- return false;
- }
- return false;
- }
- bool TokenAnnotator::mustBreakForReturnType(const AnnotatedLine &Line) const {
- assert(Line.MightBeFunctionDecl);
- if ((Style.AlwaysBreakAfterReturnType == FormatStyle::RTBS_TopLevel ||
- Style.AlwaysBreakAfterReturnType ==
- FormatStyle::RTBS_TopLevelDefinitions) &&
- Line.Level > 0)
- return false;
- switch (Style.AlwaysBreakAfterReturnType) {
- case FormatStyle::RTBS_None:
- return false;
- case FormatStyle::RTBS_All:
- case FormatStyle::RTBS_TopLevel:
- return true;
- case FormatStyle::RTBS_AllDefinitions:
- case FormatStyle::RTBS_TopLevelDefinitions:
- return Line.mightBeFunctionDefinition();
- }
- return false;
- }
- void TokenAnnotator::calculateFormattingInformation(AnnotatedLine &Line) {
- for (SmallVectorImpl<AnnotatedLine *>::iterator I = Line.Children.begin(),
- E = Line.Children.end();
- I != E; ++I) {
- calculateFormattingInformation(**I);
- }
- Line.First->TotalLength =
- Line.First->IsMultiline ? Style.ColumnLimit
- : Line.FirstStartColumn + Line.First->ColumnWidth;
- FormatToken *Current = Line.First->Next;
- bool InFunctionDecl = Line.MightBeFunctionDecl;
- while (Current) {
- if (isFunctionDeclarationName(*Current, Line))
- Current->Type = TT_FunctionDeclarationName;
- if (Current->is(TT_LineComment)) {
- if (Current->Previous->BlockKind == BK_BracedInit &&
- Current->Previous->opensScope())
- Current->SpacesRequiredBefore = Style.Cpp11BracedListStyle ? 0 : 1;
- else
- Current->SpacesRequiredBefore = Style.SpacesBeforeTrailingComments;
- // If we find a trailing comment, iterate backwards to determine whether
- // it seems to relate to a specific parameter. If so, break before that
- // parameter to avoid changing the comment's meaning. E.g. don't move 'b'
- // to the previous line in:
- // SomeFunction(a,
- // b, // comment
- // c);
- if (!Current->HasUnescapedNewline) {
- for (FormatToken *Parameter = Current->Previous; Parameter;
- Parameter = Parameter->Previous) {
- if (Parameter->isOneOf(tok::comment, tok::r_brace))
- break;
- if (Parameter->Previous && Parameter->Previous->is(tok::comma)) {
- if (!Parameter->Previous->is(TT_CtorInitializerComma) &&
- Parameter->HasUnescapedNewline)
- Parameter->MustBreakBefore = true;
- break;
- }
- }
- }
- } else if (Current->SpacesRequiredBefore == 0 &&
- spaceRequiredBefore(Line, *Current)) {
- Current->SpacesRequiredBefore = 1;
- }
- Current->MustBreakBefore =
- Current->MustBreakBefore || mustBreakBefore(Line, *Current);
- if (!Current->MustBreakBefore && InFunctionDecl &&
- Current->is(TT_FunctionDeclarationName))
- Current->MustBreakBefore = mustBreakForReturnType(Line);
- Current->CanBreakBefore =
- Current->MustBreakBefore || canBreakBefore(Line, *Current);
- unsigned ChildSize = 0;
- if (Current->Previous->Children.size() == 1) {
- FormatToken &LastOfChild = *Current->Previous->Children[0]->Last;
- ChildSize = LastOfChild.isTrailingComment() ? Style.ColumnLimit
- : LastOfChild.TotalLength + 1;
- }
- const FormatToken *Prev = Current->Previous;
- if (Current->MustBreakBefore || Prev->Children.size() > 1 ||
- (Prev->Children.size() == 1 &&
- Prev->Children[0]->First->MustBreakBefore) ||
- Current->IsMultiline)
- Current->TotalLength = Prev->TotalLength + Style.ColumnLimit;
- else
- Current->TotalLength = Prev->TotalLength + Current->ColumnWidth +
- ChildSize + Current->SpacesRequiredBefore;
- if (Current->is(TT_CtorInitializerColon))
- InFunctionDecl = false;
- // FIXME: Only calculate this if CanBreakBefore is true once static
- // initializers etc. are sorted out.
- // FIXME: Move magic numbers to a better place.
- Current->SplitPenalty = 20 * Current->BindingStrength +
- splitPenalty(Line, *Current, InFunctionDecl);
- Current = Current->Next;
- }
- calculateUnbreakableTailLengths(Line);
- unsigned IndentLevel = Line.Level;
- for (Current = Line.First; Current != nullptr; Current = Current->Next) {
- if (Current->Role)
- Current->Role->precomputeFormattingInfos(Current);
- if (Current->MatchingParen &&
- Current->MatchingParen->opensBlockOrBlockTypeList(Style)) {
- assert(IndentLevel > 0);
- --IndentLevel;
- }
- Current->IndentLevel = IndentLevel;
- if (Current->opensBlockOrBlockTypeList(Style))
- ++IndentLevel;
- }
- DEBUG({ printDebugInfo(Line); });
- }
- void TokenAnnotator::calculateUnbreakableTailLengths(AnnotatedLine &Line) {
- unsigned UnbreakableTailLength = 0;
- FormatToken *Current = Line.Last;
- while (Current) {
- Current->UnbreakableTailLength = UnbreakableTailLength;
- if (Current->CanBreakBefore ||
- Current->isOneOf(tok::comment, tok::string_literal)) {
- UnbreakableTailLength = 0;
- } else {
- UnbreakableTailLength +=
- Current->ColumnWidth + Current->SpacesRequiredBefore;
- }
- Current = Current->Previous;
- }
- }
- unsigned TokenAnnotator::splitPenalty(const AnnotatedLine &Line,
- const FormatToken &Tok,
- bool InFunctionDecl) {
- const FormatToken &Left = *Tok.Previous;
- const FormatToken &Right = Tok;
- if (Left.is(tok::semi))
- return 0;
- if (Style.Language == FormatStyle::LK_Java) {
- if (Right.isOneOf(Keywords.kw_extends, Keywords.kw_throws))
- return 1;
- if (Right.is(Keywords.kw_implements))
- return 2;
- if (Left.is(tok::comma) && Left.NestingLevel == 0)
- return 3;
- } else if (Style.Language == FormatStyle::LK_JavaScript) {
- if (Right.is(Keywords.kw_function) && Left.isNot(tok::comma))
- return 100;
- if (Left.is(TT_JsTypeColon))
- return 35;
- if ((Left.is(TT_TemplateString) && Left.TokenText.endswith("${")) ||
- (Right.is(TT_TemplateString) && Right.TokenText.startswith("}")))
- return 100;
- // Prefer breaking call chains (".foo") over empty "{}", "[]" or "()".
- if (Left.opensScope() && Right.closesScope())
- return 200;
- }
- if (Right.is(tok::identifier) && Right.Next && Right.Next->is(TT_DictLiteral))
- return 1;
- if (Right.is(tok::l_square)) {
- if (Style.Language == FormatStyle::LK_Proto)
- return 1;
- if (Left.is(tok::r_square))
- return 200;
- // Slightly prefer formatting local lambda definitions like functions.
- if (Right.is(TT_LambdaLSquare) && Left.is(tok::equal))
- return 35;
- if (!Right.isOneOf(TT_ObjCMethodExpr, TT_LambdaLSquare,
- TT_ArrayInitializerLSquare,
- TT_DesignatedInitializerLSquare))
- return 500;
- }
- if (Right.isOneOf(TT_StartOfName, TT_FunctionDeclarationName) ||
- Right.is(tok::kw_operator)) {
- if (Line.startsWith(tok::kw_for) && Right.PartOfMultiVariableDeclStmt)
- return 3;
- if (Left.is(TT_StartOfName))
- return 110;
- if (InFunctionDecl && Right.NestingLevel == 0)
- return Style.PenaltyReturnTypeOnItsOwnLine;
- return 200;
- }
- if (Right.is(TT_PointerOrReference))
- return 190;
- if (Right.is(TT_LambdaArrow))
- return 110;
- if (Left.is(tok::equal) && Right.is(tok::l_brace))
- return 160;
- if (Left.is(TT_CastRParen))
- return 100;
- if (Left.is(tok::coloncolon) ||
- (Right.is(tok::period) && Style.Language == FormatStyle::LK_Proto))
- return 500;
- if (Left.isOneOf(tok::kw_class, tok::kw_struct))
- return 5000;
- if (Left.is(tok::comment))
- return 1000;
- if (Left.isOneOf(TT_RangeBasedForLoopColon, TT_InheritanceColon,
- TT_CtorInitializerColon))
- return 2;
- if (Right.isMemberAccess()) {
- // Breaking before the "./->" of a chained call/member access is reasonably
- // cheap, as formatting those with one call per line is generally
- // desirable. In particular, it should be cheaper to break before the call
- // than it is to break inside a call's parameters, which could lead to weird
- // "hanging" indents. The exception is the very last "./->" to support this
- // frequent pattern:
- //
- // aaaaaaaa.aaaaaaaa.bbbbbbb().ccccccccccccccccccccc(
- // dddddddd);
- //
- // which might otherwise be blown up onto many lines. Here, clang-format
- // won't produce "hanging" indents anyway as there is no other trailing
- // call.
- //
- // Also apply higher penalty is not a call as that might lead to a wrapping
- // like:
- //
- // aaaaaaa
- // .aaaaaaaaa.bbbbbbbb(cccccccc);
- return !Right.NextOperator || !Right.NextOperator->Previous->closesScope()
- ? 150
- : 35;
- }
- if (Right.is(TT_TrailingAnnotation) &&
- (!Right.Next || Right.Next->isNot(tok::l_paren))) {
- // Moving trailing annotations to the next line is fine for ObjC method
- // declarations.
- if (Line.startsWith(TT_ObjCMethodSpecifier))
- return 10;
- // Generally, breaking before a trailing annotation is bad unless it is
- // function-like. It seems to be especially preferable to keep standard
- // annotations (i.e. "const", "final" and "override") on the same line.
- // Use a slightly higher penalty after ")" so that annotations like
- // "const override" are kept together.
- bool is_short_annotation = Right.TokenText.size() < 10;
- return (Left.is(tok::r_paren) ? 100 : 120) + (is_short_annotation ? 50 : 0);
- }
- // In for-loops, prefer breaking at ',' and ';'.
- if (Line.startsWith(tok::kw_for) && Left.is(tok::equal))
- return 4;
- // In Objective-C method expressions, prefer breaking before "param:" over
- // breaking after it.
- if (Right.is(TT_SelectorName))
- return 0;
- if (Left.is(tok::colon) && Left.is(TT_ObjCMethodExpr))
- return Line.MightBeFunctionDecl ? 50 : 500;
- if (Left.is(tok::l_paren) && InFunctionDecl &&
- Style.AlignAfterOpenBracket != FormatStyle::BAS_DontAlign)
- return 100;
- if (Left.is(tok::l_paren) && Left.Previous &&
- (Left.Previous->isOneOf(tok::kw_if, tok::kw_for) ||
- Left.Previous->endsSequence(tok::kw_constexpr, tok::kw_if)))
- return 1000;
- if (Left.is(tok::equal) && InFunctionDecl)
- return 110;
- if (Right.is(tok::r_brace))
- return 1;
- if (Left.is(TT_TemplateOpener))
- return 100;
- if (Left.opensScope()) {
- if (Style.AlignAfterOpenBracket == FormatStyle::BAS_DontAlign)
- return 0;
- return Left.ParameterCount > 1 ? Style.PenaltyBreakBeforeFirstCallParameter
- : 19;
- }
- if (Left.is(TT_JavaAnnotation))
- return 50;
- if (Left.isOneOf(tok::plus, tok::comma) && Left.Previous &&
- Left.Previous->isLabelString() &&
- (Left.NextOperator || Left.OperatorIndex != 0))
- return 45;
- if (Right.is(tok::plus) && Left.isLabelString() &&
- (Right.NextOperator || Right.OperatorIndex != 0))
- return 25;
- if (Left.is(tok::comma))
- return 1;
- if (Right.is(tok::lessless) && Left.isLabelString() &&
- (Right.NextOperator || Right.OperatorIndex != 1))
- return 25;
- if (Right.is(tok::lessless)) {
- // Breaking at a << is really cheap.
- if (!Left.is(tok::r_paren) || Right.OperatorIndex > 0)
- // Slightly prefer to break before the first one in log-like statements.
- return 2;
- return 1;
- }
- if (Left.is(TT_ConditionalExpr))
- return prec::Conditional;
- prec::Level Level = Left.getPrecedence();
- if (Level == prec::Unknown)
- Level = Right.getPrecedence();
- if (Level == prec::Assignment)
- return Style.PenaltyBreakAssignment;
- if (Level != prec::Unknown)
- return Level;
- return 3;
- }
- bool TokenAnnotator::spaceRequiredBetween(const AnnotatedLine &Line,
- const FormatToken &Left,
- const FormatToken &Right) {
- if (Left.is(tok::kw_return) && Right.isNot(tok::semi))
- return true;
- if (Left.is(Keywords.kw_assert) && Style.Language == FormatStyle::LK_Java)
- return true;
- if (Style.ObjCSpaceAfterProperty && Line.Type == LT_ObjCProperty &&
- Left.Tok.getObjCKeywordID() == tok::objc_property)
- return true;
- if (Right.is(tok::hashhash))
- return Left.is(tok::hash);
- if (Left.isOneOf(tok::hashhash, tok::hash))
- return Right.is(tok::hash);
- if (Left.is(tok::l_paren) && Right.is(tok::r_paren))
- return Style.SpaceInEmptyParentheses;
- if (Left.is(tok::l_paren) || Right.is(tok::r_paren))
- return (Right.is(TT_CastRParen) ||
- (Left.MatchingParen && Left.MatchingParen->is(TT_CastRParen)))
- ? Style.SpacesInCStyleCastParentheses
- : Style.SpacesInParentheses;
- if (Right.isOneOf(tok::semi, tok::comma))
- return false;
- if (Right.is(tok::less) && Line.Type == LT_ObjCDecl &&
- Style.ObjCSpaceBeforeProtocolList)
- return true;
- if (Right.is(tok::less) && Left.is(tok::kw_template))
- return Style.SpaceAfterTemplateKeyword;
- if (Left.isOneOf(tok::exclaim, tok::tilde))
- return false;
- if (Left.is(tok::at) &&
- Right.isOneOf(tok::identifier, tok::string_literal, tok::char_constant,
- tok::numeric_constant, tok::l_paren, tok::l_brace,
- tok::kw_true, tok::kw_false))
- return false;
- if (Left.is(tok::colon))
- return !Left.is(TT_ObjCMethodExpr);
- if (Left.is(tok::coloncolon))
- return false;
- if (Left.is(tok::less) || Right.isOneOf(tok::greater, tok::less))
- return false;
- if (Right.is(tok::ellipsis))
- return Left.Tok.isLiteral() || (Left.is(tok::identifier) && Left.Previous &&
- Left.Previous->is(tok::kw_case));
- if (Left.is(tok::l_square) && Right.is(tok::amp))
- return false;
- if (Right.is(TT_PointerOrReference)) {
- if (Left.is(tok::r_paren) && Line.MightBeFunctionDecl) {
- if (!Left.MatchingParen)
- return true;
- FormatToken *TokenBeforeMatchingParen =
- Left.MatchingParen->getPreviousNonComment();
- if (!TokenBeforeMatchingParen ||
- !TokenBeforeMatchingParen->isOneOf(tok::kw_typeof, tok::kw_decltype))
- return true;
- }
- return (Left.Tok.isLiteral() ||
- (!Left.isOneOf(TT_PointerOrReference, tok::l_paren) &&
- (Style.PointerAlignment != FormatStyle::PAS_Left ||
- (Line.IsMultiVariableDeclStmt &&
- (Left.NestingLevel == 0 ||
- (Left.NestingLevel == 1 && Line.First->is(tok::kw_for)))))));
- }
- if (Right.is(TT_FunctionTypeLParen) && Left.isNot(tok::l_paren) &&
- (!Left.is(TT_PointerOrReference) ||
- (Style.PointerAlignment != FormatStyle::PAS_Right &&
- !Line.IsMultiVariableDeclStmt)))
- return true;
- if (Left.is(TT_PointerOrReference))
- return Right.Tok.isLiteral() || Right.is(TT_BlockComment) ||
- (Right.isOneOf(Keywords.kw_override, Keywords.kw_final) &&
- !Right.is(TT_StartOfName)) ||
- (Right.is(tok::l_brace) && Right.BlockKind == BK_Block) ||
- (!Right.isOneOf(TT_PointerOrReference, TT_ArraySubscriptLSquare,
- tok::l_paren) &&
- (Style.PointerAlignment != FormatStyle::PAS_Right &&
- !Line.IsMultiVariableDeclStmt) &&
- Left.Previous &&
- !Left.Previous->isOneOf(tok::l_paren, tok::coloncolon));
- if (Right.is(tok::star) && Left.is(tok::l_paren))
- return false;
- if (Left.is(tok::l_square))
- return (Left.is(TT_ArrayInitializerLSquare) &&
- Style.SpacesInContainerLiterals && Right.isNot(tok::r_square)) ||
- (Left.isOneOf(TT_ArraySubscriptLSquare,
- TT_StructuredBindingLSquare) &&
- Style.SpacesInSquareBrackets && Right.isNot(tok::r_square));
- if (Right.is(tok::r_square))
- return Right.MatchingParen &&
- ((Style.SpacesInContainerLiterals &&
- Right.MatchingParen->is(TT_ArrayInitializerLSquare)) ||
- (Style.SpacesInSquareBrackets &&
- Right.MatchingParen->isOneOf(TT_ArraySubscriptLSquare,
- TT_StructuredBindingLSquare)));
- if (Right.is(tok::l_square) &&
- !Right.isOneOf(TT_ObjCMethodExpr, TT_LambdaLSquare,
- TT_DesignatedInitializerLSquare,
- TT_StructuredBindingLSquare) &&
- !Left.isOneOf(tok::numeric_constant, TT_DictLiteral))
- return false;
- if (Left.is(tok::l_brace) && Right.is(tok::r_brace))
- return !Left.Children.empty(); // No spaces in "{}".
- if ((Left.is(tok::l_brace) && Left.BlockKind != BK_Block) ||
- (Right.is(tok::r_brace) && Right.MatchingParen &&
- Right.MatchingParen->BlockKind != BK_Block))
- return !Style.Cpp11BracedListStyle;
- if (Left.is(TT_BlockComment))
- return !Left.TokenText.endswith("=*/");
- if (Right.is(tok::l_paren)) {
- if (Left.is(tok::r_paren) && Left.is(TT_AttributeParen))
- return true;
- return Line.Type == LT_ObjCDecl || Left.is(tok::semi) ||
- (Style.SpaceBeforeParens != FormatStyle::SBPO_Never &&
- (Left.isOneOf(tok::kw_if, tok::pp_elif, tok::kw_for, tok::kw_while,
- tok::kw_switch, tok::kw_case, TT_ForEachMacro,
- TT_ObjCForIn) ||
- Left.endsSequence(tok::kw_constexpr, tok::kw_if) ||
- (Left.isOneOf(tok::kw_try, Keywords.kw___except, tok::kw_catch,
- tok::kw_new, tok::kw_delete) &&
- (!Left.Previous || Left.Previous->isNot(tok::period))))) ||
- (Style.SpaceBeforeParens == FormatStyle::SBPO_Always &&
- (Left.is(tok::identifier) || Left.isFunctionLikeKeyword() ||
- Left.is(tok::r_paren)) &&
- Line.Type != LT_PreprocessorDirective);
- }
- if (Left.is(tok::at) && Right.Tok.getObjCKeywordID() != tok::objc_not_keyword)
- return false;
- if (Right.is(TT_UnaryOperator))
- return !Left.isOneOf(tok::l_paren, tok::l_square, tok::at) &&
- (Left.isNot(tok::colon) || Left.isNot(TT_ObjCMethodExpr));
- if ((Left.isOneOf(tok::identifier, tok::greater, tok::r_square,
- tok::r_paren) ||
- Left.isSimpleTypeSpecifier()) &&
- Right.is(tok::l_brace) && Right.getNextNonComment() &&
- Right.BlockKind != BK_Block)
- return false;
- if (Left.is(tok::period) || Right.is(tok::period))
- return false;
- if (Right.is(tok::hash) && Left.is(tok::identifier) && Left.TokenText == "L")
- return false;
- if (Left.is(TT_TemplateCloser) && Left.MatchingParen &&
- Left.MatchingParen->Previous &&
- Left.MatchingParen->Previous->is(tok::period))
- // A.<B<C<...>>>DoSomething();
- return false;
- if (Left.is(TT_TemplateCloser) && Right.is(tok::l_square))
- return false;
- return true;
- }
- bool TokenAnnotator::spaceRequiredBefore(const AnnotatedLine &Line,
- const FormatToken &Right) {
- const FormatToken &Left = *Right.Previous;
- if (Right.Tok.getIdentifierInfo() && Left.Tok.getIdentifierInfo())
- return true; // Never ever merge two identifiers.
- if (Style.isCpp()) {
- if (Left.is(tok::kw_operator))
- return Right.is(tok::coloncolon);
- } else if (Style.Language == FormatStyle::LK_Proto ||
- Style.Language == FormatStyle::LK_TextProto) {
- if (Right.is(tok::period) &&
- Left.isOneOf(Keywords.kw_optional, Keywords.kw_required,
- Keywords.kw_repeated, Keywords.kw_extend))
- return true;
- if (Right.is(tok::l_paren) &&
- Left.isOneOf(Keywords.kw_returns, Keywords.kw_option))
- return true;
- if (Right.isOneOf(tok::l_brace, tok::less) && Left.is(TT_SelectorName))
- return true;
- } else if (Style.Language == FormatStyle::LK_JavaScript) {
- if (Left.is(TT_JsFatArrow))
- return true;
- // for await ( ...
- if (Right.is(tok::l_paren) && Left.is(Keywords.kw_await) && Left.Previous &&
- Left.Previous->is(tok::kw_for))
- return true;
- if (Left.is(Keywords.kw_async) && Right.is(tok::l_paren) &&
- Right.MatchingParen) {
- const FormatToken *Next = Right.MatchingParen->getNextNonComment();
- // An async arrow function, for example: `x = async () => foo();`,
- // as opposed to calling a function called async: `x = async();`
- if (Next && Next->is(TT_JsFatArrow))
- return true;
- }
- if ((Left.is(TT_TemplateString) && Left.TokenText.endswith("${")) ||
- (Right.is(TT_TemplateString) && Right.TokenText.startswith("}")))
- return false;
- // In tagged template literals ("html`bar baz`"), there is no space between
- // the tag identifier and the template string. getIdentifierInfo makes sure
- // that the identifier is not a pseudo keyword like `yield`, either.
- if (Left.is(tok::identifier) && Keywords.IsJavaScriptIdentifier(Left) &&
- Right.is(TT_TemplateString))
- return false;
- if (Right.is(tok::star) &&
- Left.isOneOf(Keywords.kw_function, Keywords.kw_yield))
- return false;
- if (Right.isOneOf(tok::l_brace, tok::l_square) &&
- Left.isOneOf(Keywords.kw_function, Keywords.kw_yield,
- Keywords.kw_extends, Keywords.kw_implements))
- return true;
- if (Right.is(tok::l_paren)) {
- // JS methods can use some keywords as names (e.g. `delete()`).
- if (Line.MustBeDeclaration && Left.Tok.getIdentifierInfo())
- return false;
- // Valid JS method names can include keywords, e.g. `foo.delete()` or
- // `bar.instanceof()`. Recognize call positions by preceding period.
- if (Left.Previous && Left.Previous->is(tok::period) &&
- Left.Tok.getIdentifierInfo())
- return false;
- // Additional unary JavaScript operators that need a space after.
- if (Left.isOneOf(tok::kw_throw, Keywords.kw_await, Keywords.kw_typeof,
- tok::kw_void))
- return true;
- }
- if ((Left.isOneOf(Keywords.kw_let, Keywords.kw_var, Keywords.kw_in,
- tok::kw_const) ||
- // "of" is only a keyword if it appears after another identifier
- // (e.g. as "const x of y" in a for loop).
- (Left.is(Keywords.kw_of) && Left.Previous &&
- Left.Previous->Tok.getIdentifierInfo())) &&
- (!Left.Previous || !Left.Previous->is(tok::period)))
- return true;
- if (Left.isOneOf(tok::kw_for, Keywords.kw_as) && Left.Previous &&
- Left.Previous->is(tok::period) && Right.is(tok::l_paren))
- return false;
- if (Left.is(Keywords.kw_as) &&
- Right.isOneOf(tok::l_square, tok::l_brace, tok::l_paren))
- return true;
- if (Left.is(tok::kw_default) && Left.Previous &&
- Left.Previous->is(tok::kw_export))
- return true;
- if (Left.is(Keywords.kw_is) && Right.is(tok::l_brace))
- return true;
- if (Right.isOneOf(TT_JsTypeColon, TT_JsTypeOptionalQuestion))
- return false;
- if (Left.is(TT_JsTypeOperator) || Right.is(TT_JsTypeOperator))
- return false;
- if ((Left.is(tok::l_brace) || Right.is(tok::r_brace)) &&
- Line.First->isOneOf(Keywords.kw_import, tok::kw_export))
- return false;
- if (Left.is(tok::ellipsis))
- return false;
- if (Left.is(TT_TemplateCloser) &&
- !Right.isOneOf(tok::equal, tok::l_brace, tok::comma, tok::l_square,
- Keywords.kw_implements, Keywords.kw_extends))
- // Type assertions ('<type>expr') are not followed by whitespace. Other
- // locations that should have whitespace following are identified by the
- // above set of follower tokens.
- return false;
- if (Right.is(TT_JsNonNullAssertion))
- return false;
- if (Left.is(TT_JsNonNullAssertion) && Right.is(Keywords.kw_as))
- return true; // "x! as string"
- } else if (Style.Language == FormatStyle::LK_Java) {
- if (Left.is(tok::r_square) && Right.is(tok::l_brace))
- return true;
- if (Left.is(Keywords.kw_synchronized) && Right.is(tok::l_paren))
- return Style.SpaceBeforeParens != FormatStyle::SBPO_Never;
- if ((Left.isOneOf(tok::kw_static, tok::kw_public, tok::kw_private,
- tok::kw_protected) ||
- Left.isOneOf(Keywords.kw_final, Keywords.kw_abstract,
- Keywords.kw_native)) &&
- Right.is(TT_TemplateOpener))
- return true;
- }
- if (Left.is(TT_ImplicitStringLiteral))
- return Right.WhitespaceRange.getBegin() != Right.WhitespaceRange.getEnd();
- if (Line.Type == LT_ObjCMethodDecl) {
- if (Left.is(TT_ObjCMethodSpecifier))
- return true;
- if (Left.is(tok::r_paren) && Right.is(tok::identifier))
- // Don't space between ')' and <id>
- return false;
- }
- if (Line.Type == LT_ObjCProperty &&
- (Right.is(tok::equal) || Left.is(tok::equal)))
- return false;
- if (Right.isOneOf(TT_TrailingReturnArrow, TT_LambdaArrow) ||
- Left.isOneOf(TT_TrailingReturnArrow, TT_LambdaArrow))
- return true;
- if (Right.is(TT_OverloadedOperatorLParen))
- return Style.SpaceBeforeParens == FormatStyle::SBPO_Always;
- if (Left.is(tok::comma))
- return true;
- if (Right.is(tok::comma))
- return false;
- if (Right.isOneOf(TT_CtorInitializerColon, TT_ObjCBlockLParen))
- return true;
- if (Right.is(tok::colon)) {
- if (Line.First->isOneOf(tok::kw_case, tok::kw_default) ||
- !Right.getNextNonComment() || Right.getNextNonComment()->is(tok::semi))
- return false;
- if (Right.is(TT_ObjCMethodExpr))
- return false;
- if (Left.is(tok::question))
- return false;
- if (Right.is(TT_InlineASMColon) && Left.is(tok::coloncolon))
- return false;
- if (Right.is(TT_DictLiteral))
- return Style.SpacesInContainerLiterals;
- return true;
- }
- if (Left.is(TT_UnaryOperator))
- return Right.is(TT_BinaryOperator);
- // If the next token is a binary operator or a selector name, we have
- // incorrectly classified the parenthesis as a cast. FIXME: Detect correctly.
- if (Left.is(TT_CastRParen))
- return Style.SpaceAfterCStyleCast ||
- Right.isOneOf(TT_BinaryOperator, TT_SelectorName);
- if (Left.is(tok::greater) && Right.is(tok::greater))
- return Right.is(TT_TemplateCloser) && Left.is(TT_TemplateCloser) &&
- (Style.Standard != FormatStyle::LS_Cpp11 || Style.SpacesInAngles);
- if (Right.isOneOf(tok::arrow, tok::arrowstar, tok::periodstar) ||
- Left.isOneOf(tok::arrow, tok::period, tok::arrowstar, tok::periodstar) ||
- (Right.is(tok::period) && Right.isNot(TT_DesignatedInitializerPeriod)))
- return false;
- if (!Style.SpaceBeforeAssignmentOperators &&
- Right.getPrecedence() == prec::Assignment)
- return false;
- if (Right.is(tok::coloncolon) && Left.is(tok::identifier))
- // Generally don't remove existing spaces between an identifier and "::".
- // The identifier might actually be a macro name such as ALWAYS_INLINE. If
- // this turns out to be too lenient, add analysis of the identifier itself.
- return Right.WhitespaceRange.getBegin() != Right.WhitespaceRange.getEnd();
- if (Right.is(tok::coloncolon) && !Left.isOneOf(tok::l_brace, tok::comment))
- return (Left.is(TT_TemplateOpener) &&
- Style.Standard == FormatStyle::LS_Cpp03) ||
- !(Left.isOneOf(tok::l_paren, tok::r_paren, tok::l_square,
- tok::kw___super, TT_TemplateCloser,
- TT_TemplateOpener));
- if ((Left.is(TT_TemplateOpener)) != (Right.is(TT_TemplateCloser)))
- return Style.SpacesInAngles;
- // Space before TT_StructuredBindingLSquare.
- if (Right.is(TT_StructuredBindingLSquare))
- return !Left.isOneOf(tok::amp, tok::ampamp) ||
- Style.PointerAlignment != FormatStyle::PAS_Right;
- // Space before & or && following a TT_StructuredBindingLSquare.
- if (Right.Next && Right.Next->is(TT_StructuredBindingLSquare) &&
- Right.isOneOf(tok::amp, tok::ampamp))
- return Style.PointerAlignment != FormatStyle::PAS_Left;
- if ((Right.is(TT_BinaryOperator) && !Left.is(tok::l_paren)) ||
- (Left.isOneOf(TT_BinaryOperator, TT_ConditionalExpr) &&
- !Right.is(tok::r_paren)))
- return true;
- if (Left.is(TT_TemplateCloser) && Right.is(tok::l_paren) &&
- Right.isNot(TT_FunctionTypeLParen))
- return Style.SpaceBeforeParens == FormatStyle::SBPO_Always;
- if (Right.is(TT_TemplateOpener) && Left.is(tok::r_paren) &&
- Left.MatchingParen && Left.MatchingParen->is(TT_OverloadedOperatorLParen))
- return false;
- if (Right.is(tok::less) && Left.isNot(tok::l_paren) &&
- Line.startsWith(tok::hash))
- return true;
- if (Right.is(TT_TrailingUnaryOperator))
- return false;
- if (Left.is(TT_RegexLiteral))
- return false;
- return spaceRequiredBetween(Line, Left, Right);
- }
- // Returns 'true' if 'Tok' is a brace we'd want to break before in Allman style.
- static bool isAllmanBrace(const FormatToken &Tok) {
- return Tok.is(tok::l_brace) && Tok.BlockKind == BK_Block &&
- !Tok.isOneOf(TT_ObjCBlockLBrace, TT_DictLiteral);
- }
- bool TokenAnnotator::mustBreakBefore(const AnnotatedLine &Line,
- const FormatToken &Right) {
- const FormatToken &Left = *Right.Previous;
- if (Right.NewlinesBefore > 1 && Style.MaxEmptyLinesToKeep > 0)
- return true;
- if (Style.Language == FormatStyle::LK_JavaScript) {
- // FIXME: This might apply to other languages and token kinds.
- if (Right.is(tok::string_literal) && Left.is(tok::plus) && Left.Previous &&
- Left.Previous->is(tok::string_literal))
- return true;
- if (Left.is(TT_DictLiteral) && Left.is(tok::l_brace) && Line.Level == 0 &&
- Left.Previous && Left.Previous->is(tok::equal) &&
- Line.First->isOneOf(tok::identifier, Keywords.kw_import, tok::kw_export,
- tok::kw_const) &&
- // kw_var/kw_let are pseudo-tokens that are tok::identifier, so match
- // above.
- !Line.First->isOneOf(Keywords.kw_var, Keywords.kw_let))
- // Object literals on the top level of a file are treated as "enum-style".
- // Each key/value pair is put on a separate line, instead of bin-packing.
- return true;
- if (Left.is(tok::l_brace) && Line.Level == 0 &&
- (Line.startsWith(tok::kw_enum) ||
- Line.startsWith(tok::kw_const, tok::kw_enum) ||
- Line.startsWith(tok::kw_export, tok::kw_enum) ||
- Line.startsWith(tok::kw_export, tok::kw_const, tok::kw_enum)))
- // JavaScript top-level enum key/value pairs are put on separate lines
- // instead of bin-packing.
- return true;
- if (Right.is(tok::r_brace) && Left.is(tok::l_brace) &&
- !Left.Children.empty())
- // Support AllowShortFunctionsOnASingleLine for JavaScript.
- return Style.AllowShortFunctionsOnASingleLine == FormatStyle::SFS_None ||
- Style.AllowShortFunctionsOnASingleLine == FormatStyle::SFS_Empty ||
- (Left.NestingLevel == 0 && Line.Level == 0 &&
- Style.AllowShortFunctionsOnASingleLine &
- FormatStyle::SFS_InlineOnly);
- } else if (Style.Language == FormatStyle::LK_Java) {
- if (Right.is(tok::plus) && Left.is(tok::string_literal) && Right.Next &&
- Right.Next->is(tok::string_literal))
- return true;
- } else if (Style.Language == FormatStyle::LK_Cpp ||
- Style.Language == FormatStyle::LK_ObjC ||
- Style.Language == FormatStyle::LK_Proto) {
- if (Left.isStringLiteral() && Right.isStringLiteral())
- return true;
- }
- // If the last token before a '}', ']', or ')' is a comma or a trailing
- // comment, the intention is to insert a line break after it in order to make
- // shuffling around entries easier. Import statements, especially in
- // JavaScript, can be an exception to this rule.
- if (Style.JavaScriptWrapImports || Line.Type != LT_ImportStatement) {
- const FormatToken *BeforeClosingBrace = nullptr;
- if ((Left.isOneOf(tok::l_brace, TT_ArrayInitializerLSquare) ||
- (Style.Language == FormatStyle::LK_JavaScript &&
- Left.is(tok::l_paren))) &&
- Left.BlockKind != BK_Block && Left.MatchingParen)
- BeforeClosingBrace = Left.MatchingParen->Previous;
- else if (Right.MatchingParen &&
- (Right.MatchingParen->isOneOf(tok::l_brace,
- TT_ArrayInitializerLSquare) ||
- (Style.Language == FormatStyle::LK_JavaScript &&
- Right.MatchingParen->is(tok::l_paren))))
- BeforeClosingBrace = &Left;
- if (BeforeClosingBrace && (BeforeClosingBrace->is(tok::comma) ||
- BeforeClosingBrace->isTrailingComment()))
- return true;
- }
- if (Right.is(tok::comment))
- return Left.BlockKind != BK_BracedInit &&
- Left.isNot(TT_CtorInitializerColon) &&
- (Right.NewlinesBefore > 0 && Right.HasUnescapedNewline);
- if (Left.isTrailingComment())
- return true;
- if (Right.Previous->IsUnterminatedLiteral)
- return true;
- if (Right.is(tok::lessless) && Right.Next &&
- Right.Previous->is(tok::string_literal) &&
- Right.Next->is(tok::string_literal))
- return true;
- if (Right.Previous->ClosesTemplateDeclaration &&
- Right.Previous->MatchingParen &&
- Right.Previous->MatchingParen->NestingLevel == 0 &&
- Style.AlwaysBreakTemplateDeclarations)
- return true;
- if (Right.is(TT_CtorInitializerComma) &&
- Style.BreakConstructorInitializers == FormatStyle::BCIS_BeforeComma &&
- !Style.ConstructorInitializerAllOnOneLineOrOnePerLine)
- return true;
- if (Right.is(TT_CtorInitializerColon) &&
- Style.BreakConstructorInitializers == FormatStyle::BCIS_BeforeComma &&
- !Style.ConstructorInitializerAllOnOneLineOrOnePerLine)
- return true;
- // Break only if we have multiple inheritance.
- if (Style.BreakBeforeInheritanceComma && Right.is(TT_InheritanceComma))
- return true;
- if (Right.is(tok::string_literal) && Right.TokenText.startswith("R\""))
- // Raw string literals are special wrt. line breaks. The author has made a
- // deliberate choice and might have aligned the contents of the string
- // literal accordingly. Thus, we try keep existing line breaks.
- return Right.NewlinesBefore > 0;
- if ((Right.Previous->is(tok::l_brace) ||
- (Right.Previous->is(tok::less) && Right.Previous->Previous &&
- Right.Previous->Previous->is(tok::equal))) &&
- Right.NestingLevel == 1 && Style.Language == FormatStyle::LK_Proto) {
- // Don't put enums or option definitions onto single lines in protocol
- // buffers.
- return true;
- }
- if (Right.is(TT_InlineASMBrace))
- return Right.HasUnescapedNewline;
- if (isAllmanBrace(Left) || isAllmanBrace(Right))
- return (Line.startsWith(tok::kw_enum) && Style.BraceWrapping.AfterEnum) ||
- (Line.startsWith(tok::kw_typedef, tok::kw_enum) &&
- Style.BraceWrapping.AfterEnum) ||
- (Line.startsWith(tok::kw_class) && Style.BraceWrapping.AfterClass) ||
- (Line.startsWith(tok::kw_struct) && Style.BraceWrapping.AfterStruct);
- if (Left.is(TT_ObjCBlockLBrace) && !Style.AllowShortBlocksOnASingleLine)
- return true;
- if ((Style.Language == FormatStyle::LK_Java ||
- Style.Language == FormatStyle::LK_JavaScript) &&
- Left.is(TT_LeadingJavaAnnotation) &&
- Right.isNot(TT_LeadingJavaAnnotation) && Right.isNot(tok::l_paren) &&
- (Line.Last->is(tok::l_brace) || Style.BreakAfterJavaFieldAnnotations))
- return true;
- return false;
- }
- bool TokenAnnotator::canBreakBefore(const AnnotatedLine &Line,
- const FormatToken &Right) {
- const FormatToken &Left = *Right.Previous;
- // Language-specific stuff.
- if (Style.Language == FormatStyle::LK_Java) {
- if (Left.isOneOf(Keywords.kw_throws, Keywords.kw_extends,
- Keywords.kw_implements))
- return false;
- if (Right.isOneOf(Keywords.kw_throws, Keywords.kw_extends,
- Keywords.kw_implements))
- return true;
- } else if (Style.Language == FormatStyle::LK_JavaScript) {
- const FormatToken *NonComment = Right.getPreviousNonComment();
- if (NonComment &&
- NonComment->isOneOf(tok::kw_return, tok::kw_continue, tok::kw_break,
- tok::kw_throw, Keywords.kw_interface,
- Keywords.kw_type, tok::kw_static, tok::kw_public,
- tok::kw_private, tok::kw_protected,
- Keywords.kw_readonly, Keywords.kw_abstract,
- Keywords.kw_get, Keywords.kw_set))
- return false; // Otherwise automatic semicolon insertion would trigger.
- if (Left.is(TT_JsFatArrow) && Right.is(tok::l_brace))
- return false;
- if (Left.is(TT_JsTypeColon))
- return true;
- if (Right.NestingLevel == 0 && Right.is(Keywords.kw_is))
- return false;
- if (Left.is(Keywords.kw_in))
- return Style.BreakBeforeBinaryOperators == FormatStyle::BOS_None;
- if (Right.is(Keywords.kw_in))
- return Style.BreakBeforeBinaryOperators != FormatStyle::BOS_None;
- if (Right.is(Keywords.kw_as))
- return false; // must not break before as in 'x as type' casts
- if (Left.is(Keywords.kw_as))
- return true;
- if (Left.is(TT_JsNonNullAssertion))
- return true;
- if (Left.is(Keywords.kw_declare) &&
- Right.isOneOf(Keywords.kw_module, tok::kw_namespace,
- Keywords.kw_function, tok::kw_class, tok::kw_enum,
- Keywords.kw_interface, Keywords.kw_type, Keywords.kw_var,
- Keywords.kw_let, tok::kw_const))
- // See grammar for 'declare' statements at:
- // https://github.com/Microsoft/TypeScript/blob/master/doc/spec.md#A.10
- return false;
- if (Left.isOneOf(Keywords.kw_module, tok::kw_namespace) &&
- Right.isOneOf(tok::identifier, tok::string_literal))
- return false; // must not break in "module foo { ...}"
- if (Right.is(TT_TemplateString) && Right.closesScope())
- return false;
- if (Left.is(TT_TemplateString) && Left.opensScope())
- return true;
- }
- if (Left.is(tok::at))
- return false;
- if (Left.Tok.getObjCKeywordID() == tok::objc_interface)
- return false;
- if (Left.isOneOf(TT_JavaAnnotation, TT_LeadingJavaAnnotation))
- return !Right.is(tok::l_paren);
- if (Right.is(TT_PointerOrReference))
- return Line.IsMultiVariableDeclStmt ||
- (Style.PointerAlignment == FormatStyle::PAS_Right &&
- (!Right.Next || Right.Next->isNot(TT_FunctionDeclarationName)));
- if (Right.isOneOf(TT_StartOfName, TT_FunctionDeclarationName) ||
- Right.is(tok::kw_operator))
- return true;
- if (Left.is(TT_PointerOrReference))
- return false;
- if (Right.isTrailingComment())
- // We rely on MustBreakBefore being set correctly here as we should not
- // change the "binding" behavior of a comment.
- // The first comment in a braced lists is always interpreted as belonging to
- // the first list element. Otherwise, it should be placed outside of the
- // list.
- return Left.BlockKind == BK_BracedInit ||
- (Left.is(TT_CtorInitializerColon) &&
- Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon);
- if (Left.is(tok::question) && Right.is(tok::colon))
- return false;
- if (Right.is(TT_ConditionalExpr) || Right.is(tok::question))
- return Style.BreakBeforeTernaryOperators;
- if (Left.is(TT_ConditionalExpr) || Left.is(tok::question))
- return !Style.BreakBeforeTernaryOperators;
- if (Right.is(TT_InheritanceColon))
- return true;
- if (Right.is(TT_ObjCMethodExpr) && !Right.is(tok::r_square) &&
- Left.isNot(TT_SelectorName))
- return true;
- if (Right.is(tok::colon) &&
- !Right.isOneOf(TT_CtorInitializerColon, TT_InlineASMColon))
- return false;
- if (Left.is(tok::colon) && Left.isOneOf(TT_DictLiteral, TT_ObjCMethodExpr))
- return true;
- if (Right.is(TT_SelectorName) || (Right.is(tok::identifier) && Right.Next &&
- Right.Next->is(TT_ObjCMethodExpr)))
- return Left.isNot(tok::period); // FIXME: Properly parse ObjC calls.
- if (Left.is(tok::r_paren) && Line.Type == LT_ObjCProperty)
- return true;
- if (Left.ClosesTemplateDeclaration || Left.is(TT_FunctionAnnotationRParen))
- return true;
- if (Right.isOneOf(TT_RangeBasedForLoopColon, TT_OverloadedOperatorLParen,
- TT_OverloadedOperator))
- return false;
- if (Left.is(TT_RangeBasedForLoopColon))
- return true;
- if (Right.is(TT_RangeBasedForLoopColon))
- return false;
- if (Left.is(TT_TemplateCloser) && Right.is(TT_TemplateOpener))
- return true;
- if (Left.isOneOf(TT_TemplateCloser, TT_UnaryOperator) ||
- Left.is(tok::kw_operator))
- return false;
- if (Left.is(tok::equal) && !Right.isOneOf(tok::kw_default, tok::kw_delete) &&
- Line.Type == LT_VirtualFunctionDecl && Left.NestingLevel == 0)
- return false;
- if (Left.is(tok::l_paren) && Left.is(TT_AttributeParen))
- return false;
- if (Left.is(tok::l_paren) && Left.Previous &&
- (Left.Previous->isOneOf(TT_BinaryOperator, TT_CastRParen)))
- return false;
- if (Right.is(TT_ImplicitStringLiteral))
- return false;
- if (Right.is(tok::r_paren) || Right.is(TT_TemplateCloser))
- return false;
- if (Right.is(tok::r_square) && Right.MatchingParen &&
- Right.MatchingParen->is(TT_LambdaLSquare))
- return false;
- // We only break before r_brace if there was a corresponding break before
- // the l_brace, which is tracked by BreakBeforeClosingBrace.
- if (Right.is(tok::r_brace))
- return Right.MatchingParen && Right.MatchingParen->BlockKind == BK_Block;
- // Allow breaking after a trailing annotation, e.g. after a method
- // declaration.
- if (Left.is(TT_TrailingAnnotation))
- return !Right.isOneOf(tok::l_brace, tok::semi, tok::equal, tok::l_paren,
- tok::less, tok::coloncolon);
- if (Right.is(tok::kw___attribute))
- return true;
- if (Left.is(tok::identifier) && Right.is(tok::string_literal))
- return true;
- if (Right.is(tok::identifier) && Right.Next && Right.Next->is(TT_DictLiteral))
- return true;
- if (Left.is(TT_CtorInitializerColon))
- return Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon;
- if (Right.is(TT_CtorInitializerColon))
- return Style.BreakConstructorInitializers != FormatStyle::BCIS_AfterColon;
- if (Left.is(TT_CtorInitializerComma) &&
- Style.BreakConstructorInitializers == FormatStyle::BCIS_BeforeComma)
- return false;
- if (Right.is(TT_CtorInitializerComma) &&
- Style.BreakConstructorInitializers == FormatStyle::BCIS_BeforeComma)
- return true;
- if (Left.is(TT_InheritanceComma) && Style.BreakBeforeInheritanceComma)
- return false;
- if (Right.is(TT_InheritanceComma) && Style.BreakBeforeInheritanceComma)
- return true;
- if ((Left.is(tok::greater) && Right.is(tok::greater)) ||
- (Left.is(tok::less) && Right.is(tok::less)))
- return false;
- if (Right.is(TT_BinaryOperator) &&
- Style.BreakBeforeBinaryOperators != FormatStyle::BOS_None &&
- (Style.BreakBeforeBinaryOperators == FormatStyle::BOS_All ||
- Right.getPrecedence() != prec::Assignment))
- return true;
- if (Left.is(TT_ArrayInitializerLSquare))
- return true;
- if (Right.is(tok::kw_typename) && Left.isNot(tok::kw_const))
- return true;
- if ((Left.isBinaryOperator() || Left.is(TT_BinaryOperator)) &&
- !Left.isOneOf(tok::arrowstar, tok::lessless) &&
- Style.BreakBeforeBinaryOperators != FormatStyle::BOS_All &&
- (Style.BreakBeforeBinaryOperators == FormatStyle::BOS_None ||
- Left.getPrecedence() == prec::Assignment))
- return true;
- return Left.isOneOf(tok::comma, tok::coloncolon, tok::semi, tok::l_brace,
- tok::kw_class, tok::kw_struct, tok::comment) ||
- Right.isMemberAccess() ||
- Right.isOneOf(TT_TrailingReturnArrow, TT_LambdaArrow, tok::lessless,
- tok::colon, tok::l_square, tok::at) ||
- (Left.is(tok::r_paren) &&
- Right.isOneOf(tok::identifier, tok::kw_const)) ||
- (Left.is(tok::l_paren) && !Right.is(tok::r_paren)) ||
- (Left.is(TT_TemplateOpener) && !Right.is(TT_TemplateCloser));
- }
- void TokenAnnotator::printDebugInfo(const AnnotatedLine &Line) {
- llvm::errs() << "AnnotatedTokens(L=" << Line.Level << "):\n";
- const FormatToken *Tok = Line.First;
- while (Tok) {
- llvm::errs() << " M=" << Tok->MustBreakBefore
- << " C=" << Tok->CanBreakBefore
- << " T=" << getTokenTypeName(Tok->Type)
- << " S=" << Tok->SpacesRequiredBefore
- << " B=" << Tok->BlockParameterCount
- << " BK=" << Tok->BlockKind << " P=" << Tok->SplitPenalty
- << " Name=" << Tok->Tok.getName() << " L=" << Tok->TotalLength
- << " PPK=" << Tok->PackingKind << " FakeLParens=";
- for (unsigned i = 0, e = Tok->FakeLParens.size(); i != e; ++i)
- llvm::errs() << Tok->FakeLParens[i] << "/";
- llvm::errs() << " FakeRParens=" << Tok->FakeRParens;
- llvm::errs() << " Text='" << Tok->TokenText << "'\n";
- if (!Tok->Next)
- assert(Tok == Line.Last);
- Tok = Tok->Next;
- }
- llvm::errs() << "----\n";
- }
- } // namespace format
- } // namespace clang
|