habitica/website/client/js/services/pusherService.js

'use strict';

angular.module('habitrpg')
.factory('Pusher', ['$rootScope', 'STORAGE_SETTINGS_ID', 'Groups', 'Shared',
  function($rootScope, STORAGE_SETTINGS_ID, Groups, Shared) {
    var settings = JSON.parse(localStorage.getItem(STORAGE_SETTINGS_ID));
    var IS_PUSHER_ENABLED = window.env['PUSHER:ENABLED'] === 'true';

    var partyId;
    var onActivityEvent;

    var api = {
      pusher: undefined,
      socketId: undefined, // when defined the user is connected
    };
    var tabIdKey = 'habitica-active-tab';
    var tabId = Shared.uuid();

    function connectToPusher (partyId) {
      localStorage.setItem(tabIdKey, tabId);
      window.onbeforeunload = function () {
        localStorage.removeItem(tabIdKey);
      }

      api.pusher = new Pusher(window.env['PUSHER:KEY'], {
        encrypted: true,
        authEndpoint: '/api/v3/user/auth/pusher',
        auth: {
          headers: {
            'x-api-user': settings && settings.auth && settings.auth.apiId,
            'x-api-key': settings && settings.auth && settings.auth.apiToken,
          },
        },
      });

      // Disconnect after 30m of inactivity
      var DISCONNECTION_AFTER = 1800000; // 30m
      var disconnectionTimeout;

      var awaitIdle = function() {
        if(disconnectionTimeout) clearTimeout(disconnectionTimeout);
        disconnectionTimeout = setTimeout(function () {
          $(document).off('mousemove keydown mousedown touchstart', awaitIdle);
          disconnectPusher();
        }, DISCONNECTION_AFTER);
      };

      awaitIdle();
      $(document).on('mousemove keydown mousedown touchstart', awaitIdle);

      api.pusher.connection.bind('error', function(err) {
        console.error(err);
        // TODO if( err.data.code === 4004 ) detected connection limit
      });

      api.pusher.connection.bind('connected', function () {
        api.socketId = api.pusher.connection.socket_id;
      });

      if (!partyId) return;

      var partyChannelName = 'presence-group-' + partyId;
      var partyChannel = api.pusher.subscribe(partyChannelName);

      // When an error occurs while joining the channel
      partyChannel.bind('pusher:subscription_error', function(status) {
        console.error('Impossible to join the Pusher channel for your party, status: ', status);
      });

      // When the user correctly enters the party channel
      partyChannel.bind('pusher:subscription_succeeded', function(members) {
        // TODO members = [{id, info}]
      });

      // When a member enters the party channel
      partyChannel.bind('pusher:member_added', function(member) {
        // TODO member = {id, info}
      });

      // When a member leaves the party channel
      partyChannel.bind('pusher:member_removed', function(member) {
        // TODO member = {id, info}
      });

      // When the user is booted from the party, they get disconnected from Pusher
      partyChannel.bind('user-removed', function (data) {
        if (data.userId === user._id) {
          api.pusher.unsubscribe(partyChannelName);
        }
      });

      // Same when the user leaves the party
      partyChannel.bind('user-left', function (data) {
        if (data.userId === user._id) {
          api.pusher.unsubscribe(partyChannelName);
        }
      });

      // When a new chat message is posted
      partyChannel.bind('new-chat', function (data) {
        Groups.party().then(function () {
          // Groups.data.party.chat.unshift(data);
          // Groups.data.party.chat.splice(200);
        });
      });
    };

    function disconnectPusher () {
      api.pusher.disconnect();

      var awaitActivity = function() {
        $(document).off('mousemove keydown mousedown touchstart', awaitActivity);
        connectToPusher(partyId);
        if (!localStorage.getItem(tabIdKey) || localStorage.getItem(tabIdKey) === tabId) {
          connectToPusher(partyId);
        }
      };

      $(document).on('mousemove keydown mousedown touchstart', awaitActivity);
    };

    // Setup chat channels once app is ready, only for parties for now
    var clearAppLoadedListener = $rootScope.$watch('appLoaded', function (after) {
      if (!after) return;
      clearAppLoadedListener(); // clean the event listerner

      if (!IS_PUSHER_ENABLED) return;

      var user = $rootScope.user;

      // Connect the user to Pusher and to the party's chat channel
      partyId = user && $rootScope.user.party && $rootScope.user.party._id;
      if (!partyId) return;

      // See if another tab is already connected to Pusher
      if (!localStorage.getItem(tabIdKey)) {
        connectToPusher(partyId);
      }

      // when a tab is closed, connect the next one
      // wait between 100 and 500ms to avoid two tabs connecting at the same time
      window.addEventListener('storage', function(e) {
        if (e.key === tabIdKey && e.newValue === null) {
          setTimeout(function () {
            if (!localStorage.getItem(tabIdKey)) {
              connectToPusher(partyId);
            }
          }, Math.floor(Math.random() * 501) + 100);
        }
      });
    });

    return api;
  }]);