# /// script
# requires-python = ">=3.12"
# dependencies = ["requests<3"]
# ///

import json
import os
import requests

MANIFEST_URL = "https://null.formulatrix.dev/Contest/stitch-a-ton-answer/raw/branch/main/answer.json"
API_ENDPOINT_URL = os.getenv("CONTEST_HOST")+ os.getenv("CONTEST_ENDPOINT")
DOWNLOAD_DIR = os.getenv("CONTEST_OUTPUT")

def fetch_manifest(url):
    print(f"fetching manifest from {url}...")
    try:
        response = requests.get(url)
        response.raise_for_status()
        return response.json()
    except requests.exceptions.RequestException as e:
        print(f"  ERROR: Could not fetch manifest. Reason: {e}")
    except json.JSONDecodeError:
        print("  ERROR: Failed to parse JSON from manifest. The content may not be valid JSON.")
    return None

def download_image(url, payload, filepath):
    try:
        # Use a session for potential connection pooling
        with requests.Session() as s:
            with s.post(url, json=payload, stream=True) as response:
                response.raise_for_status()

                # Write the content to the file in chunks
                with open(filepath, 'wb') as f:
                    for chunk in response.iter_content(chunk_size=8192):
                        f.write(chunk)
                
                print(f"  SUCCESS: Saved '{os.path.basename(filepath)}'")

    except requests.exceptions.HTTPError as e:
        print(f"  ERROR: HTTP error for '{os.path.basename(filepath)}': {e.response.status_code} {e.response.reason}")
    except requests.exceptions.RequestException as e:
        print(f"  ERROR: Could not download '{os.path.basename(filepath)}'. Reason: {e}")

def main():
    os.makedirs(DOWNLOAD_DIR, exist_ok=True)

    manifest_data = fetch_manifest(MANIFEST_URL)
    if not manifest_data:
        return

    for test_case in manifest_data.get("tests", []):
        filename = test_case.get("name")
        request_payload = test_case.get("request")

        if not filename or not request_payload:
            print(f"Skipping invalid test case: {test_case}")
            continue

        print(f"Requesting '{filename}'...")
        output_filepath = os.path.join(DOWNLOAD_DIR, filename)
        download_image(API_ENDPOINT_URL, request_payload, output_filepath)

if __name__ == "__main__":
    main()