improvement(db): finish workspace file size cutover

Author: waleedlatif1

apps/sim/app/api/files/uploads/finalizers.ts

@@ -14,7 +14,7 @@ import {
   registerUploadedWorkspaceFile,
   type WorkspaceFileRecord,
 } from '@/lib/uploads/contexts/workspace'
-import { type StorageContext, toLegacyWorkspaceFileSize } from '@/lib/uploads/shared/types'
+import { getWorkspaceFileSize, type StorageContext } from '@/lib/uploads/shared/types'
 import { UploadSessionError, type UploadSessionRecord } from '@/lib/uploads/upload-session/service'
 import { toV2File } from '@/app/api/v2/files/utils'
 
@@ -364,7 +364,6 @@ async function insertOrLoadFileMetadata(
       originalName: input.originalName,
       displayName: input.originalName,
       contentType: input.contentType,
-      size: toLegacyWorkspaceFileSize(input.size),
       sizeBytes: input.size,
       deletedAt: null,
       uploadedAt: now,
@@ -396,7 +395,7 @@ async function findFileMetadataByKey(key: string): Promise<FileMetadataRecord |
 }
 
 function assertMatchingMetadata(existing: FileMetadataRecord, input: FinalizedMetadataInput): void {
-  const existingSize = existing.sizeBytes ?? existing.size
+  const existingSize = getWorkspaceFileSize(existing)
   if (
     existing.key !== input.key ||
     existing.userId !== input.userId ||

apps/sim/app/api/mothership/local-files/stage/route.test.ts

@@ -39,7 +39,7 @@ vi.mock('@sim/db/schema', () => ({
     displayName: 'workspaceFiles.displayName',
     originalName: 'workspaceFiles.originalName',
     contentType: 'workspaceFiles.contentType',
-    size: 'workspaceFiles.size',
+    sizeBytes: 'workspaceFiles.sizeBytes',
     deletedAt: 'workspaceFiles.deletedAt',
   },
 }))
@@ -95,7 +95,7 @@ describe('POST /api/mothership/local-files/stage', () => {
         displayName: null,
         originalName: 'report.pdf',
         contentType: 'application/pdf',
-        size: 42,
+        sizeBytes: 42,
       },
     ])
     mockWhere.mockReturnValue({ limit: mockLimit })
@@ -139,7 +139,7 @@ describe('POST /api/mothership/local-files/stage', () => {
         displayName: 'report (2).pdf',
         originalName: 'report.pdf',
         contentType: 'application/pdf',
-        size: 42,
+        sizeBytes: 42,
       },
     ])
     const response = await POST(request())

apps/sim/app/api/mothership/local-files/stage/route.ts

@@ -17,6 +17,7 @@ import {
   trackChatUpload,
   WorkspaceFileKeyOwnershipError,
 } from '@/lib/uploads/contexts/workspace/workspace-file-manager'
+import { getWorkspaceFileSize } from '@/lib/uploads/shared/types'
 import { getUserEntityPermissions } from '@/lib/workspaces/permissions/utils'
 
 const logger = createLogger('StageLocalFileUploadAPI')
