umpack-express/models/userDevice.js

user management pack for express framework app

var mongoose = require('mongoose');
var Promise = require('bluebird');
var config = require('../config');

mongoose.Promise = Promise;

var UserDeviceSchema = new mongoose.Schema({
  userName: String,
  devices: []
}, { usePushEach: true });

UserDeviceSchema.methods.canAccess = function(deviceToken) {
  for (var i = 0; i < this.devices.length; i++) {
    if (this.devices[i].deviceToken === deviceToken) return this.devices[i].canAccess;
  }

  return false;
};

UserDeviceSchema.methods.markDeviceUsage = function(deviceToken) {
  for (var i = 0; i < this.devices.length; i++) {
    if (this.devices[i].deviceToken === deviceToken) {
      this.devices[i].lastUsageDate = new Date();

      this.markModified('devices');

      return;
    }
  }
};

UserDeviceSchema.methods.addNewDevice = function(device) {
  this.devices.push(device);

  this.markModified('devices');
};

UserDeviceSchema.methods.deviceExists = function(deviceToken) {
  for (var i = 0; i < this.devices.length; i++) {
    if (this.devices[i].deviceToken === deviceToken) return true;
  }

  return false;
};

UserDeviceSchema.methods.getDevice = function(deviceToken) {
  for (var i = 0; i < this.devices.length; i++) {
    if (this.devices[i].deviceToken === deviceToken) return this.devices[i];
  }

  return null;
};

UserDeviceSchema.methods.grantDeviceAccess = function(deviceToken) {
  var device = this.getDevice(deviceToken);

  if (!device) this.devices.push({
    deviceToken: deviceToken,
    canAccess: true
  });
  else device.canAccess = true;

  this.markModified('devices');
};

UserDeviceSchema.methods.restrictDeviceAccess = function(deviceToken) {
  var device = this.getDevice(deviceToken);

  if (!device) this.devices.push({
    deviceToken: deviceToken,
    canAccess: false
  });
  else device.canAccess = false;

  this.markModified('devices');
};

UserDeviceSchema.statics.findOrCreateNew = function(userName) {
  return this.findOne({
      userName: userName.toString()
    }).exec()
    .then(function(userDevice) {
      if (!userDevice) return new this({
        userName: userName.toString(),
        devices: []
      });

      return userDevice;
    }.bind(this));
};

UserDeviceSchema.statics.initUserDevice = function(deviceToken) {
  if (!config.deviceControl) return Promise.resolve();

  return this.findOrCreateNew('root')
    .then(function(userDevice) {
      userDevice.grantDeviceAccess(deviceToken);

      return userDevice.save();
    });
};


module.exports = mongoose.model('userdevices', UserDeviceSchema);