add role deduplication logic

This commit is contained in:
Maidul Islam
2023-09-15 11:44:48 -04:00
parent adfa90340d
commit 0ce5aaf61c

View File

@@ -696,6 +696,9 @@ export const backfillPermission = async () => {
.populate<{ workspace: IWorkspace }>("workspace")
.lean();
// group memberships that need the same permission set
const roleMap = new Map<string, { membershipIds: string[], permissions: any[], organizationId: string, workspaceId: string }>();
for (const membership of memberships) {
// get permissions of members except secret permission
const customPermissions = memberProjectPermissions.rules.filter(
@@ -711,16 +714,17 @@ export const backfillPermission = async () => {
if (ability === "read") secretAccessRule[environmentSlug].read = false;
});
const secretPermissions: any = [];
Object.entries(secretAccessRule).forEach(([envSlug, { read, write }]) => {
if (read) {
customPermissions.push({
secretPermissions.push({
subject: ProjectPermissionSub.Secrets,
action: ProjectPermissionActions.Read,
conditions: { environment: envSlug }
});
}
if (write) {
customPermissions.push(
secretPermissions.push(
{
subject: ProjectPermissionSub.Secrets,
action: ProjectPermissionActions.Edit,
@@ -740,23 +744,44 @@ export const backfillPermission = async () => {
}
});
const key = `${JSON.stringify(secretPermissions)}-${membership.workspace.organization.toString()}-${membership.workspace._id.toString()}`; // group roles that have same permission with in the same org and workspace
const value = roleMap.get(key);
if (value) {
value.membershipIds.push(membership._id.toString());
value.organizationId = membership.workspace.organization.toString()
value.workspaceId = membership.workspace._id.toString()
} else {
roleMap.set(key, { membershipIds: [membership._id.toString()], permissions: [...customPermissions, ...secretPermissions], organizationId: membership.workspace.organization.toString(), workspaceId: membership.workspace._id.toString() });
}
}
for (const [key, value] of roleMap.entries()) {
const { membershipIds, permissions, workspaceId, organizationId } = value
const role = new Role({
name: "Migrated Role",
organization: membership.workspace.organization,
workspace: membership.workspace._id,
organization: organizationId,
workspace: workspaceId,
description: "This role was auto generated by Infisical with the release of our new permission system",
isOrgRole: false,
slug: `custom-role-${crypto.randomBytes(3).toString("hex")}`,
permissions: customPermissions
permissions: permissions
});
await role.save();
await Membership.findByIdAndUpdate(membership._id, {
$set: {
role: CUSTOM,
customRole: role
}
});
for (const id of membershipIds) {
await Membership.findByIdAndUpdate(id, { // document db doesn't support update many so we must loop
$set: {
role: CUSTOM,
customRole: role
}
});
}
}
console.log("Backfill: Finished converting old denied permission in workspace to viewers");
await MembershipOrg.updateMany(
{
role: OWNER
@@ -767,5 +792,6 @@ export const backfillPermission = async () => {
}
}
);
console.log("Backfill: Finishing converting old denied permission in workspace to viewers");
console.log("Backfill: Finished converting owner role to member");
};