mirror of
https://github.com/shlinkio/shlink.git
synced 2025-01-22 14:33:16 -06:00
Migrated QrCodeCacheMiddleware to psr-15 middleware
This commit is contained in:
parent
734dac9456
commit
6c87436a96
@ -3,9 +3,11 @@ namespace Shlinkio\Shlink\Core\Middleware;
|
|||||||
|
|
||||||
use Acelaya\ZsmAnnotatedServices\Annotation\Inject;
|
use Acelaya\ZsmAnnotatedServices\Annotation\Inject;
|
||||||
use Doctrine\Common\Cache\Cache;
|
use Doctrine\Common\Cache\Cache;
|
||||||
|
use Interop\Http\ServerMiddleware\DelegateInterface;
|
||||||
|
use Interop\Http\ServerMiddleware\MiddlewareInterface;
|
||||||
use Psr\Http\Message\ResponseInterface as Response;
|
use Psr\Http\Message\ResponseInterface as Response;
|
||||||
use Psr\Http\Message\ServerRequestInterface as Request;
|
use Psr\Http\Message\ServerRequestInterface as Request;
|
||||||
use Zend\Stratigility\MiddlewareInterface;
|
use Zend\Diactoros\Response as DiactResp;
|
||||||
|
|
||||||
class QrCodeCacheMiddleware implements MiddlewareInterface
|
class QrCodeCacheMiddleware implements MiddlewareInterface
|
||||||
{
|
{
|
||||||
@ -26,44 +28,29 @@ class QrCodeCacheMiddleware implements MiddlewareInterface
|
|||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Process an incoming request and/or response.
|
* Process an incoming server request and return a response, optionally delegating
|
||||||
*
|
* to the next middleware component to create the response.
|
||||||
* Accepts a server-side request and a response instance, and does
|
|
||||||
* something with them.
|
|
||||||
*
|
|
||||||
* If the response is not complete and/or further processing would not
|
|
||||||
* interfere with the work done in the middleware, or if the middleware
|
|
||||||
* wants to delegate to another process, it can use the `$out` callable
|
|
||||||
* if present.
|
|
||||||
*
|
|
||||||
* If the middleware does not return a value, execution of the current
|
|
||||||
* request is considered complete, and the response instance provided will
|
|
||||||
* be considered the response to return.
|
|
||||||
*
|
|
||||||
* Alternately, the middleware may return a response instance.
|
|
||||||
*
|
|
||||||
* Often, middleware will `return $out();`, with the assumption that a
|
|
||||||
* later middleware will return a response.
|
|
||||||
*
|
*
|
||||||
* @param Request $request
|
* @param Request $request
|
||||||
* @param Response $response
|
* @param DelegateInterface $delegate
|
||||||
* @param null|callable $out
|
*
|
||||||
* @return null|Response
|
* @return Response
|
||||||
*/
|
*/
|
||||||
public function __invoke(Request $request, Response $response, callable $out = null)
|
public function process(Request $request, DelegateInterface $delegate)
|
||||||
{
|
{
|
||||||
$cacheKey = $request->getUri()->getPath();
|
$cacheKey = $request->getUri()->getPath();
|
||||||
|
|
||||||
// If this QR code is already cached, just return it
|
// If this QR code is already cached, just return it
|
||||||
if ($this->cache->contains($cacheKey)) {
|
if ($this->cache->contains($cacheKey)) {
|
||||||
$qrData = $this->cache->fetch($cacheKey);
|
$qrData = $this->cache->fetch($cacheKey);
|
||||||
|
$response = new DiactResp();
|
||||||
$response->getBody()->write($qrData['body']);
|
$response->getBody()->write($qrData['body']);
|
||||||
return $response->withHeader('Content-Type', $qrData['content-type']);
|
return $response->withHeader('Content-Type', $qrData['content-type']);
|
||||||
}
|
}
|
||||||
|
|
||||||
// If not, call the next middleware and cache it
|
// If not, call the next middleware and cache it
|
||||||
/** @var Response $resp */
|
/** @var Response $resp */
|
||||||
$resp = $out($request, $response);
|
$resp = $delegate->process($request);
|
||||||
$this->cache->save($cacheKey, [
|
$this->cache->save($cacheKey, [
|
||||||
'body' => $resp->getBody()->__toString(),
|
'body' => $resp->getBody()->__toString(),
|
||||||
'content-type' => $resp->getHeaderLine('Content-Type'),
|
'content-type' => $resp->getHeaderLine('Content-Type'),
|
||||||
|
@ -4,7 +4,9 @@ namespace ShlinkioTest\Shlink\Core\Middleware;
|
|||||||
use Doctrine\Common\Cache\ArrayCache;
|
use Doctrine\Common\Cache\ArrayCache;
|
||||||
use Doctrine\Common\Cache\Cache;
|
use Doctrine\Common\Cache\Cache;
|
||||||
use PHPUnit\Framework\TestCase;
|
use PHPUnit\Framework\TestCase;
|
||||||
|
use Prophecy\Argument;
|
||||||
use Shlinkio\Shlink\Core\Middleware\QrCodeCacheMiddleware;
|
use Shlinkio\Shlink\Core\Middleware\QrCodeCacheMiddleware;
|
||||||
|
use ShlinkioTest\Shlink\Common\Util\TestUtils;
|
||||||
use Zend\Diactoros\Response;
|
use Zend\Diactoros\Response;
|
||||||
use Zend\Diactoros\ServerRequestFactory;
|
use Zend\Diactoros\ServerRequestFactory;
|
||||||
use Zend\Diactoros\Uri;
|
use Zend\Diactoros\Uri;
|
||||||
@ -29,18 +31,15 @@ class QrCodeCacheMiddlewareTest extends TestCase
|
|||||||
/**
|
/**
|
||||||
* @test
|
* @test
|
||||||
*/
|
*/
|
||||||
public function noCachedPathFallbacksToNextMiddleware()
|
public function noCachedPathFallsBackToNextMiddleware()
|
||||||
{
|
{
|
||||||
$isCalled = false;
|
$delegate = TestUtils::createDelegateMock();
|
||||||
$this->middleware->__invoke(
|
$this->middleware->process(ServerRequestFactory::fromGlobals()->withUri(
|
||||||
ServerRequestFactory::fromGlobals(),
|
new Uri('/foo/bar')
|
||||||
new Response(),
|
), $delegate->reveal());
|
||||||
function ($req, $resp) use (&$isCalled) {
|
|
||||||
$isCalled = true;
|
$this->assertTrue($this->cache->contains('/foo/bar'));
|
||||||
return $resp;
|
$delegate->process(Argument::any())->shouldHaveBeenCalledTimes(1);
|
||||||
}
|
|
||||||
);
|
|
||||||
$this->assertTrue($isCalled);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -51,19 +50,17 @@ class QrCodeCacheMiddlewareTest extends TestCase
|
|||||||
$isCalled = false;
|
$isCalled = false;
|
||||||
$uri = (new Uri())->withPath('/foo');
|
$uri = (new Uri())->withPath('/foo');
|
||||||
$this->cache->save('/foo', ['body' => 'the body', 'content-type' => 'image/png']);
|
$this->cache->save('/foo', ['body' => 'the body', 'content-type' => 'image/png']);
|
||||||
|
$delegate = TestUtils::createDelegateMock();
|
||||||
|
|
||||||
$resp = $this->middleware->__invoke(
|
$resp = $this->middleware->process(
|
||||||
ServerRequestFactory::fromGlobals()->withUri($uri),
|
ServerRequestFactory::fromGlobals()->withUri($uri),
|
||||||
new Response(),
|
$delegate->reveal()
|
||||||
function ($req, $resp) use (&$isCalled) {
|
|
||||||
$isCalled = true;
|
|
||||||
return $resp;
|
|
||||||
}
|
|
||||||
);
|
);
|
||||||
|
|
||||||
$this->assertFalse($isCalled);
|
$this->assertFalse($isCalled);
|
||||||
$resp->getBody()->rewind();
|
$resp->getBody()->rewind();
|
||||||
$this->assertEquals('the body', $resp->getBody()->getContents());
|
$this->assertEquals('the body', $resp->getBody()->getContents());
|
||||||
$this->assertEquals('image/png', $resp->getHeaderLine('Content-Type'));
|
$this->assertEquals('image/png', $resp->getHeaderLine('Content-Type'));
|
||||||
|
$delegate->process(Argument::any())->shouldHaveBeenCalledTimes(0);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user