import contextlib
import os
import re
import traceback
from typing import Type


class PrettyException:
    def __init__(self, e: Type[BaseException] | Exception):
        self.long = f"""
😍 {e.__class__.__name__}
👉 {"".join(traceback.format_exception_only(e)).strip()}
🐊 <code>{e.__traceback__.tb_frame.f_code.co_filename.replace(os.getcwd(), "")}\r
</code>:{e.__traceback__.tb_frame.f_lineno}
"""
        self.short = (
            f"{e.__class__.__name__}: "
            f'{"".join(traceback.format_exception_only(e)).strip()}'
        )

        self.pretty_exception = (
            f"{self.long}\n\n" f"⬇️ Trace:" f"{self.get_full_stack()}"
        )

    @staticmethod
    def get_full_stack():
        full_stack = traceback.format_exc().replace(
            "Traceback (most recent call last):\n", ""
        )

        line_regex = r'  File "(.*?)", line ([0-9]+), in (.+)'

        def format_line(line: str) -> str:
            filename_, lineno_, name_ = re.search(line_regex, line).groups()
            with contextlib.suppress(Exception):
                filename_ = os.path.basename(filename_)

            return (
                f"🤯 <code>{filename_}:{lineno_}</code> (<b>in</b>"
                f" <code>{name_}</code> call)"
            )

        full_stack = "\n".join(
            [
                (
                    format_line(line)
                    if re.search(line_regex, line)
                    else f"<code>{line}</code>"
                )
                for line in full_stack.splitlines()
            ]
        )

        return full_stack

    def __str__(self):
        return self.pretty_exception