mirror of
https://github.com/Dispatcharr/Dispatcharr.git
synced 2026-01-23 10:45:27 +00:00
509f2be3a8
Reworks websocket connection to be more robust and notify user of connection errors. Will retry if websocket connection dies.
66 lines
2.5 KiB
Python
66 lines
2.5 KiB
Python
import json
|
|
from channels.generic.websocket import AsyncWebsocketConsumer
|
|
import re, logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class MyWebSocketConsumer(AsyncWebsocketConsumer):
|
|
async def connect(self):
|
|
try:
|
|
await self.accept()
|
|
self.room_name = "updates"
|
|
await self.channel_layer.group_add(self.room_name, self.channel_name)
|
|
# Send a connection confirmation to the client with consistent format
|
|
await self.send(text_data=json.dumps({
|
|
'type': 'connection_established',
|
|
'data': {
|
|
'success': True,
|
|
'message': 'WebSocket connection established successfully'
|
|
}
|
|
}))
|
|
except Exception as e:
|
|
import logging
|
|
logger = logging.getLogger(__name__)
|
|
logger.error(f"Error in WebSocket connect: {str(e)}")
|
|
# If an error occurs during connection, attempt to close
|
|
try:
|
|
await self.close(code=1011) # Internal server error
|
|
except:
|
|
pass
|
|
|
|
async def disconnect(self, close_code):
|
|
try:
|
|
await self.channel_layer.group_discard(self.room_name, self.channel_name)
|
|
except Exception as e:
|
|
import logging
|
|
logger = logging.getLogger(__name__)
|
|
logger.error(f"Error in WebSocket disconnect: {str(e)}")
|
|
|
|
async def receive(self, text_data):
|
|
data = json.loads(text_data)
|
|
|
|
if data["type"] == "m3u_profile_test":
|
|
from apps.proxy.ts_proxy.url_utils import transform_url
|
|
|
|
def replace_with_mark(match):
|
|
# Wrap the match in <mark> tags
|
|
return f"<mark>{match.group(0)}</mark>"
|
|
|
|
# Apply the transformation using the replace_with_mark function
|
|
try:
|
|
search_preview = re.sub(data["search"], replace_with_mark, data["url"])
|
|
except Exception as e:
|
|
search_preview = data["search"]
|
|
logger.error(f"Failed to generate replace preview: {e}")
|
|
|
|
result = transform_url(data["url"], data["search"], data["replace"])
|
|
await self.send(text_data=json.dumps({
|
|
"data": {
|
|
'type': 'm3u_profile_test',
|
|
'search_preview': search_preview,
|
|
'result': result,
|
|
}
|
|
}))
|
|
|
|
async def update(self, event):
|
|
await self.send(text_data=json.dumps(event))
|