averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-21 12:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

[Refactoring] generate both '-dump-text' and dump-rewritten from a single swift-refactor invocation (SR-14587)

Browse Source
This commit is contained in:
jiaren wang
2021-05-20 14:28:21 +08:00
parent 6d47d3240d
commit 68eb0d7087
4 changed files with 43 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
tools/swift-refactor/swift-refactor.cpp
View File

@@ -98,6 +98,10 @@ static llvm::cl::list<std::string>
InputFilenames(llvm::cl::Positional, llvm::cl::desc("[input files...]"),
llvm::cl::ZeroOrMore);
static llvm::cl::opt<std::string>
RewrittenOutputFile("rewritten-output-file",
llvm::cl::desc("Name of the rewritten output file"));
static llvm::cl::opt<std::string>
LineColumnPair("pos", llvm::cl::desc("Line:Column pair or /*label*/"));
@@ -402,14 +406,27 @@ int main(int argc, char *argv[]) {
switch (options::DumpIn) {
case options::DumpType::REWRITTEN:
pConsumer.reset(new SourceEditOutputConsumer(SF->getASTContext().SourceMgr,
BufferID,
llvm::outs()));
BufferID, llvm::outs()));
break;
case options::DumpType::JSON:
pConsumer.reset(new SourceEditJsonConsumer(llvm::outs()));
break;
case options::DumpType::TEXT:
pConsumer.reset(new SourceEditTextConsumer(llvm::outs()));
if (options::RewrittenOutputFile.empty()) {
pConsumer.reset(new SourceEditTextConsumer(llvm::outs()));
} else {
std::error_code EC;
static llvm::raw_fd_ostream FileStream(options::RewrittenOutputFile, EC,
llvm::sys::fs::OF_None);
if (FileStream.has_error() || EC) {
llvm::errs() << "Could not open rewritten output file";
return 1;
}
pConsumer.reset(new DuplicatingSourceEditConsumer(
new SourceEditTextConsumer(llvm::outs()),
new SourceEditOutputConsumer(SF->getASTContext().SourceMgr, BufferID,
FileStream)));
}
break;
}