From 8f9fd7f1497c546be481ef39cbc40c4d26dd0563 Mon Sep 17 00:00:00 2001
From: hamed <genius.ha@gmail.com>
Date: Sun, 28 Sep 2025 18:42:23 +0330
Subject: [PATCH] fix: enhance domain handling in sitemap generation and add
 XML escaping utility

---
 app/sitemap.js   | 18 +++++++++++++++---
 utils/sitemap.js | 16 ++++++++++++++--
 2 files changed, 29 insertions(+), 5 deletions(-)

diff --git a/app/sitemap.js b/app/sitemap.js
index 4de41bb..4758ff2 100644
--- a/app/sitemap.js
+++ b/app/sitemap.js
@@ -13,9 +13,15 @@ import {
 
 // Get current domain from headers
 function getCurrentDomain() {
-    const headersList = headers();
-    const host = headersList.get('host');
-    return host || DOMAIN_CONFIG.MAIN_DOMAIN;
+    try {
+        const headersList = headers();
+        const host = headersList.get('host');
+        return host || DOMAIN_CONFIG.MAIN_DOMAIN;
+    } catch (error) {
+        // Fallback for static generation
+        console.warn('Headers not available during static generation, using main domain');
+        return DOMAIN_CONFIG.MAIN_DOMAIN;
+    }
 }
 
 // Get city data based on domain
@@ -225,12 +231,17 @@ export default async function sitemap() {
         const domain = getCurrentDomain();
         const baseUrl = getBaseUrl(domain);
 
+        console.log(`Generating sitemap for domain: ${domain}, baseUrl: ${baseUrl}`);
+        console.log(`City data length: ${cityData.length}, Specialty data length: ${specialtyData.length}`);
+
         // Get all URLs
         const staticPages = getStaticPages(baseUrl);
         const doctorSearchUrls = getDoctorSearchUrls(domain, baseUrl);
         const clinicSearchUrls = getClinicSearchUrls(domain, baseUrl);
         const specialtyUrls = getSpecialtyUrls(domain, baseUrl);
 
+        console.log(`Generated URLs - Static: ${staticPages.length}, Doctors: ${doctorSearchUrls.length}, Clinics: ${clinicSearchUrls.length}, Specialties: ${specialtyUrls.length}`);
+
         // Combine all URLs
         const allUrls = [
             ...staticPages,
@@ -244,6 +255,7 @@ export default async function sitemap() {
             arr.findIndex(i => i.url === item.url) === index
         );
 
+        console.log(`Total unique URLs: ${uniqueUrls.length}`);
         return uniqueUrls;
     } catch (error) {
         console.error('Error generating sitemap:', error);
diff --git a/utils/sitemap.js b/utils/sitemap.js
index bc74e98..bfc9dc4 100644
--- a/utils/sitemap.js
+++ b/utils/sitemap.js
@@ -6,7 +6,7 @@ export const DOMAIN_CONFIG = {
 };
 
 export function isMainDomain(domain) {
-    return domain === DOMAIN_CONFIG.MAIN_DOMAIN;
+    return domain === DOMAIN_CONFIG.MAIN_DOMAIN || domain.includes('localhost') || domain.includes('127.0.0.1');
 }
 
 export function encodeUrlParam(param) {
@@ -24,7 +24,19 @@ export function createSitemapUrl(baseUrl, path, params = {}) {
     });
 
     const queryString = searchParams.toString();
-    return queryString ? `${url}?${queryString}` : url;
+    const fullUrl = queryString ? `${url}?${queryString}` : url;
+
+    // Escape XML entities
+    return escapeXml(fullUrl);
+}
+
+export function escapeXml(str) {
+    return str
+        .replace(/&/g, '&amp;')
+        .replace(/</g, '&lt;')
+        .replace(/>/g, '&gt;')
+        .replace(/"/g, '&quot;')
+        .replace(/'/g, '&apos;');
 }
 
 export function createSitemapEntry(url, priority = 0.8, changeFrequency = 'weekly') {