import os
import hjson

from matrix_client.client import MatrixClient

from .api import CustomMatrixHttpApi
from .room import CustomRoom

MESSAGES_DIR = os.path.join(os.path.dirname(__file__), 'messages')
MESSAGES_LOCATION = MESSAGES_DIR + '/messages.dutch.hjson'

class CustomMatrixClient(MatrixClient):
    def __init__(self, base_url, token=None):
        super().__init__(base_url, token)

        self.api = CustomMatrixHttpApi(base_url, token)

        # Load messages
        with open(MESSAGES_LOCATION) as hjson_data:
            self.messages = hjson.load(hjson_data)


    def _mkroom(self, room_id):
        room = CustomRoom(self, room_id)

        self.rooms[room_id] = room
        return self.rooms[room_id]

    def create_direct_room(self, invitees=None):
        content = {
            "visibility": "private",
            "is_direct": True,
            "invite": [invitees]
        }

        room_dict = self.api._send("POST", "/createRoom", content)

        return self._mkroom(room_dict["room_id"])


    def send_message_private_public(self, room, event, message):
        """This method takes a room, event, and message and makes sure
        that the message is sent in a private room and not in a public
        room. If no private room exists, it will create a private room
        with the sender of the event.
        """

        orig_room = room
        found_room = False

        for room_id, room in self.rooms.items():
            joined_members = room.get_joined_members()

            # Check for rooms with only two members
            if len(joined_members) == 2:
                # Check if sender is in that room
                for member in joined_members:
                    if event['sender'] == member.user_id:
                        found_room = True

            if found_room:
                # If the flag is set, we do not need to check further rooms
                break

        # Send help message to an existing room or to a new room
        if found_room:
            room.send_html(message)
        else:
            room = self.create_direct_room(event['sender']);
            room.send_html(message)

        if room != orig_room:
            display_name_sender = self.api.get_display_name(event['sender'])
            orig_room.send_text(self.messages['private_message'].format(
                display_name_sender))