"""Hatchling build hook for ``openbb_us_eia/assets/eia_catalog.json.xz``."""

from __future__ import annotations

import os
import subprocess
import sys
from contextlib import suppress
from pathlib import Path
from typing import TextIO

from hatchling.builders.hooks.plugin.interface import (  # ty: ignore[unresolved-import]
    BuildHookInterface,
)

_ROOT = Path(__file__).resolve().parent
_CATALOG_PATH = _ROOT / "openbb_us_eia" / "assets" / "eia_catalog.json.xz"
_GENERATE_CATALOG = _ROOT / "openbb_us_eia" / "utils" / "generate_catalog.py"
_FORCE_ENV = "OPENBB_EIA_FORCE_CATALOG_REBUILD"


def _open_tty() -> TextIO | None:
    """Open ``/dev/tty`` for direct terminal output."""
    with suppress(OSError):
        return open("/dev/tty", "w", buffering=1, encoding="utf-8")  # noqa: SIM115
    return None


class EiaCatalogBuildHook(BuildHookInterface):
    """Materialize ``eia_catalog.json.xz`` before packaging."""

    PLUGIN_NAME = "eia-catalog"

    _generated_this_session: bool = False

    def initialize(self, version: str, build_data: dict) -> None:
        """Generate or reuse the catalog, then mark it for inclusion."""
        tty = _open_tty()

        def _say(msg: str) -> None:
            """Emit ``msg`` to ``/dev/tty`` and stderr."""
            line = msg if msg.endswith("\n") else msg + "\n"
            if tty is not None:
                with suppress(OSError):
                    tty.write(line)
                    tty.flush()
            sys.stderr.write(line)
            sys.stderr.flush()

        cls = type(self)

        force = (
            os.environ.get(_FORCE_ENV, "").lower() in {"1", "true", "yes"}
            and not cls._generated_this_session
        )
        try:
            if _CATALOG_PATH.exists() and not force:
                reuse_reason = (
                    "this build session already regenerated it"
                    if cls._generated_this_session
                    else f"set {_FORCE_ENV}=1 to rebuild"
                )
                _say(f"eia-catalog: reusing {_CATALOG_PATH.name} ({reuse_reason})")
            else:
                _say(
                    "eia-catalog: regenerating from the EIA APIv2 metadata tree"
                    " (api.eia.gov/v2, requires EIA_API_KEY or the OpenBB"
                    " 'eia_api_key' credential)..."
                )
                env = {**os.environ, "PYTHONUNBUFFERED": "1"}
                proc = subprocess.Popen(  # noqa: S603
                    [sys.executable, str(_GENERATE_CATALOG), "--asset-only"],
                    cwd=str(_ROOT),
                    env=env,
                    stdout=subprocess.PIPE,
                    stderr=subprocess.STDOUT,
                    text=True,
                    bufsize=1,
                )
                assert proc.stdout is not None  # noqa: S101
                for line in proc.stdout:
                    if tty is not None:
                        with suppress(OSError):
                            tty.write(line)
                            tty.flush()
                    sys.stderr.write(line)
                    sys.stderr.flush()
                returncode = proc.wait()

                if returncode != 0 or not _CATALOG_PATH.exists():
                    self.app.abort(
                        "eia-catalog: failed to generate eia_catalog.json.xz "
                        f"(exit code {returncode})."
                    )

                cls._generated_this_session = True
        finally:
            if tty is not None:
                tty.close()

        rel = _CATALOG_PATH.relative_to(_ROOT).as_posix()
        build_data.setdefault("force_include", {})[str(_CATALOG_PATH)] = rel
        build_data.setdefault("artifacts", []).append(rel)