tiny-stdio-mcp-server 0.1.14 → 0.1.15
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/jsonrpc.js +2 -1
- package/dist/server.js +43 -18
- package/package.json +1 -1
package/dist/jsonrpc.js
CHANGED
package/dist/server.js
CHANGED
|
@@ -292,7 +292,21 @@ export function createServer(options) {
|
|
|
292
292
|
}) + "\n");
|
|
293
293
|
return;
|
|
294
294
|
}
|
|
295
|
-
|
|
295
|
+
let handled;
|
|
296
|
+
try {
|
|
297
|
+
handled = await messageHandler(request.method, request.params);
|
|
298
|
+
}
|
|
299
|
+
catch {
|
|
300
|
+
if (!isNotification) {
|
|
301
|
+
const requestWithId = request;
|
|
302
|
+
write(formatErrorResponse(requestWithId.id, {
|
|
303
|
+
code: JSON_RPC_ERROR_CODES.INTERNAL_ERROR,
|
|
304
|
+
message: "Internal error",
|
|
305
|
+
}) + "\n");
|
|
306
|
+
}
|
|
307
|
+
return;
|
|
308
|
+
}
|
|
309
|
+
const { result, error } = handled;
|
|
296
310
|
if (isNotification) {
|
|
297
311
|
return;
|
|
298
312
|
}
|
|
@@ -412,20 +426,17 @@ export function createServer(options) {
|
|
|
412
426
|
},
|
|
413
427
|
async connect(transport) {
|
|
414
428
|
return new Promise((resolve) => {
|
|
415
|
-
const lifecycle = { initialized: false, initializeAccepted: false, notificationReady: false, resourceSubscriptions: new Set() };
|
|
416
|
-
const messageHandler = (method, params) => handleMessageWithLifecycle(method, lifecycle, params);
|
|
417
|
-
messageLifecycles.add(lifecycle);
|
|
418
429
|
const listener = (notification) => {
|
|
419
430
|
transport.writable.write(`${JSON.stringify(notification)}\n`);
|
|
420
431
|
};
|
|
421
|
-
|
|
432
|
+
const session = server.createMessageSession(listener);
|
|
422
433
|
const rl = readline.createInterface({
|
|
423
434
|
input: transport.readable,
|
|
424
435
|
crlfDelay: Infinity,
|
|
425
436
|
});
|
|
426
437
|
const pendingMessages = new Set();
|
|
427
438
|
rl.on("line", (line) => {
|
|
428
|
-
const message = processLine(line, (data) => transport.writable.write(data),
|
|
439
|
+
const message = processLine(line, (data) => transport.writable.write(data), session.handleMessage);
|
|
429
440
|
pendingMessages.add(message);
|
|
430
441
|
void message.finally(() => {
|
|
431
442
|
pendingMessages.delete(message);
|
|
@@ -433,19 +444,15 @@ export function createServer(options) {
|
|
|
433
444
|
});
|
|
434
445
|
rl.on("close", async () => {
|
|
435
446
|
await Promise.all([...pendingMessages]);
|
|
436
|
-
|
|
437
|
-
messageLifecycles.delete(lifecycle);
|
|
447
|
+
session.close();
|
|
438
448
|
resolve();
|
|
439
449
|
});
|
|
440
450
|
});
|
|
441
451
|
},
|
|
442
452
|
async connectSDK(transport) {
|
|
443
453
|
return new Promise((resolve, reject) => {
|
|
444
|
-
const lifecycle = { initialized: false, initializeAccepted: false, notificationReady: false, resourceSubscriptions: new Set() };
|
|
445
|
-
const messageHandler = (method, params) => handleMessageWithLifecycle(method, lifecycle, params);
|
|
446
|
-
messageLifecycles.add(lifecycle);
|
|
447
454
|
const listener = (notification) => transport.send(notification);
|
|
448
|
-
|
|
455
|
+
const session = server.createMessageSession(listener);
|
|
449
456
|
transport.onmessage = async (message) => {
|
|
450
457
|
// Ignore responses (we only handle requests/notifications)
|
|
451
458
|
if (!("method" in message)) {
|
|
@@ -456,7 +463,12 @@ export function createServer(options) {
|
|
|
456
463
|
if (message.method === "initialize") {
|
|
457
464
|
return;
|
|
458
465
|
}
|
|
459
|
-
|
|
466
|
+
try {
|
|
467
|
+
await session.handleMessage(message.method, message.params);
|
|
468
|
+
}
|
|
469
|
+
catch {
|
|
470
|
+
return;
|
|
471
|
+
}
|
|
460
472
|
return;
|
|
461
473
|
}
|
|
462
474
|
if (message.method === "notifications/initialized") {
|
|
@@ -471,7 +483,22 @@ export function createServer(options) {
|
|
|
471
483
|
return;
|
|
472
484
|
}
|
|
473
485
|
const request = message;
|
|
474
|
-
|
|
486
|
+
let handled;
|
|
487
|
+
try {
|
|
488
|
+
handled = await session.handleMessage(request.method, request.params);
|
|
489
|
+
}
|
|
490
|
+
catch {
|
|
491
|
+
await transport.send({
|
|
492
|
+
jsonrpc: "2.0",
|
|
493
|
+
id: request.id,
|
|
494
|
+
error: {
|
|
495
|
+
code: JSON_RPC_ERROR_CODES.INTERNAL_ERROR,
|
|
496
|
+
message: "Internal error",
|
|
497
|
+
},
|
|
498
|
+
});
|
|
499
|
+
return;
|
|
500
|
+
}
|
|
501
|
+
const { result, error } = handled;
|
|
475
502
|
if (error) {
|
|
476
503
|
const response = {
|
|
477
504
|
jsonrpc: "2.0",
|
|
@@ -490,13 +517,11 @@ export function createServer(options) {
|
|
|
490
517
|
}
|
|
491
518
|
};
|
|
492
519
|
transport.onclose = () => {
|
|
493
|
-
|
|
494
|
-
messageLifecycles.delete(lifecycle);
|
|
520
|
+
session.close();
|
|
495
521
|
resolve();
|
|
496
522
|
};
|
|
497
523
|
void transport.start().catch((error) => {
|
|
498
|
-
|
|
499
|
-
messageLifecycles.delete(lifecycle);
|
|
524
|
+
session.close();
|
|
500
525
|
reject(error);
|
|
501
526
|
});
|
|
502
527
|
});
|