import os

from sqlmesh import macro


@macro()
def psd_glob(evaluator) -> str:
    """Return a quoted glob path for all PSD CSV gzip files under LANDING_DIR."""
    landing_dir = evaluator.var("LANDING_DIR") or os.environ.get("LANDING_DIR", "data/landing")
    return f"'{landing_dir}/psd/**/*.csv.gzip'"


@macro()
def cot_glob(evaluator) -> str:
    """Return a quoted glob path for all COT CSV gzip files under LANDING_DIR."""
    landing_dir = evaluator.var("LANDING_DIR") or os.environ.get("LANDING_DIR", "data/landing")
    return f"'{landing_dir}/cot/**/*.csv.gzip'"


@macro()
def prices_glob(evaluator) -> str:
    """Return a quoted glob path for all coffee price CSV gzip files under LANDING_DIR."""
    landing_dir = evaluator.var("LANDING_DIR") or os.environ.get("LANDING_DIR", "data/landing")
    return f"'{landing_dir}/prices/coffee_kc/**/*.csv.gzip'"


@macro()
def ice_stocks_glob(evaluator) -> str:
    """Return a quoted glob path for all ICE warehouse stock CSV gzip files under LANDING_DIR."""
    landing_dir = evaluator.var("LANDING_DIR") or os.environ.get("LANDING_DIR", "data/landing")
    return f"'{landing_dir}/ice_stocks/**/*.csv.gzip'"