task/resources/notify/chrome/js/notify.js

/**
 * 连接已有会员
 * @type {*}
 */
var tagId = 0;
const instances = {};
const updateBadgeNum = function () {
    var badgeNum = 0;
    for (var index in instances) {
        if (!instances.hasOwnProperty(index)) {
            continue;
        }
        badgeNum+= instances[index].unread;
    }
    if (badgeNum == 0) {
        chrome.browserAction.setBadgeText({text: ''});
    } else {
        chrome.browserAction.setBadgeText({text: (badgeNum > 99 ? '99+' : badgeNum) + ''});
    }
    chrome.runtime.sendMessage({
        act: 'instances',
        instances: instances
    });
}
const getBadgeNum = function () {
    var configLists = $A.jsonParse($A.getStorage("configLists"), {});
    for (var index in configLists) {
        if (!configLists.hasOwnProperty(index)) {
            continue;
        }
        const key = index;
        const config = configLists[key];
        if (typeof instances[key] === "undefined") {
            instances[key] = {};
        }
        //
        if (instances[key].username !== config.username) {
            instances[key].username = config.username;
            instances[key].token = config.token;
            instances[key].unread = 0;
            instances[key].open = false;
            updateBadgeNum();
            if (typeof instances[key].ws !== "undefined") {
                instances[key].ws.config(null).close();
            }
            instances[key].ws = new WTWS({
                username: config.username,
                token: config.token,
                url: config.url,
                channel: 'chromeExtend'
            }).setOnMsgListener('notify', ['open', 'unread', 'user'], function (msgDetail) {
                let body = msgDetail.body;
                if (['taskA'].indexOf(body.type) !== -1) {
                    return;
                }
                var tempLists = $A.jsonParse($A.getStorage("configLists"), {});
                if (typeof tempLists[key] == "object" && tempLists[key].disabled === true) {
                    return;
                }
                switch (msgDetail.messageType) {
                    case 'open':
                        instances[key].open = true;
                        break;
                    case 'unread':
                        instances[key].unread = msgDetail.body.unread;
                        break;
                    case 'user':
                        instances[key].unread++;
                        chrome.tabs.query({active: true, currentWindow: true}, function (tabs) {
                            if ($A.getHost(tabs[0].url) != key) {
                                var opurl = 'http://' + key + '/todo?token=' + encodeURIComponent(instances[key].token) + '&open=chat';
                                $A.showNotify(key, {
                                    body: $A.getMsgDesc(body),
                                    icon: body.userimg
                                }, opurl);
                            }
                        });
                        break;
                }
                updateBadgeNum();
            }).sendTo('unread', function (res) {
                if (res.status === 1) {
                    instances[key].unread = $A.runNum(res.message);
                    updateBadgeNum();
                }
            });
        }
    }
    //
    tagId++;
    const tmpID = tagId;
    setTimeout(function () {
        if (tmpID === tagId) {
            getBadgeNum();
        }
    }, 5000);
}
getBadgeNum();

/**
 * 监听来自网站的会员信息
 */
chrome.runtime.onMessage.addListener(function (request, sender, sendResponse) {
    var configLists;
    if (request.act === "config") {
        if (sender.tab) {
            var hostname = $A.getHost(sender.tab.url);
            if (hostname) {
                configLists = $A.jsonParse($A.getStorage("configLists"), {});
                if (typeof configLists[hostname] !== "object") {
                    configLists[hostname] = {};
                }
                configLists[hostname] = Object.assign(request.config, {
                    hostname: hostname,
                });
                $A.setStorage("configLists", $A.jsonStringify(configLists));
                sendResponse(configLists);
            }
        }
    } else if (request.act === "getInstances") {
        sendResponse(instances);
    } else if (request.act === "clickInstances") {
        if (typeof instances[request.index] === "object") {
            instances[request.index].ws.sendTo('unread', function (res) {
                if (res.status === 1) {
                    instances[request.index].unread = $A.runNum(res.message);
                    updateBadgeNum();
                }
                sendResponse(res);
            })
        }
    } else if (request.act === "delInstances") {
        configLists = $A.jsonParse($A.getStorage("configLists"), {});
        if (typeof configLists[request.index] === "object") {
            delete configLists[request.index];
            $A.setStorage("configLists", $A.jsonStringify(configLists));
        }
        if (typeof instances[request.index] === "object") {
            if (typeof  instances[request.index].ws !== "undefined") {
                 instances[request.index].ws.config(null).close();
            }
            delete instances[request.index];
        }
        updateBadgeNum();
    }
});