habitica/website/server/controllers/api-v3/pushNotifications.js

import { authWithHeaders } from '../../middlewares/auth';
import {
  NotAuthorized,
  NotFound,
} from '../../libs/errors';

let api = {};

/**
 * @apiIgnore
 * @api {post} /api/v3/user/push-devices Add a push device to a user
 * @apiName UserAddPushDevice
 * @apiGroup User
 *
 * @apiParam (Body) {String} regId The id of the push device
 * @apiParam (Body) {String} type The type of push device
 *
 * @apiSuccess {Object} data List of push devices
 * @apiSuccess {String} message Success message
 */
api.addPushDevice = {
  method: 'POST',
  url: '/user/push-devices',
  middlewares: [authWithHeaders({
    userFieldsToExclude: ['inbox'],
  })],
  async handler (req, res) {
    let user = res.locals.user;

    req.checkBody('regId', res.t('regIdRequired')).notEmpty();
    req.checkBody('type', res.t('typeRequired')).notEmpty().isIn(['ios', 'android']);

    let validationErrors = req.validationErrors();
    if (validationErrors) throw validationErrors;

    let pushDevices = user.pushDevices;

    let item = {
      regId: req.body.regId,
      type: req.body.type,
    };

    if (pushDevices.find(device => device.regId === item.regId)) {
      throw new NotAuthorized(res.t('pushDeviceAlreadyAdded'));
    }

    pushDevices.push(item);

    await user.save();

    res.respond(200, user.pushDevices, res.t('pushDeviceAdded'));
  },
};

/**
 * @apiIgnore
 * @api {delete} /api/v3/user/push-devices/:regId remove a push device from a user
 * @apiName UserRemovePushDevice
 * @apiGroup User
 *
 * @apiParam (Path) {String} regId The id of the push device
 *
 * @apiSuccess {Object} data List of push devices
 * @apiSuccess {String} message Success message
 */
api.removePushDevice = {
  method: 'DELETE',
  url: '/user/push-devices/:regId',
  middlewares: [authWithHeaders({
    userFieldsToExclude: ['inbox'],
  })],
  async handler (req, res) {
    let user = res.locals.user;

    req.checkParams('regId', res.t('regIdRequired')).notEmpty();
    let validationErrors = req.validationErrors();
    if (validationErrors) throw validationErrors;
    let regId = req.params.regId;

    let pushDevices = user.pushDevices;

    let indexOfPushDevice = pushDevices.findIndex((element) => {
      return element.regId === regId;
    });

    if (indexOfPushDevice === -1) {
      throw new NotFound(res.t('pushDeviceNotFound'));
    }

    pushDevices.splice(indexOfPushDevice, 1);
    await user.save();

    res.respond(200, user.pushDevices, res.t('pushDeviceRemoved'));
  },
};

module.exports = api;