tools/fidl/fidlc/src/reporter.h

// Copyright 2017 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef TOOLS_FIDL_FIDLC_SRC_REPORTER_H_
#define TOOLS_FIDL_FIDLC_SRC_REPORTER_H_

#include <cassert>
#include <string>
#include <string_view>
#include <vector>

#include "tools/fidl/fidlc/src/diagnostic_types.h"
#include "tools/fidl/fidlc/src/diagnostics.h"
#include "tools/fidl/fidlc/src/source_span.h"
#include "tools/fidl/fidlc/src/utils.h"

namespace fidlc {

class Reporter {
 public:
  Reporter() = default;
  Reporter(const Reporter&) = delete;

  class Counts {
   public:
    explicit Counts(const Reporter* reporter)
        : reporter_(reporter), num_errors_(reporter->errors().size()) {}
    bool NoNewErrors() const { return NumNewErrors() == 0; }
    size_t NumNewErrors() const { return reporter_->errors().size() - num_errors_; }

   private:
    const Reporter* reporter_;
    const size_t num_errors_;
  };

  template <ErrorId Id, typename... Args>
  bool Fail(const ErrorDef<Id, Args...>& def, SourceSpan span,
            const cpp20::type_identity_t<Args>&... args) {
    static_assert(Id <= kNumDiagnosticDefs,
                  "please add this ErrorDef to kAllDiagnosticDefs in diagnostics.h");
    Report(Diagnostic::MakeError(def, span, args...));
    return false;
  }

  template <ErrorId Id, typename... Args>
  void Warn(const WarningDef<Id, Args...>& def, SourceSpan span,
            const cpp20::type_identity_t<Args>&... args) {
    static_assert(Id <= kNumDiagnosticDefs,
                  "please add this WarningDef to kAllDiagnosticDefs in diagnostics.h");
    Report(Diagnostic::MakeWarning(def, span, args...));
  }

  // Reports an error or warning.
  void Report(std::unique_ptr<Diagnostic> diag);

  // Combines errors and warnings and sorts by (file, span).
  std::vector<Diagnostic*> Diagnostics() const;
  // Prints a report based on Diagnostics() in text format, with ANSI color
  // escape codes if enable_color is true.
  void PrintReports(bool enable_color) const;
  // Prints a report based on Diagnostics() in JSON format.
  void PrintReportsJson() const;
  // Creates a checkpoint. This lets you detect how many new errors
  // have been added since the checkpoint.
  Counts Checkpoint() const { return Counts(this); }
  const std::vector<std::unique_ptr<Diagnostic>>& errors() const { return errors_; }
  const std::vector<std::unique_ptr<Diagnostic>>& warnings() const { return warnings_; }
  void set_warnings_as_errors(bool value) { warnings_as_errors_ = value; }

  // Formats a diagnostic message for the command line, displaying the filename,
  // line, column, diagnostic kind, and the full line where the span occurs,
  // with the span indicated by an ASCII "squiggle" below it. Optionally adds
  // color via ANSI escape codes.
  static std::string Format(std::string_view qualifier, SourceSpan span, std::string_view message,
                            bool color);

 private:
  void AddError(std::unique_ptr<Diagnostic> error);
  void AddWarning(std::unique_ptr<Diagnostic> warning);

  bool warnings_as_errors_ = false;

  // The reporter collects error and warnings separately so that we can easily
  // keep track of the current number of errors during compilation. The number
  // of errors is used to determine whether the parser is in an `Ok` state.
  std::vector<std::unique_ptr<Diagnostic>> errors_;
  std::vector<std::unique_ptr<Diagnostic>> warnings_;
};

}  // namespace fidlc

#endif  // TOOLS_FIDL_FIDLC_SRC_REPORTER_H_