3b583e1f45
git-subtree-dir: packages/mcp-typescript git-subtree-split: 2cc7dd104307d48bab8d27760f16c63c119d8a88
55 lines
2.0 KiB
JavaScript
Generated
55 lines
2.0 KiB
JavaScript
Generated
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 {
|
|
connect(url) {
|
|
return new Promise((resolve, reject) => {
|
|
this._socket = new WebSocket(url, SUBPROTOCOL);
|
|
this._socket.onerror = (event) => {
|
|
var _a;
|
|
const error = "error" in event
|
|
? event.error
|
|
: new Error(`WebSocket error: ${JSON.stringify(event)}`);
|
|
reject(error);
|
|
(_a = this.onerror) === null || _a === void 0 ? void 0 : _a.call(this, error);
|
|
};
|
|
this._socket.onopen = () => {
|
|
resolve();
|
|
};
|
|
this._socket.onclose = () => {
|
|
var _a;
|
|
(_a = this.onclose) === null || _a === void 0 ? void 0 : _a.call(this);
|
|
};
|
|
this._socket.onmessage = (event) => {
|
|
var _a, _b;
|
|
let message;
|
|
try {
|
|
message = JSONRPCMessageSchema.parse(JSON.parse(event.data));
|
|
}
|
|
catch (error) {
|
|
(_a = this.onerror) === null || _a === void 0 ? void 0 : _a.call(this, error);
|
|
return;
|
|
}
|
|
(_b = this.onmessage) === null || _b === void 0 ? void 0 : _b.call(this, message);
|
|
};
|
|
});
|
|
}
|
|
async close() {
|
|
var _a;
|
|
(_a = this._socket) === null || _a === void 0 ? void 0 : _a.close();
|
|
}
|
|
send(message) {
|
|
return new Promise((resolve, reject) => {
|
|
var _a;
|
|
if (!this._socket) {
|
|
reject(new Error("Not connected"));
|
|
return;
|
|
}
|
|
(_a = this._socket) === null || _a === void 0 ? void 0 : _a.send(JSON.stringify(message));
|
|
resolve();
|
|
});
|
|
}
|
|
}
|
|
//# sourceMappingURL=websocket.js.map
|