# 把本机 .env 补进进程环境，不覆盖已经有值的变量。

from __future__ import annotations

import os
from pathlib import Path


# 按顺序读 .env，只填当前为空的变量。
#
# 文件不存在则跳过。已有非空环境变量不覆盖。值两边成对的单双引号会去掉。
# 多个路径时，先读到的非空值留下。
def load_env(*paths: Path) -> None:
    for path in paths:
        _fill_env(path)


# 该班次要读的 .env：本包，加购再加选品，最后 rpa_oms。
def job_env_files(package_dir: Path) -> tuple[Path, ...]:
    repo = package_dir.resolve().parent
    files = [package_dir / ".env"]
    if package_dir.name == "rpa_cart":
        files.append(repo / "rpa_selection" / ".env")
    files.append(repo / "rpa_oms" / ".env")
    return tuple(files)


# 先读班次 .env，空缺再补后面的文件。
#
# 默认顺序见 job_env_files。paths 给定时只读这一列，供测试注入。
# 已有非空环境变量不覆盖。
def load_job_env(package_dir: Path, paths: list[Path] | None = None) -> None:
    load_env(*(paths if paths is not None else job_env_files(package_dir)))


# 把单个 .env 里尚未占用的键写进 os.environ。
def _fill_env(path: Path) -> None:
    if not path.is_file():
        return
    for raw in path.read_text(encoding="utf-8").splitlines():
        line = raw.strip()
        if not line or line.startswith("#") or "=" not in line:
            continue
        key, _, value = line.partition("=")
        key = key.strip()
        value = value.strip()
        if len(value) >= 2 and value[0] == value[-1] and value[0] in {'"', "'"}:
            value = value[1:-1]
        if not key or not value:
            continue
        current = os.environ.get(key, "")
        if not current:
            os.environ[key] = value