Function fastUpsertManyMessages

packages/core/src/message-node.ts:169–243

(data: BaseMessageWithRelations[])

}

export async function fastUpsertManyMessages(data: BaseMessageWithRelations[]) {
	const attachments = new Set<typeof dbAttachments.$inferInsert>();
	const emojis = new Set<typeof dbEmojis.$inferInsert>();
	const reactions = new Set<typeof dbReactions.$inferInsert>();
	const msgs = new Set<typeof dbMessages.$inferInsert>();
	for (const msg of data) {
		const { attachments: a, reactions: r, embeds: e, ...m } = msg;
		msgs.add({
			...messageSchema.parse(m),
			embeds: e,
		});
		if (a) {
			for (const attachment of a) {
				attachments.add(attachmentSchema.parse(attachment));
			}
		}
		if (r) {
			for (const reaction of r) {
				if (!reaction.emoji.id || !reaction.emoji.name) continue;
				emojis.add(emojiSchema.parse(reaction.emoji));
				reactions.add(reactionSchema.parse(reaction));
			}
		}
	}

	if (msgs.size > 0)
		await db
			.insert(dbMessages)
			.values(Array.from(msgs))
			.onDuplicateKeyUpdate({ set: { id: sql.raw('id') } });
	if (attachments.size > 0) {
		void db
			.select()
			.from(dbAttachments)
			.where(
				inArray(
					dbAttachments.id,
					[...attachments].map((a) => a.id),
				),
			)
			.then(async (existingAttachments) => {
				const existingAttachmentIds = new Set(
					existingAttachments.map((a) => a.id),
				);
				for await (const attachment of attachments) {
					if (!existingAttachmentIds.has(attachment.id)) {
						void uploadFileFromUrl({
							url: attachment.url,
							id: attachment.id,
							contentType: attachment.contentType ?? undefined,
							filename: attachment.filename,
						}).then(async (uploaded) => {
							if (!uploaded) return;
							await db.insert(dbAttachments).values(
								attachmentSchema.parse({
									...attachment,
									proxyUrl: `https://cdn.answeroverflow.com/${attachment.id}/${attachment.filename}`,
								}),