#!/usr/bin/env -S uv run
# /// script
# requires-python = ">=3.12"
# dependencies = [
#     "aiohttp",
#     "aiohttp-socks",
# ]
# ///
from aiohttp_socks import ProxyType, ProxyConnector, ChainProxyConnector
import aiohttp

import asyncio
import os

async def main():
    if socks_proxy := os.environ.get("SOCKS_PROXY"):
        # usage with `ssh ec2-server… -D 9999` to enable a socks5 proxy
        connector = ProxyConnector.from_url(socks_proxy)
    else:
        connector = None
    async with aiohttp.ClientSession(connector=connector) as session:
        async with session.get("https://httpbin.rs/ip") as resp:
            print(f"Status: {resp.status}")
            print(await resp.json())

if __name__ == "__main__":
    asyncio.run(main())