| 1 | //===-- PragmaCommentHandler.cpp - find all symbols -----------------------===// |
|---|---|
| 2 | // |
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 4 | // See https://llvm.org/LICENSE.txt for license information. |
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 6 | // |
| 7 | //===----------------------------------------------------------------------===// |
| 8 | |
| 9 | #include "PragmaCommentHandler.h" |
| 10 | #include "FindAllSymbols.h" |
| 11 | #include "HeaderMapCollector.h" |
| 12 | #include "clang/Lex/Preprocessor.h" |
| 13 | #include "llvm/Support/Regex.h" |
| 14 | |
| 15 | namespace clang { |
| 16 | namespace find_all_symbols { |
| 17 | namespace { |
| 18 | const char IWYUPragma[] = "// IWYU pragma: private, include "; |
| 19 | } // namespace |
| 20 | |
| 21 | bool PragmaCommentHandler::HandleComment(Preprocessor &PP, SourceRange Range) { |
| 22 | StringRef Text = |
| 23 | Lexer::getSourceText(Range: CharSourceRange::getCharRange(R: Range), |
| 24 | SM: PP.getSourceManager(), LangOpts: PP.getLangOpts()); |
| 25 | size_t Pos = Text.find(Str: IWYUPragma); |
| 26 | if (Pos == StringRef::npos) |
| 27 | return false; |
| 28 | StringRef RemappingFilePath = Text.substr(Start: Pos + std::strlen(s: IWYUPragma)); |
| 29 | Collector->addHeaderMapping( |
| 30 | OrignalHeaderPath: PP.getSourceManager().getFilename(SpellingLoc: Range.getBegin()), |
| 31 | MappingHeaderPath: RemappingFilePath.trim(Chars: "\"<>")); |
| 32 | return false; |
| 33 | } |
| 34 | |
| 35 | } // namespace find_all_symbols |
| 36 | } // namespace clang |
| 37 |