Add downloadPath to attachments after import to support resumable download

This commit is contained in:
trevor-signal 2024-09-03 11:55:30 -04:00 committed by GitHub
parent 01581b04d8
commit 34ab08aeeb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 52 additions and 6 deletions

View file

@ -4,7 +4,7 @@
import { assert } from 'chai';
import path from 'path';
import { tmpdir } from 'os';
import { sortBy } from 'lodash';
import { omit, sortBy } from 'lodash';
import { createReadStream } from 'fs';
import { mkdtemp, rm } from 'fs/promises';
import * as sinon from 'sinon';
@ -65,6 +65,11 @@ function sortAndNormalize(
reactions,
sendStateByConversationId,
verifiedChanged,
attachments,
preview,
contact,
quote,
sticker,
// This is not in the backup
// eslint-disable-next-line @typescript-eslint/no-unused-vars
@ -122,6 +127,36 @@ function sortAndNormalize(
};
}),
attachments: attachments?.map(attachment =>
omit(attachment, 'downloadPath')
),
preview: preview?.map(previewItem => ({
...previewItem,
image: omit(previewItem.image, 'downloadPath'),
})),
contact: contact?.map(contactItem => ({
...contactItem,
avatar: {
...contactItem.avatar,
avatar: omit(contactItem.avatar?.avatar, 'downloadPath'),
},
})),
quote: quote
? {
...quote,
attachments: quote?.attachments.map(quotedAttachment => ({
...quotedAttachment,
thumbnail: omit(quotedAttachment.thumbnail, 'downloadPath'),
})),
}
: undefined,
sticker: sticker
? {
...sticker,
data: omit(sticker.data, 'downloadPath'),
}
: undefined,
// Not an original property, but useful
isUnsupported: isUnsupportedMessage(message),
})