TextDiagnosticPrinter.h

//===--- TextDiagnosticPrinter.h - Text Diagnostic Client -------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// This is a concrete diagnostic client. In terminals that support it, the
// diagnostics are pretty-printed (colors + bold). The printing/flushing
// happens in HandleDiagnostics (usually called at the point when the
// diagnostic is generated).
//
//===----------------------------------------------------------------------===//
//
// Coding style: https://mlir.llvm.org/getting_started/DeveloperGuide/
//
//===----------------------------------------------------------------------===//
 
#ifndef FORTRAN_FRONTEND_TEXTDIAGNOSTICPRINTER_H
#define FORTRAN_FRONTEND_TEXTDIAGNOSTICPRINTER_H
 
#include "clang/Basic/Diagnostic.h"
#include "llvm/ADT/IntrusiveRefCntPtr.h"
#include "llvm/Support/raw_ostream.h"
 
namespace clang {
class DiagnosticOptions;
class DiagnosticsEngine;
} // namespace clang
 
using llvm::IntrusiveRefCntPtr;
using llvm::raw_ostream;
 
namespace Fortran::frontend {
class TextDiagnostic;
 
class TextDiagnosticPrinter : public clang::DiagnosticConsumer {
  raw_ostream &os;
  clang::DiagnosticOptions &diagOpts;

  std::string prefix;
 
public:
  TextDiagnosticPrinter(raw_ostream &os, clang::DiagnosticOptions &diags);
  ~TextDiagnosticPrinter() override;

  void setPrefix(std::string value) { prefix = std::move(value); }
 
  void HandleDiagnostic(clang::DiagnosticsEngine::Level level,
                        const clang::Diagnostic &info) override;
 
  void printLocForRemarks(llvm::raw_svector_ostream &diagMessageStream,
                          llvm::StringRef &diagMsg);
};
 
} // namespace Fortran::frontend
 
#endif