Fixes awcodes#529

config/curator.php

@@ -30,10 +30,10 @@
         'fallbacks' => [],
         'route_path' => 'curator',
     ],
+    'intermediate_model' => null,
     'image_crop_aspect_ratio' => null,
     'image_resize_mode' => null,
     'image_resize_target_height' => null,
-    'image_resize_target_width' => null,
     'is_limited_to_directory' => false,
     'is_tenant_aware' => true,
     'tenant_ownership_relationship_name' => 'tenant',

src/Components/Forms/CuratorPicker.php

@@ -549,6 +549,7 @@ public function relationship(string | Closure $relationshipName, string | Closur
                 return;
             }
 
+
             if ($component->isMultiple()) {
                 if ($relationship instanceof BelongsToMany) {
                     $orderColumn = $component->getOrderColumn();

src/Components/Tables/CuratorColumn.php

@@ -25,11 +25,12 @@ public function getMedia(): Media | Collection | array | null
 
         if (! is_a($record, Media::class)) {
             $state = $this->getState();
+
             if (is_a($state, Collection::class)) {
                 $state = $state->take($this->limit);
 
-                if (! is_null($state) && is_related_to_media_through_pivot(get_class(($state->first())), Media::class)) {
-                    $mediaIds = collect($state)->map(fn ($model) => $model?->media_id)->toArray();
+                if (! is_null($state) && is_related_to_media_through_pivot($state->first())) {
+                    $mediaIds = $state->map(fn ($model) => $model?->media_id)->toArray();
 
                     return Media::whereIn('id', $mediaIds)->get();
                 }

src/helpers.php

@@ -4,7 +4,6 @@
 
 use Awcodes\Curator\Models\Media;
 use Illuminate\Database\Eloquent\Collection;
-use Illuminate\Database\Eloquent\Relations\BelongsTo;
 use Illuminate\Support\Facades\Request;
 use Illuminate\Support\Str;
 
@@ -18,14 +17,17 @@ function is_media_resizable(string $ext): bool
 if (! function_exists('get_media_items')) {
     function get_media_items(array | Media | int $ids): Collection | array
     {
-        if ($ids instanceof Media) {
+        $mediaModel = config('curator.model');
+
+        if ($ids instanceof $mediaModel) {
             return [$ids];
         }
 
-        if (is_array($ids) && is_related_to_media_through_pivot(get_class(current($ids)), Media::class)) {
+
+        if (is_array($ids) && is_related_to_media_through_pivot(current($ids))) {
             $mediaIds = collect($ids)->map(fn ($model) => $model?->media_id)->toArray();
 
-            return Media::whereIn('id', $mediaIds)->get();
+            return config('curator.model')::whereIn('id', $mediaIds)->get();
         }
 
         $ids = array_values($ids);
@@ -47,25 +49,11 @@ function get_media_items(array | Media | int $ids): Collection | array
 }
 
 if (! function_exists('is_related_to_media_through_pivot')) {
-    function is_related_to_media_through_pivot(string $modelClass, string $relatedClass): bool
+    function is_related_to_media_through_pivot(mixed $type): bool
     {
-        $model = new $modelClass;
-        $reflector = new \ReflectionClass($model);
-        $methods = $reflector->getMethods(\ReflectionMethod::IS_PUBLIC);
-        foreach ($methods ?? [] as $method) {
-            if ($method?->class === $modelClass) {
-                $returnType = $method?->getReturnType();
-                if ($returnType) {
-                    $relationInstance = $model->{$method->getName()}();
-                    if ($relationInstance instanceof BelongsTo &&
-                        get_class($relationInstance->getRelated()) === $relatedClass) {
-                        return true;
-                    }
-                }
-            }
-        }
+        $intermediateModelType = config('curator.intermediate_model');
 
-        return false;
+        return (! is_null(config('curator.intermediate_model'))) && $type instanceof $intermediateModelType;
     }
 }