simplify embed

dubinc · Jan 31, 2025 · 74aa7c7 · 74aa7c7
1 parent c5cec6d
commit 74aa7c7
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 16 deletions.
diff --git a/apps/web/app/api/tokens/embed/route.ts b/apps/web/app/api/tokens/embed/route.ts
@@ -44,8 +44,7 @@ export const POST = withWorkspace(
 
     const response = await embedToken.create({
       programId,
-      partnerId,
-      tenantId,
+      partnerId: programEnrollment.partnerId,
     });
 
     return NextResponse.json(EmbedTokenSchema.parse(response), {

diff --git a/apps/web/app/app.dub.co/embed/utils.ts b/apps/web/app/app.dub.co/embed/utils.ts
@@ -4,17 +4,16 @@ import { prisma } from "@dub/prisma";
 import { notFound } from "next/navigation";
 
 export const getEmbedData = async (token: string) => {
-  const { programId, partnerId, tenantId } =
-    (await embedToken.get(token)) ?? {};
+  const { programId, partnerId } = (await embedToken.get(token)) ?? {};
 
-  if (!programId || (!tenantId && !partnerId)) {
+  if (!programId || !partnerId) {
     notFound();
   }
 
   const programEnrollment = await prisma.programEnrollment.findUnique({
-    where: tenantId
-      ? { tenantId_programId: { tenantId, programId } }
-      : { partnerId_programId: { partnerId: partnerId!, programId } },
+    where: {
+      partnerId_programId: { partnerId, programId },
+    },
     include: {
       links: true,
       program: true,

diff --git a/apps/web/lib/embed/auth.ts b/apps/web/lib/embed/auth.ts
@@ -54,10 +54,10 @@ export const withEmbedToken = (handler: WithEmbedTokenHandler) => {
           });
         }
 
-        const { programId, partnerId, tenantId } =
+        const { programId, partnerId } =
           (await embedToken.get(tokenFromCookie)) ?? {};
 
-        if (!programId || (!partnerId && !tenantId)) {
+        if (!programId || !partnerId) {
           throw new DubApiError({
             code: "unauthorized",
             message: "Invalid embed public token.",
@@ -85,9 +85,9 @@ export const withEmbedToken = (handler: WithEmbedTokenHandler) => {
 
         const programEnrollment =
           await prisma.programEnrollment.findUniqueOrThrow({
-            where: tenantId
-              ? { tenantId_programId: { tenantId, programId } }
-              : { partnerId_programId: { partnerId: partnerId!, programId } },
+            where: {
+              partnerId_programId: { partnerId, programId },
+            },
             include: {
               links: true,
               program: true,

diff --git a/apps/web/lib/embed/embed-token.ts b/apps/web/lib/embed/embed-token.ts
@@ -1,14 +1,15 @@
 import { redis } from "@/lib/upstash";
 import { createId } from "../api/utils";
-import z from "../zod";
-import { createEmbedTokenSchema } from "../zod/schemas/token";
 import {
   EMBED_PUBLIC_TOKEN_EXPIRY,
   EMBED_PUBLIC_TOKEN_LENGTH,
   EMBED_PUBLIC_TOKEN_PREFIX,
 } from "./constants";
 
-type EmbedTokenProps = z.infer<typeof createEmbedTokenSchema>;
+interface EmbedTokenProps {
+  programId: string;
+  partnerId: string;
+}
 
 class EmbedToken {
   async create(props: EmbedTokenProps) {