BeforeModelWrapper.php

<?php

declare(strict_types=1);

namespace Cortex\Agents\Middleware;

use Closure;
use Throwable;
use Cortex\Events\MiddlewareEnd;
use Cortex\Events\MiddlewareError;
use Cortex\Events\MiddlewareStart;
use Cortex\Pipeline\RuntimeConfig;
use Cortex\Support\Traits\CanPipe;
use Cortex\Support\Traits\DispatchesEvents;
use Cortex\Events\Contracts\MiddlewareEvent;
use Cortex\Agents\Contracts\BeforeModelMiddleware;

/**
 * Wrapper that delegates to beforeModel() if it exists, otherwise handlePipeable().
 * Used when middleware implements both BeforeModelMiddleware and AfterModelMiddleware.
 */
class BeforeModelWrapper implements BeforeModelMiddleware
{
    use CanPipe;
    use DispatchesEvents;

    public function __construct(
        protected BeforeModelMiddleware $middleware,
    ) {}

    public function handlePipeable(mixed $payload, RuntimeConfig $config, Closure $next): mixed
    {
        $this->dispatchEvent(new MiddlewareStart($this->middleware, $config, 'beforeModel'));

        try {
            $result = $this->middleware->beforeModel($payload, $config, $next);
        } catch (Throwable $e) {
            $this->dispatchEvent(new MiddlewareError($this->middleware, $config, 'beforeModel', $e));

            throw $e;
        }

        $this->dispatchEvent(new MiddlewareEnd($this->middleware, $config, 'beforeModel'));

        return $result;
    }

    public function beforeModel(mixed $payload, RuntimeConfig $config, Closure $next): mixed
    {
        return $this->handlePipeable($payload, $config, $next);
    }

    protected function eventBelongsToThisInstance(object $event): bool
    {
        return $event instanceof MiddlewareEvent && $event->middleware === $this->middleware;
    }
}