111

2021-08-28 01:37:47 +08:00
parent a4823d5d41
commit fc634c94ed
2 changed files with 75 additions and 54 deletions
@@ -24,12 +24,12 @@ class ExceptionHandlerDispatcher implements ExceptionHandlerInterface
    {
        if ($exception->getCode() == 404) {
            return $response->withBody(new Stream($exception->getMessage()))
-                ->withHeader('Content-Type', 'text/html')
+                ->withContentType(Message\Response::CONTENT_TYPE_HTML)
                ->withStatus(404);
        }
        $code = $exception->getCode() == 0 ? 500 : $exception->getCode();
        return $response->withBody(new Stream(jTraceEx($exception, null, true)))
-            ->withHeader('Content-Type', 'text/html')
+            ->withContentType(Message\Response::CONTENT_TYPE_HTML)
            ->withStatus($code);
    }

@@ -10,73 +10,94 @@ use Psr\Http\Message\ResponseInterface;
 */
 class Response implements ResponseInterface, \Server\ResponseInterface
 {
-	use Message;
+    use Message;

-	public int $statusCode = 200;
+    public int $statusCode = 200;


-	public string $reasonPhrase = '';
+    public string $reasonPhrase = '';
+
+    const CONTENT_TYPE_JSON = 'application/json;charset=utf-8';
+    const CONTENT_TYPE_HTML = 'text/html;charset=utf-8';
+    const CONTENT_TYPE_STREAM = 'octet-stream';
+    const CONTENT_TYPE_XML = 'application/xml;charset=utf-8';


-	/**
-	 * @return int
-	 */
-	public function getStatusCode(): int
-	{
-		// TODO: Implement getStatusCode() method.
-		return $this->statusCode;
-	}
+    /**
+     * @param string $type
+     * @return $this
+     */
+    public function withContentType(string $type): static
+    {
+        if (!in_array($type, [
+            Response::CONTENT_TYPE_HTML, Response::CONTENT_TYPE_JSON,
+            Response::CONTENT_TYPE_STREAM, Response::CONTENT_TYPE_XML
+        ])) {
+            throw new \Exception('Wrong content type.');
+        }
+        return $this->withHeader('Content-Type', $type);
+    }

-	/**
-	 * @param int $code
-	 * @param string $reasonPhrase
-	 * @return static
-	 */
-	public function withStatus($code, $reasonPhrase = ''): static
-	{
+
+    /**
+     * @return int
+     */
+    public function getStatusCode(): int
+    {
+        // TODO: Implement getStatusCode() method.
+        return $this->statusCode;
+    }
+
+    /**
+     * @param int $code
+     * @param string $reasonPhrase
+     * @return static
+     */
+    public function withStatus($code, $reasonPhrase = ''): static
+    {
        $this->statusCode = $code;
        $this->reasonPhrase = $reasonPhrase;
-		return $this;
-	}
+        return $this;
+    }


-	/**
-	 * @return string
-	 */
-	public function getReasonPhrase(): string
-	{
-		// TODO: Implement getReasonPhrase() method.
-		return $this->reasonPhrase;
-	}
+    /**
+     * @return string
+     */
+    public function getReasonPhrase(): string
+    {
+        // TODO: Implement getReasonPhrase() method.
+        return $this->reasonPhrase;
+    }


-	/**
-	 * @param $value
-	 * @return ResponseInterface
-	 */
-	public function withAccessControlAllowHeaders($value): ResponseInterface
-	{
-		return $this->withHeader('Access-Control-Allow-Headers', $value);
-	}
+    /**
+     * @param $value
+     * @return ResponseInterface
+     */
+    public function withAccessControlAllowHeaders($value): ResponseInterface
+    {
+        return $this->withHeader('Access-Control-Allow-Headers', $value);
+    }


-	/**
-	 * @param $value
-	 * @return ResponseInterface
-	 */
-	public function withAccessControlRequestMethod($value): ResponseInterface
-	{
-		return $this->withHeader('Access-Control-Request-Method', $value);
-	}
+    /**
+     * @param $value
+     * @return ResponseInterface
+     */
+    public function withAccessControlRequestMethod($value): ResponseInterface
+    {
+        return $this->withHeader('Access-Control-Request-Method', $value);
+    }


-	/**
-	 * @param $value
-	 * @return ResponseInterface
-	 */
-	public function withAccessControlAllowOrigin($value): ResponseInterface
-	{
-		return $this->withHeader('Access-Control-Allow-Origin', $value);
-	}
+    /**
+     * @param $value
+     * @return ResponseInterface
+     */
+    public function withAccessControlAllowOrigin($value): ResponseInterface
+    {
+        return $this->withHeader('Access-Control-Allow-Origin', $value);
+    }

 }