File: OutputHelper.cpp

package info (click to toggle)
chromium 138.0.7204.157-1
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 6,071,864 kB
  • sloc: cpp: 34,936,859; ansic: 7,176,967; javascript: 4,110,704; python: 1,419,953; asm: 946,768; xml: 739,967; pascal: 187,324; sh: 89,623; perl: 88,663; objc: 79,944; sql: 50,304; cs: 41,786; fortran: 24,137; makefile: 21,806; php: 13,980; tcl: 13,166; yacc: 8,925; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (74 lines) | stat: -rw-r--r-- 2,997 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
// 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"

void OutputHelper::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`.
void OutputHelper::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`.
void OutputHelper::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
bool OutputHelper::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
  return true;
}

// This is run automatically at the end of the file.
void OutputHelper::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";
    }
  }
}