import re
import xml.etree.ElementTree as etree
import xml.etree.ElementTree as ET
from typing import Tuple

import favicon
import requests
import streamlit as st
from markdown.inlinepatterns import InlineProcessor
from streamlit_extras.mention import mention

try:
    from streamlit import cache_data  # streamlit >= 1.18.0
except ImportError:
    from streamlit import experimental_memo as cache_data  # streamlit >= 0.89

SUPPORTED_PLATFORMS = ("github", "notion", "twitter", "streamlit")
GITHUB_ICON = "https://cdn-icons-png.flaticon.com/512/25/25231.png"
NOTION_ICON = "https://upload.wikimedia.org/wikipedia/commons/4/45/Notion_app_logo.png"
TWITTER_ICON = "https://seeklogo.com/images/T/twitter-icon-circle-blue-logo-0902F48837-seeklogo.com.png"
STREAMLIT_ICON = "https://aws1.discourse-cdn.com/business7/uploads/streamlit/original/2X/f/f0d0d26db1f2d99da8472951c60e5a1b782eb6fe.png"

AT_SIGN_RE = r"@(?P<a>\([^)]+\))(?P<b>\([^)]+\))?(?P<c>\([^)]+\))?"


class AtSignProcessor(InlineProcessor):
    """Transforms '@(icon)(label)(url)' into HTML.
    Also works with '@(label)(url)' or '@(url)' in some cases.
    """

    @staticmethod
    def _add_https(url: str) -> str:
        if not url.startswith("https://"):
            url = "https://" + url
        return url

    def handleMatch(self, m: re.Match, data=None) -> Tuple[etree.Element, int, int]:
        """This function is called whenever a match is found.
        We want it to replace the '@(...)' pattern with an HTML element for the link.

        Args:
            m (re.Match): Match object
            data (_type_): [Not used - not sure what this does]

        Returns:
            (etree.Element, int, int): HTML element, with its starting and ending index.
        """
        groups = [group.strip("()") for group in m.groups() if group is not None]

        num_groups = len(groups)
        if num_groups == 1:
            url = groups[0]
            icon, label = self._guess_icon_and_label(url)
        elif num_groups == 2:
            label, url = groups
            icon, _ = self._guess_icon_and_label(url)
        elif num_groups == 3:
            icon, label, url = groups

        url = self._add_https(url)

        html = mention(label=label, icon=icon, url=url, write=False)
        # Remove <style> from mention to ensure a valid XML
        # Instead, we add the <style> in the main mdlit() function
        if html.find("</style>") != -1:
            html = html[html.find("</style>") + len("</style>") :]
        el = ET.ElementTree(ET.fromstring(str(html))).getroot()
        el.set("style", "display: inline; color:inherit; text-decoration:inherit;")
        return el, m.start(0), m.end(0)

    @staticmethod
    @cache_data
    def _get_favicon(url: str) -> str:
        favicons = favicon.get(url, timeout=2)
        return favicons[0].url

    @staticmethod
    @cache_data
    def _get_page_title(url: str) -> str:
        n = requests.get(url)
        al = n.text
        return al[al.find("<title>") + 7 : al.find("</title>")]

    def _guess_icon_and_label(self, url: str) -> Tuple[str, str]:
        """Find out plausible icon and label from URL alone

        Args:
            url (str): URL of the link

        Returns:
            str: Icon of the link
            str: Label of the link
        """
        if "twitter.com" in url:
            icon = "twitter"
            # Tweet
            if "status" in url:
                label = url.split("twitter.com/")[1].split("/")[0]
            # Username
            else:
                label = url.split("/")[-1]
        elif (
            "streamlitapp.com" in url or "streamlit.io" in url or "streamlit.app" in url
        ):
            icon = "streamlit"
            label = "Streamlit App"
        elif "github.com" in url:
            icon = "github"
            label = "/".join(url.split("/")[-2:])
        elif "notion.so" in url:
            icon = "notion"
            label = "Notion page"
        else:
            try:
                icon = self._get_favicon(url)
                label = self._get_page_title(url)
            except Exception as e:
                # st.write(e)
                icon = "🔗"
                label = "Link"
        return icon, label