src/webview/badge.ts


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

import { ipcRenderer } from 'electron';

const debug = require('debug')('Ferdi:Plugin:BadgeHandler');

export class BadgeHandler {
  countCache: { direct: number; indirect: number; };

  constructor() {
    this.countCache = {
      direct: 0,
      indirect: 0,
    };
  }

  _normalizeNumber(count: string | number) {
    // Parse number to integer
    // This will correct errors that recipes may introduce, e.g.
    // by sending a String instead of an integer
    const parsedNumber = parseInt(count.toString(), 10);
    const adjustedNumber = Number.isNaN(parsedNumber) ? 0 : parsedNumber;
    return Math.max(adjustedNumber, 0);
  }

  setBadge(direct: number, indirect: number) {
    if (this.countCache.direct === direct
        && this.countCache.indirect === indirect) {
      return;
    }

    const count = {
      direct: this._normalizeNumber(direct),
      indirect: this._normalizeNumber(indirect),
    };

    debug('Sending badge count to host', count);
    ipcRenderer.sendToHost('message-counts', count);

    Object.assign(this.countCache, count);
  }
}