Function run

packages/db/scripts/migrate-block-api-keys-to-byok.ts:601–746

()

// ---------- Main ----------
async function run() {
  console.log(`Mode: ${DRY_RUN ? 'DRY RUN (audit + preview)' : 'LIVE'}`)
  console.log(
    `Mappings: ${Object.entries(BLOCK_TYPE_TO_PROVIDER)
      .map(([b, p]) => `${b}=${p}`)
      .join(', ')}`
  )
  console.log(`Users: ${USER_FILTER.length > 0 ? USER_FILTER.join(', ') : 'all'}`)
  if (FROM_FILE) console.log(`From file: ${FROM_FILE}`)
  console.log('---\n')

  const stats = createEmptyStats()
  let workspaceIdsWritten = 0

  try {
    // 1. Get distinct workspace IDs that have matching blocks
    const mappedBlockTypes = Object.keys(BLOCK_TYPE_TO_PROVIDER)
    const agentTypes = Object.keys(TOOL_INPUT_SUBBLOCK_IDS)
    const allBlockTypes = [...new Set([...mappedBlockTypes, ...agentTypes])]

    const userFilter =
      USER_FILTER.length > 0
        ? sql` AND ${workflow.userId} IN (${sql.join(
            USER_FILTER.map((id) => sql`${id}`),
            sql`, `
          )})`
        : sql``

    let workspaceIds: string[]

    if (DRY_RUN) {
      const workspaceIdRows = await db
        .selectDistinct({ workspaceId: workflow.workspaceId })
        .from(workflowBlocks)
        .innerJoin(workflow, eq(workflowBlocks.workflowId, workflow.id))
        .where(
          sql`${workflow.workspaceId} IS NOT NULL AND ${workflowBlocks.type} IN (${sql.join(
            allBlockTypes.map((t) => sql`${t}`),
            sql`, `
          )})${userFilter}`
        )

      workspaceIds = workspaceIdRows
        .map((r) => r.workspaceId)
        .filter((id): id is string => id !== null)

      console.log(`Found ${workspaceIds.length} workspaces with candidate blocks\n`)

      const outPath = resolve('migrate-byok-workspace-ids.txt')
      writeFileSync(outPath, '')
      console.log(`[DRY RUN] Will write workspace IDs with keys to ${outPath}\n`)
    } else {
      const raw = readFileSync(resolve(FROM_FILE!), 'utf-8')
      workspaceIds = raw
        .split('\n')
        .map((l) => l.trim())
        .filter(Boolean)
      console.log(`Loaded ${workspaceIds.length} workspace IDs from ${FROM_FILE}\n`)