Fix: Cross origin websockets not allowed (#463)

web/handlers/subscribe.py

@@ -12,13 +12,13 @@
 import time
 import traceback
 from typing import Any, Dict
-from urllib.parse import quote
+from urllib.parse import quote, urlparse
 
 import aiohttp
 from tornado import httputil
 from tornado.web import Application
 
-from config import proxies
+from config import proxies, domain
 
 from .base import *
 
@@ -57,6 +57,10 @@ class SubscribeUpdatingHandler(BaseWebSocketHandler):
     updating = False
     updating_start_time = 0
 
+    def check_origin(self, origin):
+        parsed_origin = urlparse(origin)                              
+        return parsed_origin.netloc.endswith(domain)
+
     async def update(self, userid):
         SubscribeUpdatingHandler.updating = True
         SubscribeUpdatingHandler.updating_start_time = int(time.time())