feat(regions): repo functions for copying project branches and commits

This commit is contained in:
Chuck Driesler
2025-01-12 00:03:14 +00:00
parent b98160f037
commit 4d5f96bb8e
6 changed files with 751 additions and 595 deletions
@@ -1,5 +1,8 @@
import { StreamRecord } from '@/modules/core/helpers/types'
import {
CopyProjectModels,
CopyProjects,
CopyProjectVersions,
GetDefaultRegion,
GetWorkspace,
GetWorkspaceRoleForUser,
@@ -97,23 +100,23 @@ type GetWorkspaceProjectsReturnValue = {
export const getWorkspaceProjectsFactory =
({ getStreams }: { getStreams: GetUserStreamsPage }) =>
async (
args: GetWorkspaceProjectsArgs,
opts: GetWorkspaceProjectsOptions
): Promise<GetWorkspaceProjectsReturnValue> => {
const { streams, cursor } = await getStreams({
cursor: opts.cursor,
limit: opts.limit || 25,
searchQuery: opts.filter?.search || undefined,
workspaceId: args.workspaceId,
userId: opts.filter.userId
})
async (
args: GetWorkspaceProjectsArgs,
opts: GetWorkspaceProjectsOptions
): Promise<GetWorkspaceProjectsReturnValue> => {
const { streams, cursor } = await getStreams({
cursor: opts.cursor,
limit: opts.limit || 25,
searchQuery: opts.filter?.search || undefined,
workspaceId: args.workspaceId,
userId: opts.filter.userId
})
return {
items: streams,
cursor
return {
items: streams,
cursor
}
}
}
type MoveProjectToWorkspaceArgs = {
projectId: string
@@ -138,66 +141,78 @@ export const moveProjectToWorkspaceFactory =
getWorkspaceRoleToDefaultProjectRoleMapping: GetWorkspaceRoleToDefaultProjectRoleMapping
updateWorkspaceRole: UpdateWorkspaceRole
}) =>
async ({
projectId,
workspaceId
}: MoveProjectToWorkspaceArgs): Promise<StreamRecord> => {
const project = await getProject({ projectId })
async ({
projectId,
workspaceId
}: MoveProjectToWorkspaceArgs): Promise<StreamRecord> => {
const project = await getProject({ projectId })
if (!project) throw new ProjectNotFoundError()
if (project.workspaceId?.length) {
// We do not currently support moving projects between workspaces
throw new WorkspaceInvalidProjectError(
'Specified project already belongs to a workspace. Moving between workspaces is not yet supported.'
)
}
// Update roles for current project members
const projectTeam = await getProjectCollaborators({ projectId })
const workspaceTeam = await getWorkspaceRoles({ workspaceId })
const defaultProjectRoleMapping = await getWorkspaceRoleToDefaultProjectRoleMapping(
{ workspaceId }
)
for (const projectMembers of chunk(projectTeam, 5)) {
await Promise.all(
projectMembers.map(
async ({ id: userId, role: serverRole, streamRole: currentProjectRole }) => {
// Update workspace role. Prefer existing workspace role if there is one.
const currentWorkspaceRole = workspaceTeam.find(
(role) => role.userId === userId
)
const nextWorkspaceRole = currentWorkspaceRole ?? {
userId,
workspaceId,
role:
serverRole === Roles.Server.Guest
? Roles.Workspace.Guest
: Roles.Workspace.Member,
createdAt: new Date()
}
await updateWorkspaceRole(nextWorkspaceRole)
// Update project role. Prefer default workspace project role if more permissive.
const defaultProjectRole =
defaultProjectRoleMapping[nextWorkspaceRole.role] ?? Roles.Stream.Reviewer
const nextProjectRole = orderByWeight(
[currentProjectRole, defaultProjectRole],
coreUserRoles
)[0]
await upsertProjectRole({
userId,
projectId,
role: nextProjectRole.name as StreamRoles
})
}
if (!project) throw new ProjectNotFoundError()
if (project.workspaceId?.length) {
// We do not currently support moving projects between workspaces
throw new WorkspaceInvalidProjectError(
'Specified project already belongs to a workspace. Moving between workspaces is not yet supported.'
)
}
// Update roles for current project members
const projectTeam = await getProjectCollaborators({ projectId })
const workspaceTeam = await getWorkspaceRoles({ workspaceId })
const defaultProjectRoleMapping = await getWorkspaceRoleToDefaultProjectRoleMapping(
{ workspaceId }
)
for (const projectMembers of chunk(projectTeam, 5)) {
await Promise.all(
projectMembers.map(
async ({ id: userId, role: serverRole, streamRole: currentProjectRole }) => {
// Update workspace role. Prefer existing workspace role if there is one.
const currentWorkspaceRole = workspaceTeam.find(
(role) => role.userId === userId
)
const nextWorkspaceRole = currentWorkspaceRole ?? {
userId,
workspaceId,
role:
serverRole === Roles.Server.Guest
? Roles.Workspace.Guest
: Roles.Workspace.Member,
createdAt: new Date()
}
await updateWorkspaceRole(nextWorkspaceRole)
// Update project role. Prefer default workspace project role if more permissive.
const defaultProjectRole =
defaultProjectRoleMapping[nextWorkspaceRole.role] ?? Roles.Stream.Reviewer
const nextProjectRole = orderByWeight(
[currentProjectRole, defaultProjectRole],
coreUserRoles
)[0]
await upsertProjectRole({
userId,
projectId,
role: nextProjectRole.name as StreamRoles
})
}
)
)
}
// Assign project to workspace
return await updateProject({ projectUpdate: { id: projectId, workspaceId } })
}
// Assign project to workspace
return await updateProject({ projectUpdate: { id: projectId, workspaceId } })
}
export const moveProjectToRegionFactory =
(deps: {
copyProjects: CopyProjects,
copyProjectModels: CopyProjectModels,
copyProjectVersions: CopyProjectVersions
}) =>
async (args: { projectId: string }): Promise<void> => {
const projectIds = await deps.copyProjects({ projectIds: [args.projectId] })
const modelIdsByProjectId = await deps.copyProjectModels({ projectIds })
const versionIdsByProjectId = await deps.copyProjectVersions({ projectIds })
}
export const getWorkspaceRoleToDefaultProjectRoleMappingFactory =
({
@@ -205,19 +220,19 @@ export const getWorkspaceRoleToDefaultProjectRoleMappingFactory =
}: {
getWorkspace: GetWorkspace
}): GetWorkspaceRoleToDefaultProjectRoleMapping =>
async ({ workspaceId }) => {
const workspace = await getWorkspace({ workspaceId })
async ({ workspaceId }) => {
const workspace = await getWorkspace({ workspaceId })
if (!workspace) {
throw new WorkspaceNotFoundError()
}
if (!workspace) {
throw new WorkspaceNotFoundError()
}
return {
[Roles.Workspace.Guest]: null,
[Roles.Workspace.Member]: workspace.defaultProjectRole,
[Roles.Workspace.Admin]: Roles.Stream.Owner
return {
[Roles.Workspace.Guest]: null,
[Roles.Workspace.Member]: workspace.defaultProjectRole,
[Roles.Workspace.Admin]: Roles.Stream.Owner
}
}
}
export const updateWorkspaceProjectRoleFactory =
({
@@ -229,63 +244,63 @@ export const updateWorkspaceProjectRoleFactory =
getWorkspaceRoleForUser: GetWorkspaceRoleForUser
updateStreamRoleAndNotify: UpdateStreamRole
}): UpdateWorkspaceProjectRole =>
async ({ role, updater }) => {
const { workspaceId } = (await getStream({ streamId: role.projectId })) ?? {}
if (!workspaceId) throw new WorkspaceInvalidProjectError()
async ({ role, updater }) => {
const { workspaceId } = (await getStream({ streamId: role.projectId })) ?? {}
if (!workspaceId) throw new WorkspaceInvalidProjectError()
const currentWorkspaceRole = await getWorkspaceRoleForUser({
workspaceId,
userId: role.userId
})
const currentWorkspaceRole = await getWorkspaceRoleForUser({
workspaceId,
userId: role.userId
})
if (currentWorkspaceRole?.role === Roles.Workspace.Admin) {
// User is workspace admin and cannot have their project roles changed
throw new WorkspaceAdminError()
if (currentWorkspaceRole?.role === Roles.Workspace.Admin) {
// User is workspace admin and cannot have their project roles changed
throw new WorkspaceAdminError()
}
if (
currentWorkspaceRole?.role === Roles.Workspace.Guest &&
role.role === Roles.Stream.Owner
) {
// Workspace guests cannot be project owners
throw new WorkspaceInvalidRoleError('Workspace guests cannot be project owners.')
}
return await updateStreamRoleAndNotify(
role,
updater.userId!,
updater.resourceAccessRules
)
}
if (
currentWorkspaceRole?.role === Roles.Workspace.Guest &&
role.role === Roles.Stream.Owner
) {
// Workspace guests cannot be project owners
throw new WorkspaceInvalidRoleError('Workspace guests cannot be project owners.')
}
return await updateStreamRoleAndNotify(
role,
updater.userId!,
updater.resourceAccessRules
)
}
export const createWorkspaceProjectFactory =
(deps: { getDefaultRegion: GetDefaultRegion }) =>
async (params: { input: WorkspaceProjectCreateInput; ownerId: string }) => {
const { input, ownerId } = params
const workspaceDefaultRegion = await deps.getDefaultRegion({
workspaceId: input.workspaceId
})
const regionKey = workspaceDefaultRegion?.key
const projectDb = await getDb({ regionKey })
const db = mainDb
async (params: { input: WorkspaceProjectCreateInput; ownerId: string }) => {
const { input, ownerId } = params
const workspaceDefaultRegion = await deps.getDefaultRegion({
workspaceId: input.workspaceId
})
const regionKey = workspaceDefaultRegion?.key
const projectDb = await getDb({ regionKey })
const db = mainDb
// todo, use the command factory here, but for that, we need to migrate to the event bus
// deps not injected to ensure proper DB injection
const createNewProject = createNewProjectFactory({
storeProject: storeProjectFactory({ db: projectDb }),
getProject: getProjectFactory({ db }),
deleteProject: deleteProjectFactory({ db: projectDb }),
storeModel: storeModelFactory({ db: projectDb }),
// THIS MUST GO TO THE MAIN DB
storeProjectRole: storeProjectRoleFactory({ db }),
projectsEventsEmitter: ProjectsEmitter.emit
})
// todo, use the command factory here, but for that, we need to migrate to the event bus
// deps not injected to ensure proper DB injection
const createNewProject = createNewProjectFactory({
storeProject: storeProjectFactory({ db: projectDb }),
getProject: getProjectFactory({ db }),
deleteProject: deleteProjectFactory({ db: projectDb }),
storeModel: storeModelFactory({ db: projectDb }),
// THIS MUST GO TO THE MAIN DB
storeProjectRole: storeProjectRoleFactory({ db }),
projectsEventsEmitter: ProjectsEmitter.emit
})
const project = await createNewProject({
...input,
regionKey,
ownerId
})
const project = await createNewProject({
...input,
regionKey,
ownerId
})
return project
}
return project
}