Function processWorkspace

packages/db/scripts/migrate-block-api-keys-to-byok.ts:370–598

(
  workspaceId: string,
  allBlockTypes: string[],
  userFilter: ReturnType<typeof sql>,
  total: number,
  index: number
)

}

async function processWorkspace(
  workspaceId: string,
  allBlockTypes: string[],
  userFilter: ReturnType<typeof sql>,
  total: number,
  index: number
): Promise<WorkspaceResult> {
  const stats = createEmptyStats()

  try {
    const [blocks, wsRows] = await Promise.all([
      db
        .select({
          blockId: workflowBlocks.id,
          blockName: workflowBlocks.name,
          blockType: workflowBlocks.type,
          subBlocks: workflowBlocks.subBlocks,
          workflowId: workflow.id,
          workflowName: workflow.name,
          userId: workflow.userId,
        })
        .from(workflowBlocks)
        .innerJoin(workflow, eq(workflowBlocks.workflowId, workflow.id))
        .where(
          sql`${workflow.workspaceId} = ${workspaceId} AND ${workflowBlocks.type} IN (${sql.join(
            allBlockTypes.map((t) => sql`${t}`),
            sql`, `
          )})${userFilter}`
        ),
      db
        .select({ ownerId: workspaceTable.ownerId })
        .from(workspaceTable)
        .where(eq(workspaceTable.id, workspaceId))
        .limit(1),
    ])

    const workspaceOwnerId = wsRows[0]?.ownerId ?? null

    console.log(
      `[${index}/${total}] [Workspace ${workspaceId}] ${blocks.length} blocks, owner=${workspaceOwnerId ?? 'unknown'}`
    )

    const providerKeys = new Map<string, RawKeyRef[]>()

    for (const block of blocks) {
      const subBlocks = block.subBlocks as Record<string, { value?: any }>

      const providerId = BLOCK_TYPE_TO_PROVIDER[block.blockType]
      if (providerId) {
        const val = subBlocks?.apiKey?.value
        if (typeof val === 'string' && val.trim()) {
          const refs = providerKeys.get(providerId) ?? []
          refs.push({
            rawValue: val,
            blockName: block.blockName,
            workflowId: block.workflowId,
            workflowName: block.workflowName,
            userId: block.userId,