aboutsummaryrefslogtreecommitdiffstats
path: root/recipes/twitter-dm/webview.js
blob: 25f9baaee2f64246e564a8b1e9a42f51dfce1944 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
module.exports = Ferdi => {
  const getMessages = function getMessages() {
    let count = 0;
    const elem = document.querySelector('a[href="/messages"] div div');

    if (elem) {
      count = parseInt(elem.innerText, 10);
    }

    Ferdi.setBadge(count);
  };

  Ferdi.loop(getMessages);
};