1
|
|
|
<?php |
2
|
|
|
|
3
|
|
|
namespace hiapi\Core\Http\Psr15\Middleware; |
4
|
|
|
|
5
|
|
|
use Psr\Http\Message\ResponseFactoryInterface; |
6
|
|
|
use Psr\Http\Message\ResponseInterface; |
7
|
|
|
use Psr\Http\Message\ServerRequestInterface; |
8
|
|
|
use Psr\Http\Server\MiddlewareInterface; |
9
|
|
|
use Psr\Http\Server\RequestHandlerInterface; |
10
|
|
|
|
11
|
|
|
class CorsMiddleware implements MiddlewareInterface |
12
|
|
|
{ |
13
|
|
|
protected $headers = [ |
14
|
|
|
'Access-Control-Allow-Origin' => ['*'], |
15
|
|
|
'Access-Control-Request-Method' => ['GET POST'], |
16
|
|
|
'Access-Control-Allow-Headers' => ['content-type'], |
17
|
|
|
'Access-Control-Max-Age' => ['600'], |
18
|
|
|
]; |
19
|
|
|
public bool $interceptOptionsRequests = false; |
|
|
|
|
20
|
|
|
private ResponseFactoryInterface $responseFactory; |
21
|
|
|
|
22
|
|
|
public function __construct(array $headers = [], ResponseFactoryInterface $responseFactory) |
23
|
|
|
{ |
24
|
|
|
foreach ($headers as $name => $value) { |
25
|
|
|
$this->addHeader($name, $value); |
26
|
|
|
} |
27
|
|
|
$this->responseFactory = $responseFactory; |
28
|
|
|
} |
29
|
|
|
|
30
|
|
|
public function addHeader($name, $value): self |
31
|
|
|
{ |
32
|
|
|
$this->headers[$name] = array_merge($this->headers[$name] ?? [], [$value]); |
33
|
|
|
|
34
|
|
|
return $this; |
35
|
|
|
} |
36
|
|
|
|
37
|
|
|
public function withHeader($name, $value): self |
38
|
|
|
{ |
39
|
|
|
return (clone $this)->addHeader($name, $value); |
40
|
|
|
} |
41
|
|
|
|
42
|
|
|
/** |
43
|
|
|
* @inheritDoc |
44
|
|
|
*/ |
45
|
|
|
public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface |
46
|
|
|
{ |
47
|
|
|
if ($this->interceptOptionsRequests && $request->getMethod() === 'OPTIONS') { |
48
|
|
|
return $this->addHeaders($this->responseFactory->createResponse(201)); |
49
|
|
|
} |
50
|
|
|
|
51
|
|
|
$response = $handler->handle($request); |
52
|
|
|
|
53
|
|
|
return $this->addHeaders($response); |
54
|
|
|
} |
55
|
|
|
|
56
|
|
|
private function addHeaders(ResponseInterface $response) |
57
|
|
|
{ |
58
|
|
|
foreach ($this->headers as $name => $headers) { |
59
|
|
|
$response = $response->withHeader($name, $headers); |
60
|
|
|
} |
61
|
|
|
|
62
|
|
|
return $response; |
63
|
|
|
} |
64
|
|
|
} |
65
|
|
|
|