41486bfaa4* refactor: reorganize server handlers and transport structure * feat: move the test files to corresponding directories * fix: linter issues * fix: correct PHPDoc formatting and streamline handler imports * refactor: update import statement for ServerBuilder to Builder
79 lines
1.9 KiB
PHP
79 lines
1.9 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the official PHP MCP SDK.
|
|
*
|
|
* A collaboration between Symfony and the PHP Foundation.
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Mcp\Server\Transport;
|
|
|
|
use Symfony\Component\Uid\Uuid;
|
|
|
|
/**
|
|
* @author Tobias Nyholm <tobias.nyholm@gmail.com>
|
|
*/
|
|
class InMemoryTransport implements TransportInterface
|
|
{
|
|
/** @var callable(string, ?Uuid): void */
|
|
private $messageListener;
|
|
|
|
/** @var callable(Uuid): void */
|
|
private $sessionDestroyListener;
|
|
|
|
private ?Uuid $sessionId = null;
|
|
|
|
/**
|
|
* @param list<string> $messages
|
|
*/
|
|
public function __construct(
|
|
private readonly array $messages = [],
|
|
) {
|
|
}
|
|
|
|
public function initialize(): void
|
|
{
|
|
}
|
|
|
|
public function onMessage(callable $listener): void
|
|
{
|
|
$this->messageListener = $listener;
|
|
}
|
|
|
|
public function send(string $data, array $context): void
|
|
{
|
|
if (isset($context['session_id'])) {
|
|
$this->sessionId = $context['session_id'];
|
|
}
|
|
}
|
|
|
|
public function listen(): mixed
|
|
{
|
|
foreach ($this->messages as $message) {
|
|
if (\is_callable($this->messageListener)) {
|
|
\call_user_func($this->messageListener, $message, $this->sessionId);
|
|
}
|
|
}
|
|
|
|
if (\is_callable($this->sessionDestroyListener) && null !== $this->sessionId) {
|
|
\call_user_func($this->sessionDestroyListener, $this->sessionId);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public function onSessionEnd(callable $listener): void
|
|
{
|
|
$this->sessionDestroyListener = $listener;
|
|
}
|
|
|
|
public function close(): void
|
|
{
|
|
if (\is_callable($this->sessionDestroyListener) && null !== $this->sessionId) {
|
|
\call_user_func($this->sessionDestroyListener, $this->sessionId);
|
|
}
|
|
}
|
|
}
|