#!/usr/bin/python3
import base64
import requests
import sys

def main(hostname, admin_port, proto_port, data):
    print(f"Using hostname {hostname}, admin-port {admin_port}, and proto-port {proto_port}\n")

    # Step 1: Log in to /api/v1/auth and retrieve the JWT
    print("Step 1: Authenticate as admin")
    basic_auth = base64.b64encode(b"admin:admin-password").decode("utf-8")
    auth_url = f"http://{hostname}:{admin_port}/api/v1/auth"
    print(f"Equivalent curl command:\n"
          f"curl -X POST -H \"Authorization: Basic {basic_auth}\" -H \"Content-Type: application/json\" {auth_url}\n")

    try:
        auth_response = requests.post(
            auth_url,
            headers={"Authorization": f"Basic {basic_auth}", "Content-Type": "application/json"}
        )
        auth_response.raise_for_status()
        jwt = auth_response.json().get("token")

        if not jwt:
            print("Error: Failed to retrieve JWT")
            sys.exit(1)

        print(f"Retrieved JWT: {jwt}\n")
    except requests.exceptions.RequestException as e:
        print(f"Error during login: {e}")
        sys.exit(1)

    # Step 2: POST to /api/v1/policy with the JWT
    print("Step 2: Allow anonymous users to create topics")
    policy_url = f"http://{hostname}:{admin_port}/api/v1/policy"
    policy_data = {"subject": "anon", "object": "topic-any", "action": "create"}
    print(f"Equivalent curl command:\n"
          f"curl -X POST -H \"Authorization: Bearer {jwt}\" -H \"Content-Type: application/json\" "
          f"-d '{policy_data}' {policy_url}\n")

    try:
        policy_response = requests.post(
            policy_url,
            headers={
                "Authorization": f"Bearer {jwt}",
                "Content-Type": "application/json"
            },
            json=policy_data
        )
        print(f"Policy POST response status: {policy_response.status_code}")
        print(policy_response.text + "\n")
    except requests.exceptions.RequestException as e:
        print(f"Error during policy POST: {e}")
        sys.exit(1)

    # Step 3: Send POST request to /test-topic
    print("Step 3: Post a message as an anonymous user, creating the topic")
    test_topic_post_url = f"http://{hostname}:{proto_port}/test-topic"
    print(f"Equivalent curl command:\n"
          f"curl -X POST -H \"Content-Type: application/json\" -d '{data}' {test_topic_post_url}\n")

    try:
        post_response = requests.post(
            test_topic_post_url,
            headers={"Content-Type": "application/json"},
            json=data
        )
        print(f"POST response status: {post_response.status_code}")
        print(post_response.text + "\n")
    except requests.exceptions.RequestException as e:
        print(f"Error during POST to /test-topic: {e}")
        sys.exit(1)

    # Step 4: Send GET request to /test-topic
    print("Step 4: Consume the message from the topic")
    test_topic_get_url = f"http://{hostname}:{proto_port}/test-topic"
    print(f"Equivalent curl command:\n"
          f"curl -X GET {test_topic_get_url}\n")

    try:
        get_response = requests.get(test_topic_get_url)
        print(f"GET response status: {get_response.status_code}")
        print(get_response.text + "\n")
    except requests.exceptions.RequestException as e:
        print(f"Error during GET from /test-topic: {e}")
        sys.exit(1)

if __name__ == "__main__":
    import argparse
    parser = argparse.ArgumentParser(description="Run API test script.")
    parser.add_argument("--hostname", type=str, default="cloud.rapidrecast.io", help="Hostname for the API")
    parser.add_argument("--admin-port", type=int, default=80, help="Port for admin API (default: 80)")
    parser.add_argument("--proto-port", type=int, default=81, help="Port for the main API (default: 81)")
    parser.add_argument("--data", type=str, required=True, help="JSON data for POST request to /test-topic")

    args = parser.parse_args()
    main(args.hostname, args.admin_port, args.proto_port, args.data)