fix(db): cover dev size cutover

Author: waleedlatif1

.github/workflows/migrations.yml

@@ -80,6 +80,7 @@ jobs:
               echo "ERROR: db:push needs an interactive rename decision; land it as a versioned migration instead of relying on push." >&2
               exit 1
             fi
+            bun run ./scripts/apply-dev-workspace-file-size-cutover.ts
           else
             echo "Applying versioned migrations (db:migrate)"
             bun run ./scripts/migrate.ts

packages/db/package.json

@@ -25,6 +25,7 @@
   "scripts": {
     "db:push": "bunx drizzle-kit push --config=./drizzle.config.ts",
     "db:migrate": "bun --env-file=.env run ./scripts/migrate.ts",
+    "db:apply-dev-workspace-file-size-cutover": "bun --env-file=.env run ./scripts/apply-dev-workspace-file-size-cutover.ts",
     "db:reconcile-fork-kb-file-ownership": "bun --env-file=.env run ./scripts/reconcile-fork-kb-file-ownership.ts",
     "db:reconcile-workspace-storage": "bun --env-file=.env run ./scripts/reconcile-workspace-storage.ts",
     "db:studio": "bunx drizzle-kit studio --config=./drizzle.config.ts",

packages/db/schema.ts

@@ -2105,7 +2105,7 @@ export const workspaceFiles = pgTable(
      */
     displayName: text('display_name'),
     contentType: text('content_type').notNull(),
-    // contract-pending(after #7112 is fully deployed): drop size and workspace_files_sync_size_columns — the compatibility trigger is then the only remaining consumer
+    /** contract-pending(after #7112 is fully deployed): drop size, workspace_files_sync_size_columns, and the temporary dev cutover runner */
     size: integer('size').notNull().default(0),
     /** Exact byte size. The deploy migration backfills existing rows before this release serves traffic. */
     sizeBytes: bigint('size_bytes', { mode: 'number' }),

packages/db/script-migrations/0008_backfill_workspace_file_size_bytes.ts

@@ -1,6 +1,9 @@
+import { createLogger } from '@sim/logger'
 import type { Sql } from 'postgres'
 import type { ScriptMigration } from './types'
 
+const logger = createLogger('WorkspaceFileSizeBytesBackfill')
+
 export const WORKSPACE_FILE_SIZE_BYTES_BATCH_SIZE = 1000
 
 export interface WorkspaceFileSizeBytesBackfillStore {
@@ -78,6 +81,6 @@ export const backfillWorkspaceFileSizeBytesMigration: ScriptMigration = {
     const backfilled = await backfillWorkspaceFileSizeBytes(
       createPostgresWorkspaceFileSizeBytesBackfillStore(sql)
     )
-    console.log(`Workspace file size_bytes backfill complete: ${backfilled} file(s) updated.`)
+    logger.info(`Workspace file size_bytes backfill complete: ${backfilled} file(s) updated.`)
   },
 }

packages/db/scripts/apply-dev-workspace-file-size-cutover.ts

@@ -0,0 +1,46 @@
+import { createLogger } from '@sim/logger'
+import postgres from 'postgres'
+import {
+  backfillWorkspaceFileSizeBytes,
+  createPostgresWorkspaceFileSizeBytesBackfillStore,
+} from '../script-migrations/0008_backfill_workspace_file_size_bytes'
+
+const logger = createLogger('DevWorkspaceFileSizeCutover')
+const MIGRATION_NAME = '0008_backfill_workspace_file_size_bytes'
+const MIGRATION_SQL_URL = new URL(
+  '../migrations/0308_workspace_file_size_cutover.sql',
+  import.meta.url
+)
+const url = process.env.MIGRATION_DATABASE_URL || process.env.DATABASE_URL
+
+if (!url) {
+  throw new Error('Missing MIGRATION_DATABASE_URL or DATABASE_URL')
+}
+
+const sql = postgres(url, {
+  max: 1,
+  connect_timeout: 10,
+  max_lifetime: null,
+  connection: { application_name: 'sim-dev-workspace-file-size-cutover' },
+})
+
+try {
+  const migrationSql = await Bun.file(MIGRATION_SQL_URL).text()
+  await sql.unsafe(migrationSql)
+  const backfilled = await backfillWorkspaceFileSizeBytes(
+    createPostgresWorkspaceFileSizeBytesBackfillStore(sql)
+  )
+  await sql`
+    CREATE TABLE IF NOT EXISTS script_migrations (
+      name text PRIMARY KEY,
+      applied_at timestamptz NOT NULL DEFAULT now()
+    )
+  `
+  await sql`
+    INSERT INTO script_migrations (name) VALUES (${MIGRATION_NAME})
+    ON CONFLICT (name) DO NOTHING
+  `
+  logger.info('Dev workspace file size cutover completed', { backfilled })
+} finally {
+  await sql.end()
+}