import os
import traceback
import contextlib
import re


class PrettyException:
    def __init__(self, e: 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