Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Side by Side Diff: ext/background.js

Issue 29417597: Issue 5161 - Use maps and sets where appropriate (Closed) Base URL: https://hg.adblockplus.org/adblockpluschrome/
Patch Set: Rebased Created May 20, 2017, 5:44 p.m.
Left:
Right:
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View unified diff | Download patch
« no previous file with comments | « no previous file | lib/devtools.js » ('j') | lib/popupBlocker.js » ('J')
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
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-2017 eyeo GmbH 3 * Copyright (C) 2006-2017 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
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details. 12 * GNU General Public License for more details.
13 * 13 *
14 * You should have received a copy of the GNU General Public License 14 * You should have received a copy of the GNU General Public License
15 * along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. 15 * along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>.
16 */ 16 */
17 17
18 "use strict"; 18 "use strict";
19 19
20 (function() 20 (function()
21 { 21 {
22 let nonEmptyPageMaps = Object.create(null); 22 let nonEmptyPageMaps = new Set();
23 let pageMapCounter = 0;
24 23
25 let PageMap = ext.PageMap = function() 24 let PageMap = ext.PageMap = function()
26 { 25 {
27 this._map = Object.create(null); 26 this._map = new Map();
28 this._id = ++pageMapCounter;
29 }; 27 };
30 PageMap.prototype = { 28 PageMap.prototype = {
31 _delete(id) 29 _delete(id)
32 { 30 {
33 delete this._map[id]; 31 this._map.delete(id);
34 32
35 if (Object.keys(this._map).length == 0) 33 if (this._map.size == 0)
36 delete nonEmptyPageMaps[this._id]; 34 nonEmptyPageMaps.delete(this);
37 }, 35 },
38 keys() 36 keys()
39 { 37 {
40 return Object.keys(this._map).map(ext.getPage); 38 return this._map.keys();
41 }, 39 },
42 get(page) 40 get(page)
43 { 41 {
44 return this._map[page.id]; 42 return this._map.get(page.id);
45 }, 43 },
46 set(page, value) 44 set(page, value)
47 { 45 {
48 this._map[page.id] = value; 46 this._map.set(page.id, value);
49 nonEmptyPageMaps[this._id] = this; 47 nonEmptyPageMaps.add(this);
50 }, 48 },
51 has(page) 49 has(page)
52 { 50 {
53 return page.id in this._map; 51 return this._map.has(page.id);
54 }, 52 },
55 clear() 53 clear()
56 { 54 {
57 for (let id in this._map) 55 this._map.clear();
58 this._delete(id); 56 nonEmptyPageMaps.delete(this);
59 }, 57 },
60 delete(page) 58 delete(page)
61 { 59 {
62 this._delete(page.id); 60 this._delete(page.id);
63 } 61 }
64 }; 62 };
65 63
66 ext._removeFromAllPageMaps = pageId => 64 ext._removeFromAllPageMaps = pageId =>
67 { 65 {
68 for (let pageMapId in nonEmptyPageMaps) 66 for (let pageMap of nonEmptyPageMaps)
69 nonEmptyPageMaps[pageMapId]._delete(pageId); 67 pageMap._delete(pageId);
70 }; 68 };
71 69
72 /* Pages */ 70 /* Pages */
73 71
74 let Page = ext.Page = function(tab) 72 let Page = ext.Page = function(tab)
75 { 73 {
76 this.id = tab.id; 74 this.id = tab.id;
77 this._url = tab.url && new URL(tab.url); 75 this._url = tab.url && new URL(tab.url);
78 76
79 this.browserAction = new BrowserAction(tab.id); 77 this.browserAction = new BrowserAction(tab.id);
80 this.contextMenus = new ContextMenus(this); 78 this.contextMenus = new ContextMenus(this);
81 }; 79 };
82 Page.prototype = { 80 Page.prototype = {
83 get url() 81 get url()
84 { 82 {
85 // usually our Page objects are created from Chrome's Tab objects, which 83 // usually our Page objects are created from Chrome's Tab objects, which
86 // provide the url. So we can return the url given in the constructor. 84 // provide the url. So we can return the url given in the constructor.
87 if (this._url) 85 if (this._url)
88 return this._url; 86 return this._url;
89 87
90 // but sometimes we only have the tab id when we create a Page object. 88 // but sometimes we only have the tab id when we create a Page object.
91 // In that case we get the url from top frame of the tab, recorded by 89 // In that case we get the url from top frame of the tab, recorded by
92 // the onBeforeRequest handler. 90 // the onBeforeRequest handler.
93 let frames = framesOfTabs[this.id]; 91 let frames = framesOfTabs.get(this.id);
94 if (frames) 92 if (frames)
95 { 93 {
96 let frame = frames[0]; 94 let frame = frames.get(0);
97 if (frame) 95 if (frame)
98 return frame.url; 96 return frame.url;
99 } 97 }
100 }, 98 },
101 sendMessage(message, responseCallback) 99 sendMessage(message, responseCallback)
102 { 100 {
103 chrome.tabs.sendMessage(this.id, message, responseCallback); 101 chrome.tabs.sendMessage(this.id, message, responseCallback);
104 } 102 }
105 }; 103 };
106 104
(...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after
151 }; 149 };
152 150
153 chrome.tabs.onUpdated.addListener((tabId, changeInfo, tab) => 151 chrome.tabs.onUpdated.addListener((tabId, changeInfo, tab) =>
154 { 152 {
155 if (changeInfo.status == "loading") 153 if (changeInfo.status == "loading")
156 ext.pages.onLoading._dispatch(new Page(tab)); 154 ext.pages.onLoading._dispatch(new Page(tab));
157 }); 155 });
158 156
159 function createFrame(tabId, frameId) 157 function createFrame(tabId, frameId)
160 { 158 {
161 let frames = framesOfTabs[tabId]; 159 let frames = framesOfTabs.get(tabId);
162 if (!frames) 160 if (!frames)
163 frames = framesOfTabs[tabId] = Object.create(null); 161 {
162 frames = new Map();
163 framesOfTabs.set(tabId, frames);
164 }
164 165
165 let frame = frames[frameId]; 166 let frame = frames.get(frameId);
166 if (!frame) 167 if (!frame)
167 frame = frames[frameId] = {}; 168 {
169 frame = {};
170 frames.set(frameId, frame);
171 }
168 172
169 return frame; 173 return frame;
170 } 174 }
171 175
172 function updatePageFrameStructure(frameId, tabId, url, parentFrameId) 176 function updatePageFrameStructure(frameId, tabId, url, parentFrameId)
173 { 177 {
174 if (frameId == 0) 178 if (frameId == 0)
175 { 179 {
176 let page = new Page({id: tabId, url}); 180 let page = new Page({id: tabId, url});
177 181
178 ext._removeFromAllPageMaps(tabId); 182 ext._removeFromAllPageMaps(tabId);
179 183
180 chrome.tabs.get(tabId, () => 184 chrome.tabs.get(tabId, () =>
181 { 185 {
182 // If the tab is prerendered, chrome.tabs.get() sets 186 // If the tab is prerendered, chrome.tabs.get() sets
183 // chrome.runtime.lastError and we have to dispatch the onLoading event, 187 // chrome.runtime.lastError and we have to dispatch the onLoading event,
184 // since the onUpdated event isn't dispatched for prerendered tabs. 188 // since the onUpdated event isn't dispatched for prerendered tabs.
185 // However, we have to keep relying on the unUpdated event for tabs that 189 // However, we have to keep relying on the unUpdated event for tabs that
186 // are already visible. Otherwise browser action changes get overridden 190 // are already visible. Otherwise browser action changes get overridden
187 // when Chrome automatically resets them on navigation. 191 // when Chrome automatically resets them on navigation.
188 if (chrome.runtime.lastError) 192 if (chrome.runtime.lastError)
189 ext.pages.onLoading._dispatch(page); 193 ext.pages.onLoading._dispatch(page);
190 }); 194 });
191 } 195 }
192 196
193 // Update frame parent and URL in frame structure 197 // Update frame URL and parent in frame structure
194 let frame = createFrame(tabId, frameId); 198 let frame = createFrame(tabId, frameId);
195 frame.url = new URL(url); 199 frame.url = new URL(url);
196 frame.parent = framesOfTabs[tabId][parentFrameId] || null; 200
201 let parentFrame = framesOfTabs.get(tabId).get(parentFrameId);
202 if (parentFrame)
203 frame.parent = parentFrame;
197 } 204 }
198 205
199 chrome.webRequest.onHeadersReceived.addListener(details => 206 chrome.webRequest.onHeadersReceived.addListener(details =>
200 { 207 {
201 // We have to update the frame structure when switching to a new 208 // We have to update the frame structure when switching to a new
202 // document, so that we process any further requests made by that 209 // document, so that we process any further requests made by that
203 // document in the right context. Unfortunately, we cannot rely 210 // document in the right context. Unfortunately, we cannot rely
204 // on webNavigation.onCommitted since it isn't guaranteed to fire 211 // on webNavigation.onCommitted since it isn't guaranteed to fire
205 // before any subresources start downloading[1]. As an 212 // before any subresources start downloading[1]. As an
206 // alternative we use webRequest.onHeadersReceived for HTTP(S) 213 // alternative we use webRequest.onHeadersReceived for HTTP(S)
(...skipping 78 matching lines...) Expand 10 before | Expand all | Expand 10 after
285 updatePageFrameStructure(details.frameId, details.tabId, details.url, 292 updatePageFrameStructure(details.frameId, details.tabId, details.url,
286 details.parentFrameId); 293 details.parentFrameId);
287 } 294 }
288 }); 295 });
289 296
290 function forgetTab(tabId) 297 function forgetTab(tabId)
291 { 298 {
292 ext.pages.onRemoved._dispatch(tabId); 299 ext.pages.onRemoved._dispatch(tabId);
293 300
294 ext._removeFromAllPageMaps(tabId); 301 ext._removeFromAllPageMaps(tabId);
295 delete framesOfTabs[tabId]; 302 framesOfTabs.delete(tabId);
296 } 303 }
297 304
298 chrome.tabs.onReplaced.addListener((addedTabId, removedTabId) => 305 chrome.tabs.onReplaced.addListener((addedTabId, removedTabId) =>
299 { 306 {
300 forgetTab(removedTabId); 307 forgetTab(removedTabId);
301 }); 308 });
302 309
303 chrome.tabs.onRemoved.addListener(forgetTab); 310 chrome.tabs.onRemoved.addListener(forgetTab);
304 311
305 chrome.tabs.onActivated.addListener(details => 312 chrome.tabs.onActivated.addListener(details =>
(...skipping 177 matching lines...) Expand 10 before | Expand all | Expand 10 after
483 490
484 chrome.windows.onFocusChanged.addListener(windowId => 491 chrome.windows.onFocusChanged.addListener(windowId =>
485 { 492 {
486 if (windowId != chrome.windows.WINDOW_ID_NONE) 493 if (windowId != chrome.windows.WINDOW_ID_NONE)
487 updateContextMenu(); 494 updateContextMenu();
488 }); 495 });
489 496
490 497
491 /* Web requests */ 498 /* Web requests */
492 499
493 let framesOfTabs = Object.create(null); 500 let framesOfTabs = new Map();
494 501
495 ext.getFrame = (tabId, frameId) => 502 ext.getFrame = (tabId, frameId) =>
496 { 503 {
497 return (framesOfTabs[tabId] || {})[frameId]; 504 let frames = framesOfTabs.get(tabId);
505 return frames && frames.get(frameId);
498 }; 506 };
499 507
500 let handlerBehaviorChangedQuota = 508 let handlerBehaviorChangedQuota =
501 chrome.webRequest.MAX_HANDLER_BEHAVIOR_CHANGED_CALLS_PER_10_MINUTES; 509 chrome.webRequest.MAX_HANDLER_BEHAVIOR_CHANGED_CALLS_PER_10_MINUTES;
502 510
503 function propagateHandlerBehaviorChange() 511 function propagateHandlerBehaviorChange()
504 { 512 {
505 // Make sure to not call handlerBehaviorChanged() more often than allowed 513 // Make sure to not call handlerBehaviorChanged() more often than allowed
506 // by chrome.webRequest.MAX_HANDLER_BEHAVIOR_CHANGED_CALLS_PER_10_MINUTES. 514 // by chrome.webRequest.MAX_HANDLER_BEHAVIOR_CHANGED_CALLS_PER_10_MINUTES.
507 // Otherwise Chrome notifies the user that this extension is causing issues. 515 // Otherwise Chrome notifies the user that this extension is causing issues.
(...skipping 24 matching lines...) Expand all
532 }; 540 };
533 541
534 chrome.tabs.query({}, tabs => 542 chrome.tabs.query({}, tabs =>
535 { 543 {
536 tabs.forEach(tab => 544 tabs.forEach(tab =>
537 { 545 {
538 chrome.webNavigation.getAllFrames({tabId: tab.id}, details => 546 chrome.webNavigation.getAllFrames({tabId: tab.id}, details =>
539 { 547 {
540 if (details && details.length > 0) 548 if (details && details.length > 0)
541 { 549 {
542 let frames = framesOfTabs[tab.id] = Object.create(null); 550 let frames = new Map();
551 framesOfTabs.set(tab.id, frames);
543 552
544 for (let i = 0; i < details.length; i++) 553 for (let detail of details)
554 frames.set(detail.frameId, {url: new URL(detail.url)});
555
556 for (let detail of details)
545 { 557 {
546 frames[details[i].frameId] = { 558 let {parentFrameId} = detail;
547 url: new URL(details[i].url),
548 parent: null
549 };
550 }
551
552 for (let i = 0; i < details.length; i++)
553 {
554 let {parentFrameId} = details[i];
555 559
556 if (parentFrameId != -1) 560 if (parentFrameId != -1)
557 frames[details[i].frameId].parent = frames[parentFrameId]; 561 frames.get(detail.frameId).parent = frames.get(parentFrameId);
558 } 562 }
559 } 563 }
560 }); 564 });
561 }); 565 });
562 }); 566 });
563 567
564 chrome.webRequest.onBeforeRequest.addListener(details => 568 chrome.webRequest.onBeforeRequest.addListener(details =>
565 { 569 {
566 // The high-level code isn't interested in requests that aren't 570 // The high-level code isn't interested in requests that aren't
567 // related to a tab or requests loading a top-level document, 571 // related to a tab or requests loading a top-level document,
(...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after
611 615
612 // Add "page" and "frame" if the message was sent by a content script. 616 // Add "page" and "frame" if the message was sent by a content script.
613 // If sent by popup or the background page itself, there is no "tab". 617 // If sent by popup or the background page itself, there is no "tab".
614 if ("tab" in rawSender) 618 if ("tab" in rawSender)
615 { 619 {
616 sender.page = new Page(rawSender.tab); 620 sender.page = new Page(rawSender.tab);
617 sender.frame = { 621 sender.frame = {
618 url: new URL(rawSender.url), 622 url: new URL(rawSender.url),
619 get parent() 623 get parent()
620 { 624 {
621 let frames = framesOfTabs[rawSender.tab.id]; 625 let frames = framesOfTabs.get(rawSender.tab.id);
622 626
623 if (!frames) 627 if (!frames)
624 return null; 628 return null;
625 629
626 let frame = frames[rawSender.frameId]; 630 let frame = frames.get(rawSender.frameId);
627 if (frame) 631 if (frame)
628 return frame.parent; 632 return frame.parent || null;
629 633
630 return frames[0]; 634 return frames.get(0) || null;
631 } 635 }
632 }; 636 };
633 } 637 }
634 638
635 return ext.onMessage._dispatch( 639 return ext.onMessage._dispatch(
636 message, sender, sendResponse 640 message, sender, sendResponse
637 ).indexOf(true) != -1; 641 ).indexOf(true) != -1;
638 }); 642 });
639 643
640 644
(...skipping 90 matching lines...) Expand 10 before | Expand all | Expand 10 after
731 ext.windows = { 735 ext.windows = {
732 create(createData, callback) 736 create(createData, callback)
733 { 737 {
734 chrome.windows.create(createData, createdWindow => 738 chrome.windows.create(createData, createdWindow =>
735 { 739 {
736 afterTabLoaded(callback)(createdWindow.tabs[0]); 740 afterTabLoaded(callback)(createdWindow.tabs[0]);
737 }); 741 });
738 } 742 }
739 }; 743 };
740 }()); 744 }());
OLDNEW
« no previous file with comments | « no previous file | lib/devtools.js » ('j') | lib/popupBlocker.js » ('J')

Powered by Google App Engine
This is Rietveld