<?php

namespace Modules\Core\Customer\Extensions;

use Lunar\Admin\Filament\Resources\CustomerResource\RelationManagers\AddressRelationManager as BaseAddressRelationManager;
use Lunar\Admin\Filament\Resources\CustomerResource\RelationManagers\UserRelationManager as BaseUserRelationManager;
use Lunar\Admin\Support\Extending\BaseExtension;
use Modules\Core\Customer\RelationManagers\AddressRelationManager;
use Modules\Core\Customer\RelationManagers\UserRelationManager;

class CustomerResourceExtension extends BaseExtension
{
    public function getRelations(array $relations): array
    {
        return array_map(
            fn ($relation) => match ($relation) {
                BaseAddressRelationManager::class => AddressRelationManager::class,
                BaseUserRelationManager::class => UserRelationManager::class,
                default => $relation,
            },
            $relations
        );
    }
}