Skip to main content

Client SDK for the Airlock Gateway — submit artifacts, poll for decisions, manage pairing and presence.

Project description

airlock-gateway (Python)

An async Python client SDK for the Airlock Integrations Gateway API.

Installation

PyPI: airlock-gateway

pip install airlock-gateway

API reference

AirlockGatewayClient (Integrations Gateway)

HTTP Method
GET /echo echo
POST /v1/artifacts submit_artifact
GET /v1/exchanges/{requestId} get_exchange_status
GET /v1/exchanges/{requestId}/wait wait_for_decision
POST /v1/exchanges/{requestId}/withdraw withdraw_exchange
POST /v1/pairing/initiate initiate_pairing
GET /v1/pairing/{nonce}/status get_pairing_status
POST /v1/pairing/revoke revoke_pairing
POST /v1/pairing/claim claim_pairing
POST /v1/presence/heartbeat send_heartbeat
GET /v1/policy/dnd/effective get_effective_dnd_policies
GET /v1/consent/status check_consent

Helper: encrypt_and_submit_artifact — canonicalizes, encrypts, and submits via POST /v1/artifacts. Lower-level crypto helpers live in airlock_gateway.crypto_helpers.

AirlockAuthClient (IdP / OAuth)

Purpose Method
OIDC discovery discover
Device code login login
Auth code + PKCE (local callback) login_with_auth_code
Auth code + PKCE (manual redirect) get_authorization_url, exchange_code
Tokens refresh_token, get_access_token
Sign out logout

Quick Start

With Bearer Token

import asyncio
from airlock_gateway import (
    AirlockGatewayClient,
    ArtifactSubmitRequest,
    EncryptedPayload,
)


async def main():
    async with AirlockGatewayClient(
        "https://igw.airlocks.io", token="your-token"
    ) as client:
        # Submit an artifact for approval
        request_id = await client.submit_artifact(
            ArtifactSubmitRequest(
                enforcer_id="my-enforcer",
                artifact_hash="sha256-hash",
                ciphertext=EncryptedPayload(
                    alg="aes-256-gcm",
                    data="base64-encrypted-content",
                    nonce="nonce",
                    tag="tag",
                ),
                metadata={"routingToken": "rt-abc"},
            )
        )

        # Wait for a decision (long-poll)
        decision = await client.wait_for_decision(request_id, timeout_seconds=30)
        if decision and decision.body and decision.body.is_approved:
            print(f"Approved: {decision.body.reason}")


asyncio.run(main())

With Enforcer App Credentials

async with AirlockGatewayClient(
    "https://igw.airlocks.io",
    client_id="your-client-id",
    client_secret="your-client-secret",
) as client:
    echo = await client.echo()

With Personal Access Token (PAT)

PAT is the recommended authentication for user-scoped operations. It replaces the Bearer token and is sent via the X-PAT header:

# After obtaining a PAT from the mobile app (Settings → Access Tokens)
client.set_pat("airlock_pat_...")

# Clear PAT when no longer needed
client.set_pat(None)

Dual Auth (set_bearer_token)

After creating a client with credentials, set a user's Bearer token to enable user-scoped operations:

# After user login (Device Auth Grant or Auth Code + PKCE)
client.set_bearer_token(access_token)

Authentication by Enforcer App Kind

EnforcerAppKind OAuth2 Flow SDK Methods Reason
Agent Device Authorization Grant (RFC 8628) login(on_user_code) Headless/CLI — no embedded browser, user opens URL + enters code separately
Desktop Device Authorization Grant (RFC 8628) login(on_user_code) Desktop app — delegates to external browser for user code entry
VsCodeExtension Device Authorization Grant (RFC 8628) login(on_user_code) VS Code extension — no embedded browser, uses device code flow
Web Auth Code + PKCE (RFC 7636) login_with_auth_code(on_browser_url, port) or get_authorization_url(redirect_uri) + exchange_code(code, redirect_uri, verifier) Browser-capable — can handle redirects and local callback
Mobile Auth Code + PKCE (RFC 7636) get_authorization_url(redirect_uri) + exchange_code(code, redirect_uri, verifier) Uses system browser + deep-link callback (manages redirect externally)

Pairing

Standard Pairing (Enforcer-Initiated)

from airlock_gateway import PairingInitiateRequest

# 1. Initiate a pairing session
resp = await client.initiate_pairing(PairingInitiateRequest(
    enforcer_id="my-enforcer",
    workspace_name="my-project",
    x25519_public_key=my_public_key,
))

# 2. Display pairing code to user
print(f"Pairing code: {resp.pairing_code}")

# 3. Poll for approval from the mobile app
status = await client.get_pairing_status(resp.nonce)
# status.state == "Completed" → save status.routing_token

Pre-Generated Code Pairing (Approver-Initiated)

When the mobile app pre-generates a pairing code, the enforcer claims it:

from airlock_gateway import PairingClaimRequest

