hooks.py

from __future__ import annotations

import importlib
import shutil
from pathlib import Path
import sys

from usr.plugins.rlm.helpers.bootstrap import ensure_rlm_dependency
from usr.plugins.rlm.helpers.docker_setup import (
    activate_docker_cli_shim,
    deactivate_docker_cli_shim,
)

PLUGIN_PACKAGE_PREFIX = "usr.plugins.rlm"
DEPENDENCY_PACKAGE_PREFIX = "rlm"
PLUGIN_ROOT = Path(__file__).resolve().parent


def install() -> None:
    _ensure_rlm_dependency()
    activate_docker_cli_shim()
    _clear_dependency_modules()
    _clear_plugin_modules()
    _clear_plugin_bytecode()


def pre_update() -> None:
    deactivate_docker_cli_shim()
    _clear_dependency_modules()
    _clear_plugin_modules()
    _clear_plugin_bytecode()


def uninstall() -> None:
    deactivate_docker_cli_shim()
    _clear_dependency_modules()
    _clear_plugin_modules()
    _clear_plugin_bytecode()


def _ensure_rlm_dependency() -> None:
    ensure_rlm_dependency()


def _clear_plugin_modules() -> None:
    for module_name in list(sys.modules):
        if module_name == PLUGIN_PACKAGE_PREFIX or module_name.startswith(
            f"{PLUGIN_PACKAGE_PREFIX}."
        ):
            sys.modules.pop(module_name, None)
    importlib.invalidate_caches()


def _clear_dependency_modules() -> None:
    for module_name in list(sys.modules):
        if module_name == DEPENDENCY_PACKAGE_PREFIX or module_name.startswith(
            f"{DEPENDENCY_PACKAGE_PREFIX}."
        ):
            sys.modules.pop(module_name, None)
    importlib.invalidate_caches()


def _clear_plugin_bytecode() -> None:
    for cache_dir in PLUGIN_ROOT.rglob("__pycache__"):
        shutil.rmtree(cache_dir, ignore_errors=True)