aboutsummaryrefslogtreecommitdiffstats
path: root/recipes/github/webview.js
blob: b73e96e263cdf206705685f8532576369f634e2e (plain) (blame)
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
module.exports = Ferdium => {
  const _parseNewCount = (text) => {
    const match = text.match(/\d+/);
    return match ? Ferdium.safeParseInt(match[0]) : 0;
  };

  const getMessages = () => {
    const directCountElement = document.querySelector(
      '.filter-list.js-notification-inboxes .count',
    );
    let directCount = directCountElement
      ? Ferdium.safeParseInt(directCountElement.textContent)
      : 0;
    
    const newCountElement = document.querySelector('a.h6[href="/notifications?query="]');
    const newCount = newCountElement ?
      _parseNewCount(newCountElement.textContent) : 0;
    directCount += newCount;

    const indirectCount = document.querySelector(
      '[class*="mail-status unread"]:not([hidden])',
    ) ? 1 : 0;
    Ferdium.setBadge(directCount, indirectCount);
  };

  Ferdium.loop(getMessages);
};