1<?php 2 3/** 4 * webtrees: online genealogy 5 * Copyright (C) 2021 webtrees development team 6 * This program is free software: you can redistribute it and/or modify 7 * it under the terms of the GNU General Public License as published by 8 * the Free Software Foundation, either version 3 of the License, or 9 * (at your option) any later version. 10 * This program is distributed in the hope that it will be useful, 11 * but WITHOUT ANY WARRANTY; without even the implied warranty of 12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 13 * GNU General Public License for more details. 14 * You should have received a copy of the GNU General Public License 15 * along with this program. If not, see <https://www.gnu.org/licenses/>. 16 */ 17 18declare(strict_types=1); 19 20namespace Fisharebest\Webtrees\Http\Middleware; 21 22use Psr\Http\Message\ResponseInterface; 23use Psr\Http\Message\ServerRequestInterface; 24use Psr\Http\Server\MiddlewareInterface; 25use Psr\Http\Server\RequestHandlerInterface; 26 27/** 28 * Middleware to add a "Content-Length" header to a response. 29 */ 30class ContentLength implements MiddlewareInterface 31{ 32 /** 33 * @param ServerRequestInterface $request 34 * @param RequestHandlerInterface $handler 35 * 36 * @return ResponseInterface 37 */ 38 public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface 39 { 40 $response = $handler->handle($request); 41 42 if ($response->hasHeader('content-length')) { 43 return $response; 44 } 45 46 $content_length = $response->getBody()->getSize(); 47 48 if ($content_length === null) { 49 return $response; 50 } 51 52 return $response 53 ->withHeader('content-length', (string) $content_length); 54 } 55} 56