aboutsummaryrefslogtreecommitdiffstats
path: root/uncompressed/telegram-react/webview.js
blob: 223e9142dd22f6c089de522f7dc7dc7c46ffb533 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
"use strict";

const path = require("path");

module.exports = Franz => {
  const getMessages = function getMessages() {
    let count = 0;
    const elements = document.querySelectorAll(".dialog-badge:not(.dialog-badge-muted)");
    if (elements) {
      for (let i = 0; i < elements.length; i += 1) {
        if (elements[i].querySelector("span").innerHTML !== 0) {
          count += parseInt(elements[i].querySelector("span").innerHTML);
        }
      }
    }
    Franz.setBadge(count);
  };
  Franz.loop(getMessages);
};