forked from OSchip/llvm-project
87 lines
3.3 KiB
C++
87 lines
3.3 KiB
C++
//===- ClangTidyPlugin.cpp - clang-tidy as a clang plugin -----------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "../ClangTidy.h"
|
|
#include "../ClangTidyDiagnosticConsumer.h"
|
|
#include "../ClangTidyForceLinker.h"
|
|
#include "../ClangTidyModule.h"
|
|
#include "clang/Frontend/CompilerInstance.h"
|
|
#include "clang/Frontend/FrontendPluginRegistry.h"
|
|
#include "clang/Frontend/MultiplexConsumer.h"
|
|
|
|
namespace clang {
|
|
namespace tidy {
|
|
|
|
/// The core clang tidy plugin action. This just provides the AST consumer and
|
|
/// command line flag parsing for using clang-tidy as a clang plugin.
|
|
class ClangTidyPluginAction : public PluginASTAction {
|
|
/// Wrapper to grant the context and diagnostics engine the same lifetime as
|
|
/// the action.
|
|
/// We use MultiplexConsumer to avoid writing out all the forwarding methods.
|
|
class WrapConsumer : public MultiplexConsumer {
|
|
std::unique_ptr<ClangTidyContext> Context;
|
|
std::unique_ptr<DiagnosticsEngine> DiagEngine;
|
|
|
|
public:
|
|
WrapConsumer(std::unique_ptr<ClangTidyContext> Context,
|
|
std::unique_ptr<DiagnosticsEngine> DiagEngine,
|
|
std::vector<std::unique_ptr<ASTConsumer>> Consumer)
|
|
: MultiplexConsumer(std::move(Consumer)), Context(std::move(Context)),
|
|
DiagEngine(std::move(DiagEngine)) {}
|
|
};
|
|
|
|
public:
|
|
std::unique_ptr<ASTConsumer> CreateASTConsumer(CompilerInstance &Compiler,
|
|
StringRef File) override {
|
|
// Create and set diagnostics engine
|
|
auto *DiagConsumer =
|
|
new ClangTidyDiagnosticConsumer(*Context, &Compiler.getDiagnostics());
|
|
auto DiagEngine = std::make_unique<DiagnosticsEngine>(
|
|
new DiagnosticIDs, new DiagnosticOptions, DiagConsumer);
|
|
Context->setDiagnosticsEngine(DiagEngine.get());
|
|
|
|
// Create the AST consumer.
|
|
ClangTidyASTConsumerFactory Factory(*Context);
|
|
std::vector<std::unique_ptr<ASTConsumer>> Vec;
|
|
Vec.push_back(Factory.createASTConsumer(Compiler, File));
|
|
|
|
return std::make_unique<WrapConsumer>(
|
|
std::move(Context), std::move(DiagEngine), std::move(Vec));
|
|
}
|
|
|
|
bool ParseArgs(const CompilerInstance &,
|
|
const std::vector<std::string> &Args) override {
|
|
ClangTidyGlobalOptions GlobalOptions;
|
|
ClangTidyOptions DefaultOptions;
|
|
ClangTidyOptions OverrideOptions;
|
|
|
|
// Parse the extra command line args.
|
|
// FIXME: This is very limited at the moment.
|
|
for (StringRef Arg : Args)
|
|
if (Arg.startswith("-checks="))
|
|
OverrideOptions.Checks = std::string(Arg.substr(strlen("-checks=")));
|
|
|
|
auto Options = std::make_unique<FileOptionsProvider>(
|
|
GlobalOptions, DefaultOptions, OverrideOptions);
|
|
Context = std::make_unique<ClangTidyContext>(std::move(Options));
|
|
return true;
|
|
}
|
|
|
|
private:
|
|
std::unique_ptr<ClangTidyContext> Context;
|
|
};
|
|
} // namespace tidy
|
|
} // namespace clang
|
|
|
|
// This anchor is used to force the linker to link in the generated object file
|
|
// and thus register the clang-tidy plugin.
|
|
volatile int ClangTidyPluginAnchorSource = 0;
|
|
|
|
static clang::FrontendPluginRegistry::Add<clang::tidy::ClangTidyPluginAction>
|
|
X("clang-tidy", "clang-tidy");
|