twitter-api.js

  /***************************************************************************/
  /* Augeo.io is a web application that uses Natural Language Processing to  */
  /* classify a user's internet activity into different 'skills'.            */
  /* Copyright (C) 2016 Brian Redd                                           */
  /*                                                                         */
  /* This program is free software: you can redistribute it and/or modify    */
  /* it under the terms of the GNU General Public License as published by    */
  /* the Free Software Foundation, either version 3 of the License, or       */
  /* (at your option) any later version.                                     */
  /*                                                                         */
  /* This program is distributed in the hope that it will be useful,         */
  /* but WITHOUT ANY WARRANTY; without even the implied warranty of          */
  /* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the           */
  /* GNU General Public License for more details.                            */
  /*                                                                         */
  /* You should have received a copy of the GNU General Public License       */
  /* along with this program.  If not, see <http://www.gnu.org/licenses/>.   */
  /***************************************************************************/

  /***************************************************************************/
  /* Description: Handles requests to Augeo's twitter-api                    */
  /***************************************************************************/

  // Required libraries
  var Mongoose = require('mongoose');
  var TwitterRouter = require('express').Router();

  // Required local modules
  var AugeoUtility = require('../utility/augeo-utility');
  var AugeoValidator = require('../validator/augeo-validator');
  var Logger = require('../module/logger');
  var QueueService = require('../service/queue-service');
  var TwitterInterfaceService = require('../interface-service/twitter-interface-service');
  var TwitterMentionTask = require('../queue-task/twitter/event/twitter-mention-task');
  var TwitterService = require('../service/twitter-service');
  var TwitterTweetTask = require('../queue-task/twitter/event/twitter-tweet-task');
  var UserService = require('../service/user-service');

  // Constants
  var API = 'twitter-api';
  var CALLBACK = '/callback';
  var GET_AUTHENTICATION_DATA = '/getAuthenticationData';
  var GET_QUEUE_WAIT_TIMES = '/getQueueWaitTimes';
  var INVALID_SESSION = 'Invalid session';

  // Global variables
  var log = new Logger();

  /***************************************************************************/
  /* GET Requests                                                            */
  /***************************************************************************/

  TwitterRouter.get(CALLBACK, function(request, response) {
    var username = AugeoValidator.isSessionValid(request) ? request.session.user.username : null;

    var rollback = function(message) {
      log.functionError(API, CALLBACK, username, message);

      // Remove invalid entry from TWITTER_USER
      if(username && request.session.user._id) {
        var logData = AugeoUtility.formatLogData(API+CALLBACK, username);
        TwitterService.removeUser(request.session.user._id, logData, function(){}, function(){});
      }
      response.redirect(process.env.AUGEO_HOME + '/signup/error'); // Redirect to signup error page
    };

    if(username) {
      log.functionCall(API, CALLBACK, null, username);
      var logData = AugeoUtility.formatLogData(API+CALLBACK, username);

      var userId = request.session.user._id;

      // Get user's oauth secret token
      TwitterService.getUserSecretToken(userId, logData, function (oauthSecretToken) {

        TwitterInterfaceService.getOAuthAccessToken(request.query, oauthSecretToken, logData, function (oauth_access_token, oauth_access_token_secret, screenName) {

          // Check if access token exists
          TwitterService.checkExistingAccessToken(oauth_access_token, logData, function (accessTokenExists) {

            if (accessTokenExists === false) {

              // Initialize twitterMessenger
              var twitterMessenger = TwitterInterfaceService.createTwitterMessenger(oauth_access_token, oauth_access_token_secret, logData);

              // Get user's Twitter information
              TwitterInterfaceService.getTwitterUser(twitterMessenger, screenName, logData, function (userData) {

                userData.accessToken = oauth_access_token;
                userData.secretAccessToken = oauth_access_token_secret;

                // Update user's Twitter information
                TwitterService.updateTwitterInfo(userId, request.session.user, userData, logData, function (updatedUser) {

                  // Update session user object
                  request.session.user = updatedUser;

                  if (process.env.TEST != 'true') {

                    var tweetTask = new TwitterTweetTask(updatedUser, userData, null, logData);
                    QueueService.tweetEventQueue.addTask(tweetTask, logData);

                    var mentionTask = new TwitterMentionTask(updatedUser, userData, null, logData);
                    QueueService.mentionEventQueue.addTask(mentionTask, logData);

                    QueueService.twitterConnectQueue.connectToTwitter(logData, function(){});
                  }

                  response.redirect(process.env.AUGEO_HOME + '/interfaceHistory');
                }, rollback); // End updateTwitterInfo

              }, rollback); // End getTwitterUser
            } else { // end accessTokenExists
              rollback();
            }
          }, rollback); // End checkExistingAccessToken
        }, rollback);
      }, rollback);
    } else {
      rollback(INVALID_SESSION);
    }
  });

  TwitterRouter.get(GET_AUTHENTICATION_DATA, function(request, response) {
    var username = AugeoValidator.isSessionValid(request) ? request.session.user.username : null;
    var logData;

    var rollback = function(message) {
      log.functionError(API, GET_AUTHENTICATION_DATA, username, message);
      response.status(401).send('Twitter authentication failed. Please try again.');
    };

    if(username) {
      log.functionCall(API, GET_AUTHENTICATION_DATA, null, request.session.user.username);
      logData = AugeoUtility.formatLogData(API+GET_AUTHENTICATION_DATA, username);
      TwitterInterfaceService.getOAuthRequestToken(logData, function (oauthToken, oauthTokenSecret) {
        TwitterService.addUserSecretToken(request.session.user._id, oauthTokenSecret, logData, function () {
          response.status(200).send({token: oauthToken});
        }, rollback);
      }, rollback);
    } else {
      rollback(INVALID_SESSION);
    }
  });

  TwitterRouter.get(GET_QUEUE_WAIT_TIMES, function(request, response) {
    var username = AugeoValidator.isSessionValid(request) ? request.session.user.username : null;

    var rollback = function(message) {
      log.functionError(API, GET_QUEUE_WAIT_TIMES, username, message);
      response.sendStatus(401);
    };

    // If user exists in session get profile data
    if(username) {
      log.functionCall(API, GET_QUEUE_WAIT_TIMES, null, username);
      var logData = AugeoUtility.formatLogData(API+GET_QUEUE_WAIT_TIMES, username);

      var waitTimes = new Array();
      if(request.session.user.twitter) {
        var userId = Mongoose.Types.ObjectId(request.session.user._id);
        waitTimes.push(QueueService.tweetEventQueue.getUserWaitTime(userId, logData));
        waitTimes.push(QueueService.mentionEventQueue.getUserWaitTime(userId, logData));
      } else {
        waitTimes.push(QueueService.tweetEventQueue.getWaitTime(logData));
        waitTimes.push(QueueService.mentionEventQueue.getWaitTime(logData));
      }

      response.status(200).json({waitTimes:waitTimes});
    } else { // If the user doesn't exist in session respond with "Unauthorized" HTTP code
      rollback(INVALID_SESSION);
    }
  });

  module.exports = TwitterRouter;