feat(transport): add phpstan generics for typed transport returns

Author: CodeWithKyrian

examples/custom-method-handlers/server.php

@@ -141,4 +141,4 @@ public function handle(CallToolRequest|HasMethodInterface $message, SessionInter
 
 logger()->info('Server listener stopped gracefully.', ['result' => $result]);
 
-exit((int) $result);
+exit($result);

examples/stdio-cached-discovery/server.php

@@ -35,4 +35,4 @@
 
 logger()->info('Server listener stopped gracefully.', ['result' => $result]);
 
-exit((int) $result);
+exit($result);

examples/stdio-custom-dependencies/server.php

@@ -43,4 +43,4 @@
 
 logger()->info('Server listener stopped gracefully.', ['result' => $result]);
 
-exit((int) $result);
+exit($result);

examples/stdio-discovery-calculator/server.php

@@ -32,4 +32,4 @@
 
 logger()->info('Server listener stopped gracefully.', ['result' => $result]);
 
-exit((int) $result);
+exit($result);

examples/stdio-env-variables/server.php

@@ -61,4 +61,4 @@
 
 logger()->info('Server listener stopped gracefully.', ['result' => $result]);
 
-exit((int) $result);
+exit($result);

examples/stdio-explicit-registration/server.php

@@ -35,4 +35,4 @@
 
 logger()->info('Server listener stopped gracefully.', ['result' => $result]);
 
-exit((int) $result);
+exit($result);

src/Server.php

@@ -35,6 +35,13 @@ public static function builder(): Builder
         return new Builder();
     }
 
+    /**
+     * @template TResult
+     *
+     * @param TransportInterface<TResult> $transport
+     *
+     * @return TResult
+     */
     public function run(TransportInterface $transport): mixed
     {
         $transport->initialize();

src/Server/Transport/InMemoryTransport.php

@@ -14,6 +14,8 @@
 use Symfony\Component\Uid\Uuid;
 
 /**
+ * @implements TransportInterface<null>
+ *
  * @author Tobias Nyholm <[email protected]>
  */
 class InMemoryTransport implements TransportInterface
@@ -50,7 +52,7 @@ public function send(string $data, array $context): void
         }
     }
 
-    public function listen(): mixed
+    public function listen(): null
     {
         foreach ($this->messages as $message) {
             if (\is_callable($this->messageListener)) {
@@ -63,7 +65,7 @@ public function listen(): mixed
             $this->sessionId = null;
         }
 
-        return 0;
+        return null;
     }
 
     public function onSessionEnd(callable $listener): void

src/Server/Transport/StdioTransport.php

@@ -16,6 +16,8 @@
 use Symfony\Component\Uid\Uuid;
 
 /**
+ * @implements TransportInterface<int>
+ *
  * @author Kyrian Obikwelu <[email protected]>
  */
 class StdioTransport implements TransportInterface
@@ -59,7 +61,7 @@ public function send(string $data, array $context): void
         fwrite($this->output, $data.\PHP_EOL);
     }
 
-    public function listen(): mixed
+    public function listen(): int
     {
         $this->logger->info('StdioTransport is listening for messages on STDIN...');
 

src/Server/Transport/StreamableHttpTransport.php

@@ -21,6 +21,8 @@
 use Symfony\Component\Uid\Uuid;
 
 /**
+ * @implements TransportInterface<ResponseInterface>
+ *
  * @author Kyrian Obikwelu <[email protected]>
  */
 class StreamableHttpTransport implements TransportInterface
@@ -78,7 +80,7 @@ public function send(string $data, array $context): void
         ]);
     }
 
-    public function listen(): mixed
+    public function listen(): ResponseInterface
     {
         return match ($this->request->getMethod()) {
             'OPTIONS' => $this->handleOptionsRequest(),