import time
from enum import Enum
import logging

import requests

_username = None
_password = None
_auth = None
_url = None


def initialize(config):
    global _username, _password, _url, _auth
    if config.es_auth:
        _auth = True
        _username = config.es_username
        _password = config.es_password
    _url = config.es_url


def es_connection(uuid, reason, count=0):
    if ConnectionReason(reason).name != "SEEN":
        es_payload = {
            "uuid": uuid,
            "time": (lambda: int(round(time.time() * 1000)))(),
            "reason": ConnectionReason(reason).name,
            "count": count,
        }
    else:
        es_payload = {
            "uuid": uuid,
            "time": (lambda: int(round(time.time() * 1000)))(),
            "reason": ConnectionReason(reason).name,
        }
    es_post_request("connections/_doc/", es_payload)


def es_chat_message(uuid, display_name, message, message_unformatted):
    es_payload = {
        "uuid": uuid,
        "display_name": display_name,
        "message": message,
        "message_unformatted": message_unformatted,
        "time": (lambda: int(round(time.time() * 1000)))(),
    }
    es_post_request("chat_messages/_doc/", es_payload)


def es_raw_message(type, message):
    es_payload = {
        "time": (lambda: int(round(time.time() * 1000)))(),
        "type": type,
        "message": message,
    }
    es_post_request("raw_messages/_doc/", es_payload)


def es_post_request(endpoint, payload):
    the_url = "{}{}".format(_url, endpoint)
    if _auth:
        post = requests.post(the_url, auth=(_username, _password), json=payload)
    else:
        post = requests.post(the_url, json=payload)
    logging.debug("[Elasticsearch POST] {}".format(post.text))


class ConnectionReason(Enum):
    CONNECTED = "CONNECTED"
    DISCONNECTED = "DISCONNECTED"
    SEEN = "SEEN"