@eventcatalog/core 3.0.0-beta.8 → 3.0.0

package/eventcatalog/src/components/Grids/DomainGrid.tsx

@@ -8,10 +8,13 @@ import {
   CircleStackIcon,
   ChevronDownIcon,
   ChevronUpIcon,
-  ArrowsPointingOutIcon,
+  ArrowTopRightOnSquareIcon,
+  ArrowLongRightIcon,
+  ArrowLongLeftIcon,
 } from '@heroicons/react/24/outline';
 import { buildUrl } from '@utils/url-builder';
-import { BoxIcon, ArrowRight, ArrowLeft } from 'lucide-react';
+import { BoxIcon } from 'lucide-react';
+import { getSpecUrl, getSpecIcon, getSpecLabel, getServiceSpecifications } from './specification-utils';
 
 // ============================================
 // Types
@@ -32,14 +35,14 @@ const getMessageIcon = (collection: string) => {
     case 'commands':
       return { Icon: ChatBubbleLeftIcon, color: 'blue' };
     case 'queries':
-      return { Icon: MagnifyingGlassIcon, color: 'green' };
+      return { Icon: MagnifyingGlassIcon, color: 'emerald' };
     default:
       return { Icon: BoltIcon, color: 'gray' };
   }
 };
 
 // ============================================