claim = await client.claim_pairing(PairingClaimRequest(
    code="ABCD-1234",
    enforcer_id="my-enforcer",
    workspace_name="my-project",
    x25519_public_key=my_public_key,
))
# claim.routing_token is ready to use

Consent Check

Enforcer apps must verify user consent before submitting artifacts:

from airlock_gateway import AirlockGatewayError

try:
    status = await client.check_consent()
    # status == "approved" — proceed normally
except AirlockGatewayError as e:
    if e.error_code == "app_consent_required":
        print("User hasn't granted consent")
    elif e.error_code == "app_consent_pending":
        print("Consent request sent, waiting for approval")

API Reference

Method Description
echo() Gateway discovery/health
set_pat(pat) Set Personal Access Token (X-PAT header)
set_bearer_token(token) Set Bearer token for user-scoped operations
check_consent() Check if user has consented to this enforcer app
submit_artifact(request) Submit artifact for approval
get_exchange_status(request_id) Get exchange status
wait_for_decision(request_id, timeout) Long-poll for decision
withdraw_exchange(request_id) Withdraw pending exchange
initiate_pairing(request) Start pairing session
claim_pairing(request) Claim a pre-generated pairing code
get_pairing_status(nonce) Poll pairing status
revoke_pairing(routing_token) Revoke a pairing
send_heartbeat(request) Presence heartbeat
get_effective_dnd_policies(enforcer_id, workspace_id, session_id=None) Fetch effective DND policies

Error Handling

All errors raise AirlockGatewayError with helper properties:

from airlock_gateway import AirlockGatewayError

try:
    await client.submit_artifact(request)
except AirlockGatewayError as e:
    if e.is_quota_exceeded:
        print("Quota exceeded")
    elif e.is_pairing_revoked:
        print("Pairing revoked")
    elif e.is_conflict:
        print("Idempotency conflict")
    else:
        print(f"Error {e.status_code}: {e}")

Encryption

The SDK includes crypto_helpers for X25519 ECDH key exchange and AES-256-GCM encryption/decryption using cryptography:

  • generate_x25519_keypair() — generates a raw 32-byte X25519 keypair (base64url encoded)
  • derive_shared_key(my_private, peer_public) — derives a shared AES-256 key via ECDH + HKDF-SHA256 (info: HARP-E2E-AES256GCM)
  • aes_gcm_encrypt(key, plaintext) / aes_gcm_decrypt(key, payload) — AES-256-GCM with detached nonce and tag

During pairing, the enforcer generates an X25519 keypair, sends the public key in the pairing request, and derives the shared encryption key from the approver's public key returned in the pairing response.

Test Enforcer CLI

A fully interactive TUI application that demonstrates the complete enforcer lifecycle — setup wizard, Device Auth Grant sign-in, PAT configuration, consent check, workspace pairing (both standard and pre-generated code), background presence heartbeat, artifact submission with decision polling, withdrawal, unpairing, and sign-out.

Prerequisites

  • Python 3.9+
  • A running Airlock platform (Gateway + Keycloak)

Run

# From the repo root
cd src/python

# Create and activate virtual environment (required once)
python -m venv .venv
.venv\Scripts\Activate.ps1   # Windows PowerShell
# source .venv/bin/activate  # macOS / Linux

# Install dependencies (required once)
pip install -r requirements.txt

# Run the test enforcer
python test_enforcer.py

On first run, the setup wizard will prompt for Gateway URL, Client ID, Client Secret, Enforcer ID, and Workspace Name. Configuration is saved to ~/.airlock/test-enforcer-python.json and restored on subsequent runs.

Requirements

  • Python 3.9+
  • httpx >= 0.25.0
  • pydantic >= 2.0.0

Development

pip install -e ".[dev]"
pytest

License

MIT

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

airlock_gateway-0.3.5.tar.gz (22.6 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

airlock_gateway-0.3.5-py3-none-any.whl (19.1 kB view details)

Uploaded Python 3

File details

Details for the file airlock_gateway-0.3.5.tar.gz.

File metadata

  • Download URL: airlock_gateway-0.3.5.tar.gz
  • Upload date:
  • Size: 22.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for airlock_gateway-0.3.5.tar.gz
Algorithm Hash digest
SHA256 fc488ab48c0cb94e632d6bca3ab723c23bff286c5a6832617dcd04111e80e41c
MD5 6c2550196415fe9f3dce6d98d20f9e06
BLAKE2b-256 0ec805ddae94f58033ddc71f80b77093df8b412ea628b0e39ab26b14add69e1f

See more details on using hashes here.

File details

Details for the file airlock_gateway-0.3.5-py3-none-any.whl.

File metadata

File hashes

Hashes for airlock_gateway-0.3.5-py3-none-any.whl
Algorithm Hash digest
SHA256 feba2df6b0cddaf04ad5f6bd20a0aa71bc60c981e927e7ca8d42218758fd5db0
MD5 a1b72701e72e4ce438ae6619acb5b54a
BLAKE2b-256 823da8361f20b283c814780c1bee9861f7f5c01e3e437ca2ecfe9a28ba31dace

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page