From bda2fc36c8a178dd2a03226d429173159f865f6d Mon Sep 17 00:00:00 2001
From: DIYgod <i@diygod.me>
Date: Wed, 17 Aug 2022 02:56:16 +0100
Subject: [PATCH] feat: custom domain setting check

---
 src/lib/tenant.server.ts | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/lib/tenant.server.ts b/src/lib/tenant.server.ts
index bb06e671..79a1dcfa 100644
--- a/src/lib/tenant.server.ts
+++ b/src/lib/tenant.server.ts
@@ -1,4 +1,5 @@
 import { OUR_DOMAIN } from "./env"
+import { getSite } from "~/models/site.model"
 
 export const getTenant = async (request: Request, search: URLSearchParams) => {
   const host = request.headers.get("host")
@@ -30,7 +31,13 @@ export const getTenant = async (request: Request, search: URLSearchParams) => {
         },
       )
       const txt = await res.json()
-      return txt?.Answer?.[0]?.data.replace(/^"|"$/g, "")
+      const tenant = txt?.Answer?.[0]?.data.replace(/^"|"$/g, "")
+      if (tenant) {
+        const site = await getSite(tenant)
+        if (site.custom_domain === host) {
+          return tenant
+        }
+      }
     }
   }
   return