from telegram import Update, InlineKeyboardMarkup, InlineKeyboardButton
from telegram.constants import ParseMode
from Config import INLINE_SEP, FEEDBACK_TIMEOUT
import logging, re, json
from asyncio import sleep
logger = logging.getLogger(__name__)

re_flags = re.I | re.M

replacers = [
    {
        "regex": re.compile(r"(?:https?:\/\/)?(?:www\.)?youtu\.?be(?:\.com)?\/?.*(?:watch|embed)?(?:.*v=|v\/|\/)([\w\-_]+)\&?", re_flags),
        "becomes": "https://y.outube.duckdns.org/{}",
    },
    {
        "regex": re.compile(r"(?:https?:\/\/)?(?:www\.)?twitter\.com\/(?:#!\/)?(.*)\/status(?:es)?\/([^\/\?\s]+)", re_flags),
        "becomes": "https://fxtwitter.com/{}/status/{}",
    },
    {
        "regex": re.compile(r"(?:https?:\/\/)?(?:www\.)?x\.com\/(?:#!\/)?(.*)\/status(?:es)?\/([^\/\?\s]+)", re_flags),
        "becomes": "https://fixupx.com/{}/status/{}",
    },
]

link_message = "Da {}[\.]({})"

def get_callback_data(feedback: bool) -> str:
    payload = { "feedback": feedback }
    return "feedback" + INLINE_SEP + json.dumps(payload)

def get_message_markup() -> InlineKeyboardMarkup:
    buttons = [
        [
            InlineKeyboardButton(text="✅", callback_data=get_callback_data(True)),
            InlineKeyboardButton(text="❌", callback_data=get_callback_data(False)),
        ]
    ]
    return InlineKeyboardMarkup(buttons)

def format_template(template: str, regex_result) -> str:
    result_type = type(regex_result)
    if result_type is str:
        return template.format(regex_result)
    elif result_type is tuple or result_type is list:
        return template.format(*regex_result)
    elif result_type is dict:
        return template.format(**regex_result)
    else:
        return ""

def parse_text(message: str) -> list:
    output = []
    for site in replacers:
        regex = site["regex"]
        res = regex.findall(message)
        for r in res:
            link = format_template(site["becomes"], r)
            output.append(link)
    return output

async def replace(update: Update, _) -> None:
    try:
        links = parse_text(update.message.text)
    except TypeError:
        links = parse_text(update.message.caption)

    for link in links:
        logger.info(link)
        text = link_message.format(update.effective_user.mention_markdown_v2(update.effective_user.name), link)
        message = await update.effective_chat.send_message(text, parse_mode=ParseMode.MARKDOWN_V2)
        await sleep(FEEDBACK_TIMEOUT)
        await message.edit_reply_markup(reply_markup=get_message_markup())

async def feedback(update: Update, _, data_json: str) -> None:
    data = json.loads(data_json)

    if data["feedback"]:
        await update.callback_query.answer("Bene!")
        await update.effective_message.edit_reply_markup()
        return

    await update.callback_query.answer("Ci ho provato...")
    await update.effective_message.delete()
    return