module.exports = Ferdium => {
  const _parseNewCount = (text) => {
    const match = text.match(/\d+/);
    return match ? Ferdium.safeParseInt(match[0]) : 0;
  };

  const getMessages = () => {
    const directCountElement = document.querySelector(
      '.filter-list.js-notification-inboxes .count',
    );
    let directCount = directCountElement
      ? Ferdium.safeParseInt(directCountElement.textContent)
      : 0;
    
    const newCountElement = document.querySelector('a.h6[href="/notifications?query="]');
    const newCount = newCountElement ?
      _parseNewCount(newCountElement.textContent) : 0;
    directCount += newCount;

    const indirectCount = document.querySelector(
      '[class*="mail-status unread"]:not([hidden])',
    ) ? 1 : 0;
    Ferdium.setBadge(directCount, indirectCount);
  };

  Ferdium.loop(getMessages);
};