meanTorrent/modules/messages/server/controllers/messages.server.controller.js

'use strict';

/**
 * Module dependencies
 */
var path = require('path'),
  config = require(path.resolve('./config/config')),
  mongoose = require('mongoose'),
  errorHandler = require(path.resolve('./modules/core/server/controllers/errors.server.controller')),
  User = mongoose.model('User'),
  Message = mongoose.model('Message'),
  AdminMessage = mongoose.model('AdminMessage'),
  async = require('async');

/**
 * create a Message
 * @param req
 * @param res
 */
exports.create = function (req, res) {
  var msg = new Message(req.body);
  msg.from_user = req.user._id;
  msg.from_status = 1;
  msg.to_status = 0;

  if (!msg.to_user || !mongoose.Types.ObjectId.isValid(msg.to_user)) {
    return res.status(400).send({
      message: 'Receiver user id is invalid'
    });
  }

  msg.save(function (err) {
    if (err) {
      return res.status(422).send({
        message: errorHandler.getErrorMessage(err)
      });
    } else {
      res.json(msg);
    }
  });
};

/**
 * list Messages
 * @param req
 * @param res
 */
exports.list = function (req, res) {
  var findMessage = function (callback) {
    Message.find({
      $or: [
        {from_user: req.user._id},
        {to_user: req.user._id}
      ]
    })
      .sort('-updatedat -createdat')
      .populate('from_user', 'displayName profileImageURL uploaded downloaded')
      .populate('to_user', 'displayName profileImageURL uploaded downloaded')
      .populate({
        path: '_replies.from_user',
        select: 'displayName profileImageURL uploaded downloaded',
        model: 'User'
      })
      .populate({
        path: '_replies.to_user',
        select: 'displayName profileImageURL uploaded downloaded',
        model: 'User'
      })
      .exec(function (err, messages) {
        if (err) {
          callback(err, null);
        } else {
          callback(null, messages);
        }
      });
  };

  var findAdminMessage = function (callback) {
    AdminMessage.find({
      createdat: {$gt: req.user.created}
    })
      .sort('-createdat')
      .populate('from_user', 'displayName profileImageURL uploaded downloaded')
      .exec(function (err, messages) {
        if (err) {
          callback(err, null);
        } else {
          callback(null, messages);
        }
      });
  };

  async.parallel([findMessage, findAdminMessage], function (err, results) {
    if (err) {
      return res.status(422).send(err);
    } else {
      res.json(results[0].concat(results[1]));
    }
  });
};

/**
 * delete Message
 * @param req
 * @param res
 */
exports.delete = function (req, res) {
  if (req.params.messageId) {
    var message = req.message;
    message.remove(function (err) {
      if (err) {
        return res.status(422).send({
          message: errorHandler.getErrorMessage(err)
        });
      } else {
        res.json(message);
      }
    });

  } else {
    if (req.query.ids) {
      Message.remove({
        _id: {$in: req.query.ids}
      }, function (err) {
        if (err) {
          return res.status(422).send({
            message: errorHandler.getErrorMessage(err)
          });
        } else {
          return res.status(200).send({
            message: 'delete successfully'
          });
        }
      });
    }
  }
};

/**
 * Update
 */
exports.update = function (req, res) {
  var message = req.message;

  if (req.body.from_status) {
    message.from_status = req.body.from_status;
  }
  if (req.body.to_status) {
    message.to_status = req.body.to_status;
  }

  message.save(function (err) {
    if (err) {
      return res.status(422).send({
        message: errorHandler.getErrorMessage(err)
      });
    } else {
      res.json(message);
    }
  });
};

/**
 * createReply
 * @param req
 * @param res
 */
