build/clang-plugin/BaseCheck.h
author Brindusan Cristian <cbrindusan@mozilla.com>
Fri, 21 Feb 2020 18:27:05 +0200
changeset 514948 b72f6888ad96b094136bec46eb320b7434e240bb
parent 500003 9d02871eff57f2fcce7c574ad23c1b78f4e1e012
permissions -rw-r--r--
Merge autoland to mozilla-central. a=merge

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef BaseCheck_h__
#define BaseCheck_h__

class MozContext {};
typedef MozContext ContextType;

class BaseCheck : public MatchFinder::MatchCallback {
public:
  BaseCheck(StringRef CheckName, ContextType *Context) {}
  virtual void registerMatchers(MatchFinder *Finder) {}
  virtual void registerPPCallbacks(CompilerInstance &CI) {}
  virtual void check(const MatchFinder::MatchResult &Result) {}
  DiagnosticBuilder diag(SourceLocation Loc, StringRef Description,
                         DiagnosticIDs::Level Level = DiagnosticIDs::Warning) {
    DiagnosticsEngine &Diag = Context->getDiagnostics();
    unsigned ID = Diag.getDiagnosticIDs()->getCustomDiagID(Level, Description);
    return Diag.Report(Loc, ID);
  }

private:
  void run(const MatchFinder::MatchResult &Result) override {
    Context = Result.Context;
    check(Result);
  }

private:
  ASTContext *Context;
};

#endif