Clang plugin that rewrites PostTask(_, NewRunnableMethod(...)) to PostTask(_, base::Bind(...));

tools/clang/BindMigrate/BindMigrateConsumer.h

+// Copyright 2011 Google Inc. All Rights Reserved.
+// Author: ajwong@google.com (Albert Wong)
+
+#ifndef BIND_MIGRATE_CONSUSUMER_H_
+#define BIND_MIGRATE_CONSUSUMER_H_
+
+#include "clang/AST/ASTConsumer.h"
+#include "clang/Basic/SourceLocation.h"
+#include "clang/Rewrite/Rewriter.h"
+
+#include "DiagnosticEmitter.h"
+
+namespace clang {
+
+namespace arcmt_hack {
+class FileRemapper;
+}  // namespace arcmt_hack
+
+class DiagnosticEmitter;
+
+class BindMigrateConsumer : public clang::ASTConsumer {
+public:
+  BindMigrateConsumer(clang::CompilerInstance* instance,
+                      arcmt_hack::FileRemapper* remapper);
+  virtual ~BindMigrateConsumer();
+
+  virtual void Initialize(clang::ASTContext &context);
+  virtual void HandleTranslationUnit(clang::ASTContext &context);
+
+private:
+  clang::CompilerInstance* instance_;
+  DiagnosticEmitter error_emitter_;
+  clang::FileID source_file_;
+  clang::Rewriter rewriter_;
+  arcmt_hack::FileRemapper* remapper_;
+};
+
+}  // namespace clang
+
+
+#endif  // BIND_MIGRATE_CONSUSUMER_H_