helpers.py

import re
import textwrap
from functools import partial
from pathlib import Path
from typing import NoReturn, Pattern

import click
from async_eval.ext.pydevd import generate_main_script
from click.exceptions import Exit

PLUGIN_CHECK_CODE_REGEX: Pattern[str] = re.compile(
    r"(?<=val PYDEVD_ASYNC_PLUGIN = \"\"\").*(?=\"\"\")",
    re.DOTALL,
)
PLUGIN_UPDATE_CODE_REGEX: Pattern[str] = re.compile(
    r"val PYDEVD_INLINE_ASYNC_PLUGIN.*",
    re.DOTALL,
)

err = partial(click.secho, fg="red", err=True)


def cli_exit(code: int = 128) -> NoReturn:
    raise Exit(code)


def find_path(pattern: str) -> Path:
    try:
        path, *_ = Path.cwd().rglob(pattern)
        return path
    except ValueError:
        err(f"Can't resolve path with pattern {pattern}")
        cli_exit()


@click.group()
def entry_point():
    pass


@entry_point.command(name="plugin")
@click.option("--check", type=bool, is_flag=True, default=False)
def plugin_entry_point(check: bool) -> None:
    plugin: Path = find_path("AsyncPyDebugUtils.kt")
    plugin_code = generate_main_script().replace('"""', "'''")

    if check:
        code: str = PLUGIN_CHECK_CODE_REGEX.search(plugin.read_text("utf-8")).group()

        if code.strip() != plugin_code.strip():
            err("Kotlin async-pydevd plugin is outdated, please run 'helpers plugin'")
            cli_exit()
    else:
        async_pydevd_plugin = "\n".join([
            f'val PYDEVD_INLINE_ASYNC_PLUGIN = """{repr(plugin_code)}"""',
            f'val PYDEVD_ASYNC_PLUGIN = """\n{plugin_code}\n""".trimStart()',
        ])

        plugin.write_text(
            PLUGIN_UPDATE_CODE_REGEX.sub("", plugin.read_text("utf-8")) + async_pydevd_plugin,
            "utf-8",
        )


README_SEPARATOR: str = "<!-- Plugin Examples -->"

IMAGES: list[str] = [
    "evaluate_expression.jpeg",
    "conditional_breakpoint.jpeg",
    "debugger_console.jpeg",
    "python_console.jpeg",
]


@entry_point.command(name="readme")
def readme_entry_point() -> None:
    readme: Path = find_path("README.md")

    def generate_html_for_img(img: str) -> str:
        name, *_ = img.split(".")
        title = name.replace("_", " ").title()

        return textwrap.dedent(
            f"""
                <!-- Start - {title} -->
                <div>
                  <h2 align="center">{title}</h2>
                
                  <h1 align="center">
                    <img width="80%" alt="{name}" src="/images/{name}.jpeg">
                  </h1>
                </div>
                <!-- End - {title} -->
            """
        )

    main_content, *_ = readme.read_text("utf-8").split(README_SEPARATOR)

    readme.write_text(
        "\n".join(
            (
                main_content.rstrip(),
                "",
                README_SEPARATOR,
                "".join(generate_html_for_img(img) for img in IMAGES),
            )
        ),
        "utf-8",
    )


__all__ = ["entry_point"]