| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354 |
- import { JSONRPCMessageSchema } from '../types.js';
- const SUBPROTOCOL = 'mcp';
- /**
- * Client transport for WebSocket: this will connect to a server over the WebSocket protocol.
- */
- export class WebSocketClientTransport {
- constructor(url) {
- this._url = url;
- }
- start() {
- if (this._socket) {
- throw new Error('WebSocketClientTransport already started! If using Client class, note that connect() calls start() automatically.');
- }
- return new Promise((resolve, reject) => {
- this._socket = new WebSocket(this._url, SUBPROTOCOL);
- this._socket.onerror = event => {
- const error = 'error' in event ? event.error : new Error(`WebSocket error: ${JSON.stringify(event)}`);
- reject(error);
- this.onerror?.(error);
- };
- this._socket.onopen = () => {
- resolve();
- };
- this._socket.onclose = () => {
- this.onclose?.();
- };
- this._socket.onmessage = (event) => {
- let message;
- try {
- message = JSONRPCMessageSchema.parse(JSON.parse(event.data));
- }
- catch (error) {
- this.onerror?.(error);
- return;
- }
- this.onmessage?.(message);
- };
- });
- }
- async close() {
- this._socket?.close();
- }
- send(message) {
- return new Promise((resolve, reject) => {
- if (!this._socket) {
- reject(new Error('Not connected'));
- return;
- }
- this._socket?.send(JSON.stringify(message));
- resolve();
- });
- }
- }
- //# sourceMappingURL=websocket.js.map
|