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

Pass extra object instead of authInfo directly

Browse source
This commit is contained in:
Allen Zhou 2025-04-12 16:38:55 -07:00
commit c4381b857b
No known key found for this signature in database
GPG key ID: 5817C145DE5C5556
1 changed files with 21 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/inMemory.ts
View file

@ -4,7 +4,7 @@ import { AuthInfo } from "./server/auth/types.js";
interface QueuedMessage {
message: JSONRPCMessage;
authInfo?: AuthInfo;
extra?: { authInfo?: AuthInfo };
}
/**
@ -16,7 +16,7 @@ export class InMemoryTransport implements Transport {
onclose?: () => void;
onerror?: (error: Error) => void;
onmessage?: (message: JSONRPCMessage, authInfo?: AuthInfo) => void;
onmessage?: (message: JSONRPCMessage, extra?: { authInfo?: AuthInfo }) => void;
sessionId?: string;
/**
@ -34,7 +34,7 @@ export class InMemoryTransport implements Transport {
// Process any messages that were queued before start was called
while (this._messageQueue.length > 0) {
const queuedMessage = this._messageQueue.shift()!;
this.onmessage?.(queuedMessage.message, queuedMessage.authInfo);
this.onmessage?.(queuedMessage.message, queuedMessage.extra);
}
}
@ -45,23 +45,19 @@ export class InMemoryTransport implements Transport {
this.onclose?.();
}
async send(message: JSONRPCMessage): Promise<void> {
return this.sendWithAuth(message);
}
/**
* Sends a message with optional auth info.
* This is useful for testing authentication scenarios.
*/
async send(message: JSONRPCMessage, options?: { authInfo?: AuthInfo }): Promise<void>
async send(message: JSONRPCMessage, options?: { authInfo?: AuthInfo }): Promise<void> {
if (!this._otherTransport) {
throw new Error("Not connected");
}
if (this._otherTransport.onmessage) {
this._otherTransport.onmessage(message, authInfo);
this._otherTransport.onmessage(message, { authInfo: options?.authInfo });
} else {
this._otherTransport._messageQueue.push({ message, authInfo });
this._otherTransport._messageQueue.push({ message, extra: { authInfo: options?.authInfo } });
}
}
}