1 files changed, 125 insertions, 0 deletions
diff --git a/src/linguist/lupdate/lupdatepreprocessoraction.h b/src/linguist/lupdate/lupdatepreprocessoraction.h
new file mode 100644
index 000000000..21c7baee1
--- /dev/null
+++ b/src/linguist/lupdate/lupdatepreprocessoraction.h
@@ -0,0 +1,125 @@
+/****************************************************************************
+**
+** Copyright (C) 2020 The Qt Company Ltd.
+** Contact: https://www.qt.io/licensing/
+**
+** This file is part of the Qt Linguist of the Qt Toolkit.
+**
+** $QT_BEGIN_LICENSE:GPL-EXCEPT$
+** Commercial License Usage
+** Licensees holding valid commercial Qt licenses may use this file in
+** accordance with the commercial license agreement provided with the
+** Software or, alternatively, in accordance with the terms contained in
+** a written agreement between you and The Qt Company. For licensing terms
+** and conditions see https://www.qt.io/terms-conditions. For further
+** information use the contact form at https://www.qt.io/contact-us.
+**
+** GNU General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU
+** General Public License version 3 as published by the Free Software
+** Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT
+** included in the packaging of this file. Please review the following
+** information to ensure the GNU General Public License requirements will
+** be met: https://www.gnu.org/licenses/gpl-3.0.html.
+**
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+
+#ifndef LUPDATEPREPROCESSORACTION_H
+#define LUPDATEPREPROCESSORACTION_H
+
+#include "cpp_clang.h"
+
+#if defined(Q_CC_MSVC)
+# pragma warning(push)
+# pragma warning(disable: 4100)
+# pragma warning(disable: 4146)
+# pragma warning(disable: 4267)
+# pragma warning(disable: 4624)
+#endif
+
+#include <clang/Frontend/CompilerInstance.h>
+#include <clang/Frontend/FrontendActions.h>
+#include <clang/Tooling/Tooling.h>
+#include <clang/Lex/PPCallbacks.h>
+#include <clang/Lex/Preprocessor.h>
+
+#if defined(Q_CC_MSVC)
+# pragma warning(pop)
+#endif
+
+#include <memory>
+
+QT_BEGIN_NAMESPACE
+
+class LupdatePPCallbacks : public clang::PPCallbacks
+{
+public:
+    LupdatePPCallbacks(TranslationStores &stores, clang::Preprocessor &pp)
+        : m_stores(stores)
+        , m_preprocessor(pp)
+    {
+        const auto &sm = m_preprocessor.getSourceManager();
+        m_inputFile = sm.getFileEntryForID(sm.getMainFileID())->getName();
+    }
+
+private:
+    void MacroExpands(const clang::Token &token, const clang::MacroDefinition &macroDefinition,
+        clang::SourceRange sourceRange, const clang::MacroArgs *macroArgs) override;
+
+    void storeMacroArguments(const std::vector<QString> &args, TranslationRelatedStore *store);
+
+    std::string m_inputFile;
+    TranslationStores &m_stores;
+    clang::Preprocessor &m_preprocessor;
+};
+
+class LupdatePreprocessorAction : public clang::PreprocessOnlyAction
+{
+public:
+    LupdatePreprocessorAction(TranslationStores &stores)
+        : m_stores(stores)
+    {}
+
+private:
+    void ExecuteAction() override
+    {
+        auto &preprocessor = getCompilerInstance().getPreprocessor();
+        preprocessor.SetSuppressIncludeNotFoundError(true);
+        auto callbacks = new LupdatePPCallbacks(m_stores, preprocessor);
+        preprocessor.addPPCallbacks(std::unique_ptr<clang::PPCallbacks>(callbacks));
+
+        clang::PreprocessOnlyAction::ExecuteAction();
+    }
+
+private:
+    TranslationStores &m_stores;
+};
+
+class LupdatePreprocessorActionFactory : public clang::tooling::FrontendActionFactory
+{
+public:
+    LupdatePreprocessorActionFactory(TranslationStores &stores)
+        : m_stores(stores)
+    {}
+
+#if (LUPDATE_CLANG_VERSION >= LUPDATE_CLANG_VERSION_CHECK(10,0,0))
+    std::unique_ptr<clang::FrontendAction> create() override
+    {
+        return std::make_unique<LupdatePreprocessorAction>(m_stores);
+    }
+#else
+    clang::FrontendAction *create() override
+    {
+        return new LupdatePreprocessorAction(m_stores);
+    }
+#endif
+
+private:
+    TranslationStores &m_stores;
+};
+
+QT_END_NAMESPACE
+
+#endif