push aws

push.js

module.exports.push = coroutine(function*(data, users){
    let params = {
        MessageStructure: 'json',
        Message: JSON.stringify({
            default: data.subject,
            APNS: JSON.stringify(assign({
                aps: assign({alert: data.alert, sound: data.sound}, data.apns||{}),
            },_.omit(data, 'alert', 'collapseKey', 'subject', 'apns', 'sound'))),
            GCM: JSON.stringify({
                collapseKey: data.collapseKey,
                data: assign({timestamp: Date.now()},
                    _.omit(data, 'alert', 'collapseKey', 'apns'), {sound: (data.sound||'').replace('.caf', '')}),
            }),
        }),
    };
    if (!users)
    {
        return assign(yield publish(assign({TopicArn: topic}, params)),
            {Message: JSON.parse(params.Message)});
    }
    return yield Promise.all(users.map(coroutine(function*(arn){
        try {
            return assign(yield publish(assign({TargetArn: arn}, params)),
                {Message: JSON.parse(params.Message)});
        } catch(err) {
            logger.error('failed to push', {arn, error: err});
            if (err.code!='EndpointDisabled')
                return;
        }
        try {
            let account = yield db.account.findOne({'sns.arn': arn});
            if (!account)
                return;
            for (let sns of account.sns)
            {
                if (sns.arn!=arn)
                    continue;
                yield unsubscribe({SubscriptionArn: sns.subscription});
                yield deleteEndpoint({EndpointArn: arn});
                yield db.account.update({_id: account._id},
                    {$pull: {sns: sns}});
                break;
            }
        } catch(err) {
            logger.error('failed to remove device', {arn, error: err});
        }
    })));
});