Chuck Driesler
90 lines
2.9 KiB
TypeScript
90 lines
2.9 KiB
TypeScript
import { GetStreamBranchCount } from '@/modules/core/domain/branches/operations'
|
|
import { GetStreamCommitCount } from '@/modules/core/domain/commits/operations'
|
|
import { GetProject } from '@/modules/core/domain/projects/operations'
|
|
import {
|
|
CopyProjectModels,
|
|
CopyProjects,
|
|
CopyProjectVersions,
|
|
CopyWorkspace,
|
|
GetAvailableRegions,
|
|
UpdateProjectRegion
|
|
} from '@/modules/workspaces/domain/operations'
|
|
import { ProjectRegionAssignmentError } from '@/modules/workspaces/errors/regions'
|
|
|
|
export const updateProjectRegionFactory =
|
|
(deps: {
|
|
getProject: GetProject
|
|
countProjectModels: GetStreamBranchCount
|
|
countProjectVersions: GetStreamCommitCount
|
|
getAvailableRegions: GetAvailableRegions
|
|
copyWorkspace: CopyWorkspace
|
|
copyProjects: CopyProjects
|
|
copyProjectModels: CopyProjectModels
|
|
copyProjectVersions: CopyProjectVersions
|
|
}): UpdateProjectRegion =>
|
|
async (params) => {
|
|
const { projectId, regionKey } = params
|
|
|
|
const project = await deps.getProject({ projectId })
|
|
if (!project) {
|
|
throw new ProjectRegionAssignmentError('Project not found', {
|
|
info: { params }
|
|
})
|
|
}
|
|
if (!project.workspaceId) {
|
|
throw new ProjectRegionAssignmentError('Project not a part of a workspace', {
|
|
info: { params }
|
|
})
|
|
}
|
|
|
|
const availableRegions = await deps.getAvailableRegions({
|
|
workspaceId: project.workspaceId
|
|
})
|
|
if (!availableRegions.find((region) => region.key === regionKey)) {
|
|
throw new ProjectRegionAssignmentError(
|
|
'Specified region not available for workspace',
|
|
{
|
|
info: {
|
|
params,
|
|
workspaceId: project.workspaceId
|
|
}
|
|
}
|
|
)
|
|
}
|
|
|
|
// Move workspace
|
|
await deps.copyWorkspace({ workspaceId: project.workspaceId })
|
|
|
|
// Move commits
|
|
const projectIds = await deps.copyProjects({ projectIds: [projectId] })
|
|
const modelIds = await deps.copyProjectModels({ projectIds })
|
|
const versionIds = await deps.copyProjectVersions({ projectIds })
|
|
|
|
// TODO: Move objects
|
|
// TODO: Move automations
|
|
// TODO: Move comments
|
|
// TODO: Move file blobs
|
|
// TODO: Move webhooks
|
|
|
|
// TODO: Validate state after move captures latest state of project
|
|
const sourceProjectModelCount = await deps.countProjectModels(projectId)
|
|
const sourceProjectVersionCount = await deps.countProjectVersions(projectId)
|
|
|
|
const tests = [
|
|
modelIds[projectId] === sourceProjectModelCount,
|
|
versionIds[projectId] === sourceProjectVersionCount
|
|
]
|
|
|
|
const isReconciled = tests.every((test) => !!test)
|
|
|
|
if (!isReconciled) {
|
|
// TODO: Move failed or source project added data while changing regions. Retry move.
|
|
throw new ProjectRegionAssignmentError(
|
|
'Missing data from source project in target region copy after move.'
|
|
)
|
|
}
|
|
|
|
// TODO: Update project region in db
|
|
return { ...project, regionKey }
|
|
}
|