jwt.py

import os
import asyncio
from eth_account import Account
from eth_account.messages import encode_defunct
import httpx
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

# Create a wallet to sign the message (must be the orders' `maker`)
private_key = os.getenv("WALLET_PRIVATE_KEY")
account = Account.from_key(private_key)

# Load API key from environment variables
api_key = os.getenv("API_KEY")

async def main():
    async with httpx.AsyncClient() as client:
        # step-1 GET auth/message

        # Send the `GET auth/message` request
        message_response = await client.get(
            "https://api.predict.fun/v1/auth/message",
            headers={
                "x-api-key": api_key,
            }
        )

        # Await for the response
        message_data = message_response.json()
        print(f"Message Response Status: {message_response.status_code}")
        print(f"Message Response: {message_data}")

        # Retrieve the message to sign
        message = message_data["data"]["message"]

        # Sign the message
        message_hash = encode_defunct(text=message)
        signed_message = account.sign_message(message_hash)
        signature = signed_message.signature.hex()

        # Add 0x prefix if not present
        if not signature.startswith('0x'):
            signature = '0x' + signature

        print(f"\nSigner Address: {account.address}")
        print(f"Message to sign: {message}")
        print(f"Signature: {signature}")

        # Send the `POST auth` request
        jwt_response = await client.post(
            "https://api.predict.fun/v1/auth",
            headers={
                "Content-Type": "application/json",
                "x-api-key": api_key,
            },
            json={
                "signer": account.address,
                "message": message,
                "signature": signature,
            },
        )

        # Await for the response
        jwt_data = jwt_response.json()
        print(f"\nJWT Response Status: {jwt_response.status_code}")
        print(f"JWT Response: {jwt_data}")

        # Fetch the JWT token
        if jwt_response.status_code == 200 and "data" in jwt_data:
            jwt = jwt_data["data"]["token"]
            print(f"\nJWT Token: {jwt}")
            return jwt
        else:
            print(f"\nError: {jwt_data.get('message', 'Unknown error')}")
            return None


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