70 lines · cpp
1//===----------------------------------------------------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "DurationAdditionCheck.h"10#include "DurationRewriter.h"11#include "clang/AST/ASTContext.h"12#include "clang/ASTMatchers/ASTMatchFinder.h"13#include "clang/Tooling/FixIt.h"14#include <optional>15 16using namespace clang::ast_matchers;17 18namespace clang::tidy::abseil {19 20void DurationAdditionCheck::registerMatchers(MatchFinder *Finder) {21 Finder->addMatcher(22 binaryOperator(hasOperatorName("+"),23 hasEitherOperand(expr(ignoringParenImpCasts(24 callExpr(callee(functionDecl(timeConversionFunction())25 .bind("function_decl")))26 .bind("call")))))27 .bind("binop"),28 this);29}30 31void DurationAdditionCheck::check(const MatchFinder::MatchResult &Result) {32 const auto *Binop = Result.Nodes.getNodeAs<clang::BinaryOperator>("binop");33 const auto *Call = Result.Nodes.getNodeAs<clang::CallExpr>("call");34 35 // Don't try to replace things inside of macro definitions.36 if (Binop->getExprLoc().isMacroID() || Binop->getExprLoc().isInvalid())37 return;38 39 std::optional<DurationScale> Scale = getScaleForTimeInverse(40 Result.Nodes.getNodeAs<clang::FunctionDecl>("function_decl")->getName());41 if (!Scale)42 return;43 44 const llvm::StringRef TimeFactory = getTimeInverseForScale(*Scale);45 46 FixItHint Hint;47 if (Call == Binop->getLHS()->IgnoreParenImpCasts()) {48 Hint = FixItHint::CreateReplacement(49 Binop->getSourceRange(),50 (llvm::Twine(TimeFactory) + "(" +51 tooling::fixit::getText(*Call->getArg(0), *Result.Context) + " + " +52 rewriteExprFromNumberToDuration(Result, *Scale, Binop->getRHS()) + ")")53 .str());54 } else {55 assert(Call == Binop->getRHS()->IgnoreParenImpCasts() &&56 "Call should be found on the RHS");57 Hint = FixItHint::CreateReplacement(58 Binop->getSourceRange(),59 (llvm::Twine(TimeFactory) + "(" +60 rewriteExprFromNumberToDuration(Result, *Scale, Binop->getLHS()) +61 " + " + tooling::fixit::getText(*Call->getArg(0), *Result.Context) +62 ")")63 .str());64 }65 66 diag(Binop->getBeginLoc(), "perform addition in the duration domain") << Hint;67}68 69} // namespace clang::tidy::abseil70