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

[incrParse] Add option to force coloured output

Browse Source
This commit is contained in:
Alex Hoppen
2018-05-07 14:20:27 -07:00
parent 6135f10151
commit 4da37b1702
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tools/swift-syntax-test/swift-syntax-test.cpp
View File

@@ -144,6 +144,13 @@ PrintVisualReuseInfo("print-visual-reuse-info",
llvm::cl::cat(Category),
llvm::cl::init(false));
static llvm::cl::opt<bool>
ForceColoredOutput("force-colored-output",
llvm::cl::desc("Print colored output even if the shell "
"does not support it."),
llvm::cl::cat(Category),
llvm::cl::init(false));
static llvm::cl::opt<bool>
PrintNodeKind("print-node-kind",
llvm::cl::desc("To print syntax node kind"),
@@ -258,18 +265,22 @@ bool parseIncrementalEditArguments(SyntaxParsingCache *Cache,
return true;
}
bool useColoredOutput() {
return llvm::outs().has_colors() || options::ForceColoredOutput;
}
void printVisualNodeReuseInformation(SourceManager &SourceMgr,
unsigned BufferID,
SyntaxParsingCache *Cache) {
unsigned CurrentOffset = 0;
auto SourceText = SourceMgr.getEntireTextForBuffer(BufferID);
if (llvm::outs().has_colors()) {
if (useColoredOutput()) {
llvm::outs().changeColor(llvm::buffer_ostream::Colors::GREEN);
}
auto PrintReparsedRegion = [](StringRef SourceText, unsigned ReparseStart,
unsigned ReparseEnd) {
if (ReparseEnd != ReparseStart) {
if (llvm::outs().has_colors()) {
if (useColoredOutput()) {
llvm::outs().changeColor(llvm::buffer_ostream::Colors::RED);
} else {
llvm::outs() << "<reparse>";
@@ -278,7 +289,7 @@ void printVisualNodeReuseInformation(SourceManager &SourceMgr,
llvm::outs() << SourceText.substr(ReparseStart,
ReparseEnd - ReparseStart);
if (llvm::outs().has_colors()) {
if (useColoredOutput()) {
llvm::outs().changeColor(llvm::buffer_ostream::Colors::GREEN);
} else {
llvm::outs() << "</reparse>";
@@ -295,7 +306,7 @@ void printVisualNodeReuseInformation(SourceManager &SourceMgr,
CurrentOffset = ReuseRange.second;
}
PrintReparsedRegion(SourceText, CurrentOffset, SourceText.size());
if (llvm::outs().has_colors())
if (useColoredOutput())
llvm::outs().resetColor();
llvm::outs() << '\n';