import 'dotenv/config';

interface AuditResponse {
  job_id: string;
  status: string;
  url: string;
  mode: string;
  engine_version: string;
  ruleset_version: string;
  created_at: string;
  sandbox: boolean;
}

async function createAudit(url: string, mode: 'basic' | 'standard' | 'deep' = 'basic'): Promise<AuditResponse> {
  const response = await fetch('https://vertaaux.ai/api/v1/audit', {
    method: 'POST',
    headers: {
      'X-API-Key': process.env.VERTAAUX_API_KEY!,
      'Content-Type': 'application/json',
    },
    body: JSON.stringify({
      url,
      mode,
    }),
  });

  if (!response.ok) {
    const error = await response.json();
    throw new Error(`Failed to create audit: ${error.message}`);
  }

  return response.json();
}

// Usage
const audit = await createAudit('https://example.com', 'standard');
console.log(`Audit created: ${audit.job_id}`);
// Output: Audit created: clu1a2b3c4d5e6f7g8h9i0j1

curl -X POST https://vertaaux.ai/api/v1/audit \
  -H "X-API-Key: $VERTAAUX_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "url": "https://example.com",
    "mode": "standard"
  }'

{
  "job_id": "clu1a2b3c4d5e6f7g8h9i0j1",
  "status": "queued",
  "url": "https://example.com",
  "mode": "standard",
  "engine_version": "latest",
  "ruleset_version": "2024.1",
  "created_at": "2024-01-15T10:30:00Z",
  "sandbox": false
}

import os
import requests
from typing import TypedDict, Literal

class AuditResponse(TypedDict):
    job_id: str
    status: str
    url: str
    mode: str
    engine_version: str
    ruleset_version: str
    created_at: str
    sandbox: bool

def create_audit(
    url: str,
    mode: Literal['basic', 'standard', 'deep'] = 'basic'
) -> AuditResponse:
    """Create a new UX audit job."""
    response = requests.post(
        'https://vertaaux.ai/api/v1/audit',
        headers={
            'X-API-Key': os.environ['VERTAAUX_API_KEY'],
            'Content-Type': 'application/json',
        },
        json={
            'url': url,
            'mode': mode,
        }
    )

    if not response.ok:
        error = response.json()
        raise Exception(f"Failed to create audit: {error.get('message')}")

    return response.json()

# Usage
audit = create_audit('https://example.com', 'standard')
print(f"Audit created: {audit['job_id']}")
# Output: Audit created: clu1a2b3c4d5e6f7g8h9i0j1