blob: b1d8e3111018cb589f907fab41940728a705a412 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
const _path = _interopRequireDefault(require('path'));
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
module.exports = (Ferdi) => {
function getMessages() {
let directCount = 0;
let indirectCount = 0;
const chat_item = document.querySelectorAll('div.chat_item');
Array.prototype.forEach.call(chat_item, (item) => {
let count = 0;
const reddot = item.querySelector('i.web_wechat_reddot_middle');
const avatarImage = item.querySelector('img.img');
if (reddot) {
count = Ferdi.safeParseInt(reddot.innerText);
}
if (avatarImage && avatarImage.getAttribute('src').search('webwxgeticon') != -1) {
directCount += count;
} else {
indirectCount += count;
}
});
Ferdi.setBadge(directCount, indirectCount);
}
Ferdi.loop(getMessages);
Ferdi.injectCSS(_path.default.join(__dirname, 'service.css'));
};
|