pushbullet-desktop/app/scripts/browser.js

Pushbullet for Desktop

'use strict';

const { ipcRenderer }  = require('electron');

let NativeNotification = Notification;


/**
 * @description Resolves a Pushbullet Push object to an image URL.
 * @param {Object} push - Pushbullet Push object (see https://docs.pushbullet.com/#push)
 * @returns {String}
 */
var getIconForPushbulletPush = function(push) {

    var imageUrl;

    // Account image
    var accountImage,
        accountIdShort = push['receiver_iden'],
        accountList = window.pb.api.accounts.all;

    for (var account of accountList) {
        if (account['iden'].startsWith(accountIdShort)) {
            //console.log('account', account);
            accountImage = account['image_url'];
        }
    }

    // Channel image (i.e. IFTTT, Zapier)
    var channelImage,
        channelId = push['client_iden'],
        channelList = window.pb.api.grants.all;

    for (var channel of channelList) {
        if (channel['client']['iden'] === channelId) {
            //console.log('channel', channel);
            channelImage = channel['client']['image_url'];
        }
    }

    // Device image (i.e. Phone, Browser)
    var deviceImage,
        deviceId = push['source_device_iden'],
        deviceList = window.pb.api.devices.all;

    for (var device of deviceList) {
        if (device['iden'] === deviceId) {
            //console.log('device', device);
            deviceImage = 'http://www.pushbullet.com/img/deviceicons/' + device['icon'] + '.png';
        }
    }

    // Fallback behaviour
    imageUrl = channelImage || deviceImage || accountImage;

    return imageUrl;
};


Notification = function(pushTitle, push) {

    /**
     * @constant
     * @default
     */
    var DEFAULT_TITLE = null;
    var DEFAULT_BODY = null;
    var DEFAULT_URL = null;
    var DEFAULT_ICON = null;

    var pushType = push.type;

    // Populate fields for Pushbullet push types (note, link, file)
    var title = push['title'] || push['body'] || DEFAULT_TITLE,
        body = push['body'] || push['title'] || DEFAULT_BODY,
        url = DEFAULT_URL,
        icon = getIconForPushbulletPush(push) || DEFAULT_ICON;

    switch (pushType) {
        case 'link':
            title = title || push['url'];
            body = body || push['url'];
            url = push['url'];
            break;
        case 'note':
            body = push['body'] || push['title'];
            break;
        case 'file':
            title = title || push['file_name'];
            body = body || push['file_type'];
            url = push['file_url'];
            icon = push['image_url'] || icon;
            break;
    }

    // Options for native notification
    var options = {
        title: title,
        body: body,
        icon: icon,
        url: url
    };

    // Trigger native notification
    var notification = new NativeNotification(options.title, options);

    // Register event handlers for main renderer
    ipcRenderer.send('notification-received');

    notification.addEventListener('click', () => {
        ipcRenderer.send('notification-click', options);
    });

    return notification;
};

Notification.prototype = NativeNotification.prototype;
Notification.permission = NativeNotification.permission;
Notification.requestPermission = NativeNotification.requestPermission.bind(Notification);

/**
 * Pushbullet API
 * @external window.pb
 * @see {@link https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/String|String}
 */

/**
 * Requests updated Push messages by jerry-rigging into Pushbullet's web app API instance (window.pb.api).
 * Keeps reference to the current state by locally storing the most current Push objects 'modified' parameter.
 * @see {@link https://docs.pushbullet.com/#list-pushes}     {@link https://docs.pushbullet.com/#list-pushes|Pushbullet API}
 * @see Pushbullet API
 */
window.requestPushbulletPushes = function() {
    console.log('Requesting Pushes after: ', window.settings.notifyAfter);
    var notifyAfter = window.settings.notifyAfter;

    window.pb.net.get('/v2/pushes', {
        modified_after: notifyAfter
    }, function(result) {
        var newPushes = result.pushes,
            newestPush = newPushes[0];

        console.log('Newest Push', JSON.stringify(newestPush, null, 3));

        if (newPushes.length === 0) {
            console.log('Nothing to do.');
            return true;
        }

        window.settings.notifyAfter = notifyAfter = newestPush.modified;
        ipcRenderer.send('settings-set', 'notifyAfter', notifyAfter);

        newPushes.forEach(function(value) {
            return new Notification(null, value);
        });

        console.log('Updated notifyAfter', window.settings.notifyAfter);
    });
};

/**
 * Extend the Pushbullet WebSocket onmessage handler, injecting our request logic {@link requestPushbulletPushes}.
 * This enables us to hook into all API-related content update events in real time.
 */
window.extendSocketMessageHandler = function() {
    var pbOnmessage = window.pb.ws.socket.onmessage;

    window.pb.ws.socket.onmessage = function() {
        window.requestPushbulletPushes();
        return pbOnmessage.apply(pbOnmessage, arguments);
    };
};

/**
 * Replaces the WebSocket onerror handler.
 * Required in order to prevent overrides of our onmessage handler hook.
 */
window.extendSocketErrorHandler = function() {
    window.pb.ws.socket.onerror = function() {
        setTimeout(function() {
            window.pb.api['listen_for_pushes']();
            window.register();
        }, 10000);
    };
};

window.settings = {};

/**
 * Inject Pushbullet API hooks on webview page load
 */
window.onload = function() {
    ipcRenderer.send('settings-get');

    ipcRenderer.on('settings-get-reply', (event, result) => {
        window.settings = result;
        window.extendSocketMessageHandler();

        console.log('[settings-get-reply]', 'result', result);
    });
};