import requests
import json

# Base URL
base_url = "http://localhost:8000"

# Step 1: Authenticate to get token
print("Step 1: Authenticating...")
auth_url = f"{base_url}/api/v1/auth/login"
auth_payload = {
    "username": "admin",
    "password": "admin123"  # Default admin credentials
}

try:
    auth_response = requests.post(auth_url, json=auth_payload)
    print(f"Auth Status Code: {auth_response.status_code}")

    if auth_response.status_code != 200:
        print(f"Auth failed: {auth_response.text}")
        exit(1)

    auth_data = auth_response.json()
    token = auth_data.get("access_token")
    print(f"Successfully authenticated! Token: {token[:20]}...")

    # Step 2: Create action mapping
    print("\nStep 2: Testing CREATE action mapping endpoint...")
    url = f"{base_url}/api/v1/configuration/action-mappings"

    payload = {
        "name": "Test Action Mapping",
        "output_actions": [
            {
                "action": "TestAction",
                "parameters": {
                    "param1": "value1",
                    "param2": "value2"
                }
            }
        ]
    }

    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {token}"
    }

    response = requests.post(url, json=payload, headers=headers)
    print(f"Create Status Code: {response.status_code}")
    print(f"Create Response: {json.dumps(response.json(), indent=2)}")

    if response.status_code == 201:
        print("\n✓ CREATE operation successful!")

        # Step 3: Read back the action mappings to verify
        print("\nStep 3: Reading action mappings to verify...")
        read_url = f"{base_url}/api/v1/configuration/action-mappings/export"
        read_response = requests.get(read_url, headers=headers)
        print(f"Read Status Code: {read_response.status_code}")

        if read_response.status_code == 200:
            mappings = read_response.json()
            print(f"Total mappings found: {mappings.get('total_count', 0)}")

            # Find our test mapping
            for mapping in mappings.get("mappings", []):
                if mapping.get("name") == "Test Action Mapping":
                    print(f"\n✓ Found our test mapping:")
                    print(json.dumps(mapping, indent=2))
                    break
    else:
        print(f"\n✗ CREATE operation failed: {response.text}")

except Exception as e:
    print(f"Error: {e}")
    import traceback
    traceback.print_exc()