media_entity_file_redirect.patch

media_entity_file_redirect.patch

diff --git a/src/Controller/MediaEntityFileRedirectController.php b/src/Controller/MediaEntityFileRedirectController.php
index 9e9c31d..b112b8f 100644
--- a/src/Controller/MediaEntityFileRedirectController.php
+++ b/src/Controller/MediaEntityFileRedirectController.php
@@ -55,6 +55,12 @@ class MediaEntityFileRedirectController implements ContainerInjectionInterface {
     if ($mediaType && $mediaType->getSource() instanceof File && $mediaType->getThirdPartySetting('media_entity_file_redirect', 'enabled', FALSE)) {
       // Now load the file and return a redirect response to the file URL.
       $fid = $media->getSource()->getSourceFieldValue($media);
+      if (empty($fid) && !is_null($media->field_document)) {
+        $fid = $media->field_document->target_id;
+      }
+      if (empty($fid) && !is_null($media->field_media_file)) {
+        $fid = $media->field_media_file->target_id;
+      }
       if ($fid) {
         /** @var \Drupal\file\FileInterface $file */
         $file = $this->entityTypeManager->getStorage('file')->load($fid);