from hatchling.version.core import VersionFile
from hatchling.version.source.plugin.interface import VersionSourceInterface


class RegexSource(VersionSourceInterface):
    PLUGIN_NAME = "regex"

    def get_version_data(self) -> dict:
        relative_path = self.config.get("path", "")
        if not relative_path:
            message = "option `path` must be specified"
            raise ValueError(message)

        if not isinstance(relative_path, str):
            message = "option `path` must be a string"
            raise TypeError(message)

        pattern = self.config.get("pattern", "")
        if not isinstance(pattern, str):
            message = "option `pattern` must be a string"
            raise TypeError(message)

        version_file = VersionFile(self.root, relative_path)
        version = version_file.read(pattern=pattern)

        return {"version": version, "version_file": version_file}

    def set_version(self, version: str, version_data: dict) -> None:  # noqa: PLR6301
        version_data["version_file"].set_version(version)