ServiceProvider.php

<?php

namespace Butler\Audit;

use Butler\Audit\Bus\Dispatcher as BusDispatcher;
use Butler\Audit\Facades\Auditor;
use Illuminate\Bus\Dispatcher as BaseBusDispatcher;
use Illuminate\Http\Client\PendingRequest;
use Illuminate\Queue\Events\JobProcessed;
use Illuminate\Support\Facades\Queue;
use Illuminate\Support\ServiceProvider as BaseServiceProvider;

class ServiceProvider extends BaseServiceProvider
{
    public function register()
    {
        $this->addPendingRequestMacro();
        $this->addDefaultInitiatorResolver();
        $this->extendBusDispatcher();
    }

    public function boot()
    {
        $this->publishes([__DIR__ . '/../config/butler.php' => config_path('butler.php')], 'config');

        $this->listenForJobProcessedEvent();
    }

    private function addPendingRequestMacro(): void
    {
        PendingRequest::macro(
            'withCorrelation',
            fn () => $this->withHeaders(Auditor::httpHeaders())
        );
    }

    private function addDefaultInitiatorResolver(): void
    {
        if (config('butler.audit.default_initiator_resolver') === false) {
            return;
        }

        $resolver = $this->app->runningInConsole()
            ? fn () => ['console', ['hostname' => gethostname()]]
            : fn () => [request()->ip(), ['userAgent' => request()->userAgent()]];

        Auditor::initiatorResolver($resolver);
    }

    private function extendBusDispatcher()
    {
        if (config('butler.audit.extend_bus_dispatcher') === false) {
            return;
        }

        $this->app->extend(
            BaseBusDispatcher::class,
            fn ($dispatcher, $app) => new BusDispatcher($app, $dispatcher)
        );
    }

    public function listenForJobProcessedEvent()
    {
        if ($this->app->runningInConsole()) {
            Queue::after(function (JobProcessed $event) {
                Auditor::correlationId(null);
                Auditor::correlationTrail(null);
            });
        }
    }
}