From e0e545557636d36e8c6388640af4a6998cafb307 Mon Sep 17 00:00:00 2001
From: xl <as2252258@163.com>
Date: Wed, 22 Nov 2023 10:19:33 +0800
Subject: [PATCH] eee

---
 src/Format/NoBody.php | 26 ++++++++++++++++++++++++++
 src/Handler.php       |  9 +++++++--
 2 files changed, 33 insertions(+), 2 deletions(-)
 create mode 100644 src/Format/NoBody.php

diff --git a/src/Format/NoBody.php b/src/Format/NoBody.php
new file mode 100644
index 0000000..0f609d5
--- /dev/null
+++ b/src/Format/NoBody.php
@@ -0,0 +1,26 @@
+<?php
+
+namespace Kiri\Router\Format;
+
+use Kiri\Router\Constrict\Stream;
+use Psr\Http\Message\ResponseInterface;
+
+class NoBody implements IFormat
+{
+
+
+    /**
+     * @param $result
+     * @return ResponseInterface
+     */
+    public function call($result): ResponseInterface
+    {
+        // TODO: Implement call() method.
+        if ($result instanceof ResponseInterface) {
+            $result->getBody()->write('');
+        } else {
+            $result = response()->withBody(new Stream());
+        }
+        return $result;
+    }
+}
\ No newline at end of file
diff --git a/src/Handler.php b/src/Handler.php
index 8ba420f..139ef67 100644
--- a/src/Handler.php
+++ b/src/Handler.php
@@ -7,6 +7,7 @@ use Closure;
 use Kiri\Router\Format\ArrayFormat;
 use Kiri\Router\Format\IFormat;
 use Kiri\Router\Format\MixedFormat;
+use Kiri\Router\Format\NoBody;
 use Kiri\Router\Format\OtherFormat;
 use Kiri\Router\Format\ResponseFormat;
 use Kiri\Router\Format\VoidFormat;
@@ -36,13 +37,14 @@ class Handler implements RequestHandlerInterface
 
     /**
      * @param array|Closure $handler
+     * @param string $method
      * @param array $parameter
      * @param ReflectionNamedType|null $reflectionType
-     * @throws ReflectionException
      * @throws ContainerExceptionInterface
      * @throws NotFoundExceptionInterface
+     * @throws ReflectionException
      */
-    public function __construct(public array|Closure $handler, public array $parameter, public ?ReflectionNamedType $reflectionType)
+    public function __construct(public array|Closure $handler, string $method, public array $parameter, public ?ReflectionNamedType $reflectionType)
     {
         $this->container = \Kiri::getDi();
         if ($this->reflectionType != null) {
@@ -50,6 +52,9 @@ class Handler implements RequestHandlerInterface
         } else {
             $this->format = $this->container->get(MixedFormat::class);
         }
+        if ($method === 'HEAD') {
+            $this->format = $this->container->get(NoBody::class);
+        }
     }