module.exports = (Franz) => {
  function getMessages() {
    let direct = 0;
    const indirect = 0;
    const badgeDiv = document.querySelector('.notion-sidebar-container > div > div > div > :nth-child(4) > :nth-child(2) > div > :nth-child(3) > div > div');
    if (badgeDiv) {
      direct = parseInt(badgeDiv.innerText);
    }

    Franz.setBadge(direct, indirect);
  }

  Franz.loop(getMessages);
};