@@ -52,7 +53,7 @@ export const POST = withRouteHandler(async (request: NextRequest) => {
         displayName: workspaceFiles.displayName,
         originalName: workspaceFiles.originalName,
         contentType: workspaceFiles.contentType,
-        size: workspaceFiles.size,
+        sizeBytes: workspaceFiles.sizeBytes,
       })
       .from(workspaceFiles)
       .where(
@@ -87,7 +88,7 @@ export const POST = withRouteHandler(async (request: NextRequest) => {
               key,
               file.originalName,
               file.contentType,
-              file.size
+              getWorkspaceFileSize(file)
             )
           ).displayName
 

apps/sim/background/cleanup-soft-deletes.ts

@@ -113,9 +113,7 @@ async function selectExpiredWorkspaceFiles(
           key: workspaceFiles.key,
           workspaceId: workspaceFiles.workspaceId,
           context: workspaceFiles.context,
-          size: sql<number>`coalesce(${workspaceFiles.sizeBytes}, ${workspaceFiles.size})`.mapWith(
-            Number
-          ),
+          size: sql<number>`${workspaceFiles.sizeBytes}`.mapWith(Number),
         })
         .from(workspaceFiles)
         .where(
@@ -329,9 +327,7 @@ async function deleteExpiredBillableWorkspaceFileRows(
             )
             .returning({
               id: workspaceFiles.id,
-              size: sql<number>`coalesce(${workspaceFiles.sizeBytes}, ${workspaceFiles.size})`.mapWith(
-                Number
-              ),
+              size: sql<number>`${workspaceFiles.sizeBytes}`.mapWith(Number),
             })
           if (deletedRows.some(({ size }) => size < 0)) {
             throw new Error('Cannot delete workspace files with negative stored-byte metadata')

apps/sim/ee/workspace-forking/lib/copy/copy-files.test.ts

@@ -549,6 +549,7 @@ describe('planForkFileCopies', () => {
       displayName: null,
       contentType: 'text/plain',
       size: 4321,
+      sizeBytes: 4321,
       deletedAt: null,
       uploadedAt: new Date('2026-01-01'),
       updatedAt: new Date('2026-01-01'),
@@ -593,6 +594,7 @@ describe('planForkFileCopies', () => {
       displayName: null,
       contentType: 'text/plain',
       size: 4321,
+      sizeBytes: 4321,
       deletedAt: null,
       uploadedAt: new Date('2026-01-01'),
       updatedAt: new Date('2026-01-01'),

apps/sim/ee/workspace-forking/lib/copy/copy-files.ts

@@ -20,7 +20,7 @@ import {
   headObject,
   uploadFile,
 } from '@/lib/uploads/core/storage-service'
-import type { StorageContext } from '@/lib/uploads/shared/types'
+import { getWorkspaceFileSize, type StorageContext } from '@/lib/uploads/shared/types'
 import { MAX_FILE_SIZE } from '@/lib/uploads/utils/validation'
 import { resolveForkFolderMapping } from '@/ee/workspace-forking/lib/copy/copy-workflows'
 import {
@@ -219,7 +219,7 @@ export async function planForkFileCopies(params: {
       context: meta.context as StorageContext,
       fileName: meta.originalName,
       contentType: meta.contentType,
-      size: meta.size,
+      size: getWorkspaceFileSize(meta),
       targetFileId: childFileId,
       displayName: meta.displayName,
       userId,
@@ -341,7 +341,7 @@ export async function executeForkFileBlobCopies(
               originalName: targetOriginalName,
               displayName: targetDisplayName,
               contentType: task.contentType,
-              size: task.size,
+              sizeBytes: task.size,
               deletedAt: null,
               uploadedAt: new Date(),
             })
@@ -389,7 +389,7 @@ export async function executeForkFileBlobCopies(
                 originalName: targetOriginalName,
                 displayName: targetDisplayName,
                 contentType: task.contentType,
-                size: task.size,
+                sizeBytes: task.size,
                 deletedAt: null,
                 uploadedAt: new Date(),
               })

apps/sim/ee/workspace-forking/lib/copy/storage-quota.ts

@@ -25,7 +25,7 @@ export interface ForkCopyBytesSelection {
 /**
  * Byte total a fork/sync copy selection would duplicate into the target: selected
  * workspace-file blobs plus the selected knowledge bases' stored document blobs. Sizes
- * come from the metadata rows (`workspace_files.size`, `document.file_size`) - no blob
+ * come from the metadata rows (`workspace_files.size_bytes`, `document.file_size`) - no blob
  * reads. Both sums scope to the source workspace with the same filters the copy itself
  * applies, so an id that is not actually copyable can only over-count (block), never
  * under-count.
@@ -48,7 +48,7 @@ export async function sumForkCopyBytes(
     fileSelectors.length === 0
       ? sql<number>`0`
       : sql<number>`(
-          SELECT coalesce(sum(coalesce(${workspaceFiles.sizeBytes}, ${workspaceFiles.size})), 0)
+          SELECT coalesce(sum(${workspaceFiles.sizeBytes}), 0)
           FROM ${workspaceFiles}
           WHERE ${and(
             fileSelectors.length === 1 ? fileSelectors[0] : or(...fileSelectors),

apps/sim/lib/billing/storage/payer-transfer.test.ts

@@ -52,7 +52,7 @@ vi.mock('@sim/db/schema', () => ({
     __table: 'workspaceFiles',
     context: 'workspaceFiles.context',
     deletedAt: 'workspaceFiles.deletedAt',
-    size: 'workspaceFiles.size',
+    sizeBytes: 'workspaceFiles.sizeBytes',
     workspaceId: 'workspaceFiles.workspaceId',
   },
 }))

apps/sim/lib/billing/storage/payer-transfer.ts

@@ -82,7 +82,7 @@ async function getExactWorkspaceStorageBytes(tx: DbOrTx, workspaceId: string): P
   const [row] = await tx.execute<ExactWorkspaceStorageRow>(sql`
     SELECT
       COALESCE((
-        SELECT SUM(coalesce(${workspaceFiles.sizeBytes}, ${workspaceFiles.size}::bigint))
+        SELECT SUM(${workspaceFiles.sizeBytes})
         FROM ${workspaceFiles}
         WHERE ${workspaceFiles.workspaceId} = ${workspaceId}
           AND ${workspaceFiles.context} = 'workspace'
@@ -171,7 +171,7 @@ async function getExactWorkspaceStorageBytesBatch(
     FROM (
       SELECT
         ${workspaceFiles.workspaceId} AS workspace_id,
-        SUM(coalesce(${workspaceFiles.sizeBytes}, ${workspaceFiles.size}::bigint)) AS workspace_file_bytes,
+        SUM(${workspaceFiles.sizeBytes}) AS workspace_file_bytes,
         0::bigint AS document_bytes
       FROM ${workspaceFiles}
       WHERE ${inArray(workspaceFiles.workspaceId, workspaceIds)}

apps/sim/lib/copilot/chat/fork-chat-files.test.ts

@@ -51,6 +51,7 @@ function makeRow(overrides: Partial<ForkableChatFileRow> = {}): ForkableChatFile
     displayName: 'cat.png',
     contentType: 'image/png',
     size: 100,
+    sizeBytes: 100,
     deletedAt: null,
     uploadedAt: new Date('2026-06-01T00:00:00.000Z'),
     updatedAt: new Date('2026-06-01T00:00:00.000Z'),