@tangle-network/agent-gateway 0.7.0 → 0.8.0

package/src/a2a/task-submission-recovery.ts

@@ -0,0 +1,178 @@
+import type { Task } from './types'
+import {
+  compareAndSetTask,
+  cryptoRandomId,
+  type TaskStateStore,
+  withStatus,
+} from './task-state'
+
+const TASK_ORIGIN_METADATA_KEY = 'gatewayOrigin'
+const TASK_SUBMISSION_METADATA_KEY = 'gatewaySubmission'
+const TASK_SUBMISSION_RECOVERY_METADATA_KEY = 'gatewaySubmissionRecovery'
+const TASK_SUBMISSION_LEASE_MS = 5 * 60 * 1000
+
+export interface TaskOriginAgent {
+  id: string
+  slug: string
+}
+
+export interface TaskSubmissionIdentity {
+  agent: TaskOriginAgent
+  requestId: string
+  consumerId: string
+}
+
+interface TaskOriginBinding {
+  version: 1
+  agentId: string
+  agentSlug: string
+}
+
+export interface TaskSubmissionRecord {
+  version: 1
+  lease: { id: string; expiresAt: number }
+  agentId: string
+  agentSlug: string
+  requestId: string
+  consumerId: string
+}
+
+export interface SubmissionRecoveryDependencies {
+  taskStore: TaskStateStore
+  deliverPush: (task: Task) => Promise<void>
+}
+
+export function withTaskOrigin(
+  metadata: Record<string, unknown> | undefined,
+  agent: TaskOriginAgent,
+): Record<string, unknown> {
+  return {
+    ...(metadata ?? {}),
+    [TASK_ORIGIN_METADATA_KEY]: {
+      version: 1,
+      agentId: agent.id,
+      agentSlug: agent.slug,
+    } satisfies TaskOriginBinding,
+  }
+}
+
+export function withTaskSubmission(
+  metadata: Record<string, unknown> | undefined,
+  identity: TaskSubmissionIdentity,
+): Record<string, unknown> {
+  const origin = metadata?.[TASK_ORIGIN_METADATA_KEY]
+  return {
+    ...(metadata ?? {}),
+    ...(origin === undefined
+      ? {
+          [TASK_ORIGIN_METADATA_KEY]: {
+            version: 1,
+            agentId: identity.agent.id,
+            agentSlug: identity.agent.slug,
+          } satisfies TaskOriginBinding,
+        }
+      : {}),
+    [TASK_SUBMISSION_METADATA_KEY]: {
+      version: 1,
+      lease: { id: cryptoRandomId(), expiresAt: Date.now() + TASK_SUBMISSION_LEASE_MS },
+      agentId: identity.agent.id,
+      agentSlug: identity.agent.slug,
+      requestId: identity.requestId,
+      consumerId: identity.consumerId,
+    } satisfies TaskSubmissionRecord,
+  }
+}
+
+export function readTaskOrigin(task: Task): TaskOriginBinding | undefined {
+  const raw = task.metadata?.[TASK_ORIGIN_METADATA_KEY]
+  if (!raw || typeof raw !== 'object') return undefined
+  const origin = raw as Partial<TaskOriginBinding>
+  if (
+    origin.version !== 1 ||
+    typeof origin.agentId !== 'string' ||
+    origin.agentId.length === 0 ||
+    typeof origin.agentSlug !== 'string' ||
+    origin.agentSlug.length === 0
+  ) {
+    return undefined
+  }
+  return origin as TaskOriginBinding
+}
+
+export function readTaskSubmission(task: Task): TaskSubmissionRecord | undefined {
+  const raw = task.metadata?.[TASK_SUBMISSION_METADATA_KEY]
+  if (!raw || typeof raw !== 'object') return undefined
+  const submission = raw as Partial<TaskSubmissionRecord>
+  if (
+    submission.version !== 1 ||
+    !submission.lease ||
+    typeof submission.lease.id !== 'string' ||
+    submission.lease.id.length === 0 ||
+    typeof submission.lease.expiresAt !== 'number' ||
+    !Number.isFinite(submission.lease.expiresAt) ||
+    typeof submission.agentId !== 'string' ||
+    submission.agentId.length === 0 ||
+    typeof submission.agentSlug !== 'string' ||
+    submission.agentSlug.length === 0 ||
+    typeof submission.requestId !== 'string' ||
+    submission.requestId.length === 0 ||
+    typeof submission.consumerId !== 'string'
+  ) {
+    return undefined
+  }
+  return submission as TaskSubmissionRecord
+}
+
+export function clearTaskSubmission(task: Task): Task {
+  if (!task.metadata || !(TASK_SUBMISSION_METADATA_KEY in task.metadata)) return task
+  const metadata = { ...task.metadata }
+  delete metadata[TASK_SUBMISSION_METADATA_KEY]
+  if (Object.keys(metadata).length > 0) return { ...task, metadata }
+  const { metadata: _metadata, ...withoutMetadata } = task
+  return withoutMetadata
+}
+
+export async function recoverSubmissionIfNeeded(
+  task: Task,
+  deps: SubmissionRecoveryDependencies,
+): Promise<Task> {
+  const raw = task.metadata?.[TASK_SUBMISSION_METADATA_KEY]
+  if (raw === undefined) return task
+  const submission = readTaskSubmission(task)
+  if (submission && submission.lease.expiresAt > Date.now()) return task
+  if (task.status.state !== 'submitted') {
+    return (await clearTaskSubmissionMarker(deps.taskStore, task)).task
+  }
+
+  const cleanTask = clearTaskSubmission(task)
+  const failed: Task = {
+    ...withStatus(cleanTask, 'failed'),
+    metadata: {
+      ...(cleanTask.metadata ?? {}),
+      [TASK_SUBMISSION_RECOVERY_METADATA_KEY]: {
+        error: submission
+          ? 'A2A task submission lease expired before payment authorization completed'
+          : 'A2A task submission lease is invalid',
+      },
+    },
+  }
+  if (await compareAndSetTask(deps.taskStore, task, failed)) {
+    await deps.deliverPush(failed)
+    return failed
+  }
+  return await deps.taskStore.get(task.id) ?? task
+}
+
+async function clearTaskSubmissionMarker(
+  taskStore: TaskStateStore,
+  expected: Task,
+): Promise<{ task: Task; applied: boolean }> {
+  const current = await taskStore.get(expected.id)
+  if (!current || JSON.stringify(current) !== JSON.stringify(expected)) {
+    return { task: current ?? expected, applied: false }
+  }
+  const cleared = clearTaskSubmission(current)
+  if (cleared === current) return { task: current, applied: true }
+  if (await compareAndSetTask(taskStore, current, cleared)) return { task: cleared, applied: true }
+  return { task: await taskStore.get(expected.id) ?? expected, applied: false }
+}

