cli.py

import pyperclip
import typer, subprocess
from rich import print
import config
from pathlib import Path
from layers.python_manager.main import PythonManager
from layers.archive_manager.main import PackagesArchive
from layers.pack_installer.main import PackagesManager

app = typer.Typer(
    no_args_is_help=True,
    # если пользователь дал команду без аргументов то не падать с ошибкой а показать справку
    rich_markup_mode='rich',
    # добавить rich панели (группировка комманд по заголовкам)
    add_completion=False,  # убрать блок option в всплывающем меню
)


@app.callback()
def main():
    """CLI интерфейс"""


@app.command()
def version():
    """Получение текущей версии"""
    print(f'[cyan]PyOffline {config.VERSION}[/cyan]')


@app.command()
def show():
    """
    Архив, доступных python версий и пакетов.
    Пример формата архива:
    {'python_storage': ['3.12', '3.14'], 'packages_storage': {'3.12': {'requests': ['2.34.2']}, '3.14': {'requests': ['2.34.2']}}}
    Примеры команд:
        [yellow]show[/yellow]
    """
    print(PackagesArchive.get_archive_packages_by_version())


@app.command()
def get(packages: list[str]):
    """
    Скачать пакеты с PyPi.org (при наличии соединения с Pypi)
    (под капотом используется pip download)
    Примеры команд:
        [yellow]get fastapi uvicorn[/yellow] - скачает последние версии указанных библиотек
        [yellow]get python-dotenv==1.2.2[/yellow] - скачает конкретную версию пакета
        [yellow]get passlib[bcrypt][/yellow] - скачает пакет + extras указанный в скобках
    """
    packages = [p.strip().rstrip(',') for p in packages if p.strip()]
    try:
        PackagesArchive.download(packages_list=packages)
    except Exception:  # noqa
        pass


@app.command()
def get_from(
        file_name: str,
):
    """
    Скачивание списка библиотек из файла, например из requirements.txt, поддерживаются комментарии вида
    Примеры команд:
        [yellow]get-from ./requirements.txt[/yellow]
    """
    file_name = Path(file_name)
    if not file_name.exists():
        print(f'[red]Файл `{file_name}` не найден.[/red]')
        return
    with open(file=file_name, mode='r', encoding='utf8') as f:
        text = f.read()
        x = text.splitlines()
        packages_list = []
        for i in x:
            if i and i[0].isalpha():
                packages_list.append(i.split()[0].replace(',', ''))
        try:
            PackagesArchive.download(packages_list=packages_list)
        except Exception:  # noqa
            pass


@app.command()
def update(
        keep: int | None = typer.Option(None, '--keep', '-k')
):
    """
    Получение latest версий (если они есть на PyPi) для всех пакетов всех версий python в архиве.
    можно оставлять только заданное количество последних версий, чтобы архив не разрастался.
    Опции:
        -k (--keep) - оставлять только заданное количество последних версий, а более старые версии удалять.
    Примеры команд:
        [yellow]`update -k 1`[/yellow] - оставит только последнюю latest версию
    """
    PackagesArchive.update(keep=keep)


@app.command()
def add(
        packages: list[str],
        project_path: Path | None = typer.Option(Path.cwd(), "--project-path", "-p", help="Путь к проекту"),
):
    """
    Установка пакета в целевой проект, создает toml если его нет в целевом проекте и добавляет в него пакет.
    Версия python пакета определяется автоматически.
    Параметры (только в режиме разработчика, в обычном режиме указывать их не нужно):
        -p (--project-path) - путь к целевому проекту
        -pkg (--pkg-name) - название пакета, можно несколько через запятую
    Примеры команд:
        [yellow]add fastapi, uvicorn[/yellow] - указать пакеты через пробел или через запятую
        [yellow]add fastapi, uvicorn -p ./app[/yellow] - если другой проект то указать путь к нему через -p
    """
    project_path = project_path or Path.cwd()
    packages = [p.strip().rstrip(',') for p in packages if p.strip()]
    for pkg in packages:
        PackagesManager.install(
            target_project_path=Path(project_path),
            pkg_name=pkg,
        )


