Added user ID requirement to notify tool to prevent mass-notifying all users.

src/alerts.js

@@ -616,7 +616,8 @@ export async function notify(sceneIds, options = {}) {
 		return acc;
 	}, {});
 
-	const alerts = rawAlerts.map((alert) => ({
+	const alerts = rawAlerts
+		.map((alert) => ({
 			id: alert.id,
 			userId: alert.user_id,
 			username: alert.username,
@@ -634,7 +635,8 @@ export async function notify(sceneIds, options = {}) {
 			entityIds: alertEntityIdsByAlertId[alert.id] || [],
 			matches: alertsMatchesByAlertId[alert.id] || [],
 			stashIds: alertsStashesByAlertId[alert.id] || [],
-	}));
+		}))
+		.filter((alert) => options.spoofUserIds ? options.spoofUserIds.includes(alert.userId) : true);
 
 	const curatedScenes = scenes
 		.filter((scene) => scene.isNew || options.spoofNew)

tools/notify.js

@@ -9,14 +9,25 @@ const { argv } = yargs(process.argv.slice(2))
 	.option('scenes', {
 		type: 'array',
 		alias: ['scene-ids'],
+	})
+	.option('user-ids', {
+		type: 'array',
+		alias: ['users', 'user'],
 	});
 
 async function triggerNotify() {
-	if (!argv.scenes.length) {
+	if (!argv.scenes?.length) {
 		throw new Error('Must specify at least one scene ID');
 	}
 
-	await notify(argv.scenes, { spoofNew: true });
+	if (!argv.userIds?.length) {
+		throw new Error('Must specify at least one user ID');
+	}
+
+	await notify(argv.scenes, {
+		spoofNew: true,
+		spoofUserIds: argv.userIds,
+	});
 
 	await knex.destroy();
 	await redis.quit();