mirrors/signal-desktop
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages 2 Wiki Activity Actions

Improve backup conversation message request state handling

Browse source
This commit is contained in:
trevor-signal 2024-12-04 00:53:44 -05:00 committed by GitHub
parent 644f5e0fd4
commit 8655244261
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 17 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
ts/services/backups/import.ts
View file

@ -118,6 +118,7 @@ import { getEnvironment, isTestEnvironment } from '../../environment';
import { hasAttachmentDownloads } from '../../util/hasAttachmentDownloads';
import { isNightly } from '../../util/version';
import { ToastType } from '../../types/Toast';
import { isConversationAccepted } from '../../util/isConversationAccepted';
const MAX_CONCURRENCY = 10;
@ -299,8 +300,9 @@ export class BackupImportStream extends Writable {
override async _final(done: (error?: Error) => void): Promise<void> {
try {
// Finish saving remaining conversations/messages
await this.flushConversations();
// Save messages first since they depend on conversations in memory
await this.flushMessages();
await this.flushConversations();
log.info(`${this.logId}: flushed messages and conversations`);
// Store sticker packs and schedule downloads
@ -568,11 +570,14 @@ export class BackupImportStream extends Writable {
}
if (hasAttachmentDownloads(attributes)) {
attachmentDownloadJobPromises.push(
queueAttachmentDownloads(attributes, {
source: AttachmentDownloadSource.BACKUP_IMPORT,
})
);
const conversation = this.conversations.get(attributes.conversationId);
if (conversation && isConversationAccepted(conversation)) {
attachmentDownloadJobPromises.push(
queueAttachmentDownloads(attributes, {
source: AttachmentDownloadSource.BACKUP_IMPORT,
})
);
}
}
}
await Promise.allSettled(attachmentDownloadJobPromises);
@ -940,6 +945,10 @@ export class BackupImportStream extends Writable {
secretParams: Bytes.toBase64(secretParams),
publicParams: Bytes.toBase64(publicParams),
profileSharing: group.whitelisted === true,
messageRequestResponseType:
group.whitelisted === true
? SignalService.SyncMessage.MessageRequestResponse.Type.ACCEPT
: undefined,
hideStory: group.hideStory === true,
storySendMode,
avatar: avatarUrl
@ -1377,7 +1386,7 @@ export class BackupImportStream extends Writable {
if (item.revisions?.length) {
strictAssert(
item.standardMessage,
'Only standard message can have revisions'
`${logId}: Only standard message can have revisions`
);
const history = await this.fromRevisions(attributes, item.revisions);
@ -1386,7 +1395,7 @@ export class BackupImportStream extends Writable {
// Update timestamps on the parent message
const oldest = history.at(-1);
assertDev(oldest != null, 'History is non-empty');
assertDev(oldest != null, `${logId}: History is non-empty`);
attributes.editMessageReceivedAt = attributes.received_at;
attributes.editMessageReceivedAtMs = attributes.received_at_ms;