1 | // Copyright (C) 2020 The Qt Company Ltd. |
2 | // SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0 |
3 | |
4 | #ifndef LUPDATEPREPROCESSORACTION_H |
5 | #define LUPDATEPREPROCESSORACTION_H |
6 | |
7 | #include "cpp_clang.h" |
8 | #include "synchronized.h" |
9 | |
10 | QT_WARNING_PUSH |
11 | QT_WARNING_DISABLE_MSVC(4100) |
12 | QT_WARNING_DISABLE_MSVC(4146) |
13 | QT_WARNING_DISABLE_MSVC(4267) |
14 | QT_WARNING_DISABLE_MSVC(4624) |
15 | QT_WARNING_DISABLE_GCC("-Wnonnull" ) |
16 | |
17 | #include <clang/Frontend/CompilerInstance.h> |
18 | #include <clang/Frontend/FrontendActions.h> |
19 | #include <clang/Tooling/Tooling.h> |
20 | #include <clang/Lex/PPCallbacks.h> |
21 | #include <clang/Lex/Preprocessor.h> |
22 | |
23 | QT_WARNING_POP |
24 | |
25 | #include <memory> |
26 | |
27 | QT_BEGIN_NAMESPACE |
28 | |
29 | class LupdatePPCallbacks : public clang::PPCallbacks |
30 | { |
31 | public: |
32 | LupdatePPCallbacks(WriteSynchronizedRef<TranslationRelatedStore> *stores, clang::Preprocessor &pp) |
33 | : m_preprocessor(pp) |
34 | , m_stores(stores) |
35 | { |
36 | const auto &sm = m_preprocessor.getSourceManager(); |
37 | m_inputFile = sm.getFileEntryRefForID(FID: sm.getMainFileID())->getName(); |
38 | } |
39 | |
40 | ~LupdatePPCallbacks() override |
41 | { |
42 | m_stores->emplace_bulk(values: std::move(m_ppStores)); |
43 | } |
44 | |
45 | private: |
46 | void MacroExpands(const clang::Token &token, const clang::MacroDefinition ¯oDefinition, |
47 | clang::SourceRange sourceRange, const clang::MacroArgs *macroArgs) override; |
48 | |
49 | void storeMacroArguments(const std::vector<QString> &args, TranslationRelatedStore *store); |
50 | |
51 | void SourceRangeSkipped(clang::SourceRange sourceRange, clang::SourceLocation endifLoc) override; |
52 | #if (LUPDATE_CLANG_VERSION < LUPDATE_CLANG_VERSION_CHECK(14,0,0)) |
53 | void InclusionDirective(clang::SourceLocation /*hashLoc*/, const clang::Token &/*includeTok*/, |
54 | clang::StringRef /*fileName*/, bool /*isAngled*/, |
55 | clang::CharSourceRange /*filenameRange*/, |
56 | #if (LUPDATE_CLANG_VERSION >= LUPDATE_CLANG_VERSION_CHECK(16,0,0)) |
57 | const clang::OptionalFileEntryRef file, |
58 | #elif (LUPDATE_CLANG_VERSION >= LUPDATE_CLANG_VERSION_CHECK(15,0,0)) |
59 | const clang::Optional<clang::FileEntryRef> file, |
60 | #else |
61 | const clang::FileEntry *file, |
62 | #endif |
63 | clang::StringRef /*searchPath*/, clang::StringRef /*relativePath*/, |
64 | const clang::Module */*imported*/, |
65 | clang::SrcMgr::CharacteristicKind /*fileType*/) override; |
66 | #endif |
67 | |
68 | std::string m_inputFile; |
69 | clang::Preprocessor &m_preprocessor; |
70 | |
71 | TranslationStores m_ppStores; |
72 | WriteSynchronizedRef<TranslationRelatedStore> *m_stores { nullptr }; |
73 | }; |
74 | |
75 | class LupdatePreprocessorAction : public clang::PreprocessOnlyAction |
76 | { |
77 | public: |
78 | LupdatePreprocessorAction(WriteSynchronizedRef<TranslationRelatedStore> *stores) |
79 | : m_stores(stores) |
80 | {} |
81 | |
82 | private: |
83 | void ExecuteAction() override |
84 | { |
85 | auto &preprocessor = getCompilerInstance().getPreprocessor(); |
86 | preprocessor.SetSuppressIncludeNotFoundError(true); |
87 | auto callbacks = new LupdatePPCallbacks(m_stores, preprocessor); |
88 | preprocessor.addPPCallbacks(C: std::unique_ptr<clang::PPCallbacks>(callbacks)); |
89 | |
90 | clang::PreprocessOnlyAction::ExecuteAction(); |
91 | } |
92 | |
93 | private: |
94 | WriteSynchronizedRef<TranslationRelatedStore> *m_stores { nullptr }; |
95 | }; |
96 | |
97 | class LupdatePreprocessorActionFactory : public clang::tooling::FrontendActionFactory |
98 | { |
99 | public: |
100 | explicit LupdatePreprocessorActionFactory(WriteSynchronizedRef<TranslationRelatedStore> *stores) |
101 | : m_stores(stores) |
102 | {} |
103 | |
104 | #if (LUPDATE_CLANG_VERSION >= LUPDATE_CLANG_VERSION_CHECK(10,0,0)) |
105 | std::unique_ptr<clang::FrontendAction> create() override |
106 | { |
107 | return std::make_unique<LupdatePreprocessorAction>(args&: m_stores); |
108 | } |
109 | #else |
110 | clang::FrontendAction *create() override |
111 | { |
112 | return new LupdatePreprocessorAction(m_stores); |
113 | } |
114 | #endif |
115 | |
116 | private: |
117 | WriteSynchronizedRef<TranslationRelatedStore> *m_stores { nullptr }; |
118 | }; |
119 | |
120 | QT_END_NAMESPACE |
121 | |
122 | #endif |
123 | |