import os
from typing import Dict, Literal, Optional, Set

from .const import ENVIRON_KEY_BASE, LOG_POLICY, PBAR_POLICY, TMP_FOLDER_KEY_BASE


def data_folder(task_id: int) -> Optional[str]:
    idstr = str(int(task_id))
    tmp = os.getenv(TMP_FOLDER_KEY_BASE + idstr)
    return tmp


def all_environ() -> Dict[str, str]:
    """returns a dictionary of all environment variables set by any instance of scgenerator"""
    d = dict(filter(lambda el: el[0].startswith(ENVIRON_KEY_BASE), os.environ.items()))
    return d


def pbar_policy() -> Set[Literal["print", "file"]]:
    policy = os.getenv(PBAR_POLICY)
    if policy == "print" or policy is None:
        return {"print"}
    elif policy == "file":
        return {"file"}
    elif policy == "both":
        return {"file", "print"}
    else:
        return set()


def log_policy() -> Set[Literal["print", "file"]]:
    policy = os.getenv(LOG_POLICY)
    if policy == "print" or policy is None:
        return {"print"}
    elif policy == "file":
        return {"file"}
    elif policy == "both":
        return {"file", "print"}
    else:
        return set()