// Copyright (c) 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
//
// This implements a Clang tool to rewrite all instances of scoped_array<T> to
// scoped_ptr<T[]>. The former is being deprecated in favor of the latter, to
// allow for an eventual transition from scoped_ptr to unique_ptr.

#include "clang/AST/ASTContext.h"
#include "clang/ASTMatchers/ASTMatchers.h"
#include "clang/ASTMatchers/ASTMatchFinder.h"
#include "clang/Basic/SourceManager.h"
#include "clang/Frontend/FrontendActions.h"
#include "clang/Lex/Lexer.h"
#include "clang/Tooling/CommonOptionsParser.h"
#include "clang/Tooling/Refactoring.h"
#include "clang/Tooling/Tooling.h"
#include "llvm/Support/CommandLine.h"

using clang::ast_matchers::MatchFinder;
using clang::ast_matchers::hasDeclaration;
using clang::ast_matchers::hasName;
using clang::ast_matchers::id;
using clang::ast_matchers::loc;
using clang::ast_matchers::qualType;
using clang::ast_matchers::recordDecl;
using clang::tooling::CommonOptionsParser;
using clang::tooling::Replacement;
using clang::tooling::Replacements;
using llvm::StringRef;

namespace {

class RewriterCallback : public MatchFinder::MatchCallback {
 public:
  RewriterCallback(Replacements* replacements) : replacements_(replacements) {}
  virtual void run(const MatchFinder::MatchResult& result) LLVM_OVERRIDE;

 private:
  Replacements* const replacements_;
};

void RewriterCallback::run(const MatchFinder::MatchResult& result) {
  const clang::TypeLoc type_location =
      *result.Nodes.getNodeAs<clang::TypeLoc>("loc");
  clang::CharSourceRange range = clang::CharSourceRange::getTokenRange(
      result.SourceManager->getSpellingLoc(type_location.getLocStart()),
      result.SourceManager->getSpellingLoc(type_location.getLocEnd()));
  // TODO(dcheng): Log an error?
  if (!range.isValid())
    return;
  std::string replacement_text = clang::Lexer::getSourceText(
      range, *result.SourceManager, result.Context->getLangOpts());
  // TODO(dcheng): Log errors?
  if (!StringRef(replacement_text).startswith("scoped_array<") ||
      !StringRef(replacement_text).endswith(">"))
    return;
  replacement_text.replace(strlen("scoped_"), strlen("array"), "ptr");
  replacement_text.insert(replacement_text.size() - 1, "[]");
  replacements_->insert(
      Replacement(*result.SourceManager, range, replacement_text));
}

}  // namespace

static llvm::cl::extrahelp common_help(CommonOptionsParser::HelpMessage);

int main(int argc, const char* argv[]) {
  CommonOptionsParser options(argc, argv);
  clang::tooling::ClangTool tool(options.getCompilations(),
                                 options.getSourcePathList());

  Replacements replacements;
  RewriterCallback callback(&replacements);
  MatchFinder match_finder;
  match_finder.addMatcher(
      id("loc",
         loc(qualType(hasDeclaration(recordDecl(hasName("::scoped_array")))))),
      &callback);

  int result =
      tool.run(clang::tooling::newFrontendActionFactory(&match_finder));
  if (result != 0)
    return result;

  // Serialization format is documented in tools/clang/scripts/run_tool.py
  llvm::outs() << "==== BEGIN EDITS ====\n";
  for (Replacements::const_iterator it = replacements.begin();
       it != replacements.end(); ++it) {
    llvm::outs() << "r:" << it->getFilePath() << ":" << it->getOffset() << ":"
                 << it->getLength() << ":" << it->getReplacementText() << "\n";
  }
  llvm::outs() << "==== END EDITS ====\n";

  return 0;
}