1//===--- TimeComparisonCheck.cpp - clang-tidy
2//--------------------------------===//
3//
4// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
5// See https://llvm.org/LICENSE.txt for license information.
6// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
7//
8//===----------------------------------------------------------------------===//
9
10#include "TimeComparisonCheck.h"
11#include "DurationRewriter.h"
12#include "clang/AST/ASTContext.h"
13#include "clang/ASTMatchers/ASTMatchFinder.h"
14#include "clang/Tooling/FixIt.h"
15#include <optional>
16
17using namespace clang::ast_matchers;
18
19namespace clang::tidy::abseil {
20
21void TimeComparisonCheck::registerMatchers(MatchFinder *Finder) {
22 auto Matcher =
23 expr(comparisonOperatorWithCallee(funcDecl: functionDecl(
24 functionDecl(TimeConversionFunction()).bind(ID: "function_decl"))))
25 .bind(ID: "binop");
26
27 Finder->addMatcher(NodeMatch: Matcher, Action: this);
28}
29
30void TimeComparisonCheck::check(const MatchFinder::MatchResult &Result) {
31 const auto *Binop = Result.Nodes.getNodeAs<BinaryOperator>(ID: "binop");
32
33 std::optional<DurationScale> Scale = getScaleForTimeInverse(
34 Result.Nodes.getNodeAs<FunctionDecl>(ID: "function_decl")->getName());
35 if (!Scale)
36 return;
37
38 if (isInMacro(Result, E: Binop->getLHS()) || isInMacro(Result, E: Binop->getRHS()))
39 return;
40
41 // In most cases, we'll only need to rewrite one of the sides, but we also
42 // want to handle the case of rewriting both sides. This is much simpler if
43 // we unconditionally try and rewrite both, and let the rewriter determine
44 // if nothing needs to be done.
45 std::string LhsReplacement =
46 rewriteExprFromNumberToTime(Result, Scale: *Scale, Node: Binop->getLHS());
47 std::string RhsReplacement =
48 rewriteExprFromNumberToTime(Result, Scale: *Scale, Node: Binop->getRHS());
49
50 diag(Loc: Binop->getBeginLoc(), Description: "perform comparison in the time domain")
51 << FixItHint::CreateReplacement(Binop->getSourceRange(),
52 (llvm::Twine(LhsReplacement) + " " +
53 Binop->getOpcodeStr() + " " +
54 RhsReplacement)
55 .str());
56}
57
58} // namespace clang::tidy::abseil
59

source code of clang-tools-extra/clang-tidy/abseil/TimeComparisonCheck.cpp