summaryrefslogtreecommitdiff
path: root/src/logger.cc
blob: e6198da471f1c107945034c4bdc10aaee1670494 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
#include "logger.hh"

#include <cstdint>
#include <format>
#include <iostream>
#include <memory>
#include <string>
#include <string_view>
#include <utility>

namespace logger {

namespace {

class BaseLogger : public Logger {
 protected:
  enum class Level : uint8_t {
    kError,
    kWarning,
    kInfo,
    kDebug,
  };

 public:
  void err(std::string_view message) final { write(Level::kError, message); }

  void warn(std::string_view message) final { write(Level::kWarning, message); }

  void info(std::string_view message) final { write(Level::kInfo, message); }

#ifndef NDEBUG
  void dbg(std::string_view message) final { write(Level::kDebug, message); }
#endif

 protected:
  BaseLogger() = default;

  virtual void write(Level level, std::string_view message) = 0;
};

class NoopLogger : public BaseLogger {
 public:
  NoopLogger() = default;

 protected:
  void write(Level /* level */, std::string_view /* message */) override {}
};

class StderrLogger : public BaseLogger {
 public:
  explicit StderrLogger(bool verbose) : verbose_(verbose) {}

 protected:
  void write(Level level, std::string_view message) override {
    switch (level) {
      case Level::kError:
        std::cerr << "Error: ";
        break;
      case Level::kWarning:
        std::cerr << "Warning: ";
        break;
      case Level::kInfo:
        if (!verbose_)
          return;
        std::cerr << "Info: ";
        break;
      case Level::kDebug:
        if (!verbose_)
          return;
        std::cerr << "Debug: ";
        break;
    }
    std::cerr << message << '\n';
  }

 private:
  bool const verbose_;
};

class PrefixLogger : public Logger {
 public:
  PrefixLogger(Logger& logger, std::string prefix)
      : logger_(logger), prefix_(std::move(prefix)) {}

  void err(std::string_view message) override {
    logger_.err(std::format("{}: {}", prefix_, message));
  }

  void warn(std::string_view message) override {
    logger_.warn(std::format("{}: {}", prefix_, message));
  }

  void info(std::string_view message) override {
    logger_.info(std::format("{}: {}", prefix_, message));
  }

#ifndef NDEBUG
  void dbg(std::string_view message) override {
    logger_.dbg(std::format("{}: {}", prefix_, message));
  }
#endif

 private:
  Logger& logger_;
  std::string prefix_;
};

}  // namespace

[[nodiscard]]
std::unique_ptr<Logger> noop() {
  return std::make_unique<NoopLogger>();
}

[[nodiscard]]
std::unique_ptr<Logger> stderr(bool verbose) {
  return std::make_unique<StderrLogger>(verbose);
}

[[nodiscard]]
std::unique_ptr<Logger> prefix(Logger& logger, std::string prefix) {
  return std::make_unique<PrefixLogger>(logger, std::move(prefix));
}

}  // namespace logger