modelcontextprotocol/typescript-sdk
Watch
1
0
Fork
You've already forked typescript-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Use Zod for message validation at transport layer

Browse source
This commit is contained in:
Justin Spahr-Summers 2024-10-02 15:06:30 +01:00
commit 3ea255f088
No known key found for this signature in database
GPG key ID: E78498BF3C053CA7
2 changed files with 11 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/shared/message.ts
View file

@ -1,10 +0,0 @@
import { JSONRPCMessage } from "../types.js";
// Throws an exception if the message is invalid or unsupported.
export function validateMessage(message: JSONRPCMessage): void {
if (message.jsonrpc !== "2.0") {
throw new Error(
`Message is not in a supported JSON-RPC version: ${message}`,
);
}
}

7
src/shared/stdio.ts
View file

@ -1,5 +1,4 @@
import { JSONRPCMessage } from "../types.js";
import { validateMessage } from "./message.js";
import { JSONRPCMessage, JSONRPCMessageSchema } from "../types.js";
/**
* Buffers a continuous stdio stream into discrete JSON-RPC messages.
@ -32,9 +31,7 @@ export class ReadBuffer {
}
export function deserializeMessage(line: string): JSONRPCMessage {
const message = JSON.parse(line) as JSONRPCMessage;
validateMessage(message);
return message;
return JSONRPCMessageSchema.parse(JSON.parse(line));
}
export function serializeMessage(message: JSONRPCMessage): string {