@app.command()
def remove(
        packages: list[str],
        project_path: Path | None = typer.Option(None, "--project-path", "-p", help="Путь к проекту"),
):
    """
    Удаление пакетов из целевого проекта, удаляет пакет из toml проекта.
    Версия python пакета определяется автоматически.
    Примеры команд:
        [yellow]remove fastapi, uvicorn[/yellow] - указать пакеты через пробел или через запятую
        [yellow]remove fastapi, uvicorn -p ./app[/yellow] - если другой проект то указать путь к нему через -p
    """
    project_path = project_path or Path.cwd()
    packages = [p.strip().rstrip(',') for p in packages if p.strip()]
    for pkg in packages:
        PackagesManager.uninstall(
            target_project_path=Path(project_path),
            pkg_name=pkg,
        )


@app.command()
def remove_all(
        project_path: Path | None = typer.Option(None, "--project-path", "-p", help="Путь к проекту"),
):
    """
    Удаление всех пакетов из целевого проекта, очищает toml проекта.
    Версия python пакета определяется автоматически.
    Примеры команд:
        [yellow]remove-all[/yellow] - удаление всех пакетов
        [yellow]remove-all -p ./app[/yellow] - если другой проект то указать путь к нему через -p
    """
    project_path = project_path or Path.cwd()
    PackagesManager.uninstall_all(target_project_path=Path(project_path))


@app.command()
def init(
        project_path: Path | None = typer.Option(None, "--project-path", "-p", help="Путь к проекту"),
        python_version: str | None = typer.Option(..., '-pv', '--python-version'),
        replace_project: bool = typer.Option(False, '-rp', '--replace-project', is_flag=True),
):
    """
    Создание проекта виртуального окружения (папка .venv в корне проекта).
    Опции:
        -pv (--python-version) версия python
        -rp (--replace-project) заменить venv в существующем проекте если есть.
    Примеры команд:
        [yellow]init -pv 3.12[/yellow] - создаст проект в текущей директории
        [yellow]init -pv 3.12 -rp[/yellow] - пересоздаст проект в текущ директории (флаг -rp)
        [yellow]init -p ./target-project -pv 3.12 -rp[/yellow] - создание проекта по указанному пути
    """
    project_path = project_path or Path.cwd()

    PythonManager.create(
        project_path=Path(project_path),
        python_version=python_version,
        replace_project=replace_project,
    )


@app.command()
def sync(
        project_path: Path | None = typer.Option(None, "--project-path", "-p", help="Путь к проекту"),
):
    """
    Установка всех пакетов прописанных в pyproject.toml целевого проекта. Может быть полезно для стартовой инициализации
    проекта, или редактирования .toml
    Примеры команд:
        [yellow]sync[/yellow] - синхронизация пакетов для текущей директории
        [yellow]sync -p ./target-project[/yellow] - синхронизация пакетов для проекта по указанному пути
    * метод в разработке пока не слишком оптимальный по времени выполнения.
    """
    project_path = project_path or Path.cwd()
    PackagesManager.sync(target_project_path=project_path)


@app.command()
def project(
        project_path: Path | None = typer.Option(None, "--project-path", "-p", help="Путь к проекту"),
):
    """
    Получение информации о проекте (путь к интерпретатору, версия python, установленные пакеты)
    Примеры команд:
        [yellow]project[/yellow] - получение информации о проекте в текущей директории
        [yellow]project -p ./target-project[/yellow] - получение информации о проекте по указанному пути
    """
    project_path = project_path or Path.cwd()

    packages_project = PackagesManager.project_info(project_path)
    print(packages_project)


@app.command()
def get_python(python_version: str):
    """
    Загрузка python с https://github.com/astral-sh/python-build-standalone/releases
    Обязательные параметры:
        python_version - желаемая версия python
    Примеры команд:
        [yellow]get-python 3.12[/yellow] - скачает портативную версию python-standalone 3.12
    """
    PythonManager.download_python(python_version=python_version)


