yiisoft / demo

blog/src/Middleware/ApiDataWrapper.php

<?php

declare(strict_types=1);

namespace App\Middleware;

use Psr\Http\Message\ResponseInterface;

The type Psr\Http\Message\ResponseInterface was not found. Maybe you did not declare it correctly or list all dependencies?

use Psr\Http\Message\ServerRequestInterface;

The type Psr\Http\Message\ServerRequestInterface was not found. Maybe you did not declare it correctly or list all dependencies?

use Psr\Http\Server\MiddlewareInterface;

The type Psr\Http\Server\MiddlewareInterface was not found. Maybe you did not declare it correctly or list all dependencies?

use Psr\Http\Server\RequestHandlerInterface;

The type Psr\Http\Server\RequestHandlerInterface was not found. Maybe you did not declare it correctly or list all dependencies?

use Yiisoft\DataResponse\DataResponse;

The type Yiisoft\DataResponse\DataResponse was not found. Maybe you did not declare it correctly or list all dependencies?

use OpenApi\Attributes as OA;

The type OpenApi\Attributes was not found. Maybe you did not declare it correctly or list all dependencies?

#[OA\Schema(
    schema: 'Response',
    properties: [
        new OA\Property(property: 'status', type: 'string', enum: ['success', 'failed']),
        new OA\Property(property: 'error', type:'object', nullable: true),
        new OA\Property(property: 'data', type: 'object', nullable: true),
    ]
)]
final class ApiDataWrapper implements MiddlewareInterface
{
    public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface
    {
        $response = $handler->handle($request);
        if ($response instanceof DataResponse) {
            $data = $response->getData();
            if ($response->getStatusCode() !== 200) {
                if (is_string($data) && !empty($data)) {
                    $message = $data;
                } else {
                    $message = 'Unknown error';
                }

                return $response->withData([
                    'status' => 'failed',
                    'error' => ['message' => $message, 'status' => $response->getStatusCode()],
                ]);
            }

            return $response->withData(['status' => 'success', 'data' => $data]);
        }

        return $response;
    }
}