Completed
Push — master ( 2b82a4...1acfba )
by Oscar
02:59
created

Robots::__invoke()   B

Complexity

Conditions 4
Paths 4

Size

Total Lines 22
Code Lines 11

Duplication

Lines 0
Ratio 0 %

Importance

Changes 3
Bugs 0 Features 0
Metric Value
c 3
b 0
f 0
dl 0
loc 22
rs 8.9197
cc 4
eloc 11
nc 4
nop 3
1
<?php
2
3
namespace Psr7Middlewares\Middleware;
4
5
use Psr\Http\Message\RequestInterface;
6
use Psr\Http\Message\ResponseInterface;
7
8
/**
9
 * Middleware to block robots search engine.
10
 */
11
class Robots
12
{
13
    const HEADER = 'X-Robots-Tag';
14
15
    private $allow = false;
16
17
    /**
18
     * Set whether search engines are allowed or not.
19
     * 
20
     * @param bool $allow
21
     */
22
    public function __construct($allow = false)
23
    {
24
        $this->allow = $allow;
25
    }
26
27
    /**
28
     * Execute the middleware.
29
     *
30
     * @param RequestInterface  $request
31
     * @param ResponseInterface $response
32
     * @param callable          $next
33
     *
34
     * @return ResponseInterface
35
     */
36
    public function __invoke(RequestInterface $request, ResponseInterface $response, callable $next)
37
    {
38
        if ($request->getUri()->getPath() === '/robots.txt') {
39
            if ($this->allow) {
40
                $response->getBody()->write("User-Agent: *\nAllow: /");
41
42
                return $response;
43
            }
44
45
            $response->getBody()->write("User-Agent: *\nDisallow: /");
46
47
            return $response;
48
        }
49
50
        $response = $next($request, $response);
51
52
        if ($this->allow) {
53
            return $response->withHeader(self::HEADER, 'index, follow');
54
        }
55
56
        return $response->withHeader(self::HEADER, 'noindex, nofollow, noarchive');
57
    }
58
}
59