import json
from channels.generic.websocket import WebsocketConsumer
from asgiref.sync import async_to_sync
# from .funcs import *

class ChatConsumer(WebsocketConsumer):
    def connect(self):
        print('ws connect')
        self.room_group_name = 'test'

        async_to_sync(self.channel_layer.group_add)(
            self.room_group_name,
            self.channel_name
        )
        print(f'self.room_group_name = {self.room_group_name}')
        print(f'self.channel_name = {self.channel_name}')
        self.accept()

    def receive(self, text_data):
        print(f'ws receive text_data = {text_data}')

        from .funcs import send_msg
        resDict = send_msg(text_data)
        resDict.update({
            'type': 'ws_send_msg',
            'message': resDict,
            # 'sender': sender
        })
        print(f'send_msg res = {len(resDict)}')

        self.send(json.dumps(resDict))

        async_to_sync(self.channel_layer.group_send)(
            self.room_group_name,
            resDict
            # {
            #     'type': 'chat_message',
            #     'message': message,
            #     'sender': sender
            # }
        )

    def ws_send_msg(self, event):
        print('ws ws_send_msg')

        message = event['message']
        # sender = event['sender']

        self.send(text_data=json.dumps({
            'type': 'chat',
            'message': message,
            # 'sender': sender
        }))