| // Copyright 2025 The Chromium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include"OutputHelper.h" |
| |
| voidOutputHelper::Delete(const clang::CharSourceRange& replacement_range, |
| const clang::SourceManager& source_manager, |
| const clang::LangOptions& lang_opts){ |
| Replace(replacement_range,"", source_manager, lang_opts); |
| } |
| |
| // Replaces `replacement_range` with `replacement_text`. |
| voidOutputHelper::Replace(const clang::CharSourceRange& replacement_range, |
| std::string replacement_text, |
| const clang::SourceManager& source_manager, |
| const clang::LangOptions& lang_opts){ |
| clang::tooling::Replacement replacement(source_manager, replacement_range, |
| std::move(replacement_text), |
| lang_opts); |
| |
| llvm::StringRef file_path= replacement.getFilePath(); |
| if(file_path.empty()){ |
| return; |
| } |
| |
| PrintReplacement(file_path, replacement.getOffset(), replacement.getLength(), |
| replacement.getReplacementText()); |
| } |
| // Inserts `lhs` and `rhs` to the left and right of `replacement_range`. |
| voidOutputHelper::Wrap(const clang::CharSourceRange& replacement_range, |
| std::string_view lhs, |
| std::string_view rhs, |
| const clang::SourceManager& source_manager, |
| const clang::LangOptions& lang_opts){ |
| clang::tooling::Replacement replacement(source_manager, replacement_range,"", |
| lang_opts); |
| |
| llvm::StringRef file_path= replacement.getFilePath(); |
| if(file_path.empty()){ |
| return; |
| } |
| |
| PrintReplacement(file_path, replacement.getOffset(),0, lhs); |
| PrintReplacement(file_path, replacement.getOffset()+ replacement.getLength(), |
| 0, rhs); |
| } |
| |
| // This is run automatically when the tool is first invoked |
| boolOutputHelper::handleBeginSource(clang::CompilerInstance& compiler){ |
| const clang::FrontendOptions& frontend_options= compiler.getFrontendOpts(); |
| |
| // Validate our expectations about how this tool should be used |
| assert((frontend_options.Inputs.size()==1)&& |
| "run_tool.py should invoke the rewriter one file at a time"); |
| const clang::FrontendInputFile& input_file= frontend_options.Inputs[0]; |
| assert(input_file.isFile()&& |
| "run_tool.py should invoke the rewriter on actual files"); |
| |
| current_language_= input_file.getKind().getLanguage(); |
| |
| // Report that we succeeded |
| returntrue; |
| } |
| |
| // This is run automatically at the end of the file. |
| voidOutputHelper::handleEndSource(){ |
| for(auto& file: files_replaced_in_){ |
| for(auto& header: headers_to_add_){ |
| llvm::outs()<<"include-user-header:::"<< file.getKey() |
| <<":::-1:::-1:::"<< header.getKey()<<"\n"; |
| } |
| } |
| } |
