From 7f757884c5a04484820a97e592afae74beff95a9 Mon Sep 17 00:00:00 2001
From: Tim Jenssen <tim.jenssen@qt.io>
Date: Wed, 23 Nov 2016 13:13:38 +0100
Subject: Clang: Extend clang query

It's a first step to introduce clang query.

Change-Id: I4d001a8883f56066765ce6bc561fa3f49611c0a4
Reviewed-by: Tim Jenssen <tim.jenssen@qt.io>
---
 .../source/collectincludesaction.h                 | 88 ++++++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 src/tools/clangrefactoringbackend/source/collectincludesaction.h

(limited to 'src/tools/clangrefactoringbackend/source/collectincludesaction.h')

diff --git a/src/tools/clangrefactoringbackend/source/collectincludesaction.h b/src/tools/clangrefactoringbackend/source/collectincludesaction.h
new file mode 100644
index 0000000000..1586ba3f8b
--- /dev/null
+++ b/src/tools/clangrefactoringbackend/source/collectincludesaction.h
@@ -0,0 +1,88 @@
+/****************************************************************************
+**
+** Copyright (C) 2016 The Qt Company Ltd.
+** Contact: https://www.qt.io/licensing/
+**
+** This file is part of Qt Creator.
+**
+** 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.
+**
+****************************************************************************/
+
+#pragma once
+
+#include <collectincludespreprocessorcallbacks.h>
+
+#if defined(__GNUC__)
+#pragma GCC diagnostic push
+#pragma GCC diagnostic ignored "-Wunused-parameter"
+#endif
+
+#include <clang/Frontend/FrontendActions.h>
+#include <clang/Frontend/CompilerInstance.h>
+#include <clang/Lex/Preprocessor.h>
+
+#if defined(__GNUC__)
+#pragma GCC diagnostic pop
+#endif
+
+namespace ClangBackEnd {
+
+class CollectIncludesAction final : public clang::PreprocessOnlyAction
+{
+public:
+    CollectIncludesAction(Utils::SmallStringVector &includes,
+                          const std::vector<uint> &excludedIncludeUID,
+                          std::vector<uint> &alreadyIncludedFileUIDs)
+        : includes(includes),
+          excludedIncludeUID(excludedIncludeUID),
+          alreadyIncludedFileUIDs(alreadyIncludedFileUIDs)
+    {
+    }
+
+    bool BeginSourceFileAction(clang::CompilerInstance &compilerInstance,
+                               llvm::StringRef filename) override
+    {
+      if (clang::PreprocessOnlyAction::BeginSourceFileAction(compilerInstance, filename)) {
+          auto &preprocessor = compilerInstance.getPreprocessor();
+          auto &headerSearch = preprocessor.getHeaderSearchInfo();
+
+          auto macroPreprocessorCallbacks = new CollectIncludesPreprocessorCallbacks(headerSearch,
+                                                                                     includes,
+                                                                                     excludedIncludeUID,
+                                                                                     alreadyIncludedFileUIDs);
+
+          preprocessor.addPPCallbacks(std::unique_ptr<clang::PPCallbacks>(macroPreprocessorCallbacks));
+
+          return true;
+      }
+
+      return false;
+    }
+
+    void EndSourceFileAction() override
+    {
+      clang::PreprocessOnlyAction::EndSourceFileAction();
+    }
+
+private:
+    Utils::SmallStringVector &includes;
+    const std::vector<uint> &excludedIncludeUID;
+    std::vector<uint> &alreadyIncludedFileUIDs;
+};
+
+} // namespace ClangBackEnd
-- 
cgit v1.2.1