module.exports = Ferdium => {
  function getMessages() {
    const directMatches = document.querySelector('title').textContent
      .match('(?<=\\[)\\d+(?=])');
    Ferdium.setBadge(
      Ferdium.safeParseInt(directMatches !== null ? directMatches[0] : 0),
      document.querySelector('.mx_SpaceTreeLevel')
        .querySelectorAll('.mx_NotificationBadge_dot')
        .length
    );
  }

  Ferdium.loop(getMessages);
};