60 lines
1.9 KiB
TypeScript
60 lines
1.9 KiB
TypeScript
|
// Copyright 2024 Signal Messenger, LLC
|
||
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
||
|
|
|
||
|
|
import * as z from 'zod';
|
||
|
|
import type { LoggerType } from '../types/Logging';
|
||
|
|
import { applyNewAvatar } from '../groups';
|
||
|
|
import { isGroupV2 } from '../util/whatTypeOfConversation';
|
||
|
|
import Data from '../sql/Client';
|
||
|
|
|
||
|
|
import type { JOB_STATUS } from './JobQueue';
|
||
|
|
import { JobQueue } from './JobQueue';
|
||
|
|
import { jobQueueDatabaseStore } from './JobQueueDatabaseStore';
|
||
|
|
|
||
|
|
const groupAvatarJobDataSchema = z.object({
|
||
|
|
conversationId: z.string(),
|
||
|
|
newAvatarUrl: z.string().optional(),
|
||
|
|
});
|
||
|
|
|
||
|
|
export type GroupAvatarJobData = z.infer<typeof groupAvatarJobDataSchema>;
|
||
|
|
|
||
|
|
export class GroupAvatarJobQueue extends JobQueue<GroupAvatarJobData> {
|
||
|
|
protected parseData(data: unknown): GroupAvatarJobData {
|
||
|
|
return groupAvatarJobDataSchema.parse(data);
|
||
|
|
}
|
||
|
|
|
||
|
|
protected async run(
|
||
|
|
{ data }: Readonly<{ data: GroupAvatarJobData; timestamp: number }>,
|
||
|
|
{ attempt, log }: Readonly<{ attempt: number; log: LoggerType }>
|
||
|
|
): Promise<typeof JOB_STATUS.NEEDS_RETRY | undefined> {
|
||
|
|
const { conversationId, newAvatarUrl } = data;
|
||
|
|
const logId = `groupAvatarJobQueue(${conversationId}, attempt=${attempt})`;
|
||
|
|
|
||
|
|
const convo = window.ConversationController.get(conversationId);
|
||
|
|
if (!convo) {
|
||
|
|
log.warn(`${logId}: dropping ${conversationId}, not found`);
|
||
|
|
return undefined;
|
||
|
|
}
|
||
|
|
|
||
|
|
const { attributes } = convo;
|
||
|
|
if (!isGroupV2(attributes)) {
|
||
|
|
log.warn(`${logId}: dropping ${conversationId}, not a group`);
|
||
|
|
return undefined;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Generate correct attributes patch
|
||
|
|
const patch = await applyNewAvatar(newAvatarUrl, attributes, logId);
|
||
|
|
|
||
|
|
convo.set(patch);
|
||
|
|
await Data.updateConversation(convo.attributes);
|
||
|
|
|
||
|
|
return undefined;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export const groupAvatarJobQueue = new GroupAvatarJobQueue({
|
||
|
|
store: jobQueueDatabaseStore,
|
||
|
|
queueType: 'groupAvatar',
|
||
|
|
maxAttempts: 25,
|
||
|
|
});
|