From 5148c60ad8cfb78c20bec36bcf4f2b4cbf35f3e1 Mon Sep 17 00:00:00 2001
From: mortezaei <mortezaei2324@gmail.com>
Date: Sun, 30 Aug 2026 05:22:39 +0330
Subject: [PATCH] feat: implement initial server-side injection of marriage
 data into window global and improve cookie parsing robustness

---
 src/app/layout.tsx     | 26 ++++++++++++++++++++++++++
 src/lib/auth-bridge.ts | 23 +++++++++++++++++++----
 src/lib/cookies.ts     |  8 +++++++-
 3 files changed, 52 insertions(+), 5 deletions(-)
diff --git a/src/app/layout.tsx b/src/app/layout.tsx
index 45a2dfa..bbeb056 100644
--- a/src/app/layout.tsx
+++ b/src/app/layout.tsx
@@ -85,6 +85,24 @@ export default async function RootLayout({
         ? cookieLocale
         : defaultLocale;
 
+  const rawMarriageCookie =
+    cookieStore.get("HABIB_MARRIAGE_DATA")?.value ??
+    cookieStore.get("habib_marriage_data")?.value;
+
+  let initialMarriageJson: string | null = null;
+  if (rawMarriageCookie) {
+    try {
+      const decoded = decodeURIComponent(rawMarriageCookie);
+      JSON.parse(decoded);
+      initialMarriageJson = decoded;
+    } catch {
+      try {
+        JSON.parse(rawMarriageCookie);
+        initialMarriageJson = rawMarriageCookie;
+      } catch {}
+    }
+  }
+
   return (
     <html
       lang={locale}
@@ -99,6 +117,14 @@ export default async function RootLayout({
         />
         <link rel="preconnect" href="https://habibapp.com" />
         <link rel="dns-prefetch" href="https://habibapp.com" />
+        {initialMarriageJson && (
+          <script
+            id="habib-early-marriage-initial-data"
+            dangerouslySetInnerHTML={{
+              __html: `window.__HABIB_MARRIAGE_INITIAL_DATA__ = ${initialMarriageJson}; window.HABIB_MARRIAGE = window.__HABIB_MARRIAGE_INITIAL_DATA__;`,
+            }}
+          />
+        )}
         <script
           id="habib-core-bootstrap"
           dangerouslySetInnerHTML={{
diff --git a/src/lib/auth-bridge.ts b/src/lib/auth-bridge.ts
index ec17c7a..fb4542e 100644
--- a/src/lib/auth-bridge.ts
+++ b/src/lib/auth-bridge.ts
@@ -35,6 +35,9 @@ class AuthBridge {
   }
 
   public getMarriageData(): Record<string, any> | null {
+    if (!this.marriageData) {
+      this.syncFromStorage();
+    }
     return this.marriageData;
   }
 
@@ -48,19 +51,31 @@ class AuthBridge {
       HABIB_COINS?: number;
       HABIB_MARRIAGE?: Record<string, any>;
       __HABIB_MARRIAGE_INITIAL_DATA__?: Record<string, any>;
+      __HABIB_BOOTSTRAP__?: Record<string, any>;
     };
 
-    if (win.HABIB_MARRIAGE || win.__HABIB_MARRIAGE_INITIAL_DATA__) {
+    if (
+      win.HABIB_MARRIAGE ||
+      win.__HABIB_MARRIAGE_INITIAL_DATA__ ||
+      win.__HABIB_BOOTSTRAP__?.marriage
+    ) {
       this.marriageData =
-        win.HABIB_MARRIAGE ?? win.__HABIB_MARRIAGE_INITIAL_DATA__ ?? null;
+        win.HABIB_MARRIAGE ??
+        win.__HABIB_MARRIAGE_INITIAL_DATA__ ??
+        win.__HABIB_BOOTSTRAP__?.marriage ??
+        null;
     } else {
       const rawMarriageCookie =
         getClientCookie("HABIB_MARRIAGE_DATA") ??
         getClientCookie("habib_marriage_data");
       if (rawMarriageCookie) {
         try {
-          this.marriageData = JSON.parse(decodeURIComponent(rawMarriageCookie));
-        } catch (e) {}
+          this.marriageData = JSON.parse(rawMarriageCookie);
+        } catch {
+          try {
+            this.marriageData = JSON.parse(decodeURIComponent(rawMarriageCookie));
+          } catch {}
+        }
       }
     }
 
diff --git a/src/lib/cookies.ts b/src/lib/cookies.ts
index 2bd0a92..53dcebc 100644
--- a/src/lib/cookies.ts
+++ b/src/lib/cookies.ts
@@ -18,7 +18,13 @@ export function getClientCookie(name: string) {
     new RegExp(`(?:^|; )${escapeCookieName(name)}=([^;]*)`),
   );
 
-  return match ? decodeURIComponent(match[1]) : null;
+  if (!match) return null;
+
+  try {
+    return decodeURIComponent(match[1]);
+  } catch {
+    return match[1];
+  }
 }
 
 export function setClientCookie(
