# pip install pulumi
# Runs: python thread_demo.py

import os
import tempfile
from concurrent.futures import ThreadPoolExecutor, as_completed
from pulumi import automation as auto

PROJECT_NAME = "py-thread-demo"

def make_program(run_id: str):
    # Capture run_id in a closure (avoid env/global mutation per thread)
    def pulumi_program():
        import pulumi
        pulumi.export("greeting", f"hello from {run_id}")
    return pulumi_program

def run_stack(run_id: str, fail: bool = False):
    # Demo-only: local backend with no passphrase prompt
    os.environ.setdefault("PULUMI_CONFIG_PASSPHRASE", "")

    workdir = tempfile.mkdtemp(prefix=f"{run_id}-")  # unique dir per run
    stack = auto.create_or_select_stack_inline(
        stack_name=run_id,
        project_name=PROJECT_NAME,
        program=make_program(run_id),
        opts=auto.LocalWorkspaceOptions(work_dir=workdir),
    )

    if fail:
        # Prove isolation: this thread fails, siblings continue
        raise RuntimeError(f"simulated failure in {run_id}")

    result = stack.up(on_output=lambda _: None)  # quiet output
    return run_id, {k: v.value for k, v in result.outputs.items()}

if __name__ == "__main__":
    jobs = [
        ("stack-A", False),
        ("stack-B", True),   # this one will fail
        ("stack-C", False),
    ]

    with ThreadPoolExecutor(max_workers=len(jobs)) as pool:
        fut_to_name = {
            pool.submit(run_stack, name, should_fail): name
            for name, should_fail in jobs
        }
        for fut in as_completed(fut_to_name):
            name = fut_to_name[fut]
            try:
                run_id, outputs = fut.result()
                print(f"[OK] {run_id}: {outputs}")
            except Exception as e:
                print(f"[FAIL] {name}: {e}")

    print("Done.")