multi tenancy stuff

app/core/mutations/set-notification-subscription.ts

@@ -3,10 +3,13 @@ import { z } from "zod";
 
 import db from "../../../db";
 import appLogger from "../../../integrations/logger";
+import { enforceSuperAdminIfNotCurrentOrganization, setDefaultOrganizationId } from "../utils";
 
 const logger = appLogger.child({ mutation: "set-notification-subscription" });
 
 const Body = z.object({
+	organizationId: z.string().optional(),
+	phoneNumberId: z.string(),
 	subscription: z.object({
 		endpoint: z.string(),
 		expirationTime: z.number().nullable(),
@@ -17,22 +20,36 @@ const Body = z.object({
 	}),
 });
 
-export default resolver.pipe(resolver.zod(Body), resolver.authorize(), async ({ subscription }, context) => {
-	const customerId = context.session.userId;
-	try {
-		await db.notificationSubscription.create({
-			data: {
-				customerId,
-				endpoint: subscription.endpoint,
-				expirationTime: subscription.expirationTime,
-				keys_p256dh: subscription.keys.p256dh,
-				keys_auth: subscription.keys.auth,
-			},
+export default resolver.pipe(
+	resolver.zod(Body),
+	resolver.authorize(),
+	setDefaultOrganizationId,
+	enforceSuperAdminIfNotCurrentOrganization,
+	async ({ organizationId, phoneNumberId, subscription }) => {
+		const phoneNumber = await db.phoneNumber.findFirst({
+			where: { id: phoneNumberId, organizationId },
+			include: { organization: true },
 		});
-	} catch (error) {
-		if (error.code !== "P2002") {
-			logger.error(error);
-			// we might want to `throw error`;
+		if (!phoneNumber) {
+			return;
 		}
-	}
-});
+
+		try {
+			await db.notificationSubscription.create({
+				data: {
+					organizationId,
+					phoneNumberId,
+					endpoint: subscription.endpoint,
+					expirationTime: subscription.expirationTime,
+					keys_p256dh: subscription.keys.p256dh,
+					keys_auth: subscription.keys.auth,
+				},
+			});
+		} catch (error) {
+			if (error.code !== "P2002") {
+				logger.error(error);
+				// we might want to `throw error`;
+			}
+		}
+	},
+);