Function subscribeToCodingPlan

apps/web/src/lib/coding-plans/index.ts:71–266

(
  userId: string,
  planId: string,
  idempotencyKey: string
)

function databaseConstraint(error: unknown): string | null {
  let current: unknown = error;
  for (let depth = 0; depth < 5 && current && typeof current === 'object'; depth++) {
    if ('constraint' in current && typeof current.constraint === 'string') {
      return current.constraint;
    }
    current = 'cause' in current ? current.cause : null;
  }
  return null;
}

// Credential validation calls the live provider API per key. Bound the fan-out so
// large inventory uploads finish within the request budget without overwhelming
// the upstream provider with one unbounded burst of requests.
const INVENTORY_VALIDATION_CONCURRENCY = 10;
const BYTEPLUS_INVENTORY_VALIDATION_CONCURRENCY = 2;

type CancellationReason =
  | 'user_canceled'
  | 'insufficient_credits'
  | 'account_deleted'
  | 'administrative_termination';

type SubscriptionOutcome = {
  subscriptionId: string;
  charged: boolean;
};

export type CodingPlanUsageAssignmentContext =
  | { providerId: 'minimax'; apiKey: string }
  | { providerId: 'byteplus-coding'; seatId: string };

export async function getAssignedCodingPlanUsageContext(input: {
  inventoryId: string;
  userId: string;
  planId: string;
  providerId: string;
}): Promise<CodingPlanUsageAssignmentContext | null> {
  const [assignment] = await db
    .select({
      planId: coding_plan_key_inventory.plan_id,
      providerId: coding_plan_key_inventory.provider_id,
      status: coding_plan_key_inventory.status,
      assignedToUserId: coding_plan_key_inventory.assigned_to_user_id,
      encryptedApiKey: coding_plan_key_inventory.encrypted_api_key,
      upstreamUsageId: coding_plan_key_inventory.upstream_usage_id,
    })
    .from(coding_plan_key_inventory)
    .where(eq(coding_plan_key_inventory.id, input.inventoryId))
    .limit(1);
  if (
    !assignment ||
    assignment.status !== 'assigned' ||
    assignment.assignedToUserId !== input.userId ||
    assignment.planId !== input.planId ||
    assignment.providerId !== input.providerId
  ) {
    return null;
  }