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 notificationDetailService = require('./notification_detail.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) => {
            config = config || {
                scopes: [],
            };

            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).then(function(result) {
                    return new Promise(function(resolve) { resolve(result.rows) });
                });
                let saveNotificationPromise = (notification) => notificationService.create(notification, context);
                let sendNotificationsPromise = (messages) => notificationService.sendNotification(messages);
                let disableUserDevicePromise = (token) => userDeviceService.update({ params: {
                    token: token,
                }}, { valid: false }, context);

                let disableInvalidUserDevicesPromise = (userDevices) => {
                    return new Promise(function (resolve) {
                        let _userDevices = [];
                        userDevices.forEach(async function (userDevice) {
                            if (!userDeviceService.isValidPushToken(userDevice.token)) {
                                await disableUserDevicePromise(userDevice.token);
                            } else {
                                _userDevices.push(userDevice);
                            }
                        });

                        resolve(_userDevices)
                    });
                };

                let disableUserDevicesWithErrorStatus = (messages, tickets) => {
                    return new Promise(function (resolve) {
                        tickets.forEach(async function (ticket, index) {
                            if ((ticket.status === 'error') && (ticket.details.error === 'DeviceNotRegistered')) {                                
                                await disableUserDevicePromise(messages[index].to)
                            }
                        });
                        resolve(true);
                    });
                }

                let saveResponseStatusPromise = (messages, tickets) => notificationDetailService.saveNotificationDetails(messages, tickets);
                
                const notificationRecord = {
                    title: req.body.title,
                    body: req.body.message,
                    ttl: req.body.ttl || undefined,
                    priority: req.body.priority || 'default',
                    data: req.body.data || { userIds: req.body.userIds },
                };

                let buildMessagesPromise = (userDevices) => {                    
                    return new Promise(function (resolve) { 
                        let messages = [];

                        userDevices.forEach(async function (userDevice) {       
                            messages.push({
                                title: notificationRecord.title,
                                body: notificationRecord.body,
                                ttl: notificationRecord.ttl,
                                priority: notificationRecord.priority,
                                userId: userDevice.userId,
                                to: userDevice.token,
                                sound: 'default',
                                notificationId: notificationRecord.id,
                            });
                        });
                    
                        resolve(messages) 
                    });
                };
                
                let getUserDevicesList = [];
                
                userIds.forEach(function (userId) {
                    getUserDevicesList.push(getUserDevicesPromise(userId));
                });

                saveNotificationPromise(notificationRecord)
                .then(function(notification) {
                    notificationRecord.id = notification.id;
                    return Promise.all(getUserDevicesList)
                }).then(function (userDeviceList) {
                    return new Promise(function(resolve) { resolve(userDeviceList[0]); });
                })
                .then(disableInvalidUserDevicesPromise)
                .then(buildMessagesPromise)
                .then(sendNotificationsPromise)
                .then(function({ messages, tickets }) {
                    let _saveStatus = saveResponseStatusPromise(messages, tickets);
                    let _disableDevices = disableUserDevicesWithErrorStatus(messages, tickets);

                    return Promise.all([_saveStatus, _disableDevices]);
                })
                .then(function (details) {
                    console.log(details);
                });
            } catch (error) {
                console.error(error);
                return handleErrorResponse(controllerOptions.MODULE_NAME, 'sendNotification', error, res)
            } finally {
                return handleResultResponse(receipt, null, null, res, httpStatus.OK);
            }
        }
    },

    updateNotificationsWithReceipts: (config) => {
        return async (req, res, next) => {
            config = config || {
                scopes: [],
            };

            const context = buildContext(req, config);
            let params = extractParamsFromRequest(req, res);

            let getNotificationsWithoutReceipt = async () => {
                let params = {
                    where: { }
                };               
                return await notificationDetailService.fetchAll(params, context);
            }

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

    registerDevice: (config) => {
        return async (req, res, next) => {
            config = config || {
                scopes: [],
            };

            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);