from softqcos_sdk import QCOSClient

client = QCOSClient()

# Fetch all available solvers
backends = client.backends.all_backends()

print(f"Total backends registered: {len(backends)}")
print("-" * 50)
for backend in backends:
    status = "Active" if backend.get("available", True) else "Offline"
    print(f"Backend ID:   {backend['id']}")
    print(f"Provider:     {backend['provider']}")
    print(f"Qubit Count:  {backend['qubits']}")
    print(f"Status:       {status}")
    print("-" * 50)

from softqcos_sdk import QCOSClient

client = QCOSClient()

# 1. Validate credentials with the provider gateway
print("Validating IBM Quantum API token...")
validation = client.backends.validate(
    provider="ibm_quantum",
    token="YOUR_IBM_QUANTUM_API_TOKEN"
)

if validation["valid"]:
    print("✅ IBM Token validated successfully!")
    
    # 2. Save the credential connection locally
    connection = client.backends.add(
        provider="ibm_quantum",
        name="IBM Research Cluster",
        credentials={"token": "YOUR_IBM_QUANTUM_API_TOKEN"}
    )
    print(f"Credential saved. ID: {connection.get('id', connection.get('backend_id'))}")
else:
    print(f"❌ Validation failed: {validation['error_details']}")

connection = client.backends.add(
    provider="aws_braket",
    name="AWS Rigetti Solver",
    credentials={
        "access_key": "AWS_ACCESS_KEY_ID",
        "secret_key": "AWS_SECRET_ACCESS_KEY",
        "region": "us-east-1"
    }
)

# Execute circuit on a physical IBM backend using local credentials
result = client.backends.execute(
    credential_id="cred_abc123", # returned by backends.add
    backend="ibm_brisbane",
    circuit="""
    OPENQASM 2.0;
    include "qelib1.inc";
    qreg q[2];
    creg c[2];
    h q[0];
    cx q[0], q[1];
    measure q -> c;
    """,
    shots=2000
)

print("QPU results:", result)

# Dynamically register a custom physical solver adapter
client.provider_hub.register(
    module_path="my_lab_drivers.quantum_adapter",
    class_name="LabQPUAdapter",
    init_kwargs={"device_ip": "10.0.4.150"},
    overwrite=True
)
print("Custom driver adapter registered in QCOS Solver Hub.")