diff options
| author | _ov4 <ov4@ov4.org> | 2026-10-21 14:06:54 +0000 |
|---|---|---|
| committer | _ov4 <ov4@ov4.org> | 2026-10-21 14:06:54 +0000 |
| commit | 899c2af576124df2918fd2ae976434df0ebbae5b (patch) | |
| tree | bb67a979eca30a6d447c2819576b81dc84741f17 /tools/driver/EmbeddedRunner.cc | |
| parent | b4ffb2d9498560557821714e2d4651279dd6817c (diff) | |
feat(driver): color diagnostics like gcc with -fdiagnostics-color
Add -fdiagnostics-color[=auto|always|never] and -fno-diagnostics-color.
The default auto mode colors only a terminal stderr (TERM set and not
dumb on Unix; a console that accepts VT sequences on Windows), so test
harnesses, the desktop bundle and other pipes stay plain.
Every diagnostic label now comes from cps::diagnosticPrefix, which emits
gcc's default GCC_COLORS bytes: bold locus, bold red error/fatal error,
bold magenta warning, bold cyan note. compilation terminated., runtime
messages of the compiled program and lld output are unchanged.
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Diffstat (limited to 'tools/driver/EmbeddedRunner.cc')
| -rw-r--r-- | tools/driver/EmbeddedRunner.cc | 43 |
1 files changed, 29 insertions, 14 deletions
diff --git a/tools/driver/EmbeddedRunner.cc b/tools/driver/EmbeddedRunner.cc index 2833cfd..b635e0f 100644 --- a/tools/driver/EmbeddedRunner.cc +++ b/tools/driver/EmbeddedRunner.cc @@ -3,6 +3,7 @@ #include "NativeCompiler.h" #include "llvm/Object/ObjectFile.h" #include "cps/CodeGen.h" +#include "cps/Diagnostics.h" #include "llvm/ExecutionEngine/Orc/AbsoluteSymbols.h" #include "llvm/ExecutionEngine/Orc/ExecutionUtils.h" #include "llvm/ExecutionEngine/Orc/LLJIT.h" @@ -194,17 +195,20 @@ int runEmbedded(const cps::CodeGen &CG, const std::vector<std::string> &Objects) // Pipes are fully buffered by default. Use the host CRT's stdout, also // preferred by the JIT on Windows, so OUTPUT reaches desktop clients now. if (std::setvbuf(stdout, nullptr, _IONBF, 0) != 0) { - llvm::errs() << "error: cannot configure embedded stdout\n"; + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "cannot configure embedded stdout\n"; return 1; } if (llvm::InitializeNativeTarget() || llvm::InitializeNativeTargetAsmPrinter()) { - llvm::errs() << "error: cannot initialize the embedded LLVM target\n"; + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "cannot initialize the embedded LLVM target\n"; return 1; } auto JTMBOrErr = llvm::orc::JITTargetMachineBuilder::detectHost(); if (!JTMBOrErr) { - llvm::errs() << "error: cannot detect the embedded JIT target: " + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "cannot detect the embedded JIT target: " << llvm::toString(JTMBOrErr.takeError()) << "\n"; return 1; } @@ -227,19 +231,22 @@ int runEmbedded(const cps::CodeGen &CG, const std::vector<std::string> &Objects) auto JITOrErr = Builder.create(); if (!JITOrErr) { - llvm::errs() << "error: cannot create the embedded LLVM engine: " + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "cannot create the embedded LLVM engine: " << llvm::toString(JITOrErr.takeError()) << "\n"; return 1; } std::unique_ptr<llvm::orc::LLJIT> J = std::move(*JITOrErr); if (auto Err = defineRuntimeSymbols(*J)) { - llvm::errs() << "error: cannot define embedded runtime symbols: " + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "cannot define embedded runtime symbols: " << llvm::toString(std::move(Err)) << "\n"; return 1; } if (auto Err = loadCxxRuntime(*J)) { - llvm::errs() << "error: cannot load embedded C++ runtime: " + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "cannot load embedded C++ runtime: " << llvm::toString(std::move(Err)) << "\n"; return 1; } @@ -250,7 +257,8 @@ int runEmbedded(const cps::CodeGen &CG, const std::vector<std::string> &Objects) auto CRT = llvm::orc::DynamicLibrarySearchGenerator::Load( "msvcrt.dll", J->getDataLayout().getGlobalPrefix()); if (!CRT) { - llvm::errs() << "error: cannot load the embedded C runtime: " + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "cannot load the embedded C runtime: " << llvm::toString(CRT.takeError()) << "\n"; return 1; } @@ -262,7 +270,8 @@ int runEmbedded(const cps::CodeGen &CG, const std::vector<std::string> &Objects) llvm::orc::ThreadSafeModule TSM = llvm::orc::cloneExternalModuleToContext(CG.getModule(), TSCtx); if (!TSM) { - llvm::errs() << "error: cannot clone the embedded module for JIT execution\n"; + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "cannot clone the embedded module for JIT execution\n"; return 1; } TSM.withModuleDo([&](llvm::Module &M) { @@ -271,32 +280,37 @@ int runEmbedded(const cps::CodeGen &CG, const std::vector<std::string> &Objects) }); if (auto Err = J->addIRModule(J->getMainJITDylib(), std::move(TSM))) { - llvm::errs() << "error: embedded LLVM execution failed: " + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "embedded LLVM execution failed: " << llvm::toString(std::move(Err)) << "\n"; return 1; } for (const auto &Path : Objects) { auto Buffer = llvm::MemoryBuffer::getFile(Path); if (!Buffer) { - llvm::errs() << "error: cannot read object file '" << Path << "': " + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "cannot read object file '" << Path << "': " << Buffer.getError().message() << "\n"; return 1; } if (auto Err = J->addObjectFile(std::move(*Buffer))) { - llvm::errs() << "error: cannot load object file '" << Path << "': " + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "cannot load object file '" << Path << "': " << llvm::toString(std::move(Err)) << "\n"; return 1; } } if (auto Err = J->initialize(J->getMainJITDylib())) { - llvm::errs() << "error: embedded LLVM execution failed: " + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "embedded LLVM execution failed: " << llvm::toString(std::move(Err)) << "\n"; return 1; } auto MainSymOrErr = J->lookup("main"); if (!MainSymOrErr) { - llvm::errs() << "error: embedded LLVM could not resolve main: " + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "embedded LLVM could not resolve main: " << llvm::toString(MainSymOrErr.takeError()) << "\n"; return 1; } @@ -306,7 +320,8 @@ int runEmbedded(const cps::CodeGen &CG, const std::vector<std::string> &Objects) // Run platform deinitializers while JIT code is still mapped; frame // deregistration runs when the JIT releases its memory below. if (auto Err = J->deinitialize(J->getMainJITDylib())) { - llvm::errs() << "error: embedded LLVM execution failed: " + llvm::errs() << cps::diagnosticPrefix(cps::Severity::Error, "") + << "embedded LLVM execution failed: " << llvm::toString(std::move(Err)) << "\n"; return 1; } |
