clang-tools-extra/clang-tidy/abseil/DurationComparisonCheck.cpp

   1 //===--- DurationComparisonCheck.cpp - clang-tidy -------------------------===//
   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-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "DurationComparisonCheck.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
  16 using namespace clang::ast_matchers;
  17
  18 namespace clang::tidy::abseil {
  19
  20 void DurationComparisonCheck::registerMatchers(MatchFinder *Finder) {
  21   auto Matcher = expr(comparisonOperatorWithCallee(functionDecl(
  22                           functionDecl(DurationConversionFunction())
  23                               .bind("function_decl"))))
  24                      .bind("binop");
  25
  26   Finder->addMatcher(Matcher, this);
  27 }
  28
  29 void DurationComparisonCheck::check(const MatchFinder::MatchResult &Result) {
  30   const auto *Binop = Result.Nodes.getNodeAs<BinaryOperator>("binop");
  31
  32   std::optional<DurationScale> Scale = getScaleForDurationInverse(
  33       Result.Nodes.getNodeAs<FunctionDecl>("function_decl")->getName());
  34   if (!Scale)
  35     return;
  36
  37   // In most cases, we'll only need to rewrite one of the sides, but we also
  38   // want to handle the case of rewriting both sides. This is much simpler if
  39   // we unconditionally try and rewrite both, and let the rewriter determine
  40   // if nothing needs to be done.
  41   if (isInMacro(Result, Binop->getLHS()) || isInMacro(Result, Binop->getRHS()))
  42     return;
  43   std::string LhsReplacement =
  44       rewriteExprFromNumberToDuration(Result, *Scale, Binop->getLHS());
  45   std::string RhsReplacement =
  46       rewriteExprFromNumberToDuration(Result, *Scale, Binop->getRHS());
  47
  48   diag(Binop->getBeginLoc(), "perform comparison in the duration domain")
  49       << FixItHint::CreateReplacement(Binop->getSourceRange(),
  50                                       (llvm::Twine(LhsReplacement) + " " +
  51                                        Binop->getOpcodeStr() + " " +
  52                                        RhsReplacement)
  53                                           .str());
  54 }
  55
  56 } // namespace clang::tidy::abseil