import logging
import os

import requests
from jinja2 import Environment, FileSystemLoader

logger = logging.getLogger(__name__)
absolute_path = os.path.split(os.path.abspath(__file__))[0]


def markdown_escape(text):
    text = str(text)
    for special_char in ['\\', '`', '*', '_', '{', '}', '[', ']', '(', ')', '#', '+', '-', '.', '!']:
        text = text.replace(special_char, fr'\{special_char}')
    return text


def create_block_payload(chat_id, message_variables):
    return generate_payload(chat_id, message_variables, 'block.md.j2')


def create_balance_payload(chat_id, message_variables):
    return generate_payload(chat_id, message_variables, 'balance.md.j2')


def generate_payload(chat_id, message_variables, template_name):
    payload = {'chat_id': chat_id, 'parse_mode': 'MarkdownV2'}
    template_path = os.path.join(absolute_path, 'templates')
    loader = FileSystemLoader(template_path)
    env = Environment(loader=loader)
    template = env.get_template(template_name)
    template_variables = {}
    for key, value in message_variables.items():
        template_variables[key] = markdown_escape(value)
    text = template.render(**template_variables)
    payload['text'] = text
    return payload


def send_message(auth_key, payload):
    logger.debug(payload)
    r = requests.post(f'https://api.telegram.org/bot{auth_key}/sendMessage', json=payload)
    r.raise_for_status()