exports.createReply = function (req, res) {
  var reply = new Message(req.body);

  var message = req.message;
  message._replies.push(reply);
  message.updatedat = Date.now();

  if (message.from_user._id.equals(req.user._id)) {
    message.to_status = 0;
  } else {
    message.from_status = 0;
  }

  message.save(function (err) {
    if (err) {
      return res.status(422).send({
        message: errorHandler.getErrorMessage(err)
      });
    } else {
      Message.populate(message._replies, {
        path: 'from_user to_user',
        select: 'displayName profileImageURL uploaded downloaded'
      }, function (err, t) {
        if (err) {
          return res.status(422).send({
            message: errorHandler.getErrorMessage(err)
          });
        } else {
          res.json(message);
        }
      });
    }
  });
};

/**
 * countUnread
 * @param req
 * @param res
 */
exports.countUnread = function (req, res) {
  if (!req.user) {
    return res.status(422).send({
      message: 'User is not signed in'
    });
  }

  //active user update method to update some fields value
  req.user.globalUpdateMethod();

  var countFrom = function (callback) {
    Message.count({
      type: 'user',
      from_user: req.user._id,
      from_status: 0
    }, function (err, count) {
      if (err) {
        callback(err, null);
      } else {
        callback(null, count);
      }
    });
  };
  var countTo = function (callback) {
    Message.count({
      type: 'user',
      to_user: req.user._id,
      to_status: 0
    }, function (err, count) {
      if (err) {
        callback(err, null);
      } else {
        callback(null, count);
      }
    });
  };
  var countServer = function (callback) {
    Message.count({
      type: 'server',
      to_user: req.user._id,
      to_status: 0
    }, function (err, count) {
      if (err) {
        callback(err, null);
      } else {
        callback(null, count);
      }
    });
  };
  var countAdminSystemMessage = function (callback) {
    AdminMessage.count({
      type: 'system',
      createdat: {$gt: req.user.created},
      _readers: {$not: {$in: [req.user._id]}}

    }, function (err, count) {
      if (err) {
        callback(err, null);
      } else {
        callback(null, count);
      }
    });
  };
  var countAdminAdvertMessage = function (callback) {
    AdminMessage.count({
      type: 'advert',
      createdat: {$gt: req.user.created},
      _readers: {$not: {$in: [req.user._id]}}

    }, function (err, count) {
      if (err) {
        callback(err, null);
      } else {
        callback(null, count);
      }
    });
  };
  var countAdminNoticeMessage = function (callback) {
    AdminMessage.count({
      type: 'notice',
      createdat: {$gt: req.user.created},
      _readers: {$not: {$in: [req.user._id]}}

    }, function (err, count) {
      if (err) {
        callback(err, null);
      } else {
        callback(null, count);
      }
    });
  };

  async.parallel([countFrom, countTo, countServer, countAdminSystemMessage, countAdminAdvertMessage, countAdminNoticeMessage], function (err, results) {
    if (err) {
      return res.status(422).send(err);
    } else {
      res.json({
        countAll: results[0] + results[1] + results[2] + results[3] + results[4] + results[5],
        countFrom: results[0],
        countTo: results[1],
        countServer: results[2],
        countSystem: results[3],
        countAdvert: results[4],
        countNotice: results[5]
      });
    }
  });

};

/**
 * Invitation middleware
 */
exports.messageByID = function (req, res, next, id) {

  if (!mongoose.Types.ObjectId.isValid(id)) {
    return res.status(400).send({
      message: 'Message is invalid'
    });
  }

  Message.findById(id)
    .populate('from_user', 'displayName profileImageURL uploaded downloaded')
    .populate('to_user', 'displayName profileImageURL uploaded downloaded')
    .populate({
      path: '_replies.from_user',
      select: 'displayName profileImageURL uploaded downloaded',
      model: 'User'
    })
    .populate({
      path: '_replies.to_user',
      select: 'displayName profileImageURL uploaded downloaded',
      model: 'User'
    })
    .exec(function (err, message) {
      if (err) {
        return next(err);
      } else if (!message) {
        return res.status(404).send({
          message: 'No message with that identifier has been found'
        });
      }
      req.message = message;
      next();
    });
};