blob: 263c5438f95cf400bbece43286994928e84188f5 (
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
"use strict";
const {
remote
} = require('electron');
const path = require('path');
const webContents = remote.getCurrentWebContents();
const {
session
} = webContents;
window.addEventListener('beforeunload', async () => {
try {
session.flushStorageData();
session.clearStorageData({
storages: ['appcache', 'serviceworkers', 'cachestorage', 'websql', 'indexdb']
});
const registrations = await window.navigator.serviceWorker.getRegistrations();
registrations.forEach(r => {
r.unregister();
console.log('ServiceWorker unregistered');
});
} catch (err) {
console.err(err);
}
});
module.exports = Franz => {
const getMessages = function getMessages() {
const elements = document.querySelectorAll("[class=_31gEB]");
var count = 0;
var indirectCount = 0;
for (var i = 0; i < elements.length; i += 1) {
var countValue = parseInt(elements[i].textContent || '0', 10);
if (elements[i].parentNode.previousElementSibling === null || elements[i].parentNode.previousElementSibling.querySelectorAll("[data-icon=muted]").length === 0) {
count += countValue;
}
else {
indirectCount += countValue;
}
}
Franz.setBadge(count, indirectCount);
};
Franz.injectCSS(path.join(__dirname, 'service.css'));
Franz.loop(getMessages);
};
|