zkldi_Tachi/server/src/lib/notifications/notification-wrappers.ts

import { BulkSendNotification, SendNotification } from "./notifications";
import db from "external/mongo/db";
import { FormatGame } from "tachi-common";
import type { Game, integer, Playtype, UserDocument } from "tachi-common";

/**
 * A utility wrapper for sending the RIVALED_BY notification.
 *
 * @param toUserID - The user to send this notification to.
 * @param fromUser - The user who rivalled them.
 * @param game - The game they rivalled them on.
 * @param playtype - The playtype they rivalled them on.
 */
export async function SendSetRivalNotification(
	toUserID: integer,
	fromUser: UserDocument,
	game: Game,
	playtype: Playtype
) {
	const alreadyBeenPinged = await db.notifications.findOne({
		sentTo: toUserID,
		"body.type": "RIVALED_BY",
		"body.content": {
			userID: fromUser.id,
			game,
			playtype,
		},
	});

	if (alreadyBeenPinged) {
		return;
	}

	return SendNotification(
		`${fromUser.username} just added you as a rival for ${FormatGame(game, playtype)}`,
		toUserID,
		{
			type: "RIVALED_BY",
			content: {
				userID: fromUser.id,
				game,
				playtype,
			},
		}
	);
}

export async function SendSiteAnnouncementNotification(
	title: string,
	maybeGame?: Game,
	maybePlaytype?: Playtype
) {
	let toUserIDs = [];

	if (maybeGame && maybePlaytype) {
		toUserIDs = (
			await db["game-stats"].find(
				{ game: maybeGame, playtype: maybePlaytype },
				{ projection: { userID: 1 } }
			)
		).map((e) => e.userID);
	} else {
		toUserIDs = (await db.users.find({}, { projection: { id: 1 } })).map((e) => e.id);
	}

	return BulkSendNotification(title, toUserIDs, {
		type: "SITE_ANNOUNCEMENT",
		content: {},
	});
}