chore: import upstream snapshot with attribution
CI / Migrate Dev DB (push) Has been skipped
CI / Detect Version (push) Has been cancelled
CI / Migrate DB (push) Has been cancelled
CI / Build Dev ECR (./docker/app.Dockerfile, ECR_APP) (push) Has been cancelled
CI / Build Dev ECR (./docker/db.Dockerfile, ECR_MIGRATIONS) (push) Has been cancelled
CI / Build Dev ECR (./docker/pii.Dockerfile, ECR_PII) (push) Has been cancelled
CI / Build Dev ECR (./docker/realtime.Dockerfile, ECR_REALTIME) (push) Has been cancelled
CI / Deploy Trigger.dev (Dev) (push) Has been cancelled
CI / Build AMD64 (./docker/app.Dockerfile, ECR_APP, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build AMD64 (./docker/db.Dockerfile, ECR_MIGRATIONS, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build AMD64 (./docker/pii.Dockerfile, ECR_PII, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build AMD64 (./docker/realtime.Dockerfile, ECR_REALTIME, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/app.Dockerfile, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/db.Dockerfile, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/pii.Dockerfile, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/realtime.Dockerfile, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Check Docs Changes (push) Has been cancelled
CI / Process Docs (push) Has been cancelled
CI / Create GitHub Release (push) Has been cancelled
CI / Test and Build (push) Has been cancelled
Publish CLI Package / publish-npm (push) Has been cancelled
Publish Python SDK / publish-pypi (push) Has been cancelled
Publish TypeScript SDK / publish-npm (push) Has been cancelled
CI / Migrate Dev DB (push) Has been skipped
CI / Detect Version (push) Has been cancelled
CI / Migrate DB (push) Has been cancelled
CI / Build Dev ECR (./docker/app.Dockerfile, ECR_APP) (push) Has been cancelled
CI / Build Dev ECR (./docker/db.Dockerfile, ECR_MIGRATIONS) (push) Has been cancelled
CI / Build Dev ECR (./docker/pii.Dockerfile, ECR_PII) (push) Has been cancelled
CI / Build Dev ECR (./docker/realtime.Dockerfile, ECR_REALTIME) (push) Has been cancelled
CI / Deploy Trigger.dev (Dev) (push) Has been cancelled
CI / Build AMD64 (./docker/app.Dockerfile, ECR_APP, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build AMD64 (./docker/db.Dockerfile, ECR_MIGRATIONS, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build AMD64 (./docker/pii.Dockerfile, ECR_PII, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build AMD64 (./docker/realtime.Dockerfile, ECR_REALTIME, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/app.Dockerfile, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/db.Dockerfile, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/pii.Dockerfile, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/realtime.Dockerfile, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Check Docs Changes (push) Has been cancelled
CI / Process Docs (push) Has been cancelled
CI / Create GitHub Release (push) Has been cancelled
CI / Test and Build (push) Has been cancelled
Publish CLI Package / publish-npm (push) Has been cancelled
Publish Python SDK / publish-pypi (push) Has been cancelled
Publish TypeScript SDK / publish-npm (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,156 @@
|
||||
import { createLogger } from '@sim/logger'
|
||||
import { generateId } from '@sim/utils/id'
|
||||
import { type NextRequest, NextResponse } from 'next/server'
|
||||
import { importTableAsyncContract } from '@/lib/api/contracts/tables'
|
||||
import { parseRequest } from '@/lib/api/server'
|
||||
import { checkSessionOrInternalAuth } from '@/lib/auth/hybrid'
|
||||
import { isTriggerDevEnabled } from '@/lib/core/config/env-flags'
|
||||
import { runDetached } from '@/lib/core/utils/background'
|
||||
import { generateRequestId } from '@/lib/core/utils/request'
|
||||
import { withRouteHandler } from '@/lib/core/utils/with-route-handler'
|
||||
import { captureServerEvent } from '@/lib/posthog/server'
|
||||
import {
|
||||
createTable,
|
||||
deleteTable,
|
||||
getWorkspaceTableLimits,
|
||||
listTables,
|
||||
releaseJobClaim,
|
||||
sanitizeName,
|
||||
TABLE_LIMITS,
|
||||
TableConflictError,
|
||||
} from '@/lib/table'
|
||||
import { runTableImport, type TableImportPayload } from '@/lib/table/import-runner'
|
||||
import { getUserSettings } from '@/lib/users/queries'
|
||||
import { getUserEntityPermissions } from '@/lib/workspaces/permissions/utils'
|
||||
|
||||
const logger = createLogger('TableImportAsync')
|
||||
|
||||
export const runtime = 'nodejs'
|
||||
export const dynamic = 'force-dynamic'
|
||||
|
||||
export const POST = withRouteHandler(async (request: NextRequest) => {
|
||||
const requestId = generateRequestId()
|
||||
|
||||
const authResult = await checkSessionOrInternalAuth(request, { requireWorkflowId: false })
|
||||
if (!authResult.success || !authResult.userId) {
|
||||
return NextResponse.json({ error: 'Authentication required' }, { status: 401 })
|
||||
}
|
||||
const userId = authResult.userId
|
||||
|
||||
const parsed = await parseRequest(importTableAsyncContract, request, {})
|
||||
if (!parsed.success) return parsed.response
|
||||
const { workspaceId, fileKey, fileName, deleteSourceFile, timezone } = parsed.data.body
|
||||
|
||||
const permission = await getUserEntityPermissions(userId, 'workspace', workspaceId)
|
||||
if (permission !== 'write' && permission !== 'admin') {
|
||||
return NextResponse.json({ error: 'Access denied' }, { status: 403 })
|
||||
}
|
||||
// The fileKey is client-supplied — ensure it points at this workspace's storage prefix so a
|
||||
// caller can't import another workspace's uploaded object.
|
||||
if (!fileKey.startsWith(`workspace/${workspaceId}/`)) {
|
||||
return NextResponse.json({ error: 'Invalid file key for workspace' }, { status: 400 })
|
||||
}
|
||||
|
||||
const ext = fileName.split('.').pop()?.toLowerCase()
|
||||
if (ext !== 'csv' && ext !== 'tsv') {
|
||||
return NextResponse.json({ error: 'Only CSV and TSV files are supported' }, { status: 400 })
|
||||
}
|
||||
const delimiter = ext === 'tsv' ? '\t' : ','
|
||||
|
||||
const planLimits = await getWorkspaceTableLimits(workspaceId)
|
||||
const baseName = sanitizeName(fileName.replace(/\.[^.]+$/, ''), 'imported_table').slice(
|
||||
0,
|
||||
TABLE_LIMITS.MAX_TABLE_NAME_LENGTH
|
||||
)
|
||||
// Re-importing the same file shouldn't fail on a name collision — pick the next free
|
||||
// `name_2`, `name_3`, … (matching how "New table" auto-names), keeping under the cap.
|
||||
const existingNames = new Set(
|
||||
(await listTables(workspaceId, { scope: 'all' })).map((t) => t.name.toLowerCase())
|
||||
)
|
||||
let tableName = baseName
|
||||
for (let n = 2; existingNames.has(tableName.toLowerCase()); n++) {
|
||||
const suffix = `_${n}`
|
||||
tableName = `${baseName.slice(0, TABLE_LIMITS.MAX_TABLE_NAME_LENGTH - suffix.length)}${suffix}`
|
||||
}
|
||||
const importId = generateId()
|
||||
|
||||
// Placeholder schema satisfies createTable's validation; the import worker infers the
|
||||
// real columns from the file and overwrites it before any rows become visible.
|
||||
let table: Awaited<ReturnType<typeof createTable>>
|
||||
try {
|
||||
table = await createTable(
|
||||
{
|
||||
name: tableName,
|
||||
description: `Imported from ${fileName}`,
|
||||
schema: { columns: [{ name: 'column_1', type: 'string' }] },
|
||||
workspaceId,
|
||||
userId,
|
||||
maxTables: planLimits.maxTables,
|
||||
jobStatus: 'running',
|
||||
jobType: 'import',
|
||||
jobId: importId,
|
||||
},
|
||||
requestId
|
||||
)
|
||||
} catch (error) {
|
||||
if (error instanceof TableConflictError) {
|
||||
return NextResponse.json({ error: error.message }, { status: 409 })
|
||||
}
|
||||
if (error instanceof Error && error.message.includes('maximum table limit')) {
|
||||
return NextResponse.json({ error: error.message }, { status: 400 })
|
||||
}
|
||||
throw error
|
||||
}
|
||||
|
||||
const importPayload: TableImportPayload = {
|
||||
importId,
|
||||
tableId: table.id,
|
||||
workspaceId,
|
||||
userId,
|
||||
fileKey,
|
||||
fileName,
|
||||
delimiter,
|
||||
mode: 'create',
|
||||
deleteSourceFile,
|
||||
timezone: timezone ?? (await getUserSettings(userId)).timezone ?? 'UTC',
|
||||
}
|
||||
if (isTriggerDevEnabled) {
|
||||
// Trigger.dev runs the import outside the web container, so it survives app deploys.
|
||||
try {
|
||||
const [{ tableImportTask }, { tasks }, { resolveTriggerRegion }] = await Promise.all([
|
||||
import('@/background/table-import'),
|
||||
import('@trigger.dev/sdk'),
|
||||
import('@/lib/core/async-jobs/region'),
|
||||
])
|
||||
await tasks.trigger<typeof tableImportTask>('table-import', importPayload, {
|
||||
tags: [`tableId:${table.id}`, `jobId:${importId}`],
|
||||
region: await resolveTriggerRegion(),
|
||||
})
|
||||
} catch (error) {
|
||||
// A failed dispatch must not leave a ghost `running` job holding the
|
||||
// table's one-write-job slot — nor, in create mode, the placeholder
|
||||
// table itself: the user never saw it, so archive it back out of the
|
||||
// workspace (no hard-delete surface exists; archived is invisible).
|
||||
await releaseJobClaim(table.id, importId).catch(() => {})
|
||||
await deleteTable(table.id, requestId).catch(() => {})
|
||||
throw error
|
||||
}
|
||||
} else {
|
||||
runDetached('table-import', () => runTableImport(importPayload))
|
||||
}
|
||||
|
||||
captureServerEvent(
|
||||
userId,
|
||||
'table_import_started',
|
||||
{
|
||||
table_id: table.id,
|
||||
workspace_id: workspaceId,
|
||||
import_id: importId,
|
||||
file_type: ext,
|
||||
},
|
||||
{ groups: { workspace: workspaceId } }
|
||||
)
|
||||
|
||||
logger.info(`[${requestId}] Async CSV import started`, { tableId: table.id, importId, fileName })
|
||||
return NextResponse.json({ success: true, data: { tableId: table.id, importId } })
|
||||
})
|
||||
Reference in New Issue
Block a user