Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

apps/shared/src/charge-settlement.ts

@@ -0,0 +1,89 @@
+import { refusalPolicy } from './billing-policy.js'
+import type { BillingChargeStatusResponse } from './billing-types.js'
+
+export interface SettlementDeps {
+  fetchStatus(): Promise<BillingChargeStatusResponse>
+  sleep(ms: number): Promise<void>
+  isCancelled(): boolean
+  now(): number
+}
+
+export type SettlementOutcome =
+  | { kind: 'settled'; status: BillingChargeStatusResponse }
+  | { kind: 'failed'; status: BillingChargeStatusResponse }
+  | { kind: 'refused'; error: string; status: BillingChargeStatusResponse }
+  | {
+      kind: 'ambiguous'
+      error: string
+      status?: BillingChargeStatusResponse
+      cause?: unknown
+    }
+  | { kind: 'timed_out' }
+  | { kind: 'cancelled' }
+
+export const SETTLEMENT_POLL_INTERVAL_MS = 2000
+export const SETTLEMENT_POLL_CAP_MS = 5 * 60 * 1000
+export const SETTLEMENT_MAX_RETRY_AFTER_MS = 30000
+
+export async function driveChargeSettlement(deps: SettlementDeps): Promise<SettlementOutcome> {
+  const start = deps.now()
+  const timedOut = (): boolean => deps.now() - start >= SETTLEMENT_POLL_CAP_MS
+
+  while (true) {
+    if (deps.isCancelled()) {
+      return { kind: 'cancelled' }
+    }
+
+    let status: BillingChargeStatusResponse
+
+    try {
+      status = await deps.fetchStatus()
+    } catch (cause) {
+      return { kind: 'ambiguous', error: 'transport', cause }
+    }
+
+    if (!status.ok) {
+      const error = status.error ?? ''
+      const policy = refusalPolicy(error)
+
+      if (policy.ambiguousMidPoll) {
+        return { kind: 'ambiguous', error: error || 'unknown', status }
+      }
+
+      if (policy.recovery === 'retry') {
+        if (timedOut()) {
+          return { kind: 'timed_out' }
+        }
+
+        const wait = Math.min(
+          (status.retry_after ?? 5) * 1000,
+          SETTLEMENT_MAX_RETRY_AFTER_MS
+        )
+
+        await deps.sleep(wait)
+
+        continue
+      }
+
+      return {
+        kind: 'refused',
+        error: status.error ?? status.message ?? 'error',
+        status
+      }
+    }
+
+    if (status.status === 'settled') {
+      return { kind: 'settled', status }
+    }
+
+    if (status.status === 'failed') {
+      return { kind: 'failed', status }
+    }
+
+    if (timedOut()) {
+      return { kind: 'timed_out' }
+    }
+
+    await deps.sleep(SETTLEMENT_POLL_INTERVAL_MS)
+  }
+}