package/src/a2a/types.ts

@@ -51,6 +51,7 @@ export const A2A_ERROR_CODES = {
   CONTENT_TYPE_NOT_SUPPORTED: -32005,
   INVALID_AGENT_RESPONSE: -32006,
   AUTHENTICATED_EXTENDED_CARD_NOT_CONFIGURED: -32007,
+  TASK_ACCESS_DENIED: -32008,
 } as const
 
 // ── Message parts ────────────────────────────────────────────────────────

package/src/dispatch-authorization.ts

@@ -0,0 +1,437 @@
+import type { Context } from 'hono'
+
+import { filterConsumerMessagesStrict } from './filter'
+import { type RequestContext, generateRequestId } from './observer'
+import { type GatewayState, type AuthorizedRequest } from './dispatch-types'
+import { checkRateLimit } from './rate-limit'
+import {
+  maximumBillableInputTokens,
+  requiredX402Amount,
+} from './dispatch-pricing'
+import type {
+  ApiKeyInfo,
+  ChatMessage,
+  GatewayConfig,
+  PaymentMethod,
+  SandboxExecutionBudget,
+} from './types'
+import {
+  defaultVerifyApiKey,
+  isApiKeyAuthEnabled,
+  isMppAuthEnabled,
+  mppPaymentPayload,
+  mppPaymentCredential,
+  verifyMppCredential,
+  verifyX402,
+} from './verify'
+
+/**
+ * Resolve the agent, then run the full pre-dispatch pipeline: payment +
+ * rate-limit + injection filter + user-message extraction + optional
+ * `authorizeConsumer` hook. Returns the success record on the happy path
+ * or a fully-formed `Response` (402/404/429/400/403) on any short-circuit.
+ *
+ * Body parsing is the caller's responsibility — different wire formats
+ * (OpenAI chat completions vs A2A JSON-RPC) have different envelopes; both
+ * still ultimately produce a `ChatMessage[]`.
+ */
+export async function authenticateAndGuard(
+  c: Context,
+  slug: string,
+  messages: ChatMessage[],
+  config: GatewayConfig,
+  state: GatewayState,
+  requestedMaxOutputTokens?: number,
+): Promise<AuthorizedRequest | Response> {
+  const startMs = Date.now()
+  const requestId = generateRequestId()
+  const ctx: RequestContext = { requestId, agentSlug: slug, startMs }
+  await state.obs?.onRequestStart?.(ctx)
+
+  const agent = await config.resolveAgent(slug)
+  if (!agent || !agent.enabled) {
+    return c.json({ error: { message: 'Agent not found', type: 'not_found' } }, 404)
+  }
+  if (!messages?.length) {
+    return c.json(
+      { error: { message: 'messages array required', type: 'invalid_request' } },
+      400,
+    )
+  }
+
+  const maxOutputTokens = requestedMaxOutputTokens ?? state.defaultOutputTokens
+  if (
+    !Number.isInteger(maxOutputTokens) ||
+    maxOutputTokens <= 0 ||
+    maxOutputTokens > state.maxOutputTokens
+  ) {
+    return c.json(
+      {
+        error: {
+          message: `max_tokens must be an integer between 1 and ${state.maxOutputTokens}`,
+          type: 'invalid_request',
+          code: 'invalid_max_tokens',
+        },
+      },
+      400,
+    )
+  }
+
+  // Quote the maximum UTF-8 input plus every hidden provider cost before
+  // verification. The verifier must remain read-only at this point.
+  const { messages: filtered, injectionWarnings } = filterConsumerMessagesStrict(
+    messages,
+    state.maxLen,
+  )
+  const userMessage = filtered
+    .filter((m) => m.role === 'user')
+    .map((m) => m.content)
+    .join('\n\n')
+  if (!userMessage) {
+    return c.json(
+      { error: { message: 'No user message provided', type: 'invalid_request' } },
+      400,
+    )
+  }
+
+  let requiredPaymentAmount: bigint
+  const messageInputBound = maximumBillableInputTokens(agent, filtered)
+  let maxInputTokens = messageInputBound
+  if (config.inputTokenBound) {
+    let configuredBound: number
+    try {
+      configuredBound = config.inputTokenBound({ agent, messages: filtered })
+    } catch {
+      return c.json(
+        {
+          error: {
+            message: 'Agent input token bound is unavailable',
+            type: 'server_error',
+            code: 'input_token_bound_unavailable',
+          },
+        },
+        503,
+      )
+    }
+    if (!Number.isSafeInteger(configuredBound) || configuredBound < messageInputBound) {
+      return c.json(
+        {
+          error: {
+            message: 'Agent input token bound is invalid',
+            type: 'server_error',
+            code: 'invalid_input_token_bound',
+          },
+        },
+        503,
+      )
+    }
+    maxInputTokens = configuredBound
+  }
+  const maxReasoningTokens = state.maxReasoningTokens
+  const maxToolTokens = state.maxToolTokens
+  const maxToolCalls = state.maxToolCalls
+  const maxProviderCostUsd = state.maxProviderCostUsd ??
+    (maxInputTokens + maxOutputTokens + maxReasoningTokens + maxToolTokens) * agent.pricePerTokenUsd
+  const executionBudget: SandboxExecutionBudget = {
+    maxInputTokens,
+    maxOutputTokens,
+    maxReasoningTokens,
+    maxToolTokens,
+    maxToolCalls,
+    maxProviderCostUsd,
+  }
+  try {
+    requiredPaymentAmount = requiredX402Amount(
+      agent.pricePerTokenUsd,
+      maxInputTokens,
+      maxOutputTokens,
+      config.x402.currencyDecimals,
+      maxReasoningTokens,
+      maxToolTokens,
+      maxProviderCostUsd,
+    )
+  } catch {
+    return c.json(
+      {
+        error: {
+          message: 'Agent payment configuration is invalid',
+          type: 'server_error',
+          code: 'invalid_payment_configuration',
+        },
+      },
+      503,
+    )
+  }
+
+  // Payment / auth.
+  const spendAuthHeader = c.req.header('X-Payment-Signature')
+  const authHeader = c.req.header('Authorization') ?? ''
+  let consumerId: string | null = null
+  let paymentMethod: PaymentMethod = 'none'
+  let keyInfo: ApiKeyInfo | null = null
+  let x402Payload: Record<string, unknown> | null = null
+  let paymentNonceKey: string | undefined
+  let mppMethod: string | undefined
+  let mppCredential: string | undefined
+  let mppPaymentIdentity: string | undefined
+
+  if (spendAuthHeader) {
+    const signer = await verifyX402(
+      spendAuthHeader,
+      config.x402,
+      state.nonceStore,
+      requiredPaymentAmount,
+      false,
+    )
+    if (!signer) {
+      await state.obs?.onAuthFailure?.(ctx, {
+        method: 'x402',
+        code: 'invalid_spend_auth',
+        httpStatus: 402,
+      })
+      return c.json(
+        {
+          error: {
+            message: 'Invalid X-Payment-Signature',
+            type: 'authentication_error',
+            code: 'invalid_spend_auth',
+            required_amount: requiredPaymentAmount.toString(),
+            currency_decimals: config.x402.currencyDecimals ?? 6,
+          },
+        },
+        {
+          status: 402,
+          headers: { 'X-Payment-Required': 'spendauth', 'X-Request-Id': requestId },
+        },
+      )
+    }
+    x402Payload = JSON.parse(spendAuthHeader) as Record<string, unknown>
+    paymentNonceKey = `${String(x402Payload.commitment).toLowerCase()}:${BigInt(String(x402Payload.nonce)).toString()}`
+    consumerId = signer
+    paymentMethod = 'x402'
+  } else if (isMppAuthEnabled(config) && authHeader.toLowerCase().startsWith('payment ')) {
+    const authenticated = await verifyMppCredential(
+      authHeader,
+      config.mpp!,
+      config.x402,
+      state.nonceStore,
+      requiredPaymentAmount,
+      false,
+    )
+    if (!authenticated) {
+      const realm = config.mpp!.realm
+      const method = config.mpp!.method ?? 'blueprintevm'
+      await state.obs?.onAuthFailure?.(ctx, {
+        method: 'mpp',
+        code: 'invalid_mpp_credential',
+        httpStatus: 401,
+      })
+      return c.json(
+        {
+          error: {
+            message: 'Invalid Payment credential',
+            type: 'authentication_error',
+            code: 'invalid_mpp_credential',
+          },
+        },
+        {
+          status: 401,
+          headers: {
+            'WWW-Authenticate': `Payment realm="${realm}", method="${method}"`,
+            'X-Request-Id': requestId,
+          },
+        },
+      )
+    }
+    consumerId = authenticated.consumerId
+    paymentMethod = 'mpp'
+    mppMethod = authHeader.match(/^Payment\s+(\S+)\s+/i)?.[1]?.toLowerCase()
+    mppCredential = mppPaymentCredential(authHeader)
+    mppPaymentIdentity = authenticated.paymentIdentity
+    x402Payload = mppPaymentPayload(authHeader) ?? null
+    paymentNonceKey = authenticated.replayKey
+  } else if (authHeader.startsWith('Bearer ')) {
+    const verify = config.verifyApiKey ?? (config.x402.demoMode ? defaultVerifyApiKey : null)
+    if (!verify || !isApiKeyAuthEnabled(config)) {
+      await state.obs?.onAuthFailure?.(ctx, {
+        method: 'apikey',
+        code: 'api_keys_not_configured',
+        httpStatus: 401,
+      })
+      return c.json(
+        { error: { message: 'API key authentication is not configured', type: 'authentication_error' } },
+        { status: 401, headers: { 'X-Request-Id': requestId } },
+      )
+    }
+    const key = await verify(authHeader)
+    if (!key) {
+      await state.obs?.onAuthFailure?.(ctx, {
+        method: 'apikey',
+        code: 'invalid_api_key',
+        httpStatus: 401,
+      })
+      return c.json(
+        { error: { message: 'Invalid API key', type: 'authentication_error' } },
+        { status: 401, headers: { 'X-Request-Id': requestId } },
+      )
+    }
+    if (key.scopes && key.scopes.length > 0 && !key.scopes.includes(state.requiredScope)) {
+      await state.obs?.onAuthFailure?.(ctx, {
+        method: 'apikey',
+        code: 'insufficient_scope',
+        httpStatus: 403,
+      })
+      return c.json(
+        {
+          error: {
+            message: `API key missing required scope: ${state.requiredScope}`,
+            type: 'forbidden',
+            code: 'insufficient_scope',
+          },
+        },
+        { status: 403, headers: { 'X-Request-Id': requestId } },
+      )
+    }
+    consumerId = key.consumerId
+    paymentMethod = 'apikey'
+    keyInfo = key
+  } else {
+    await state.obs?.onAuthFailure?.(ctx, {
+      method: 'none',
+      code: 'payment_required',
+      httpStatus: 402,
+    })
+    const methods: string[] = ['x402']
+    if (isMppAuthEnabled(config)) methods.push('mpp')
+    if (isApiKeyAuthEnabled(config)) methods.push('api_key')
+    const headers: Record<string, string> = {
+      'X-Payment-Required': methods.join(', '),
+      'X-Request-Id': requestId,
+    }
+    if (isMppAuthEnabled(config) && config.mpp) {
+      headers['WWW-Authenticate'] =
+        `Payment realm="${config.mpp.realm}", method="${config.mpp.method ?? 'blueprintevm'}"`
+    }
+    return c.json(
+      {
+        error: {
+          message: 'Payment required',
+          type: 'payment_required',
+          payment_methods: methods,
+          x402: {
+            operator: config.x402.operatorAddress,
+            chain_id: config.x402.chainId,
+            credits_address: config.x402.creditsAddress,
+            required_amount: requiredPaymentAmount.toString(),
+            currency_decimals: config.x402.currencyDecimals ?? 6,
+            max_output_tokens: maxOutputTokens,
+          },
+          ...(isMppAuthEnabled(config) && config.mpp
+            ? { mpp: { realm: config.mpp.realm, method: config.mpp.method ?? 'blueprintevm' } }
+            : {}),
+          ...(isApiKeyAuthEnabled(config)
+            ? {
+                api_key: {
+                  purchase_url: config.baseUrl
+                    ? `${config.baseUrl}/agents/${slug}/api-keys`
+                    : undefined,
+                },
+              }
+            : {}),
+        },
+      },
+      { status: 402, headers },
+    )
+  }
+
+  // Rate limit.
+  const effectiveRateLimit = keyInfo?.rateLimitPerMinute
+    ? { limit: keyInfo.rateLimitPerMinute, windowSeconds: 60 }
+    : state.globalRateLimit
+  const rl = await checkRateLimit(consumerId, effectiveRateLimit, state.rateLimitStore)
+  if (!rl.allowed) {
+    await state.obs?.onRateLimited?.(ctx, {
+      consumerId: consumerId,
+      retryAfterSeconds: rl.retryAfterSeconds ?? 60,
+    })
+    return c.json(
+      {
+        error: {
+          message: 'Rate limit exceeded',
+          type: 'rate_limit_error',
+          retry_after: rl.retryAfterSeconds,
+        },
+      },
+      {
+        status: 429,
+        headers: {
+          'Retry-After': String(rl.retryAfterSeconds ?? 60),
+          'X-Request-Id': requestId,
+        },
+      },
+    )
+  }
+
+  // Reject or report injection only after authentication so observer events
+  // retain the authenticated consumer identity.
+  if (injectionWarnings.length > 0) {
+    await state.obs?.onInjectionDetected?.(ctx, {
+      consumerId: consumerId,
+      patterns: injectionWarnings,
+      blocked: !!config.blockInjection,
+    })
+    if (config.blockInjection) {
+      return c.json(
+        {
+          error: {
+            message: 'Request rejected: potential prompt injection detected',
+            type: 'content_policy_violation',
+          },
+        },
+        { status: 400, headers: { 'X-Request-Id': requestId } },
+      )
+    }
+  }
+
+  if (config.authorizeConsumer) {
+    const authz = await config.authorizeConsumer(agent, {
+      method: paymentMethod,
+      consumerId: consumerId,
+      keyId: keyInfo?.keyId,
+      requestId,
+    })
+    if (!authz.allow) {
+      return c.json(
+        {
+          error: {
+            message: authz.reason,
+            type: 'authorization_denied',
+            code: authz.code,
+          },
+        },
+        { status: 403, headers: { 'X-Request-Id': requestId } },
+      )
+    }
+  }
+
+  return {
+    agent,
+    consumerId,
+    paymentMethod,
+    keyInfo,
+    userMessage,
+    rateLimitRemaining: rl.remaining,
+    requestId,
+    startMs,
+    maxOutputTokens,
+    executionBudget,
+    requiredPaymentAmount,
+    paymentPayload: x402Payload,
+    paymentNonceKey,
+    mppMethod,
+    mppCredential,
+    mppPaymentIdentity,
+  }
+}
+
+export type { AuthorizedRequest, GatewayState }