UniqAI/hexabot
1
0
Fork 0
mirror of https://github.com/hexastack/hexabot synced 2025-04-09 23:44:34 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add migration for attachment blocks

Browse Source
This commit is contained in:
Mohamed Marrouchi 2025-01-09 18:31:46 +01:00
parent 0ac7d7cac5
commit c5440a949e
1 changed files with 88 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

88
api/src/migration/migrations/1735836154221-v-2-2-0.migration.ts
View File

@ -14,7 +14,9 @@ import mongoose from 'mongoose';
import attachmentSchema, {
Attachment,
} from '@/attachment/schemas/attachment.schema';
import blockSchema, { Block } from '@/chat/schemas/block.schema';
import subscriberSchema, { Subscriber } from '@/chat/schemas/subscriber.schema';
import { StdOutgoingAttachmentMessage } from '@/chat/schemas/types/message';
import { config } from '@/config';
import userSchema, { User } from '@/user/schemas/user.schema';
import { moveFile, moveFiles } from '@/utils/helpers/fs';
@ -260,15 +262,101 @@ const restoreOldAvatarsPath = async ({ logger }: MigrationServices) => {
}
};
/**
* Updates attachment documents for blocks that contain "message.attachment":
* - Rename 'attachment_id' to 'id'
*
* @returns Resolves when the migration process is complete.
*/
const updateAttachmentBlocks = async ({ logger }: MigrationServices) => {
const BlockModel = mongoose.model<Block>(Block.name, blockSchema);
// Find blocks where "message.attachment" exists
const cursor = BlockModel.find({
'message.attachment': { $exists: true },
}).cursor();
for await (const block of cursor) {
try {
const blockMessage = block.message as StdOutgoingAttachmentMessage;
const attachmentId =
blockMessage.attachment?.payload &&
'attachment_id' in blockMessage.attachment?.payload
? blockMessage.attachment?.payload?.attachment_id
: null;
await BlockModel.updateOne(
{ _id: block._id },
{
$set: {
'message.attachment.payload.id': attachmentId,
},
$unset: {
'message.attachment.payload.attachment_id': '',
},
},
);
logger.log(
`Attachment ${attachmentId} attributes successfully updated for block ${block._id}`,
);
} catch (error) {
logger.error(`Failed to update block ${block._id}: ${error.message}`);
}
}
};
/**
* Revert updates on attachment documents for blocks that contain "message.attachment":
* - Rename 'id' back to 'attachment_id'
*
* @returns Resolves when the migration process is complete.
*/
const revertAttachmentBlocks = async ({ logger }: MigrationServices) => {
const BlockModel = mongoose.model<Block>(Block.name, blockSchema);
// Find blocks where "message.attachment" exists
const cursor = BlockModel.find({
'message.attachment': { $exists: true },
}).cursor();
for await (const block of cursor) {
try {
const blockMessage = block.message as StdOutgoingAttachmentMessage;
const attachmentId =
blockMessage.attachment?.payload &&
'id' in blockMessage.attachment?.payload
? blockMessage.attachment?.payload?.id
: null;
await BlockModel.updateOne(
{ _id: block._id },
{
$set: {
'message.attachment.payload.attachment_id': attachmentId,
},
$unset: {
'message.attachment.payload.id': '',
},
},
);
logger.log(
`Attachment ${attachmentId} attributes successfully reverted for block ${block._id}`,
);
} catch (error) {
logger.error(`Failed to revert block ${block._id}: ${error.message}`);
}
}
};
module.exports = {
async up(services: MigrationServices) {
await populateSubscriberAvatar(services);
await updateOldAvatarsPath(services);
await updateAttachmentBlocks(services);
return true;
},
async down(services: MigrationServices) {
await unpopulateSubscriberAvatar(services);
await restoreOldAvatarsPath(services);
await revertAttachmentBlocks(services);
return true;
},
};