refactor: remove unused export and clean up comments in ListDatabases and useRenderData

Author: tsck

src/ui/components/ListDatabases/ListDatabases.tsx

@@ -12,29 +12,22 @@ import {
 } from "@leafygreen-ui/table";
 import { tableStyles } from "./ListDatabases.styles.js";
 
-// Cast components to fix incorrect type definitions in @leafygreen-ui/table
-// The library types these as (props, ref) => ReactElement, but JSX expects (props) => ReactElement
 const HeaderCell = LGHeaderCell as React.FC<React.ComponentPropsWithoutRef<"th">>;
 const Cell = LGCell as React.FC<React.ComponentPropsWithoutRef<"td">>;
 const Row = LGRow as React.FC<React.ComponentPropsWithoutRef<"tr">>;
 
-// Schema for a single database
 const DatabaseInfoSchema = z.object({
     name: z.string(),
     size: z.number(),
 });
 
-// Schema for list-databases tool render data
 const ListDatabasesDataSchema = z.object({
     databases: z.array(DatabaseInfoSchema),
     totalCount: z.number(),
 });
 
 type ListDatabasesRenderData = z.infer<typeof ListDatabasesDataSchema>;
 
-/**
- * Format bytes to human-readable format
- */
 function formatBytes(bytes: number): string {
     if (bytes === 0) return "0 Bytes";
 

src/ui/hooks/useRenderData.ts

@@ -2,7 +2,7 @@ import { useEffect, useState } from "react";
 
 /**
  * Hook for receiving render data from parent window via postMessage
- * This is used by iframe-based UI components that receive data from the MCP client
+ * This is used by iframe-based UI components that receive data from an MCP client
  *
  * @template T - The type of data expected in the renderData payload
  * @returns An object containing:
@@ -28,68 +28,43 @@ export function useRenderData<T = unknown>() {
     const [error, setError] = useState<string | null>(null);
 
     useEffect(() => {
-        console.log("[useRenderData] Waiting for render data from parent...");
-
-        // Listen for render data from parent window
         const handleMessage = (event: MessageEvent) => {
-            console.log("[useRenderData] Received message:", event.data);
-
-            // Note: Origin validation is intentionally NOT performed here
-            // MCP-UI is designed to be universally embeddable from any MCP client
-            // (desktop apps, browser extensions, web apps with variable origins)
-            // For private/enterprise deployments requiring origin restrictions, see next.config.ts
-
-            // Security: Message type validation - only accept expected message types
             if (event.data?.type !== "ui-lifecycle-iframe-render-data") {
                 // Silently ignore messages that aren't for us
                 return;
             }
 
-            // Security: Payload structure validation
             if (!event.data.payload || typeof event.data.payload !== "object") {
                 const errorMsg = "Invalid payload structure received";
-                console.error(`[useRenderData] ${errorMsg}`);
                 setError(errorMsg);
                 setIsLoading(false);
                 return;
             }
 
             const renderData = event.data.payload.renderData;
-            console.log("[useRenderData] Received render data:", renderData);
 
-            // Security: Validate data exists and is of expected type
             if (renderData === undefined || renderData === null) {
-                console.warn("[useRenderData] Received null/undefined renderData");
                 setIsLoading(false);
                 // Not an error - parent may intentionally send null
                 return;
             }
 
-            // Security: Basic type checking - ensure data matches expected structure
             if (typeof renderData !== "object") {
                 const errorMsg = `Expected object but received ${typeof renderData}`;
-                console.error(`[useRenderData] ${errorMsg}`);
                 setError(errorMsg);
                 setIsLoading(false);
                 return;
             }
 
-            // Note: XSS prevention is handled by React's automatic escaping when rendering
-            // If you need to render raw HTML (dangerouslySetInnerHTML), sanitize with DOMPurify
-            // Note: Schema validation is handled by the components themselves
             setData(renderData as T);
             setIsLoading(false);
             setError(null);
         };
 
         window.addEventListener("message", handleMessage);
-
-        // Notify parent we're ready to receive data
         window.parent.postMessage({ type: "ui-lifecycle-iframe-ready" }, "*");
-        console.log("[useRenderData] Sent ui-lifecycle-iframe-ready to parent");
 
         return () => {
-            console.log("[useRenderData] Cleaning up message listener");
             window.removeEventListener("message", handleMessage);
         };
     }, []);

src/ui/index.ts

@@ -1,2 +1 @@
 export { UIRegistry, getDefaultUIRegistry, setDefaultUIRegistry } from "./registry/index.js";
-export { useRenderData } from "./hooks/index.js";