feat(server): Rework transport architecture and add StreamableHttpTransport (#49)

* feat(server): Rework transport architecture and add StreamableHttpTransport

- `Server::connect()` no longer contains a processing loop.
- `TransportInterface` is updated with `setMessageHandler()` and `listen()`.
- `StdioTransport` is updated to implement the new interface.
- A new, minimal `StreamableHttpTransport` is added for stateless HTTP.

* refactor(server): use  event-driven message handling

* feat(server): Introduce a formal session management system

* fix: ensure session elements are preserved when building server (regression from #46)

* refactor: consolidate HTTP example to use shared dependencies

* feat(server): Enhance message handling with session support

- Updated `TransportInterface` to use `onMessage` for handling incoming messages with session IDs.
- Refactored `Server`, `Handler`, and transport classes to accommodate session management using `Uuid`.
- Introduced methods for creating sessions with auto-generated and specific UUIDs in `SessionFactory` and `SessionFactoryInterface`.

* feat(server): Integrate session management in message handling

- Added session support to the `Server` and `Handler` classes, allowing for session data to be managed during message processing.
- Updated `TransportInterface` to include session context in the `send` method.
- Refactored various request handlers to utilize session information, ensuring proper session handling for incoming requests.
- Introduced a file-based session store for persistent session data management

* feat: Update session builder method to use set* instead of with*

* feat: Fix test compatibility with new session management architecture

* chore: apply code style fixes

* fix: remove deprecated SSE transport and resolve PHPStan issues
This commit is contained in:
Kyrian Obikwelu 2025-09-21 11:12:40 +01:00 • committed by GitHub
commit 719262b15d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -14,13 +14,20 @@ namespace Mcp\Server\Transport;
use Mcp\Server\TransportInterface;
use Psr\Log\LoggerInterface;
use Psr\Log\NullLogger;
use Symfony\Component\Uid\Uuid;
/**
* Heavily inspired by https://jolicode.com/blog/mcp-the-open-protocol-that-turns-llm-chatbots-into-intelligent-agents.
* @author Kyrian Obikwelu <koshnawaza@gmail.com>
*/
class StdioTransport implements TransportInterface
{
private string $buffer = '';
/** @var callable(string, ?Uuid): void */
private $messageListener;
/** @var callable(Uuid): void */
private $sessionEndListener;
private ?Uuid $sessionId = null;
/**
* @param resource $input
@ -37,39 +44,67 @@ class StdioTransport implements TransportInterface
{
}
public function isConnected(): bool
public function onMessage(callable $listener): void
{
return true;
$this->messageListener = $listener;
}
public function receive(): \Generator
{
$line = fgets($this->input);
$this->logger->debug('Received message on StdioTransport.', [
'line' => $line,
]);
if (false === $line) {
return;
}
$this->buffer .= rtrim($line).\PHP_EOL;
if (str_contains($this->buffer, \PHP_EOL)) {
$lines = explode(\PHP_EOL, $this->buffer);
$this->buffer = array_pop($lines);
yield from $lines;
}
}
public function send(string $data): void
public function send(string $data, array $context): void
{
$this->logger->debug('Sending data to client via StdioTransport.', ['data' => $data]);
if (isset($context['session_id'])) {
$this->sessionId = $context['session_id'];
}
fwrite($this->output, $data.\PHP_EOL);
}
public function listen(): mixed
{
$this->logger->info('StdioTransport is listening for messages on STDIN...');
while (!feof($this->input)) {
$line = fgets($this->input);
if (false === $line) {
break;
}
$trimmedLine = trim($line);
if (!empty($trimmedLine)) {
$this->logger->debug('Received message on StdioTransport.', ['line' => $trimmedLine]);
if (\is_callable($this->messageListener)) {
\call_user_func($this->messageListener, $trimmedLine, $this->sessionId);
}
}
}
$this->logger->info('StdioTransport finished listening.');
if (\is_callable($this->sessionEndListener) && null !== $this->sessionId) {
\call_user_func($this->sessionEndListener, $this->sessionId);
}
return null;
}
public function onSessionEnd(callable $listener): void
{
$this->sessionEndListener = $listener;
}
public function close(): void
{
if (\is_callable($this->sessionEndListener) && null !== $this->sessionId) {
\call_user_func($this->sessionEndListener, $this->sessionId);
}
if (\is_resource($this->input)) {
fclose($this->input);
}
if (\is_resource($this->output)) {
fclose($this->output);
}
}
}