From 8fdfda64829d301c6cf8e8ba8c001f38071cc37a Mon Sep 17 00:00:00 2001
From: Baptiste Arnaud <contact@baptiste-arnaud.fr>
Date: Mon, 2 May 2022 07:24:24 -0700
Subject: [PATCH] =?UTF-8?q?fix(viewer):=20=F0=9F=90=9B=20Make=20custom=20d?=
 =?UTF-8?q?omain=20fetching=20more=20predictable?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/builder/components/share/ShareContent.tsx | 14 +++++++++-----
 apps/viewer/pages/[[...publicId]].tsx          |  2 +-
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/apps/builder/components/share/ShareContent.tsx b/apps/builder/components/share/ShareContent.tsx
index 82695610f2..36a61eba54 100644
--- a/apps/builder/components/share/ShareContent.tsx
+++ b/apps/builder/components/share/ShareContent.tsx
@@ -41,6 +41,14 @@ export const ShareContent = () => {
     : ''
   const isPublished = isDefined(typebot?.publishedTypebotId)
 
+  const handlePathnameChange = (pathname: string) => {
+    if (!typebot?.customDomain) return
+    const existingHost = typebot.customDomain?.split('/')[0]
+    const newDomain =
+      pathname === '' ? existingHost : existingHost + '/' + pathname
+    handleCustomDomainChange(newDomain)
+  }
+
   const handleCustomDomainChange = (customDomain: string | null) =>
     updateOnBothTypebots({ customDomain })
 
@@ -65,11 +73,7 @@ export const ShareContent = () => {
               <EditableUrl
                 hostname={'https://' + typebot.customDomain.split('/')[0]}
                 pathname={typebot.customDomain.split('/')[1]}
-                onPathnameChange={(pathname) =>
-                  handleCustomDomainChange(
-                    typebot.customDomain?.split('/')[0] + '/' + pathname
-                  )
-                }
+                onPathnameChange={handlePathnameChange}
               />
               <IconButton
                 icon={<TrashIcon />}
diff --git a/apps/viewer/pages/[[...publicId]].tsx b/apps/viewer/pages/[[...publicId]].tsx
index 179598c021..2a648c7994 100644
--- a/apps/viewer/pages/[[...publicId]].tsx
+++ b/apps/viewer/pages/[[...publicId]].tsx
@@ -50,7 +50,7 @@ const getTypebotFromPublicId = async (publicId?: string) => {
 
 const getTypebotFromCustomDomain = async (customDomain: string) => {
   const typebot = await prisma.publicTypebot.findFirst({
-    where: { customDomain: { contains: customDomain } },
+    where: { customDomain },
   })
   if (isNotDefined(typebot)) return null
   return omit(typebot as unknown as PublicTypebot, 'createdAt', 'updatedAt')