Zakharov Artem. Lab 1. Option 4.

clang/labs/lab1/zakharov_artem/CMakeLists.txt

@@ -0,0 +1,14 @@
+add_llvm_library(RenamePlugin MODULE RenameIds.cpp PLUGIN_TOOL clang)
+
+if(WIN32 OR CYGWIN)
+    set(LLVM_LINK_COMPONENTS
+            Support
+    )
+    clang_target_link_libraries(RenamePlugin PRIVATE
+            clangAST
+            clangBasic
+            clangFrontend
+    )
+endif()
+
+set(CLANG_TEST_DEPS "RenamePlugin" ${CLANG_TEST_DEPS} PARENT_SCOPE)

clang/labs/lab1/zakharov_artem/RenameIds.cpp

@@ -0,0 +1,197 @@
+#include "clang/AST/ASTConsumer.h"
+#include "clang/AST/RecursiveASTVisitor.h"
+#include "clang/Frontend/CompilerInstance.h"
+#include "clang/Frontend/FrontendPluginRegistry.h"
+#include "clang/Rewrite/Core/Rewriter.h"
+
+enum class IdType { Var, Func, Class };
+
+class RenameVisitor : public clang::RecursiveASTVisitor<RenameVisitor> {
+public:
+  explicit RenameVisitor(clang::Rewriter rewriter, IdType type,
+                         clang::StringRef cur_name, clang::StringRef new_name)
+      : rewriter(rewriter), type(type), cur_name(cur_name), new_name(new_name) {
+  }
+
+  bool VisitFunctionDecl(clang::FunctionDecl *func) {
+    if (type == IdType::Func && func->getName() == cur_name) {
+      rewriter.ReplaceText(func->getNameInfo().getSourceRange(), new_name);
+    }
+    return true;
+  }
+
+  bool VisitCallExpr(clang::CallExpr *call) {
+    if (type == IdType::Func) {
+      clang::FunctionDecl *callee = call->getDirectCallee();
+      if (callee && callee->getName() == cur_name) {
+        rewriter.ReplaceText(call->getCallee()->getSourceRange(), new_name);
+      }
+    }
+    return true;
+  }
+
+  bool VisitVarDecl(clang::VarDecl *var) {
+    if (type == IdType::Var && var->getName() == cur_name) {
+      rewriter.ReplaceText(var->getLocation(), cur_name.size(), new_name);
+    }
+    if (type == IdType::Class &&
+        var->getType().getAsString() == cur_name + " *") {
+      rewriter.ReplaceText(var->getTypeSourceInfo()->getTypeLoc().getBeginLoc(),
+                           cur_name.size(), new_name);
+    }
+    if (type == IdType::Class && var->getType().getAsString() == cur_name) {
+      rewriter.ReplaceText(
+          var->getTypeSourceInfo()->getTypeLoc().getSourceRange(), new_name);
+    }
+    return true;
+  }
+
+  bool VisitDeclRefExpr(clang::DeclRefExpr *expr) {
+    clang::VarDecl *var = clang::dyn_cast<clang::VarDecl>(expr->getDecl());
+    if (type == IdType::Var && var && var->getName() == cur_name) {
+      rewriter.ReplaceText(expr->getSourceRange(), new_name);
+    }
+    return true;
+  }
+
+  bool VisitCXXRecordDecl(clang::CXXRecordDecl *record) {
+    if (type == IdType::Class && record->getName() == cur_name) {
+      rewriter.ReplaceText(record->getLocation(), new_name);
+      const auto *destructor = record->getDestructor();
+      if (destructor) {
+        rewriter.ReplaceText(destructor->getLocation(), cur_name.size() + 1,
+                             '~' + new_name);
+      }
+    }
+    return true;
+  }
+
+  bool VisitCXXConstructorDecl(clang::CXXConstructorDecl *constructor) {
+    if (type == IdType::Class) {
+      if (constructor->getNameAsString() == cur_name) {
+        rewriter.ReplaceText(constructor->getLocation(), cur_name.size(),
+                             new_name);
+      }
+    }
+    return true;
+  }
+
+  bool VisitCXXNewExpr(clang::CXXNewExpr *newExpr) {
+    if (type == IdType::Class) {
+      if (newExpr->getConstructExpr()->getType().getAsString() == cur_name) {
+        rewriter.ReplaceText(newExpr->getExprLoc(), cur_name.size() + 4,
+                             "new " + new_name);
+      }
+    }
+    return true;
+  }
+
+  bool save_changes() { return rewriter.overwriteChangedFiles(); }
+
+private:
+  clang::Rewriter rewriter;
+  IdType type;
+  std::string cur_name;
+  std::string new_name;
+};
+
+class RenameASTConsumer : public clang::ASTConsumer {
+public:
+  explicit RenameASTConsumer(clang::CompilerInstance &CI, IdType type,
+                             clang::StringRef cur_name,
+                             clang::StringRef new_name)
+      : Visitor(clang::Rewriter(CI.getSourceManager(), CI.getLangOpts()), type,
+                cur_name, new_name) {}
+
+  void HandleTranslationUnit(clang::ASTContext &context) override {
+    Visitor.TraverseDecl(context.getTranslationUnitDecl());
+    if (Visitor.save_changes()) {
+      llvm::errs() << "An error occurred while saving changes to a file!\n";
+    }
+  }
+
+private:
+  RenameVisitor Visitor;
+};
+
+class RenamePlugin : public clang::PluginASTAction {
+public:
+  std::unique_ptr<clang::ASTConsumer>
+  CreateASTConsumer(clang::CompilerInstance &CI,
+                    clang::StringRef InFile) override {
+    return std::make_unique<RenameASTConsumer>(CI, type, cur_name, new_name);
+  }
+
+protected:
+  bool ParseArgs(const clang::CompilerInstance &CI,
+                 const std::vector<std::string> &args) override {
+    std::vector<std::pair<std::string, std::string>> params = {
+        {"type=", ""}, {"cur-name=", ""}, {"new-name=", ""}};
+
+    if (!args.empty() && args[0] == "help") {
+      PrintHelp(llvm::errs());
+      return true;
+    }
+
+    if (args.size() < 3) {
+      PrintParamsError(CI);
+      return false;
+    }
+
+    for (const auto &arg : args) {
+      bool is_found = false;
+      for (auto &param : params) {
+        if (arg.find(param.first) == 0 && param.second.empty()) {
+          param.second = arg.substr(param.first.size());
+          is_found = true;
+          break;
+        }
+      }
+      if (!is_found) {
+        PrintParamsError(CI);
+        return false;
+      }
+    }
+
+    std::vector<std::pair<std::string, IdType>> id_type = {
+        {"var", IdType::Var}, {"func", IdType::Func}, {"class", IdType::Class}};
+    size_t i;
+    for (i = 0; i < id_type.size(); i++) {
+      if (params[0].second == id_type[i].first) {
+        type = id_type[i].second;
+        break;
+      }
+    }
+    if (i == id_type.size()) {
+      PrintParamsError(CI);
+      return false;
+    }
+    cur_name = params[1].second;
+    new_name = params[2].second;
+    return true;
+  }
+
+  void PrintHelp(llvm::raw_ostream &ros) {
+    ros << "Specify three required arguments:\n"
+           "-plugin-arg-rename type=[\"var\", \"func\", \"class\"]\n"
+           "-plugin-arg-rename cur-name=\"Current identifier name\"\n"
+           "-plugin-arg-rename new-name=\"New identifier name\"\n";
+  }
+
+  void PrintParamsError(const clang::CompilerInstance &CI) {
+    clang::DiagnosticsEngine &D = CI.getDiagnostics();
+
+    D.Report(
+        D.getCustomDiagID(clang::DiagnosticsEngine::Error,
+                          "Invalid arguments\n"
+                          "Specify \"-plugin-arg-rename help\" for usage\n"));
+  }
+
+private:
+  IdType type;
+  std::string cur_name;
+  std::string new_name;
+};
+
+static clang::FrontendPluginRegistry::Add<RenamePlugin>
+    X("rename", "Rename variable, function or class");