forked from OSchip/llvm-project
62 lines
2.1 KiB
C++
62 lines
2.1 KiB
C++
//===--- DurationSubtractionCheck.cpp - clang-tidy ------------------------===//
|
|
//
|
|
// 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 "DurationSubtractionCheck.h"
|
|
#include "DurationRewriter.h"
|
|
#include "clang/AST/ASTContext.h"
|
|
#include "clang/ASTMatchers/ASTMatchFinder.h"
|
|
#include "clang/Tooling/FixIt.h"
|
|
|
|
using namespace clang::ast_matchers;
|
|
|
|
namespace clang {
|
|
namespace tidy {
|
|
namespace abseil {
|
|
|
|
void DurationSubtractionCheck::registerMatchers(MatchFinder *Finder) {
|
|
Finder->addMatcher(
|
|
binaryOperator(
|
|
hasOperatorName("-"),
|
|
hasLHS(callExpr(callee(functionDecl(DurationConversionFunction())
|
|
.bind("function_decl")),
|
|
hasArgument(0, expr().bind("lhs_arg")))))
|
|
.bind("binop"),
|
|
this);
|
|
}
|
|
|
|
void DurationSubtractionCheck::check(const MatchFinder::MatchResult &Result) {
|
|
const auto *Binop = Result.Nodes.getNodeAs<BinaryOperator>("binop");
|
|
const auto *FuncDecl = Result.Nodes.getNodeAs<FunctionDecl>("function_decl");
|
|
|
|
// Don't try to replace things inside of macro definitions.
|
|
if (Binop->getExprLoc().isMacroID() || Binop->getExprLoc().isInvalid())
|
|
return;
|
|
|
|
llvm::Optional<DurationScale> Scale =
|
|
getScaleForDurationInverse(FuncDecl->getName());
|
|
if (!Scale)
|
|
return;
|
|
|
|
std::string RhsReplacement =
|
|
rewriteExprFromNumberToDuration(Result, *Scale, Binop->getRHS());
|
|
|
|
const Expr *LhsArg = Result.Nodes.getNodeAs<Expr>("lhs_arg");
|
|
|
|
diag(Binop->getBeginLoc(), "perform subtraction in the duration domain")
|
|
<< FixItHint::CreateReplacement(
|
|
Binop->getSourceRange(),
|
|
(llvm::Twine("absl::") + FuncDecl->getName() + "(" +
|
|
tooling::fixit::getText(*LhsArg, *Result.Context) + " - " +
|
|
RhsReplacement + ")")
|
|
.str());
|
|
}
|
|
|
|
} // namespace abseil
|
|
} // namespace tidy
|
|
} // namespace clang
|