module.exports = (Ferdi) => {
  const getMessages = () => {
    let directCount = 0;
    const element = document.querySelector('.ws-navigation-button__indicator.ws-navigation-button-indicator');
    if (element) {
      directCount = Ferdi.safeParseInt(element.innerText);
    }

    Ferdi.setBadge(directCount);
  }

  Ferdi.loop(getMessages);
};