@app.command()
def folder():
    """Открыть домашнюю папку приложения"""
    from infrastructure.path_utils.open_folder import open_folder
    open_folder(config.ROOT_DIR)


@app.command()
def tree(
        project_path: Path | None = typer.Option(Path.cwd(), "--project-path", "-p", help="Путь к проекту"),
        show_tree: bool = typer.Option(False, "--show-tree", "-st", is_flag=True),
        show_content: bool = typer.Option(False, "--show-content", "-sc", is_flag=True),
        copy_buffer: bool = typer.Option(False, "--copy-buffer", "-cb", is_flag=True),
):
    """
    Показать дерево и контент по текущему пути или по указанному.
    Опции:
        -p  (--project-path) - путь к целевому каталогу
        -st (--show-tree) - показывать дерево целевого каталога
        -sc (--show-content) - показать контент (содержимое файлов)
        -cb (--copy-buffer) - копировать в буфер (например для чата с ИИ)
    Примеры команд:
        [yellow]tree -st[/yellow]  -  показать дерево каталога
        [yellow]tree -sc[/yellow]  -  показать контент (содержимое файлов)
        [yellow]tree -st -sc[/yellow]  -  показать и дерево контента и содержимое файлов
    """
    from layers.directory_walker.main import show_tree_catalog
    if not show_tree and not show_content and not copy_buffer:
        print(f'[yellow]Не выбрано ни одной опции для демонстрации дерева, укажите -st и/или -sc и/или -cp[/yellow]')
        return
    path = project_path or config.ROOT_DIR
    tree_list, content_list = show_tree_catalog(root_dir=path)
    if show_tree:
        [print(t) for t in tree_list]
    if show_content:
        [print(c) for c in content_list]
    if copy_buffer:
        copy_text = ""
        copy_text += "\n".join([i for i in tree_list])
        copy_text += "\n".join([i for i in content_list])
        try:
            pyperclip.copy(copy_text)
        except Exception as err:
            print(f'[red]Ошибка копирования текста: {err}[/red]')
            return

        print('[green]Скопировано в буфер обмена.[/green]')


@app.command()
def tree_cfg():
    """
    Открыть папку с настройками tree (иконки, исключаемые файлы и папки).
    [yellow]hidden.json[/yellow] - в исключаемых файлах и папках можно указывать *, например:
        "http_*", # скроет все папки которые начинаются на http_
        "*_http", # скроет все папки которые заканчиваются на _http
        "*_http_*", # скроет все папки которые содержат _http_
    [yellow]icons.json[/yellow]В иконках можно указывать эмодзи для обозначения файлов и папок.
    """
    from infrastructure.path_utils.open_folder import open_folder
    open_folder(config.TREE_HIDDEN_CONFIG.parent)


# доступно только в режиме разработки
if not config.EXE_MODE:
    @app.command()
    def build(
            name: str = typer.Option('pyoff', '--name', '-n')
    ):
        """
        Сборка exe (bin) приложения.
        Опции:
            -n (--name) - название приложения.
        Примеры команд:
            [yellow]build -n pyoff[/yellow] - создаст дистрибутив с названием pyoff
        """
        from build import start_build
        start_build(name=name)


    @app.command()
    def run_tests(
            v: bool = typer.Option(False, '-v', flag_value=True),
            s: bool = typer.Option(False, '-s', flag_value=True),
    ):
        """
        Запуск тестов.
        Опции:
            -v - подробный режим с путем к каждому модулю
            -s - показывать принты внутри тестов
        Примеры команд:
            [yellow]run-tests -v -s[/yellow] - запуск тестов
        """
        cmd = ['pytest']

        # # добавление опций / add options
        cmd.extend(['-v']) if v else cmd.extend([])
        cmd.extend(['-s']) if s else cmd.extend([])
        result = subprocess.run(cmd, cwd=config.ROOT_DIR)

        if result.returncode != 0:
            print(
                f"[red]Тесты не пройдены (код {result.returncode})[/red]")
            return False
        else:
            print("[green]Все тесты пройдены[/green]")
            return True

if __name__ == '__main__':
    app()