app2-api/modules/notification/notification.controller.js

'use strict';
const httpStatus = require('http-status');

const generateControllers = require('../../core/controllers');
const { buildContext } = require('../../core/controllers');
const notificationService = require('./notification.service');
const userDeviceService = require('./user_device.service');

const { extractParamsFromRequest, handleErrorResponse, handleResultResponse } = require('../../helpers/controller.helper');

// Module Name
const MODULE_NAME = '[notification.controller]';
const controllerOptions = { MODULE_NAME };

const extraControllers = {
    sendNotification: (config) => {
        return async (req, res, next) => {
            let receipt = undefined;
            const context = buildContext(req, config);
            let params = extractParamsFromRequest(req, res);

            let userIds = req.body.userIds;
            if (!userIds) {
                return handleErrorResponse(controllerOptions.MODULE_NAME, 'sendNotification', new Error('Missing user Ids'), res)
            }

            if (!req.body.title) {
                return handleErrorResponse(controllerOptions.MODULE_NAME, 'sendNotification', new Error('Missing message title'), res)
            }

            if (!req.body.message) {
                return handleErrorResponse(controllerOptions.MODULE_NAME, 'sendNotification', new Error('Missing message content'), res)
            }

            try {                               
                let getUserDevicesPromise = (userId) => userDeviceService.fetchAll({ params: { userId: userId }}, context);
                let sendNotificationsPromise = (messages) => notificationService.sendNotification(messages);
                
                let buildMessagePromise = (userDevices) => {
                    let message = undefined;
                    userDevices.rows.forEach(function (userDevice) {
                        if (notificationService.isValidPushToken(userDevice.token)) {
                            message = {
                                userId: userDevice.userId,
                                to: userDevice.token,
                                title: req.body.title,
                                body: req.body.message,
                                ttl: req.body.ttl,
                                priority: req.body.priority,
                                data: req.body.data,
                                sound: 'default',
                            };
                        }
                    });
                    return new Promise(function(resolve) { resolve(message) });
                };
                
                let getUserDevicesList = [];
                
                userIds.forEach(function (userId) {
                    getUserDevicesList.push(getUserDevicesPromise(userId));
                });

                receipt = await Promise.all(getUserDevicesList).then(
                    function(userDeviceList) { 
                        return Promise.all(userDeviceList.map(buildMessagePromise))
                    })
                .then(sendNotificationsPromise)

            } catch (error) {
                return handleErrorResponse(controllerOptions.MODULE_NAME, 'sendNotification', error, res)
            } finally {
                return handleResultResponse(receipt, null, null, res, httpStatus.OK);
            }
        }
    },

    registerUser: (config) => {
        return async (req, res, next) => {
            try {
                const context = buildContext(req, config);
                var data = {
                    token: req.body.token,
                    valid: 1,
                    userId: context.user.id
                };

                let params = extractParamsFromRequest(req, res, { 
                    includeAll: false, 
                    paginate: { limit: 1, page: 1 },
                    params: {
                        userId: context.user.id,
                        token: data.token,
                    }
                });

                // Buscamos el token y el usuario
                console.log('>> Busco el usuario y el token', params.params);
                let result = await notificationService.fetchOne(params, context);
                if (!result) {
                    // Dar de alta el token
                    console.log('>> Dar de alta el token', data);
                    result = await notificationService.create(data, context);
                } else {
                    // Actualizar el token
                    console.log('>> Actualizar el token', params.params, data, context);
                    result = await notificationService.update(params.params, data, context);
                }
            } catch(error) {
                console.error(error);
            } finally {
                // En todo caso devolver OK al cliente
                return handleResultResponse('OK', null, null, res, httpStatus.OK);    
            }
        }
    },    
};

module.exports = generateControllers(notificationService, extraControllers, controllerOptions);