Convert message cleanup services to TypeScript

This commit is contained in:
Evan Hahn 2022-05-31 23:53:14 +00:00 committed by GitHub
parent 16d180efac
commit 2a2f44a73a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 127 additions and 151 deletions

View file

@ -1,98 +0,0 @@
// Copyright 2019-2020 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
/* global
_,
MessageController,
Whisper
*/
// eslint-disable-next-line func-names
(function () {
window.Whisper = window.Whisper || {};
async function eraseTapToViewMessages() {
try {
window.SignalContext.log.info(
'eraseTapToViewMessages: Loading messages...'
);
const messages =
await window.Signal.Data.getTapToViewMessagesNeedingErase({
MessageCollection: Whisper.MessageCollection,
});
await Promise.all(
messages.map(async fromDB => {
const message = MessageController.register(fromDB.id, fromDB);
window.SignalContext.log.info(
'eraseTapToViewMessages: message data erased',
message.idForLogging()
);
await message.eraseContents();
})
);
} catch (error) {
window.SignalContext.log.error(
'eraseTapToViewMessages: Error erasing messages',
error && error.stack ? error.stack : error
);
}
window.SignalContext.log.info('eraseTapToViewMessages: complete');
}
let timeout;
async function checkTapToViewMessages() {
const SECOND = 1000;
const MINUTE = 60 * SECOND;
const HOUR = 60 * MINUTE;
const THIRTY_DAYS = 30 * 24 * HOUR;
const receivedAt =
await window.Signal.Data.getNextTapToViewMessageTimestampToAgeOut();
if (!receivedAt) {
return;
}
const nextCheck = receivedAt + THIRTY_DAYS;
Whisper.TapToViewMessagesListener.nextCheck = nextCheck;
window.SignalContext.log.info(
'checkTapToViewMessages: next check at',
new Date(nextCheck).toISOString()
);
let wait = nextCheck - Date.now();
// In the past
if (wait < 0) {
wait = 0;
}
// Too far in the future, since it's limited to a 32-bit value
if (wait > 2147483647) {
wait = 2147483647;
}
clearTimeout(timeout);
timeout = setTimeout(async () => {
await eraseTapToViewMessages();
checkTapToViewMessages();
}, wait);
}
const debouncedCheckTapToViewMessages = _.debounce(
checkTapToViewMessages,
1000
);
Whisper.TapToViewMessagesListener = {
nextCheck: null,
init(events) {
checkTapToViewMessages();
events.on('timetravel', debouncedCheckTapToViewMessages);
},
update: debouncedCheckTapToViewMessages,
};
})();

View file

