from __future__ import annotations

import os

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


class EnvSource(VersionSourceInterface):
    PLUGIN_NAME = "env"

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

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

        if variable not in os.environ:
            message = f"environment variable `{variable}` is not set"
            raise RuntimeError(message)

        return {"version": os.environ[variable]}

    def set_version(self, version: str, version_data: dict) -> None:
        message = "Cannot set environment variables"
        raise NotImplementedError(message)