jeremypeter
--- a/src/Controller/DisplayController.php
+++ b/src/Controller/DisplayController.php
+   * @param string $view_mode
+   *   The view mode in which the media entity is being displayed.
+   * @param \Drupal\Core\Config\ImmutableConfig $config
+   *   The media alias display configuration.
+   * @param int $depth
+   *   The current recursion depth (default: 0).
+   * @param int $maxDepth
+   *   The maximum allowed recursion depth (default: 10).
+   *
+   * @return \Drupal\file\FileInterface|null
+   *   The file entity if found, otherwise NULL.
+   *
+   * @throws \RuntimeException
+   *   Thrown if the maximum recursion depth is reached.
+   */
+  protected function getFileEntity($source, $media, $view_mode, $config, $depth = 0, $maxDepth = 10)
--- a/src/Controller/DisplayController.php
+++ b/src/Controller/DisplayController.php
+      $source = $field_entity->getSource();
+      $media = $field_entity;
+
+      // Recursively call the function with the updated source and media, increasing the depth.
+      return $this->getFileEntity($source, $media, $view_mode, $config, $depth + 1, $maxDepth);
+    }
+
+    // Ensure the source is an instance of FileMediaSource.
+    if ($max_depth_reached || !($source instanceof FileMediaSource)) {
+      // Log a notice if the media item does not have a file media source.
+      $this->logger
+        ->notice('Media item "@media_entity_id" does not have a file media source', [
+          '@media_entity_id' => $media->id(),
+        ]);
+      // Update the render cache and return the response.
+      return $this->updateRenderCache(parent::__invoke($media, $view_mode), $config);
--- a/src/Controller/DisplayController.php
+++ b/src/Controller/DisplayController.php
+    }
+
+    // Ensure the source is an instance of FileMediaSource.
+    if ($max_depth_reached || !($source instanceof FileMediaSource)) {
+      // Log a notice if the media item does not have a file media source.
+      $this->logger
+        ->notice('Media item "@media_entity_id" does not have a file media source', [
+          '@media_entity_id' => $media->id(),
+        ]);
+      // Update the render cache and return the response.
+      return $this->updateRenderCache(parent::__invoke($media, $view_mode), $config);
+    }
+
+    // Retrieve the file entity from the media source field.
+    $file = $media->get($source->getConfiguration()['source_field'])->entity;
+    return $file;