Completed
Pull Request — master (#8)
by Tobias
03:56 queued 01:55
created

CachePlugin::getModifiedAt()   A

Complexity

Conditions 2
Paths 2

Size

Total Lines 9
Code Lines 5

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 0
CRAP Score 6

Importance

Changes 1
Bugs 0 Features 1
Metric Value
c 1
b 0
f 1
dl 0
loc 9
ccs 0
cts 5
cp 0
rs 9.6666
cc 2
eloc 5
nc 2
nop 1
crap 6
1
<?php
2
3
namespace Http\Client\Common\Plugin;
4
5
use Http\Client\Common\Plugin;
6
use Http\Message\StreamFactory;
7
use Http\Promise\FulfilledPromise;
8
use Psr\Cache\CacheItemInterface;
9
use Psr\Cache\CacheItemPoolInterface;
10
use Psr\Http\Message\RequestInterface;
11
use Psr\Http\Message\ResponseInterface;
12
use Symfony\Component\OptionsResolver\OptionsResolver;
13
14
/**
15
 * Allow for caching a response.
16
 *
17
 * @author Tobias Nyholm <[email protected]>
18
 */
19
final class CachePlugin implements Plugin
20
{
21
    /**
22
     * @var CacheItemPoolInterface
23
     */
24
    private $pool;
25
26
    /**
27
     * @var StreamFactory
28
     */
29
    private $streamFactory;
30
31
    /**
32
     * @var array
33
     */
34
    private $config;
35
36
    /**
37
     * @param CacheItemPoolInterface $pool
38
     * @param StreamFactory          $streamFactory
39
     * @param array                  $config        {
40
     *
41
     *     @var bool $respect_cache_headers Whether to look at the cache directives or ignore them
42
     *     @var int $default_ttl If we do not respect cache headers or can't calculate a good ttl, use this value.
43
     * }
44
     */
45 6
    public function __construct(CacheItemPoolInterface $pool, StreamFactory $streamFactory, array $config = [])
46
    {
47 6
        $this->pool = $pool;
48 6
        $this->streamFactory = $streamFactory;
49
50 6
        $optionsResolver = new OptionsResolver();
51 6
        $this->configureOptions($optionsResolver);
52 6
        $this->config = $optionsResolver->resolve($config);
53 6
    }
54
55
    /**
56
     * {@inheritdoc}
57
     */
58 4
    public function handleRequest(RequestInterface $request, callable $next, callable $first)
59
    {
60 4
        $method = strtoupper($request->getMethod());
61
        // if the request not is cachable, move to $next
62 4
        if ($method !== 'GET' && $method !== 'HEAD') {
63 1
            return $next($request);
64
        }
65
66
        // If we can cache the request
67 3
        $key = $this->createCacheKey($request);
68 3
        $cacheItem = $this->pool->getItem($key);
69
70 3
        if ($cacheItem->isHit()) {
71
            $data = $cacheItem->get();
72
            if (isset($data['expiresAt']) && time() > $data['expiresAt']) {
73
                // This item is still valid according to previous cache headers
74
                return new FulfilledPromise($this->createResponseFromCacheItem($cacheItem));
75
            }
76
77
            // Add headers to ask the server if this cache is still valid
78
            if ($mod = $this->getModifiedAt($cacheItem)) {
79
                $mod = new \DateTime('@'.$mod);
80
                $mod->setTimezone(new \DateTimeZone('GMT'));
81
                $request = $request->withHeader('If-Modified-Since', sprintf('%s GMT', $mod->format('l, d-M-y H:i:s')));
82
            }
83
84
            if ($etag = $this->getETag($cacheItem)) {
85
                $request = $request->withHeader('If-None-Match', $etag);
86
            }
87
        }
88
89 3
        return $next($request)->then(function (ResponseInterface $response) use ($cacheItem) {
90 3
            if (304 === $response->getStatusCode()) {
91
                if (!$cacheItem->isHit()) {
92
                    // We do not have the item in cache. We can return the cached response.
93
                    return $response;
94
                }
95
96
                // The cached response we have is still valid
97
                $data = $cacheItem->get();
98
                $maxAge = $this->getMaxAge($response);
99
                $data['expiresAt'] = time() + $maxAge;
100
                $cacheItem->set($data)->expiresAfter($this->config['cache_lifetime'] + $maxAge);
101
                $this->pool->save($cacheItem);
102
103
                return $this->createResponseFromCacheItem($cacheItem);
104
            }
105
106 3
            if ($this->isCacheable($response)) {
107 2
                $bodyStream = $response->getBody();
108 2
                $body = $bodyStream->__toString();
109 2
                if ($bodyStream->isSeekable()) {
110 2
                    $bodyStream->rewind();
111 2
                } else {
112
                    $response = $response->withBody($this->streamFactory->createStream($body));
113
                }
114
115 2
                $maxAge = $this->getMaxAge($response);
116
                $cacheItem
117 2
                    ->expiresAfter($this->config['cache_lifetime'] + $maxAge)
118 2
                    ->set([
119 2
                    'response' => $response,
120 2
                    'body' => $body,
121 2
                    'expiresAt' => time() + $maxAge,
122 2
                    'createdAt' => time(),
123 2
                    'etag' => $response->getHeader('ETag'),
124 2
                ]);
125 2
                $this->pool->save($cacheItem);
126 2
            }
127
128 3
            return $response;
129 3
        });
130
    }
131
132
    /**
133
     * Verify that we can cache this response.
134
     *
135
     * @param ResponseInterface $response
136
     *
137
     * @return bool
138
     */
139 3
    protected function isCacheable(ResponseInterface $response)
140
    {
141 3
        if (!in_array($response->getStatusCode(), [200, 203, 300, 301, 302, 404, 410])) {
142 1
            return false;
143
        }
144 2
        if (!$this->config['respect_cache_headers']) {
145
            return true;
146
        }
147 2
        if ($this->getCacheControlDirective($response, 'no-store') || $this->getCacheControlDirective($response, 'private')) {
148
            return false;
149
        }
150
151 2
        return true;
152
    }
153
154
    /**
155
     * Get the value of a parameter in the cache control header.
156
     *
157
     * @param ResponseInterface $response
158
     * @param string            $name     The field of Cache-Control to fetch
159
     *
160
     * @return bool|string The value of the directive, true if directive without value, false if directive not present
161
     */
162 2
    private function getCacheControlDirective(ResponseInterface $response, $name)
163
    {
164 2
        $headers = $response->getHeader('Cache-Control');
165 2
        foreach ($headers as $header) {
166 1
            if (preg_match(sprintf('|%s=?([0-9]+)?|i', $name), $header, $matches)) {
167
168
                // return the value for $name if it exists
169 1
                if (isset($matches[1])) {
170 1
                    return $matches[1];
171
                }
172
173
                return true;
174
            }
175 2
        }
176
177 2
        return false;
178
    }
179
180
    /**
181
     * @param RequestInterface $request
182
     *
183
     * @return string
184
     */
185 3
    private function createCacheKey(RequestInterface $request)
186
    {
187 3
        return md5($request->getMethod().' '.$request->getUri());
188
    }
189
190
    /**
191
     * Get a ttl in seconds. It could return null if we do not respect cache headers and got no defaultTtl.
192
     *
193
     * @param ResponseInterface $response
194
     *
195
     * @return int|null
196
     */
197 2
    private function getMaxAge(ResponseInterface $response)
198
    {
199 2
        if (!$this->config['respect_cache_headers']) {
200
            return $this->config['default_ttl'];
201
        }
202
203
        // check for max age in the Cache-Control header
204 2
        $maxAge = $this->getCacheControlDirective($response, 'max-age');
205 2
        if (!is_bool($maxAge)) {
206 1
            $ageHeaders = $response->getHeader('Age');
207 1
            foreach ($ageHeaders as $age) {
208 1
                return $maxAge - ((int) $age);
209
            }
210
211
            return (int) $maxAge;
212
        }
213
214
        // check for ttl in the Expires header
215 1
        $headers = $response->getHeader('Expires');
216 1
        foreach ($headers as $header) {
217
            return (new \DateTime($header))->getTimestamp() - (new \DateTime())->getTimestamp();
218 1
        }
219
220 1
        return $this->config['default_ttl'];
221
    }
222
223
    /**
224
     * Configure an options resolver.
225
     *
226
     * @param OptionsResolver $resolver
227
     */
228 6
    private function configureOptions(OptionsResolver $resolver)
229
    {
230 6
        $resolver->setDefaults([
231 6
            'cache_lifetime' => 2592000, // 30 days
232 6
            'default_ttl' => null,
233 6
            'respect_cache_headers' => true,
234 6
        ]);
235
236 6
        $resolver->setAllowedTypes('cache_lifetime', 'int');
237 6
        $resolver->setAllowedTypes('default_ttl', ['int', 'null']);
238 6
        $resolver->setAllowedTypes('respect_cache_headers', 'bool');
239 6
    }
240
241
    /**
242
     * @param CacheItemInterface $cacheItem
243
     *
244
     * @return ResponseInterface
245
     */
246
    private function createResponseFromCacheItem(CacheItemInterface $cacheItem)
247
    {
248
        $data = $cacheItem->get();
249
250
        /** @var ResponseInterface $response */
251
        $response = $data['response'];
252
        $response = $response->withBody($this->streamFactory->createStream($data['body']));
253
254
        return $response;
255
    }
256
257
    /**
258
     * Get the timestamp when the cached response was stored.
259
     *
260
     * @param CacheItemInterface $cacheItem
261
     *
262
     * @return int|null
263
     */
264
    private function getModifiedAt(CacheItemInterface $cacheItem)
265
    {
266
        $data = $cacheItem->get();
267
        if (!isset($data['createdAt'])) {
268
            return;
269
        }
270
271
        return $data['createdAt'];
272
    }
273
274
    /**
275
     * Get the ETag from the cached response.
276
     *
277
     * @param CacheItemInterface $cacheItem
278
     *
279
     * @return string|null
280
     */
281
    private function getETag(CacheItemInterface $cacheItem)
282
    {
283
        $data = $cacheItem->get();
284
        if (!isset($data['etag'])) {
285
            return;
286
        }
287
288
        if (is_array($data['etag'])) {
289
            foreach ($data['etag'] as $etag) {
290
                if (!empty($etag)) {
291
                    return $etag;
292
                }
293
            }
294
295
            return;
296
        }
297
298
        return $data['etag'];
299
    }
300
}
301