#203 websockets

AdguardTeam · May 5, 2016 · c80af42 · c80af42
1 parent 9471e0c
commit c80af42
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/Extension/lib/content-script/preload.js b/Extension/lib/content-script/preload.js
@@ -211,7 +211,7 @@
             }
         }
 
-        // Override window.WebSocket
+        // Override window.WebSocket running the function from websocket.js
         // https://github.com/AdguardTeam/AdguardBrowserExtension/issues/203
         scriptsToApply.unshift('(' + overrideWebSocket.toString() + ')();');
 

diff --git a/Extension/lib/content-script/websocket.js b/Extension/lib/content-script/websocket.js
@@ -15,6 +15,13 @@
  * along with Adguard Browser Extension.  If not, see <http://www.gnu.org/licenses/>.
  */
 
+/**
+ * The function overrides window.WebSocket with our wrapper,
+ * that will check url with filters through messaging with content script.
+ *
+ * This function will be inlined as page script.
+ * https://github.com/AdguardTeam/AdguardBrowserExtension/issues/203
+ */
 var overrideWebSocket = function () {
 
     'use strict';
@@ -198,6 +205,7 @@ var overrideWebSocket = function () {
 
     };
 
+    // Safari doesn't have EventTarget
     var EventTarget = window.EventTarget || Element;
     WebSocket.prototype = Object.create(EventTarget.prototype, {
         CONNECTING: {value: 0},
@@ -261,7 +269,11 @@ var overrideWebSocket = function () {
     window.WebSocket = WebSocket;
 };
 
-
+/**
+ * Listener for websocket wrapper messages.
+ *
+ * @param event
+ */
 function pageMessageListener(event) {
     if (!(event.source == window &&
         event.data.direction &&