import json
import sys
import logging
from app.services.upstream_client import UpstreamClient

logging.basicConfig(level=logging.DEBUG)

def main():
    # Usage: python test_upstream.py <base_url> [api_prefix] [auth_type] [token]
    # Example: python test_upstream.py http://localhost:8000 "" bearer "sk-xxx"
    import sys
    base_url = sys.argv[1] if len(sys.argv) > 1 else "http://localhost:8000"
    api_prefix = sys.argv[2] if len(sys.argv) > 2 else ""
    auth_type = sys.argv[3] if len(sys.argv) > 3 else "bearer"
    token = sys.argv[4] if len(sys.argv) > 4 else ""
    with UpstreamClient(
        base_url=base_url,
        api_prefix=api_prefix,
        auth_type=auth_type,
        auth_config={"token": token},
        timeout=10.0,
    ) as client:
        try:
            groups = client.get_available_groups("/api/group/")
            print("Groups:", groups)
        except Exception as e:
            print("Groups Error:", e)

        try:
            rates = client.get_group_rates("/api/option/?key=GroupRatio")
            print("Rates:", rates)
        except Exception as e:
            print("Rates Error:", e)

        try:
            from app.services.upstream_client import _extract_rates_map, _unwrap_list
            print("Unwrapped Groups:", _unwrap_list(groups))
            print("Extracted Rates:", _extract_rates_map(rates))
        except Exception as e:
            pass

if __name__ == "__main__":
    main()