OLD | NEW |
1 /* | 1 /* |
2 * This file is part of Adblock Plus <https://adblockplus.org/>, | 2 * This file is part of Adblock Plus <https://adblockplus.org/>, |
3 * Copyright (C) 2006-present eyeo GmbH | 3 * Copyright (C) 2006-present eyeo GmbH |
4 * | 4 * |
5 * Adblock Plus is free software: you can redistribute it and/or modify | 5 * Adblock Plus is free software: you can redistribute it and/or modify |
6 * it under the terms of the GNU General Public License version 3 as | 6 * it under the terms of the GNU General Public License version 3 as |
7 * published by the Free Software Foundation. | 7 * published by the Free Software Foundation. |
8 * | 8 * |
9 * Adblock Plus is distributed in the hope that it will be useful, | 9 * Adblock Plus is distributed in the hope that it will be useful, |
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of | 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
(...skipping 18 matching lines...) Expand all Loading... |
29 | 29 |
30 let getBlockedPerPage = | 30 let getBlockedPerPage = |
31 /** | 31 /** |
32 * Gets the number of requests blocked on the given page. | 32 * Gets the number of requests blocked on the given page. |
33 * | 33 * |
34 * @param {Page} page | 34 * @param {Page} page |
35 * @return {Number} | 35 * @return {Number} |
36 */ | 36 */ |
37 exports.getBlockedPerPage = page => blockedPerPage.get(page) || 0; | 37 exports.getBlockedPerPage = page => blockedPerPage.get(page) || 0; |
38 | 38 |
| 39 function updateBadge(page, blockedCount) |
| 40 { |
| 41 if (Prefs.show_statsinicon) |
| 42 { |
| 43 page.browserAction.setBadge(blockedCount && { |
| 44 color: badgeColor, |
| 45 number: blockedCount |
| 46 }); |
| 47 } |
| 48 } |
| 49 |
39 // Once nagivation for the tab has been committed to (e.g. it's no longer | 50 // Once nagivation for the tab has been committed to (e.g. it's no longer |
40 // being prerendered) we clear its badge, or if some requests were already | 51 // being prerendered) we clear its badge, or if some requests were already |
41 // blocked beforehand we display those on the badge now. | 52 // blocked beforehand we display those on the badge now. |
42 browser.webNavigation.onCommitted.addListener(details => | 53 browser.webNavigation.onCommitted.addListener(details => |
43 { | 54 { |
44 if (details.frameId == 0) | 55 if (details.frameId == 0) |
45 { | 56 { |
46 let page = new ext.Page({id: details.tabId}); | 57 let page = new ext.Page({id: details.tabId}); |
47 let blocked = blockedPerPage.get(page); | 58 let blocked = blockedPerPage.get(page); |
48 | 59 |
49 page.browserAction.setBadge(blocked && { | 60 updateBadge(page, blocked); |
50 color: badgeColor, | |
51 number: blocked | |
52 }); | |
53 } | 61 } |
54 }); | 62 }); |
55 | 63 |
56 FilterNotifier.on("filter.hitCount", (filter, newValue, oldValue, tabIds) => | 64 FilterNotifier.on("filter.hitCount", (filter, newValue, oldValue, tabIds) => |
57 { | 65 { |
58 if (!(filter instanceof BlockingFilter)) | 66 if (!(filter instanceof BlockingFilter)) |
59 return; | 67 return; |
60 | 68 |
61 for (let tabId of tabIds) | 69 for (let tabId of tabIds) |
62 { | 70 { |
63 let page = new ext.Page({id: tabId}); | 71 let page = new ext.Page({id: tabId}); |
64 let blocked = blockedPerPage.get(page) || 0; | 72 let blocked = blockedPerPage.get(page) || 0; |
| 73 |
65 blockedPerPage.set(page, ++blocked); | 74 blockedPerPage.set(page, ++blocked); |
66 | 75 updateBadge(page, blocked); |
67 // Update number in icon | |
68 if (Prefs.show_statsinicon) | |
69 { | |
70 page.browserAction.setBadge({ | |
71 color: badgeColor, | |
72 number: blocked | |
73 }); | |
74 } | |
75 } | 76 } |
76 | 77 |
77 Prefs.blocked_total++; | 78 Prefs.blocked_total++; |
78 }); | 79 }); |
79 | 80 |
80 Prefs.on("show_statsinicon", () => | 81 Prefs.on("show_statsinicon", () => |
81 { | 82 { |
82 browser.tabs.query({}, tabs => | 83 browser.tabs.query({}, tabs => |
83 { | 84 { |
84 for (let tab of tabs) | 85 for (let tab of tabs) |
85 { | 86 { |
86 let page = new ext.Page(tab); | 87 let page = new ext.Page(tab); |
87 let badge = null; | |
88 | 88 |
89 if (Prefs.show_statsinicon) | 89 if (Prefs.show_statsinicon) |
90 { | 90 updateBadge(page, blockedPerPage.get(page)); |
91 let blocked = blockedPerPage.get(page); | 91 else |
92 if (blocked) | 92 page.browserAction.setBadge(null); |
93 { | |
94 badge = { | |
95 color: badgeColor, | |
96 number: blocked | |
97 }; | |
98 } | |
99 } | |
100 | |
101 page.browserAction.setBadge(badge); | |
102 } | 93 } |
103 }); | 94 }); |
104 }); | 95 }); |
105 | 96 |
106 port.on("stats.getBlockedPerPage", | 97 port.on("stats.getBlockedPerPage", |
107 message => getBlockedPerPage(new ext.Page(message.tab))); | 98 message => getBlockedPerPage(new ext.Page(message.tab))); |
OLD | NEW |