@ -39,6 +39,8 @@ import { normalizeUuid } from './util/normalizeUuid';
import { filter } from './util/iterables'; import { filter } from './util/iterables';
import { isNotNil } from './util/isNotNil'; import { isNotNil } from './util/isNotNil';
import { IdleDetector } from './IdleDetector'; import { IdleDetector } from './IdleDetector';
import { expiringMessagesDeletionService } from './services/expiringMessagesDeletion';
import { tapToViewMessagesDeletionService } from './services/tapToViewMessagesDeletionService';
import { getStoriesForRedux, loadStories } from './services/storyLoader'; import { getStoriesForRedux, loadStories } from './services/storyLoader';
import { senderCertificateService } from './services/senderCertificate'; import { senderCertificateService } from './services/senderCertificate';
import { GROUP_CREDENTIALS_KEY } from './services/groupCredentialFetcher'; import { GROUP_CREDENTIALS_KEY } from './services/groupCredentialFetcher';
@ -1731,8 +1733,12 @@ export async function startApp(): Promise<void> {
window.Whisper.events.trigger('timetravel'); window.Whisper.events.trigger('timetravel');
}); });
window.Whisper.ExpiringMessagesListener.init(window.Whisper.events); expiringMessagesDeletionService.update();
window.Whisper.TapToViewMessagesListener.init(window.Whisper.events); tapToViewMessagesDeletionService.update();
window.Whisper.events.on('timetravel', () => {
expiringMessagesDeletionService.update();
tapToViewMessagesDeletionService.update();
});
const isCoreDataValid = Boolean( const isCoreDataValid = Boolean(
window.textsecure.storage.user.getUuid() && window.textsecure.storage.user.getUuid() &&

View file

@ -1,34 +1,38 @@
// Copyright 2016-2021 Signal Messenger, LLC // Copyright 2016-2022 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only // SPDX-License-Identifier: AGPL-3.0-only
/* global import { debounce } from 'lodash';
_, import type { MessageModel } from '../models/messages';
MessageController, import { clearTimeoutIfNecessary } from '../util/clearTimeoutIfNecessary';
Whisper
*/
// eslint-disable-next-line func-names class ExpiringMessagesDeletionService {
(function () { public update: typeof this.checkExpiringMessages;
window.Whisper = window.Whisper || {};
async function destroyExpiredMessages() { private timeout?: ReturnType<typeof setTimeout>;
constructor() {
this.update = debounce(this.checkExpiringMessages, 1000);
}
private async destroyExpiredMessages() {
try { try {
window.SignalContext.log.info( window.SignalContext.log.info(
'destroyExpiredMessages: Loading messages...' 'destroyExpiredMessages: Loading messages...'
); );
const messages = await window.Signal.Data.getExpiredMessages({ const messages = await window.Signal.Data.getExpiredMessages();
MessageCollection: Whisper.MessageCollection,
});
window.SignalContext.log.info( window.SignalContext.log.info(
`destroyExpiredMessages: found ${messages.length} messages to expire` `destroyExpiredMessages: found ${messages.length} messages to expire`
); );
const messageIds = []; const messageIds: Array<string> = [];
const inMemoryMessages = []; const inMemoryMessages: Array<MessageModel> = [];
const messageCleanup = []; const messageCleanup: Array<Promise<void>> = [];
messages.forEach(dbMessage => { messages.forEach(dbMessage => {
const message = MessageController.register(dbMessage.id, dbMessage); const message = window.MessageController.register(
dbMessage.id,
dbMessage
);
messageIds.push(message.id); messageIds.push(message.id);
inMemoryMessages.push(message); inMemoryMessages.push(message);
messageCleanup.push(message.cleanup()); messageCleanup.push(message.cleanup());
@ -59,11 +63,10 @@
} }
window.SignalContext.log.info('destroyExpiredMessages: complete'); window.SignalContext.log.info('destroyExpiredMessages: complete');
checkExpiringMessages(); this.update();
} }
let timeout; private async checkExpiringMessages() {
async function checkExpiringMessages() {
window.SignalContext.log.info( window.SignalContext.log.info(
'checkExpiringMessages: checking for expiring messages' 'checkExpiringMessages: checking for expiring messages'
); );
@ -94,19 +97,10 @@
).toISOString()}; waiting ${wait} ms before clearing` ).toISOString()}; waiting ${wait} ms before clearing`
); );
clearTimeout(timeout); clearTimeoutIfNecessary(this.timeout);
timeout = setTimeout(destroyExpiredMessages, wait); this.timeout = setTimeout(this.destroyExpiredMessages.bind(this), wait);
} }
const debouncedCheckExpiringMessages = _.debounce( }
checkExpiringMessages,
1000
);
Whisper.ExpiringMessagesListener = { export const expiringMessagesDeletionService =
init(events) { new ExpiringMessagesDeletionService();
checkExpiringMessages();
events.on('timetravel', debouncedCheckExpiringMessages);
},
update: debouncedCheckExpiringMessages,
};
})();

View file

@ -0,0 +1,80 @@
// Copyright 2019-2022 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
import { debounce } from 'lodash';
import { clearTimeoutIfNecessary } from '../util/clearTimeoutIfNecessary';
import { DAY } from '../util/durations';
async function eraseTapToViewMessages() {
try {
window.SignalContext.log.info(
'eraseTapToViewMessages: Loading messages...'
);
const messages =
await window.Signal.Data.getTapToViewMessagesNeedingErase();
await Promise.all(
messages.map(async fromDB => {
const message = window.MessageController.register(fromDB.id, fromDB);
window.SignalContext.log.info(
'eraseTapToViewMessages: erasing message contents',
message.idForLogging()
);
await message.eraseContents();
})
);
} catch (error) {
window.SignalContext.log.error(
'eraseTapToViewMessages: Error erasing messages',
error && error.stack ? error.stack : error
);
}
window.SignalContext.log.info('eraseTapToViewMessages: complete');
}
class TapToViewMessagesDeletionService {
public update: typeof this.checkTapToViewMessages;
private timeout?: ReturnType<typeof setTimeout>;
constructor() {
this.update = debounce(this.checkTapToViewMessages, 1000);
}
private async checkTapToViewMessages() {
const receivedAt =
await window.Signal.Data.getNextTapToViewMessageTimestampToAgeOut();
if (!receivedAt) {
return;
}
const nextCheck = receivedAt + 30 * DAY;
window.SignalContext.log.info(
'checkTapToViewMessages: next check at',
new Date(nextCheck).toISOString()
);
let wait = nextCheck - Date.now();
// In the past
if (wait < 0) {
wait = 0;
}
// Too far in the future, since it's limited to a 32-bit value
if (wait > 2147483647) {
wait = 2147483647;
}
clearTimeoutIfNecessary(this.timeout);
this.timeout = setTimeout(async () => {
await eraseTapToViewMessages();
this.update();
}, wait);
}
}
export const tapToViewMessagesDeletionService =
new TapToViewMessagesDeletionService();

View file

@ -27,6 +27,8 @@ import {
} from 'lodash'; } from 'lodash';
import { deleteExternalFiles } from '../types/Conversation'; import { deleteExternalFiles } from '../types/Conversation';
import { expiringMessagesDeletionService } from '../services/expiringMessagesDeletion';
import { tapToViewMessagesDeletionService } from '../services/tapToViewMessagesDeletionService';
import * as Bytes from '../Bytes'; import * as Bytes from '../Bytes';
import { CURRENT_SCHEMA_VERSION } from '../../js/modules/types/message'; import { CURRENT_SCHEMA_VERSION } from '../../js/modules/types/message';
import { createBatcher } from '../util/batcher'; import { createBatcher } from '../util/batcher';
@ -1090,8 +1092,8 @@ async function saveMessage(
jobToInsert: options.jobToInsert && formatJobForInsert(options.jobToInsert), jobToInsert: options.jobToInsert && formatJobForInsert(options.jobToInsert),
}); });
window.Whisper.ExpiringMessagesListener.update(); expiringMessagesDeletionService.update();
window.Whisper.TapToViewMessagesListener.update(); tapToViewMessagesDeletionService.update();
return id; return id;
} }
@ -1105,8 +1107,8 @@ async function saveMessages(
options options
); );
window.Whisper.ExpiringMessagesListener.update(); expiringMessagesDeletionService.update();
window.Whisper.TapToViewMessagesListener.update(); tapToViewMessagesDeletionService.update();
} }
async function removeMessage(id: string) { async function removeMessage(id: string) {

View file

@ -1,4 +1,4 @@
// Copyright 2021 Signal Messenger, LLC // Copyright 2021-2022 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only // SPDX-License-Identifier: AGPL-3.0-only
import { omit } from 'lodash'; import { omit } from 'lodash';
@ -8,6 +8,8 @@ import { hasErrors } from '../state/selectors/message';
import { readReceiptsJobQueue } from '../jobs/readReceiptsJobQueue'; import { readReceiptsJobQueue } from '../jobs/readReceiptsJobQueue';
import { readSyncJobQueue } from '../jobs/readSyncJobQueue'; import { readSyncJobQueue } from '../jobs/readSyncJobQueue';
import { notificationService } from '../services/notifications'; import { notificationService } from '../services/notifications';
import { expiringMessagesDeletionService } from '../services/expiringMessagesDeletion';
import { tapToViewMessagesDeletionService } from '../services/tapToViewMessagesDeletionService';
import { isGroup } from './whatTypeOfConversation'; import { isGroup } from './whatTypeOfConversation';
import * as log from '../logging/log'; import * as log from '../logging/log';
import { getConversationIdForLogging } from './idForLogging'; import { getConversationIdForLogging } from './idForLogging';
@ -137,8 +139,8 @@ export async function markConversationRead(
); );
} }
window.Whisper.ExpiringMessagesListener.update(); expiringMessagesDeletionService.update();
window.Whisper.TapToViewMessagesListener.update(); tapToViewMessagesDeletionService.update();
return true; return true;
} }

10
ts/window.d.ts vendored
View file

@ -556,16 +556,6 @@ export type WhisperType = {
events: Backbone.Events; events: Backbone.Events;
activeConfirmationView: WhatIsThis; activeConfirmationView: WhatIsThis;
ExpiringMessagesListener: {
init: (events: Backbone.Events) => void;
update: () => void;
};
TapToViewMessagesListener: {
nextCheck: null | number;
init: (events: Backbone.Events) => void;
update: () => void;
};
// Backbone views // Backbone views
// Modernized // Modernized