import json
from pathlib import Path
from samsara.core.validation import validate_hostname

DEMO_HOST = "demo-web-01.example.test"
LINUX_DEMO_HOST = "demo-rhel-01.example.test"
DEMO_FILES = {DEMO_HOST: "demo_inventory.json", LINUX_DEMO_HOST: "demo_linux_inventory.json"}

def collect_demo(hostname):
    hostname = validate_hostname(hostname)
    if hostname not in DEMO_FILES:
        raise ValueError("Demo mode supports " + " or ".join(DEMO_FILES) + ". No real host was contacted.")
    return json.loads((Path(__file__).resolve().parents[1] / "assets" / DEMO_FILES[hostname]).read_text())