receive and display server notifications

This commit is contained in:
Ashwin Bhat
2024-10-16 14:44:21 -07:00
parent 3b4d8136b2
commit 48355e9cb2
6 changed files with 158 additions and 106 deletions

View File

@@ -1,8 +1,5 @@
#root { #root {
max-width: 1280px;
margin: 0 auto; margin: 0 auto;
padding: 2rem;
text-align: center;
} }
.logo { .logo {

View File

@@ -10,8 +10,10 @@ import {
Resource, Resource,
Tool, Tool,
ClientRequest, ClientRequest,
ProgressNotificationSchema,
ServerNotification,
} from "mcp-typescript/types.js"; } from "mcp-typescript/types.js";
import { useState } from "react"; import { useState, useRef } from "react";
import { import {
Send, Send,
Bell, Bell,
@@ -36,11 +38,11 @@ import ConsoleTab from "./components/ConsoleTab";
import Sidebar from "./components/Sidebar"; import Sidebar from "./components/Sidebar";
import RequestsTab from "./components/RequestsTabs"; import RequestsTab from "./components/RequestsTabs";
import ResourcesTab from "./components/ResourcesTab"; import ResourcesTab from "./components/ResourcesTab";
import NotificationsTab from "./components/NotificationsTab";
import PromptsTab, { Prompt } from "./components/PromptsTab"; import PromptsTab, { Prompt } from "./components/PromptsTab";
import ToolsTab from "./components/ToolsTab"; import ToolsTab from "./components/ToolsTab";
import History from "./components/History";
import { AnyZodObject } from "zod"; import { AnyZodObject } from "zod";
import HistoryAndNotifications from "./components/History";
import "./App.css";
const App = () => { const App = () => {
const [connectionStatus, setConnectionStatus] = useState< const [connectionStatus, setConnectionStatus] = useState<
@@ -57,7 +59,7 @@ const App = () => {
"/Users/ashwin/.nvm/versions/node/v18.20.4/bin/node", "/Users/ashwin/.nvm/versions/node/v18.20.4/bin/node",
); );
const [args, setArgs] = useState<string>( const [args, setArgs] = useState<string>(
"/Users/ashwin/code/example-servers/build/everything/stdio.js", "/Users/ashwin/code/mcp/example-servers/build/everything/stdio.js",
); );
const [url, setUrl] = useState<string>("http://localhost:3001/sse"); const [url, setUrl] = useState<string>("http://localhost:3001/sse");
const [transportType, setTransportType] = useState<"stdio" | "sse">("stdio"); const [transportType, setTransportType] = useState<"stdio" | "sse">("stdio");
@@ -65,12 +67,14 @@ const App = () => {
{ request: string; response: string }[] { request: string; response: string }[]
>([]); >([]);
const [mcpClient, setMcpClient] = useState<Client | null>(null); const [mcpClient, setMcpClient] = useState<Client | null>(null);
const [notifications, setNotifications] = useState<ServerNotification[]>([]);
const [selectedResource, setSelectedResource] = useState<Resource | null>( const [selectedResource, setSelectedResource] = useState<Resource | null>(
null, null,
); );
const [selectedPrompt, setSelectedPrompt] = useState<Prompt | null>(null); const [selectedPrompt, setSelectedPrompt] = useState<Prompt | null>(null);
const [selectedTool, setSelectedTool] = useState<Tool | null>(null); const [selectedTool, setSelectedTool] = useState<Tool | null>(null);
const progressTokenRef = useRef(0);
const pushHistory = (request: object, response: object) => { const pushHistory = (request: object, response: object) => {
setRequestHistory((prev) => [ setRequestHistory((prev) => [
@@ -155,7 +159,13 @@ const App = () => {
const response = await makeRequest( const response = await makeRequest(
{ {
method: "tools/call" as const, method: "tools/call" as const,
params: { name, arguments: params }, params: {
name,
arguments: params,
_meta: {
progressToken: progressTokenRef.current++,
},
},
}, },
CallToolResultSchema, CallToolResultSchema,
); );
@@ -182,6 +192,16 @@ const App = () => {
const clientTransport = new SSEClientTransport(backendUrl); const clientTransport = new SSEClientTransport(backendUrl);
await client.connect(clientTransport); await client.connect(clientTransport);
client.setNotificationHandler(
ProgressNotificationSchema,
(notification) => {
setNotifications((prevNotifications) => [
...prevNotifications,
notification,
]);
},
);
setMcpClient(client); setMcpClient(client);
setConnectionStatus("connected"); setConnectionStatus("connected");
} catch (e) { } catch (e) {
@@ -255,10 +275,6 @@ const App = () => {
<Send className="w-4 h-4 mr-2" /> <Send className="w-4 h-4 mr-2" />
Requests Requests
</TabsTrigger> </TabsTrigger>
<TabsTrigger value="notifications" disabled>
<Bell className="w-4 h-4 mr-2" />
Notifications
</TabsTrigger>
<TabsTrigger value="tools"> <TabsTrigger value="tools">
<Hammer className="w-4 h-4 mr-2" /> <Hammer className="w-4 h-4 mr-2" />
Tools Tools
@@ -279,7 +295,6 @@ const App = () => {
resourceContent={resourceContent} resourceContent={resourceContent}
error={error} error={error}
/> />
<NotificationsTab />
<PromptsTab <PromptsTab
prompts={prompts} prompts={prompts}
listPrompts={listPrompts} listPrompts={listPrompts}
@@ -315,7 +330,10 @@ const App = () => {
</div> </div>
</div> </div>
</div> </div>
<History requestHistory={requestHistory} /> <HistoryAndNotifications
requestHistory={requestHistory}
serverNotifications={notifications}
/>
</div> </div>
); );
}; };

View File

@@ -1,94 +1,163 @@
import { useState } from "react"; import { useState } from "react";
import { Copy } from "lucide-react"; import { Copy } from "lucide-react";
import { ServerNotification } from "mcp-typescript/types.js";
const History = ({ const HistoryAndNotifications = ({
requestHistory, requestHistory,
serverNotifications,
}: { }: {
requestHistory: Array<{ request: string; response: string | null }>; requestHistory: Array<{ request: string; response: string | null }>;
serverNotifications: ServerNotification[];
}) => { }) => {
const [expandedRequests, setExpandedRequests] = useState<{ const [expandedRequests, setExpandedRequests] = useState<{
[key: number]: boolean; [key: number]: boolean;
}>({}); }>({});
const [expandedNotifications, setExpandedNotifications] = useState<{
[key: number]: boolean;
}>({});
const toggleRequestExpansion = (index: number) => { const toggleRequestExpansion = (index: number) => {
setExpandedRequests((prev) => ({ ...prev, [index]: !prev[index] })); setExpandedRequests((prev) => ({ ...prev, [index]: !prev[index] }));
}; };
const toggleNotificationExpansion = (index: number) => {
setExpandedNotifications((prev) => ({ ...prev, [index]: !prev[index] }));
};
const copyToClipboard = (text: string) => { const copyToClipboard = (text: string) => {
navigator.clipboard.writeText(text); navigator.clipboard.writeText(text);
}; };
return ( return (
<div className="w-64 bg-white shadow-md p-4 overflow-y-auto"> <div className="w-64 bg-white shadow-md p-4 overflow-hidden flex flex-col h-full">
<h2 className="text-lg font-semibold mb-4">History</h2> <div className="flex-1 overflow-y-auto mb-4 border-b pb-4">
<ul className="space-y-3"> <h2 className="text-lg font-semibold mb-4">History</h2>
{requestHistory {requestHistory.length === 0 ? (
.slice() <p className="text-sm text-gray-500 italic">No history yet</p>
.reverse() ) : (
.map((request, index) => ( <ul className="space-y-3">
<li {requestHistory
key={index} .slice()
className="text-sm text-gray-600 bg-gray-100 p-2 rounded" .reverse()
> .map((request, index) => (
<div <li
className="flex justify-between items-center cursor-pointer" key={index}
onClick={() => className="text-sm text-gray-600 bg-gray-100 p-2 rounded"
toggleRequestExpansion(requestHistory.length - 1 - index) >
} <div
> className="flex justify-between items-center cursor-pointer"
<span className="font-mono"> onClick={() =>
{requestHistory.length - index}.{" "} toggleRequestExpansion(requestHistory.length - 1 - index)
{JSON.parse(request.request).method} }
</span> >
<span> <span className="font-mono">
{expandedRequests[requestHistory.length - 1 - index] {requestHistory.length - index}.{" "}
? "▼" {JSON.parse(request.request).method}
: "▶"} </span>
</span> <span>
</div> {expandedRequests[requestHistory.length - 1 - index]
{expandedRequests[requestHistory.length - 1 - index] && ( ? "▼"
<> : "▶"}
<div className="mt-2"> </span>
<div className="flex justify-between items-center mb-1">
<span className="font-semibold text-blue-600">
Request:
</span>
<button
onClick={() => copyToClipboard(request.request)}
className="text-blue-500 hover:text-blue-700"
>
<Copy size={16} />
</button>
</div>
<pre className="whitespace-pre-wrap break-words bg-blue-50 p-2 rounded">
{JSON.stringify(JSON.parse(request.request), null, 2)}
</pre>
</div> </div>
{request.response && ( {expandedRequests[requestHistory.length - 1 - index] && (
<>
<div className="mt-2">
<div className="flex justify-between items-center mb-1">
<span className="font-semibold text-blue-600">
Request:
</span>
<button
onClick={() => copyToClipboard(request.request)}
className="text-blue-500 hover:text-blue-700"
>
<Copy size={16} />
</button>
</div>
<pre className="whitespace-pre-wrap break-words bg-blue-50 p-2 rounded">
{JSON.stringify(JSON.parse(request.request), null, 2)}
</pre>
</div>
{request.response && (
<div className="mt-2">
<div className="flex justify-between items-center mb-1">
<span className="font-semibold text-green-600">
Response:
</span>
<button
onClick={() => copyToClipboard(request.response!)}
className="text-blue-500 hover:text-blue-700"
>
<Copy size={16} />
</button>
</div>
<pre className="whitespace-pre-wrap break-words bg-green-50 p-2 rounded">
{JSON.stringify(
JSON.parse(request.response),
null,
2,
)}
</pre>
</div>
)}
</>
)}
</li>
))}
</ul>
)}
</div>
<div className="flex-1 overflow-y-auto">
<h2 className="text-lg font-semibold mb-4">Server Notifications</h2>
{serverNotifications.length === 0 ? (
<p className="text-sm text-gray-500 italic">No notifications yet</p>
) : (
<ul className="space-y-3">
{serverNotifications
.slice()
.reverse()
.map((notification, index) => (
<li
key={index}
className="text-sm text-gray-600 bg-gray-100 p-2 rounded"
>
<div
className="flex justify-between items-center cursor-pointer"
onClick={() => toggleNotificationExpansion(index)}
>
<span className="font-mono">
{serverNotifications.length - index}.{" "}
{notification.method}
</span>
<span>{expandedNotifications[index] ? "▼" : "▶"}</span>
</div>
{expandedNotifications[index] && (
<div className="mt-2"> <div className="mt-2">
<div className="flex justify-between items-center mb-1"> <div className="flex justify-between items-center mb-1">
<span className="font-semibold text-green-600"> <span className="font-semibold text-purple-600">
Response: Details:
</span> </span>
<button <button
onClick={() => copyToClipboard(request.response!)} onClick={() =>
copyToClipboard(JSON.stringify(notification))
}
className="text-blue-500 hover:text-blue-700" className="text-blue-500 hover:text-blue-700"
> >
<Copy size={16} /> <Copy size={16} />
</button> </button>
</div> </div>
<pre className="whitespace-pre-wrap break-words bg-green-50 p-2 rounded"> <pre className="whitespace-pre-wrap break-words bg-purple-50 p-2 rounded">
{JSON.stringify(JSON.parse(request.response), null, 2)} {JSON.stringify(notification, null, 2)}
</pre> </pre>
</div> </div>
)} )}
</> </li>
)} ))}
</li> </ul>
))} )}
</ul> </div>
</div> </div>
); );
}; };
export default History; export default HistoryAndNotifications;

View File

@@ -1,33 +0,0 @@
import { Bell } from "lucide-react";
import { Button } from "@/components/ui/button";
import { Input } from "@/components/ui/input";
import { Textarea } from "@/components/ui/textarea";
import { TabsContent } from "@/components/ui/tabs";
const NotificationsTab = () => (
<TabsContent value="notifications" className="space-y-4">
<div className="grid grid-cols-2 gap-4">
<div className="space-y-4">
<div className="flex space-x-2">
<Input placeholder="Notification method" />
<Button>
<Bell className="w-4 h-4 mr-2" />
Send
</Button>
</div>
<Textarea
placeholder="Notification parameters (JSON)"
className="h-64 font-mono"
/>
</div>
<div className="bg-white rounded-lg shadow p-4">
<h3 className="font-semibold mb-4">Recent Notifications</h3>
<div className="space-y-2">
{/* Notification history would go here */}
</div>
</div>
</div>
</TabsContent>
);
export default NotificationsTab;

View File

@@ -36,7 +36,9 @@ const ToolsTab = ({
renderItem={(tool) => ( renderItem={(tool) => (
<> <>
<span className="flex-1">{tool.name}</span> <span className="flex-1">{tool.name}</span>
<span className="text-sm text-gray-500">{tool.description}</span> <span className="text-sm text-gray-500 text-right">
{tool.description}
</span>
</> </>
)} )}
title="Tools" title="Tools"

View File

@@ -1 +0,0 @@
{"root":["./vite.config.ts"],"version":"5.6.3"}