summaryrefslogtreecommitdiff
path: root/std/log/handlers.ts
blob: 32b0525e73d25c3b461e1a4fa0f268f39dddce85 (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
// Copyright 2018-2020 the Deno authors. All rights reserved. MIT license.
const { open } = Deno;
type File = Deno.File;
type Writer = Deno.Writer;
import { getLevelByName, LogLevel } from "./levels.ts";
import { LogRecord } from "./logger.ts";
import { red, yellow, blue, bold } from "../fmt/colors.ts";

const DEFAULT_FORMATTER = "{levelName} {msg}";
type FormatterFunction = (logRecord: LogRecord) => string;

interface HandlerOptions {
  formatter?: string | FormatterFunction;
}

export class BaseHandler {
  level: number;
  levelName: string;
  formatter: string | FormatterFunction;

  constructor(levelName: string, options: HandlerOptions = {}) {
    this.level = getLevelByName(levelName);
    this.levelName = levelName;

    this.formatter = options.formatter || DEFAULT_FORMATTER;
  }

  handle(logRecord: LogRecord): void {
    if (this.level > logRecord.level) return;

    const msg = this.format(logRecord);
    return this.log(msg);
  }

  format(logRecord: LogRecord): string {
    if (this.formatter instanceof Function) {
      return this.formatter(logRecord);
    }

    return this.formatter.replace(/{(\S+)}/g, (match, p1): string => {
      const value = logRecord[p1 as keyof LogRecord];

      // do not interpolate missing values
      if (!value) {
        return match;
      }

      return String(value);
    });
  }

  log(_msg: string): void {}
  async setup(): Promise<void> {}
  async destroy(): Promise<void> {}
}

export class ConsoleHandler extends BaseHandler {
  format(logRecord: LogRecord): string {
    let msg = super.format(logRecord);

    switch (logRecord.level) {
      case LogLevel.INFO:
        msg = blue(msg);
        break;
      case LogLevel.WARNING:
        msg = yellow(msg);
        break;
      case LogLevel.ERROR:
        msg = red(msg);
        break;
      case LogLevel.CRITICAL:
        msg = bold(red(msg));
        break;
      default:
        break;
    }

    return msg;
  }

  log(msg: string): void {
    console.log(msg);
  }
}

export abstract class WriterHandler extends BaseHandler {
  protected _writer!: Writer;
  private _encoder = new TextEncoder();

  log(msg: string): void {
    this._writer.write(this._encoder.encode(msg + "\n"));
  }
}

interface FileHandlerOptions extends HandlerOptions {
  filename: string;
}

export class FileHandler extends WriterHandler {
  private _file!: File;
  private _filename: string;

  constructor(levelName: string, options: FileHandlerOptions) {
    super(levelName, options);
    this._filename = options.filename;
  }

  async setup(): Promise<void> {
    // open file in append mode - write only
    this._file = await open(this._filename, "a");
    this._writer = this._file;
  }

  async destroy(): Promise<void> {
    await this._file.close();
  }
}