awatertrevi/infisical
1
0
Fork 0
mirror of https://github.com/awatertrevi/infisical.git synced 2026-10-07 06:28:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(infisical-pg): minor bug fixes in frontend and pg migrator secret approval req

Browse Source
This commit is contained in:
Akhil Mohan
2024-01-27 12:40:37 +05:30
parent 53f16f01b8
commit af9661b221
2 changed files with 157 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files
frontend/src/pages/org/[id]/overview/index.tsx
+16 -13
View File
@@ -542,19 +542,22 @@ const OrganizationPage = withPermission(
// not using hooks because need at this point only // not using hooks because need at this point only
const orgUsers = await fetchOrgUsers(currentOrg); const orgUsers = await fetchOrgUsers(currentOrg);
const decryptKey = await fetchUserWsKey(newWorkspaceId); const decryptKey = await fetchUserWsKey(newWorkspaceId);
await addWsUser.mutateAsync({ const members = orgUsers
workspaceId: newWorkspaceId, .filter(
decryptKey, ({ status, user: orgUser }) => status === "accepted" && user.email !== orgUser.email
userPrivateKey: PRIVATE_KEY, )
members: orgUsers .map(({ user: orgUser, id: orgMembershipId }) => ({
.filter( userPublicKey: orgUser.publicKey,
({ status, user: orgUser }) => status === "accepted" && user.email !== orgUser.email orgMembershipId
) }));
.map(({ user: orgUser, id: orgMembershipId }) => ({ if (members.length) {
userPublicKey: orgUser.publicKey, await addWsUser.mutateAsync({
orgMembershipId workspaceId: newWorkspaceId,
})) decryptKey,
}); userPrivateKey: PRIVATE_KEY,
members
});
}
} }
createNotification({ text: "Workspace created", type: "success" }); createNotification({ text: "Workspace created", type: "success" });
handlePopUpClose("addNewWs"); handlePopUpClose("addNewWs");
pg-migrator/src/index.ts
+141 -28
View File
@@ -13,6 +13,7 @@ import {
Bot, Bot,
BotKey, BotKey,
BotOrg, BotOrg,
CommitType,
Folder, Folder,
FolderVersion, FolderVersion,
GitAppInstallationSession, GitAppInstallationSession,
@@ -175,7 +176,11 @@ export const migrateCollection = async <
" Postgres table name:", " Postgres table name:",
postgresTableName, postgresTableName,
); );
const totalMongoCount = await mongooseCollection.countDocuments(); const totalMongoCount =
postgresTableName === TableName.AuditLog
? await mongooseCollection.estimatedDocumentCount() // this is because audit log is too big
: await mongooseCollection.countDocuments();
console.log("Total documents", totalMongoCount); console.log("Total documents", totalMongoCount);
console.log("Total batches", Math.ceil(totalMongoCount / 1000)); console.log("Total batches", Math.ceil(totalMongoCount / 1000));
let batch = 1; let batch = 1;
@@ -2330,33 +2335,141 @@ const main = async () => {
}, },
}); });
// console.log( console.log(
// "Migrating Mongo Secret approval request secrets -> Pg Secret Approval Request secrets table", "Migrating Mongo Secret approval request secrets -> Pg Secret Approval Request secrets table",
// ); );
// await migrateCollection({ await migrateCollection({
// db, db,
// mongooseCollection: SecretApprovalRequest, mongooseCollection: SecretApprovalRequest,
// postgresTableName: TableName.SarSecret, postgresTableName: TableName.SecretApprovalRequestSecret,
// returnKeys: ["id"], returnKeys: ["id"],
// preProcessing: async (doc) => { preProcessing: async (doc) => {
// const id = uuidV4(); const requestId = await sarKv.get(doc._id.toString()).catch(() => null);
// const requestId = await sarKv.get(doc._id.toString()); if (!requestId) return;
//
// return Promise.all( return Promise.all(
// doc.commits.map(async (commit) => { doc.commits.map(async (commit) => {
// if (commit.op === CommitType.CREATE) { const id = uuidV4();
// return {
// id, if (commit.op === CommitType.CREATE) {
// requestId, if (!commit.newVersion.secretBlindIndex) return;
// secretBlindIndex: commit.newVersion.secretBlindIndex,
// createdAt: new Date((doc as any).createdAt), return {
// updatedAt: new Date((doc as any).updatedAt), id,
// }; op: CommitType.CREATE,
// } requestId,
// }), secretBlindIndex: commit.newVersion.secretBlindIndex,
// ); keyEncoding: commit.newVersion.keyEncoding as string,
// }, algorithm: commit.newVersion.algorithm as string,
// }); version: commit.newVersion.version,
secretKeyIV: commit.newVersion.secretKeyIV,
secretKeyTag: commit.newVersion.secretKeyTag,
secretKeyCiphertext: commit.newVersion.secretKeyCiphertext,
secretValueIV: commit.newVersion.secretValueIV,
secretValueTag: commit.newVersion.secretValueTag,
secretValueCiphertext: commit.newVersion.secretValueCiphertext,
secretCommentIV: commit.newVersion.secretCommentIV,
secretCommentTag: commit.newVersion.secretCommentTag,
secretCommentCiphertext:
commit.newVersion.secretCommentCiphertext,
createdAt: new Date((doc as any).createdAt),
updatedAt: new Date((doc as any).updatedAt),
};
}
if (commit.op === CommitType.UPDATE) {
const id = uuidV4();
const secretId = await secKv
.get(commit.secret.toString())
.catch(() => null);
if (!secretId) return;
const secret = await Secret.findById(commit.secret);
if (!secret) return;
if (
!commit.newVersion.secretBlindIndex ||
!secret?.secretBlindIndex
)
return;
return {
id,
op: CommitType.UPDATE,
requestId,
secretId,
secretBlindIndex:
commit.newVersion.secretBlindIndex || secret.secretBlindIndex,
keyEncoding:
(commit.newVersion.keyEncoding as string) ||
secret.keyEncoding,
algorithm:
(commit.newVersion.algorithm as string) || secret.algorithm,
version: commit.newVersion.version || secret.version,
secretKeyIV:
commit.newVersion.secretKeyIV || secret.secretKeyIV,
secretKeyTag:
commit.newVersion.secretKeyTag || secret.secretKeyTag,
secretKeyCiphertext:
commit.newVersion.secretKeyCiphertext ||
secret.secretKeyCiphertext,
secretValueIV:
commit.newVersion.secretValueIV || secret.secretValueIV,
secretValueTag:
commit.newVersion.secretValueTag || secret.secretValueTag,
secretValueCiphertext:
commit.newVersion.secretValueCiphertext ||
secret.secretValueCiphertext,
secretCommentIV:
commit.newVersion.secretCommentIV || secret.secretCommentIV,
secretCommentTag:
commit.newVersion.secretCommentTag || secret.secretCommentTag,
secretCommentCiphertext:
commit.newVersion.secretCommentCiphertext ||
secret.secretCommentCiphertext,
createdAt: new Date((doc as any).createdAt),
updatedAt: new Date((doc as any).updatedAt),
};
}
if (commit.op === CommitType.DELETE) {
const id = uuidV4();
const secretId = await secKv
.get(commit.secret.toString())
.catch(() => null);
if (!secretId) return;
const secret = await Secret.findById(commit.secret);
if (!secret) return;
if (!secret?.secretBlindIndex) return;
return {
id,
op: CommitType.DELETE,
requestId,
secretId,
secretBlindIndex: secret.secretBlindIndex,
keyEncoding: secret.keyEncoding,
algorithm: secret.algorithm,
version: secret.version,
secretKeyIV: secret.secretKeyIV,
secretKeyTag: secret.secretKeyTag,
secretKeyCiphertext: secret.secretKeyCiphertext,
secretValueIV: secret.secretValueIV,
secretValueTag: secret.secretValueTag,
secretValueCiphertext: secret.secretValueCiphertext,
secretCommentIV: secret.secretCommentIV,
secretCommentTag: secret.secretCommentTag,
secretCommentCiphertext: secret.secretCommentCiphertext,
createdAt: new Date((doc as any).createdAt),
updatedAt: new Date((doc as any).updatedAt),
};
}
}),
);
},
});
console.log("MIGRATION SCRIPT COMPLETED SUCCESSFULLY"); console.log("MIGRATION SCRIPT COMPLETED SUCCESSFULLY");
process.exit(1); process.exit(1);