Mailspring/app/internal_packages/unread-notifications/lib/main.es6

import _ from 'underscore'
import {
  Thread,
  Actions,
  Message,
  SoundRegistry,
  NativeNotifications,
  DatabaseStore,
} from 'nylas-exports';

export class Notifier {
  constructor() {
    this.activationTime = Date.now();
    this.unnotifiedQueue = [];
    this.hasScheduledNotify = false;

    this.activeNotifications = {};
    this.unlisteners = [
      DatabaseStore.listen(this._onDatabaseChanged, this),
    ];
  }

  unlisten() {
    for (const unlisten of this.unlisteners) {
      unlisten();
    }
  }

  // async for testing
  async _onDatabaseChanged({objectClass, objects}) {
    if (objectClass === Thread.name) {
      // Ensure notifications are dismissed when the user reads a thread
      objects.forEach(({id, unread}) => {
        if (!unread && this.activeNotifications[id]) {
          this.activeNotifications[id].forEach((n) => n.close());
          delete this.activeNotifications[id];
        }
      });
    }

    if (objectClass === Message.name) {
      if (NylasEnv.config.get('core.notifications.enabled') === false) {
        return;
      }
      const newUnread = objects.filter((msg) => {
        // ensure the message is unread
        if (msg.unread !== true) return false;
        // ensure the message was just saved (eg: this is not a modification)
        if (msg.version !== 1) return false;
        // ensure the message was received after the app launched (eg: not syncing an old email)
        if (!msg.date || msg.date.valueOf() < this.activationTime) return false;
        // ensure the message is from someone else
        if (msg.isFromMe()) return false;
        return true;
      });

      if (newUnread.length > 0) {
        await this._onNewMessagesReceived(newUnread);
      }
    }
  }

  _notifyAll() {
    NativeNotifications.displayNotification({
      title: `${this.unnotifiedQueue.length} Unread Messages`,
      tag: 'unread-update',
    });
    this.unnotifiedQueue = [];
  }

  _notifyOne({message, thread}) {
    const from = (message.from[0]) ? message.from[0].displayName() : "Unknown";
    const title = from;
    let subtitle = null;
    let body = null;
    if (message.subject && message.subject.length > 0) {
      subtitle = message.subject;
      body = message.snippet;
    } else {
      subtitle = message.snippet
      body = null
    }

    const notification = NativeNotifications.displayNotification({
      title: title,
      subtitle: subtitle,
      body: body,
      canReply: true,
      tag: 'unread-update',
      onActivate: ({response, activationType}) => {
        if ((activationType === 'replied') && response && typeof response === 'string') {
          Actions.sendQuickReply({thread, message}, response);
        } else {
          NylasEnv.displayWindow()
        }

        if (!thread) {
          NylasEnv.showErrorDialog(`Can't find that thread`)
          return
        }
        Actions.ensureCategoryIsFocused('inbox', thread.accountId);
        Actions.setFocus({collection: 'thread', item: thread});
      },
    });

    if (!this.activeNotifications[thread.id]) {
      this.activeNotifications[thread.id] = [notification];
    } else {
      this.activeNotifications[thread.id].push(notification);
    }
  }

  _notifyMessages() {
    if (this.unnotifiedQueue.length >= 5) {
      this._notifyAll()
    } else if (this.unnotifiedQueue.length > 0) {
      this._notifyOne(this.unnotifiedQueue.shift());
    }

    this.hasScheduledNotify = false;
    if (this.unnotifiedQueue.length > 0) {
      setTimeout(() => this._notifyMessages(), 2000);
      this.hasScheduledNotify = true;
    }
  }

  _onNewMessagesReceived(newMessages) {
    // For each message, find it's corresponding thread. First, look to see
    // if it's already in the `incoming` payload (sent via delta sync
    // at the same time as the message.) If it's not, try loading it from
    // the local cache.

    const threadIds = {};
    for (const {threadId} of newMessages) {
      threadIds[threadId] = true;
    }

    // TODO: Use xGMLabels + folder on message to identify which ones
    // are in the inbox to avoid needing threads here.
    return DatabaseStore.findAll(Thread, Thread.attributes.id.in(Object.keys(threadIds))).then((threadsArray) => {
      const threads = {};
      for (const t of threadsArray) {
        threads[t.id] = t;
      }

      // Filter new messages to just the ones in the inbox
      const newMessagesInInbox = newMessages.filter(({threadId}) => {
        return threads[threadId] && threads[threadId].categories.find(c => c.role === 'inbox');
      })

      if (newMessagesInInbox.length === 0) {
        return;
      }

      for (const msg of newMessagesInInbox) {
        this.unnotifiedQueue.push({message: msg, thread: threads[msg.threadId]});
      }
      if (!this.hasScheduledNotify) {
        if (NylasEnv.config.get("core.notifications.sounds")) {
          this._playNewMailSound = this._playNewMailSound || _.debounce(() => SoundRegistry.playSound('new-mail'), 5000, true);
          this._playNewMailSound();
        }
        this._notifyMessages();
      }
    });
  }
}

export const config = {
  enabled: {
    'type': 'boolean',
    'default': true,
  },
};

export function activate() {
  this.notifier = new Notifier();
}

export function deactivate() {
  this.notifier.unlisten();
}