import json
import time
from typing import List, Tuple

import constants

HEADER_LENGTH = 10


def prepare(message: str, encoding='utf-8') -> bytes:
    """Prepares a message for sending through a socket by adding a proper header and encoding it."""
    header = f'{len(message):<{HEADER_LENGTH}}'
    return (header + message).encode(encoding)


def prepare_json(object) -> bytes:
    """
    Prepares a object for sending as encoded JSON with a header.

    :param object: A JSON-encodable object
    :return: Encoded JSON
    """
    return prepare(json.dumps(object))


def prepare_message(nickname: str, message: str, color: str, message_id: int, timestamp: int = None) -> bytes:
    return prepare_json(
        {
            'type': constants.Types.MESSAGE,
            'nickname': nickname,
            'content': message,
            'color': color,
            'time': timestamp or int(time.time()),
            'id': message_id,
        }
    )


def prepare_message_history(messages: List[Tuple[int, str, str, str, int]]) -> bytes:
    return prepare_json(
        {
            'type': constants.Types.MESSAGE_HISTORY,
            'messages': [
                {
                    'type': constants.Types.MESSAGE,
                    'nickname': nickname,
                    'content': message,
                    'color': color,
                    'time': timestamp,
                    'id': message_id
                } for message_id, nickname, color, message, timestamp in messages
            ]
        }
    )


def prepare_request(request: str) -> bytes:
    """Helper function for creating a request message."""
    return prepare_json({'type': constants.Types.REQUEST, 'request': request})