|
2 | 2 |
|
3 | 3 | namespace PhpMiddleware\BlockRobots; |
4 | 4 |
|
| 5 | +use Interop\Http\ServerMiddleware\DelegateInterface; |
| 6 | +use Interop\Http\ServerMiddleware\MiddlewareInterface; |
| 7 | +use PhpMiddleware\DoublePassCompatibilityTrait; |
5 | 8 | use Psr\Http\Message\ResponseInterface; |
6 | 9 | use Psr\Http\Message\ServerRequestInterface; |
7 | 10 | use Zend\Diactoros\Response; |
8 | 11 |
|
9 | | -class BlockRobotsMiddleware |
| 12 | +class BlockRobotsMiddleware implements MiddlewareInterface |
10 | 13 | { |
| 14 | + use DoublePassCompatibilityTrait; |
| 15 | + |
11 | 16 | const ROBOTS_HEADER = 'X-Robots-Tag'; |
12 | 17 |
|
13 | | - public function __invoke(ServerRequestInterface $request, ResponseInterface $response, callable $out = null) |
| 18 | + public function process(ServerRequestInterface $request, DelegateInterface $delegate) |
14 | 19 | { |
15 | | - if ($request->getUri()->getPath() === '/robots.txt') { |
16 | | - $newReponse = new Response('php://memory', 200, ['Content-Type' => 'text/plain']); |
17 | | - $newReponse->getBody()->write("User-Agent: *\nDisallow: /"); |
18 | | - |
19 | | - return $newReponse; |
| 20 | + if ($this->isRobotsTxt($request)) { |
| 21 | + return $this->createRobotsTxtResponse(); |
20 | 22 | } |
21 | | - /* @var $out ResponseInterface */ |
22 | | - $response = $out === null ? $response : $out($request, $response); |
23 | 23 |
|
| 24 | + $response = $delegate->process($request); |
| 25 | + |
| 26 | + return $this->attachRobotsHeaderTo($response); |
| 27 | + } |
| 28 | + |
| 29 | + private function createRobotsTxtResponse() |
| 30 | + { |
| 31 | + $response = new Response('php://memory', 200, ['Content-Type' => 'text/plain']); |
| 32 | + $response->getBody()->write("User-Agent: *\nDisallow: /"); |
| 33 | + |
| 34 | + return $response; |
| 35 | + } |
| 36 | + |
| 37 | + private function isRobotsTxt(ServerRequestInterface $request) |
| 38 | + { |
| 39 | + return $request->getUri()->getPath() === '/robots.txt'; |
| 40 | + } |
| 41 | + |
| 42 | + private function attachRobotsHeaderTo(ResponseInterface $response) |
| 43 | + { |
24 | 44 | return $response->withHeader(self::ROBOTS_HEADER, 'noindex, nofollow'); |
25 | 45 | } |
26 | 46 | } |
0 commit comments