from __future__ import annotations

import importlib

import pkgutil

import warnings

from collections.abc import Iterable

from importlib import metadata

from typing import TYPE_CHECKING

if TYPE_CHECKING:

from collections.abc import Iterable

from commitizen.cz.base import BaseCommitizen

def discover_plugins(

path: Iterable[str] | None = None,

) -> dict[str, type[BaseCommitizen]]:

"""Discover commitizen plugins on the path

Args:

path (Path, optional): If provided, 'path' should be either None or a list of paths to look for

modules in. If path is None, all top-level modules on sys.path.. Defaults to None.

Returns:

Dict[str, Type[BaseCommitizen]]: Registry with found plugins

"""

for _, name, _ in pkgutil.iter_modules(path):

if name.startswith("cz_"):

mod = importlib.import_module(name)

if hasattr(mod, "discover_this"):

warnings.warn(

UserWarning(

f"Legacy plugin '{name}' has been ignored: please expose it the 'commitizen.plugin' entrypoint"

)

)

return {

ep.name: ep.load() for ep in metadata.entry_points(group="commitizen.plugin")

}

registry: dict[str, type[BaseCommitizen]] = discover_plugins()