-// Simple Sub-components
+// Sub-components
 // ============================================
 
 const EntityBadge = memo(({ entity }: { entity: any }) => {
@@ -49,177 +52,269 @@ const EntityBadge = memo(({ entity }: { entity: any }) => {
   return (
     <a
       href={buildUrl(`/docs/entities/${id}`)}
-      className="inline-flex items-center gap-1.5 px-2.5 py-1 bg-purple-100 border border-purple-300 rounded-md text-xs font-medium hover:bg-purple-200 transition-colors"
+      className="inline-flex items-center gap-2 px-3 py-2 bg-white border border-gray-200 rounded-lg text-sm font-medium text-gray-700 hover:bg-gray-50 hover:border-gray-300 transition-all shadow-sm"
     >
-      <BoxIcon className="h-3.5 w-3.5 text-purple-600" />
-      <span className="text-purple-800">{name}</span>
+      <BoxIcon className="h-4 w-4 text-purple-500" />
+      <span>{name}</span>
     </a>
   );
 });
 
-const MessageBadge = memo(({ message }: { message: any }) => {
+const MessageLink = memo(({ message }: { message: any }) => {
   const data = message?.data || message;
   const collection = message?.collection || 'events';
   const { Icon, color } = getMessageIcon(collection);
   const id = data?.id || message?.id;
   const name = data?.name || data?.id || id;
-  const version = data?.version;
+  const version = data?.version || message?.data?.version || 'latest';
+
+  const iconStyles: Record<string, string> = {
+    orange: 'text-orange-500',
+    blue: 'text-blue-500',
+    emerald: 'text-emerald-500',
+    gray: 'text-gray-500',
+  };
 
   return (
     <a
       href={buildUrl(`/docs/${collection}/${id}/${version}`)}
-      className="flex items-center gap-1.5 px-2 py-1 bg-white border border-gray-200 rounded text-[11px] font-medium hover:bg-gray-50 transition-colors"
+      className="flex items-center gap-2 py-1.5 text-sm text-gray-700 hover:text-gray-900 transition-colors group"
     >
-      <Icon className={`h-3 w-3 text-${color}-500`} />
-      <span className="text-gray-700 truncate max-w-[120px]">{name}</span>
+      <Icon className={`h-4 w-4 flex-shrink-0 ${iconStyles[color]}`} />
+      <span className="group-hover:underline">{name}</span>
+      <span className="text-xs text-gray-400">v{version}</span>
     </a>
   );
 });
 
-const ContainerBadge = memo(({ container, type }: { container: any; type: 'reads' | 'writes' }) => {
+const SpecificationBadge = memo(
+  ({ spec, serviceId, serviceVersion }: { spec: any; serviceId: string; serviceVersion: string }) => {
+    return (
+      <a
+        href={getSpecUrl(spec, serviceId, serviceVersion)}
+        className="inline-flex items-center gap-1.5 px-2 py-1.5 bg-white border border-gray-200 rounded-lg text-xs font-medium text-gray-700 hover:bg-gray-50 hover:border-gray-300 transition-all shadow-sm"
+      >
+        <img src={buildUrl(`/icons/${getSpecIcon(spec.type)}.svg`, true)} alt={`${spec.type} icon`} className="h-3.5 w-3.5" />
+        <span>{getSpecLabel(spec.type)}</span>
+      </a>
+    );
+  }
+);
+
+const ContainerLink = memo(({ container, type }: { container: any; type: 'reads' | 'writes' }) => {
   const data = container?.data || container;
   const id = data?.id || container?.id;
   const name = data?.name || id;
-  const version = data?.version;
-  const colorClass = type === 'reads' ? 'orange' : 'purple';
+  const version = data?.version || 'latest';
 
   return (
     <a
       href={buildUrl(`/docs/containers/${id}/${version}`)}
-      className={`inline-flex items-center gap-1.5 px-2 py-1 bg-${colorClass}-100 border border-${colorClass}-300 rounded text-[11px] font-medium hover:bg-${colorClass}-200 transition-colors`}
+      className="flex items-center gap-2 py-1.5 text-sm text-gray-700 hover:text-gray-900 transition-colors group"
     >
-      <CircleStackIcon className={`h-3 w-3 text-${colorClass}-600`} />
-      <span className={`text-${colorClass}-800`}>{name}</span>
+      <CircleStackIcon className={`h-4 w-4 ${type === 'reads' ? 'text-amber-500' : 'text-violet-500'}`} />
+      <span className="group-hover:underline">{name}</span>
     </a>
   );
 });
 
+// Searchable scrollable box component
+const SearchableBox = memo(
+  ({
+    title,
+    icon: Icon,
+    iconColor,
+    items,
+    renderItem,
+    emptyText = '—',
+  }: {
+    title: string;
+    icon: any;
+    iconColor: string;
+    items: any[];
+    renderItem: (item: any, idx: number) => React.ReactNode;
+    emptyText?: string;
+  }) => {
+    const [search, setSearch] = useState('');
+
+    const filteredItems = items.filter((item) => {
+      if (!search) return true;
+      const data = item?.data || item;
+      const name = data?.name || data?.id || '';
+      return name.toLowerCase().includes(search.toLowerCase());
+    });
+
+    return (
+      <div className="flex flex-col">
+        <div className="flex items-center gap-2 mb-2">
+          <div className="flex items-center gap-1.5 flex-shrink-0">
+            <Icon className={`h-4 w-4 ${iconColor}`} />
+            <h4 className="text-xs font-semibold text-gray-700 uppercase tracking-wide">{title}</h4>
+            <span className="text-[10px] text-gray-400 font-medium">({items.length})</span>
+          </div>
+          {items.length > 0 && (
+            <input
+              type="text"
+              placeholder="Search..."
+              value={search}
+              onChange={(e) => setSearch(e.target.value)}
+              className="flex-1 px-2 py-0.5 text-xs border border-gray-200 rounded focus:outline-none focus:border-gray-300"
+              onClick={(e) => e.stopPropagation()}
+            />
+          )}
+        </div>
+        {items.length > 0 ? (
+          <div className="space-y-0.5 max-h-32 overflow-y-auto pr-1">
+            {filteredItems.length > 0 ? (
+              filteredItems.map((item, idx) => renderItem(item, idx))
+            ) : (
+              <p className="text-xs text-gray-400 italic">No matches</p>
+            )}
+          </div>
+        ) : (
+          <p className="text-xs text-gray-300">{emptyText}</p>
+        )}
+      </div>
+    );
+  }
+);
+
+// Expanded content for service card
+const ServiceExpandedContent = memo(
+  ({ receives, sends, readsFrom, writesTo }: { receives: any[]; sends: any[]; readsFrom: any[]; writesTo: any[] }) => {
+    const hasMessages = receives.length > 0 || sends.length > 0;
+    const hasContainers = readsFrom.length > 0 || writesTo.length > 0;
+
+    return (
+      <div className="border-t border-gray-100 px-4 py-3 space-y-4">
+        {/* Messages Row */}
+        {hasMessages && (
+          <div className="grid grid-cols-2 gap-x-6">
+            <SearchableBox
+              title="Receives"
+              icon={ArrowLongRightIcon}
+              iconColor="text-blue-400"
+              items={receives}
+              renderItem={(msg, idx) => {
+                const msgId = msg?.data?.id || msg?.id;
+                return msgId ? <MessageLink key={`${msgId}-${idx}`} message={msg} /> : null;
+              }}
+            />
+            <SearchableBox
+              title="Sends"
+              icon={ArrowLongLeftIcon}
+              iconColor="text-emerald-400 rotate-180"
+              items={sends}
+              renderItem={(msg, idx) => {
+                const msgId = msg?.data?.id || msg?.id;
+                return msgId ? <MessageLink key={`${msgId}-${idx}`} message={msg} /> : null;
+              }}
+            />
+          </div>
+        )}
+
+        {/* Data Row */}
+        {hasContainers && (
+          <div className="grid grid-cols-2 gap-x-6 pt-3 border-t border-gray-100">
+            <SearchableBox
+              title="Reads"
+              icon={CircleStackIcon}
+              iconColor="text-amber-400"
+              items={readsFrom}
+              renderItem={(container, idx) => {
+                const containerId = container?.data?.id || container?.id;
+                return containerId ? <ContainerLink key={`${containerId}-${idx}`} container={container} type="reads" /> : null;
+              }}
+            />
+            <SearchableBox
+              title="Writes"
+              icon={CircleStackIcon}
+              iconColor="text-violet-400"
+              items={writesTo}
+              renderItem={(container, idx) => {
+                const containerId = container?.data?.id || container?.id;
+                return containerId ? <ContainerLink key={`${containerId}-${idx}`} container={container} type="writes" /> : null;
+              }}
+            />
+          </div>
+        )}
+      </div>
+    );
+  }
+);
+
 const ServiceCard = memo(({ service }: { service: any }) => {
   const data = service?.data || service;
+  const [isCollapsed, setIsCollapsed] = useState(true);
+
   if (!data?.id) return null;
 
   const receives = data.receives || [];
   const sends = data.sends || [];
   const readsFrom = data.readsFrom || [];
   const writesTo = data.writesTo || [];
+  const specifications = getServiceSpecifications(data);
   const hasMessages = receives.length > 0 || sends.length > 0;
   const hasContainers = readsFrom.length > 0 || writesTo.length > 0;
+  const hasSpecs = specifications.length > 0;
+  const hasContent = hasMessages || hasContainers;
 
   return (
-    <div className="bg-white border-2 border-dashed border-pink-400 rounded-lg p-4">
+    <div className="bg-white border border-gray-200 rounded-xl shadow-sm hover:shadow-md transition-shadow">
       {/* Service Header */}
-      <div className="flex items-center justify-between">
-        <a
-          href={buildUrl(`/architecture/services/${data.id}/${data.version}`)}
-          className="flex items-center gap-2 hover:underline"
-        >
-          <ServerIcon className="h-5 w-5 text-pink-500" />
-          <span className="font-semibold text-gray-900">{data.name || data.id}</span>
-          <span className="text-xs text-gray-500">v{data.version}</span>
-        </a>
-        <a
-          href={buildUrl(`/architecture/services/${data.id}/${data.version}`)}
-          className="p-1 hover:bg-pink-100 rounded-md transition-colors duration-200"
-          title="Expand service architecture"
-          onClick={(e) => e.stopPropagation()}
-        >
-          <ArrowsPointingOutIcon className="h-4 w-4 text-gray-500 hover:text-pink-600" />
-        </a>
-      </div>
-
-      {data.summary && <p className="mt-2 text-sm text-gray-600 line-clamp-2">{data.summary}</p>}
-
-      {/* Message Flow Diagram */}
-      {hasMessages && (
-        <div className="mt-4 flex items-stretch gap-3">
-          {/* Receives (Inbound) */}
-          <div className="flex-1 bg-blue-50 border border-blue-200 rounded-lg p-3">
-            <div className="flex items-center gap-1.5 mb-2">
-              <ArrowRight className="h-3.5 w-3.5 text-blue-500" />
-              <span className="text-xs font-semibold text-blue-700 uppercase">Inbound Messages</span>
-              <span className="text-xs text-blue-500">({receives.length})</span>
-            </div>
-            {receives.length > 0 ? (
-              <div className="space-y-1.5">
-                {receives.slice(0, 4).map((msg: any, idx: number) => {
-                  const msgId = msg?.data?.id || msg?.id;
-                  return msgId ? <MessageBadge key={`${msgId}-${idx}`} message={msg} /> : null;
-                })}
-                {receives.length > 4 && <p className="text-[10px] text-gray-500 text-center">+{receives.length - 4} more</p>}
-              </div>
-            ) : (
-              <p className="text-[10px] text-gray-400 italic">No incoming messages</p>
-            )}
+      <div
+        onClick={() => hasContent && setIsCollapsed(!isCollapsed)}
+        className={`flex items-center justify-between px-4 py-3 ${hasContent ? 'cursor-pointer hover:bg-gray-50' : ''} transition-colors`}
+      >
+        <div className="flex items-center gap-2.5">
+          <div className="flex items-center justify-center w-8 h-8 bg-pink-100 rounded-lg">
+            <ServerIcon className="h-4 w-4 text-pink-600" />
           </div>
-
-          {/* Service Icon (Center) */}
-          <div className="flex items-center">
-            <div className="bg-pink-100 border-2 border-pink-300 rounded-lg p-3">
-              <ServerIcon className="h-6 w-6 text-pink-500" />
+          <div>
+            <div className="flex items-center gap-2">
+              <span className="font-semibold text-gray-900">{data.name || data.id}</span>
+              <span className="text-[11px] text-gray-500 font-medium bg-gray-100 px-1.5 py-0.5 rounded">v{data.version}</span>
             </div>
-          </div>
-
-          {/* Sends (Outbound) */}
-          <div className="flex-1 bg-green-50 border border-green-200 rounded-lg p-3">
-            <div className="flex items-center gap-1.5 mb-2">
-              <ArrowLeft className="h-3.5 w-3.5 text-green-500" />
-              <span className="text-xs font-semibold text-green-700 uppercase">Outbound Messages</span>
-              <span className="text-xs text-green-500">({sends.length})</span>
-            </div>
-            {sends.length > 0 ? (
-              <div className="space-y-1.5">
-                {sends.slice(0, 4).map((msg: any, idx: number) => {
-                  const msgId = msg?.data?.id || msg?.id;
-                  return msgId ? <MessageBadge key={`${msgId}-${idx}`} message={msg} /> : null;
-                })}
-                {sends.length > 4 && <p className="text-[10px] text-gray-500 text-center">+{sends.length - 4} more</p>}
-              </div>
-            ) : (
-              <p className="text-[10px] text-gray-400 italic">No outgoing messages</p>
-            )}
+            {data.summary && <p className="text-xs text-gray-500 line-clamp-1 mt-0.5 max-w-md">{data.summary}</p>}
           </div>
         </div>
-      )}
-
-      {/* Container Relationships */}
-      {hasContainers && (
-        <div className="mt-4 pt-4 border-t border-gray-200 grid grid-cols-2 gap-4">
-          {/* Reads From */}
-          {readsFrom.length > 0 && (
-            <div>
-              <div className="flex items-center gap-1.5 mb-2">
-                <CircleStackIcon className="h-3.5 w-3.5 text-orange-500" />
-                <span className="text-xs font-semibold text-gray-700">Reads from</span>
-              </div>
-              <div className="flex flex-wrap gap-1.5">
-                {readsFrom.slice(0, 3).map((container: any, idx: number) => {
-                  const containerId = container?.data?.id || container?.id;
-                  return containerId ? <ContainerBadge key={`${containerId}-${idx}`} container={container} type="reads" /> : null;
-                })}
-                {readsFrom.length > 3 && <span className="text-[10px] text-gray-500">+{readsFrom.length - 3} more</span>}
-              </div>
+        <div className="flex items-center gap-2">
+          {/* Specs in header - always visible */}
+          {hasSpecs && (
+            <div className="flex items-center gap-1.5">
+              {specifications.map((spec: any, idx: number) => (
+                <a
+                  key={`${spec.type}-${idx}`}
+                  href={getSpecUrl(spec, data.id, data.version)}
+                  onClick={(e) => e.stopPropagation()}
+                  className="flex items-center gap-1 px-2 py-1 bg-gray-100 hover:bg-gray-200 rounded text-xs text-gray-600 hover:text-gray-900 transition-colors"
+                  title={getSpecLabel(spec.type)}
+                >
+                  <img src={buildUrl(`/icons/${getSpecIcon(spec.type)}.svg`, true)} alt="" className="h-3.5 w-3.5" />
+                  <span className="hidden sm:inline">{getSpecLabel(spec.type)}</span>
+                </a>
+              ))}
             </div>
           )}
-
-          {/* Writes To */}
-          {writesTo.length > 0 && (
-            <div>
-              <div className="flex items-center gap-1.5 mb-2">
-                <CircleStackIcon className="h-3.5 w-3.5 text-purple-500" />
-                <span className="text-xs font-semibold text-gray-700">Writes to</span>
-              </div>
-              <div className="flex flex-wrap gap-1.5">
-                {writesTo.slice(0, 3).map((container: any, idx: number) => {
-                  const containerId = container?.data?.id || container?.id;
-                  return containerId ? (
-                    <ContainerBadge key={`${containerId}-${idx}`} container={container} type="writes" />
-                  ) : null;
-                })}
-                {writesTo.length > 3 && <span className="text-[10px] text-gray-500">+{writesTo.length - 3} more</span>}
-              </div>
+          {hasContent && (
+            <div className="p-1.5 text-gray-400">
+              {isCollapsed ? <ChevronDownIcon className="h-4 w-4" /> : <ChevronUpIcon className="h-4 w-4" />}
             </div>
           )}
+          <a
+            href={buildUrl(`/architecture/services/${data.id}/${data.version}`)}
+            onClick={(e) => e.stopPropagation()}
+            className="p-1.5 text-gray-400 hover:text-pink-600 hover:bg-pink-50 rounded-lg transition-colors"
+            title="View service architecture"
+          >
+            <ArrowTopRightOnSquareIcon className="h-4 w-4" />
+          </a>
         </div>
+      </div>
+
+      {/* Expanded Content - Compact Flow */}
+      {!isCollapsed && hasContent && (
+        <ServiceExpandedContent receives={receives} sends={sends} readsFrom={readsFrom} writesTo={writesTo} />
       )}
     </div>
   );
@@ -227,7 +322,7 @@ const ServiceCard = memo(({ service }: { service: any }) => {
 
 const SubdomainSection = memo(({ subdomain }: { subdomain: any }) => {
   const data = subdomain?.data || subdomain;
-  const [isCollapsed, setIsCollapsed] = useState(false);
+  const [isCollapsed, setIsCollapsed] = useState(true);
 
   if (!data?.id) return null;
 
@@ -235,39 +330,59 @@ const SubdomainSection = memo(({ subdomain }: { subdomain: any }) => {
   const entities = data.entities || [];
 
   return (
-    <div className="bg-orange-50 border-2 border-orange-400 rounded-lg p-6">
-      {/* Subdomain Header */}
-      <div className="flex items-center justify-between mb-4">
-        <div className="flex items-center gap-2">
-          <RectangleGroupIcon className="h-5 w-5 text-orange-500" />
-          <h3 className="text-lg font-semibold text-gray-900">{data.name || data.id}</h3>
-          <span className="text-xs text-gray-500">v{data.version}</span>
-          <span className="px-2 py-0.5 bg-orange-200 text-orange-800 text-xs rounded">Subdomain</span>
+    <div className="bg-white border border-gray-200 rounded-xl overflow-hidden shadow-sm">
+      {/* Subdomain Header - Clickable */}
+      <div
+        onClick={() => setIsCollapsed(!isCollapsed)}
+        className={`flex items-center justify-between px-5 py-4 cursor-pointer hover:bg-gray-50 transition-colors ${!isCollapsed ? 'border-b border-gray-200' : ''}`}
+      >
+        <div className="flex items-center gap-3">
+          <div className="flex items-center justify-center w-9 h-9 bg-orange-100 rounded-lg">
+            <RectangleGroupIcon className="h-5 w-5 text-orange-600" />
+          </div>
+          <div>
+            <div className="flex items-center gap-2">
+              <h3 className="text-base font-semibold text-gray-900">{data.name || data.id}</h3>
+              <span className="text-[11px] text-gray-500 font-medium bg-white px-1.5 py-0.5 rounded border border-gray-200">
+                v{data.version}
+              </span>
+              {/* Show counts when collapsed */}
+              {isCollapsed && (services.length > 0 || entities.length > 0) && (
+                <span className="text-[11px] text-gray-400 ml-1">
+                  {services.length > 0 && `${services.length} service${services.length > 1 ? 's' : ''}`}
+                  {services.length > 0 && entities.length > 0 && ', '}
+                  {entities.length > 0 && `${entities.length} entit${entities.length > 1 ? 'ies' : 'y'}`}
+                </span>
+              )}
+            </div>
+            <span className="text-[11px] text-gray-500 font-medium">Subdomain</span>
+          </div>
         </div>
-        <div className="flex gap-2">
-          <button
-            onClick={() => setIsCollapsed(!isCollapsed)}
-            className="p-1 hover:bg-orange-200 rounded-md transition-colors cursor-pointer text-gray-500 hover:text-gray-700"
-          >
+        <div className="flex items-center gap-1">
+          <div className="p-2 text-gray-400">
             {isCollapsed ? <ChevronDownIcon className="h-5 w-5" /> : <ChevronUpIcon className="h-5 w-5" />}
-          </button>
+          </div>
           <a
             href={buildUrl(`/architecture/domains/${data.id}/${data.version}`)}
-            className="p-1 hover:bg-orange-200 rounded-md transition-colors cursor-pointer text-gray-500 hover:text-gray-700"
-            title="Expand domain architecture"
             onClick={(e) => e.stopPropagation()}
+            className="p-2 text-gray-400 hover:text-gray-600 hover:bg-white rounded-lg transition-colors"
+            title="View subdomain architecture"
           >
-            <ArrowsPointingOutIcon className="h-5 w-5" />
+            <ArrowTopRightOnSquareIcon className="h-5 w-5" />
           </a>
         </div>
       </div>
 
       {!isCollapsed && (
-        <>
+        <div className="p-5 space-y-5">
           {/* Subdomain Entities */}
           {entities.length > 0 && (
-            <div className="mb-4">
-              <h4 className="text-xs font-semibold text-gray-500 uppercase tracking-wide mb-2">Entities</h4>
+            <div>
+              <div className="flex items-center gap-2 mb-3">
+                <BoxIcon className="h-4 w-4 text-purple-600" />
+                <h4 className="text-sm font-semibold text-gray-700">Entities</h4>
+                <span className="text-xs text-gray-500 bg-gray-100 px-2 py-0.5 rounded-full font-medium">{entities.length}</span>
+              </div>
               <div className="flex flex-wrap gap-2">
                 {entities.map((entity: any) => {
                   const entityId = entity?.data?.id || entity?.id;
@@ -280,11 +395,14 @@ const SubdomainSection = memo(({ subdomain }: { subdomain: any }) => {
           {/* Subdomain Services */}
           {services.length > 0 && (
             <div>
-              <h4 className="text-xs font-semibold text-gray-500 uppercase tracking-wide mb-2">Services</h4>
-              <div className="grid gap-4 xl:grid-cols-2">
+              <div className="flex items-center gap-2 mb-3">
+                <ServerIcon className="h-4 w-4 text-pink-600" />
+                <h4 className="text-sm font-semibold text-gray-700">Services</h4>
+                <span className="text-xs text-gray-500 bg-gray-100 px-2 py-0.5 rounded-full font-medium">{services.length}</span>
+              </div>
+              <div className="space-y-3">
                 {services.map((service: any) => {
                   const serviceId = service?.data?.id || service?.id;
-                  // Ensure we pass the service down with its messages populated
                   return serviceId ? <ServiceCard key={serviceId} service={service} /> : null;
                 })}
               </div>
@@ -292,9 +410,9 @@ const SubdomainSection = memo(({ subdomain }: { subdomain: any }) => {
           )}
 
           {entities.length === 0 && services.length === 0 && (
-            <p className="text-sm text-gray-500 italic">No entities or services in this subdomain</p>
+            <p className="text-sm text-gray-400 italic text-center py-4">No entities or services in this subdomain</p>
           )}
-        </>
+        </div>
       )}
     </div>
   );
@@ -306,7 +424,7 @@ const SubdomainSection = memo(({ subdomain }: { subdomain: any }) => {
 
 export default function DomainGrid({ domain }: DomainGridProps) {
   const data = domain?.data;
-  if (!data) return <div>No domain data</div>;
+  if (!data) return <div className="text-gray-500">No domain data</div>;
 
   const subdomains = data.domains || [];
   const entities = data.entities || [];
@@ -334,39 +452,43 @@ export default function DomainGrid({ domain }: DomainGridProps) {
   );
 
   return (
-    <div className="space-y-6">
-      {/* Domain Container - Yellow */}
-      <div className="bg-yellow-50 border-2 border-yellow-400 rounded-xl p-6">
-        {/* Domain Header */}
-        <div className="flex items-center justify-between mb-4">
-          <div className="flex items-center gap-3">
-            <RectangleGroupIcon className="h-7 w-7 text-yellow-600" />
-            <h1 className="text-2xl font-bold text-gray-900">{data.name || data.id}</h1>
-            <span className="px-2 py-0.5 bg-yellow-200 text-yellow-800 text-xs font-medium rounded">v{data.version}</span>
-            <span className="px-2 py-0.5 bg-yellow-300 text-yellow-900 text-xs font-medium rounded">Domain</span>
+    <div className="w-full">
+      {/* Domain Header - Doc style */}
+      <div className="border-b border-gray-200 md:pb-4">
+        <div className="flex items-start justify-between">
+          <div>
+            <h2 className="text-2xl md:text-4xl font-bold text-black">{data.name || data.id}</h2>
+            {data.summary && <p className="text-lg pt-2 text-gray-500 font-light">{data.summary}</p>}
           </div>
-          <div className="flex gap-3">
+          <div className="flex items-center gap-2 flex-shrink-0">
             <a
               href={buildUrl(`/docs/domains/${data.id}/${data.version}`)}
-              className="text-sm bg-white px-3 py-1.5 rounded border border-gray-300 hover:bg-gray-50 transition-colors"
+              className="inline-flex items-center gap-1.5 px-3 py-2 text-sm font-medium text-gray-700 bg-white border border-gray-200 rounded-lg hover:bg-gray-50 hover:border-gray-300 transition-all"
             >
               View docs
+              <ArrowTopRightOnSquareIcon className="h-4 w-4 text-gray-400" />
             </a>
             <a
               href={buildUrl(`/visualiser/domains/${data.id}/${data.version}`)}
-              className="text-sm bg-white px-3 py-1.5 rounded border border-gray-300 hover:bg-gray-50 transition-colors"
+              className="inline-flex items-center gap-1.5 px-3 py-2 text-sm font-medium text-white bg-gray-800 rounded-lg hover:bg-gray-900 transition-all"
             >
               Visualizer
+              <ArrowTopRightOnSquareIcon className="h-4 w-4 text-gray-400" />
             </a>
           </div>
         </div>
+      </div>
 
-        {data.summary && <p className="text-gray-600 mb-4">{data.summary}</p>}
-
+      {/* Domain Content */}
+      <div className="py-4 space-y-8">
         {/* Domain Entities */}
         {entities.length > 0 && (
-          <div className="mb-6">
-            <h3 className="text-xs font-semibold text-gray-500 uppercase tracking-wide mb-2">Entities</h3>
+          <div>
+            <div className="flex items-center gap-2 mb-4">
+              <BoxIcon className="h-5 w-5 text-purple-600" />
+              <h3 className="text-lg font-semibold text-gray-900">Entities</h3>
+              <span className="text-sm text-gray-500 bg-gray-100 px-2.5 py-0.5 rounded-full font-medium">{entities.length}</span>
+            </div>
             <div className="flex flex-wrap gap-2">
               {entities.map((entity: any) => {
                 const entityId = entity?.data?.id || entity?.id;
@@ -376,11 +498,17 @@ export default function DomainGrid({ domain }: DomainGridProps) {
           </div>
         )}
 
-        {/* Top-level Services (not in subdomains) */}
+        {/* Top-level Services */}
         {topLevelServices.length > 0 && (
-          <div className="mb-6">
-            <h3 className="text-xs font-semibold text-gray-500 uppercase tracking-wide mb-3">Services</h3>
-            <div className="grid gap-4 xl:grid-cols-2">
+          <div>
+            <div className="flex items-center gap-2 mb-4">
+              <ServerIcon className="h-5 w-5 text-pink-600" />
+              <h3 className="text-lg font-semibold text-gray-900">Services</h3>
+              <span className="text-sm text-gray-500 bg-gray-100 px-2.5 py-0.5 rounded-full font-medium">
+                {topLevelServices.length}
+              </span>
+            </div>
+            <div className="space-y-3">
               {topLevelServices.map((service: any) => {
                 const serviceId = service?.data?.id || service?.id;
                 return serviceId ? <ServiceCard key={serviceId} service={service} /> : null;
@@ -389,10 +517,16 @@ export default function DomainGrid({ domain }: DomainGridProps) {
           </div>
         )}
 
-        {/* Subdomains - nested inside domain */}
+        {/* Subdomains */}
         {subdomains.length > 0 && (
           <div>
-            <h3 className="text-xs font-semibold text-gray-500 uppercase tracking-wide mb-3">Subdomains</h3>
+            <div className="flex items-center gap-2 mb-4">
+              <RectangleGroupIcon className="h-5 w-5 text-orange-600" />
+              <h3 className="text-lg font-semibold text-gray-900">Subdomains</h3>
+              <span className="text-sm text-gray-500 bg-gray-100 px-2.5 py-0.5 rounded-full font-medium">
+                {subdomains.length}
+              </span>
+            </div>
             <div className="space-y-4">
               {subdomains.map((subdomain: any) => {
                 const subdomainId = subdomain?.data?.id || subdomain?.id;
@@ -404,7 +538,10 @@ export default function DomainGrid({ domain }: DomainGridProps) {
 
         {/* Empty state */}
         {entities.length === 0 && services.length === 0 && subdomains.length === 0 && (
-          <div className="text-center py-8">
+          <div className="text-center py-12">
+            <div className="flex items-center justify-center w-16 h-16 mx-auto mb-4 bg-gray-100 rounded-2xl">
+              <RectangleGroupIcon className="h-8 w-8 text-gray-400" />
+            </div>
             <p className="text-gray-500">This domain has no entities, services, or subdomains defined.</p>
           </div>
         )}