From 9a51bc6901c155ba76497219fa65fdca9be7e458 Mon Sep 17 00:00:00 2001
From: Felix <fxsth@pm.me>
Date: Sun, 24 Sep 2023 21:47:28 +0200
Subject: [PATCH] Fixed downloading selected movie file version

---
 Web/ClientApp/src/components/DownloadButton.jsx |  8 +++++++-
 Web/ClientApp/src/components/Movies.js          |  2 +-
 Web/Controllers/DownloadController.cs           |  4 ++--
 Web/Services/DownloadService.cs                 | 10 +++++++---
 4 files changed, 17 insertions(+), 7 deletions(-)

diff --git a/Web/ClientApp/src/components/DownloadButton.jsx b/Web/ClientApp/src/components/DownloadButton.jsx
index 69dc913..10e4f01 100644
--- a/Web/ClientApp/src/components/DownloadButton.jsx
+++ b/Web/ClientApp/src/components/DownloadButton.jsx
@@ -7,7 +7,8 @@ export default class DownloadButton extends React.Component {
         this.state = {
             mediaKey: this.props.mediaKey,
             isLoading: false,
-            mediatype: this.props.mediaType
+            mediatype: this.props.mediaType,
+            mediaFileKey: this.props.mediaFileKey
         };
     }
 
@@ -39,6 +40,11 @@ export default class DownloadButton extends React.Component {
                 season: this.props.season
             })
         }
+        if (typeof this.props.mediaFileKey !== 'undefined') {
+            input = input + '?' + new URLSearchParams({
+                mediaFileKey: this.props.mediaFileKey
+            })
+        }
         return fetch(input, settings)
             .then(response => {
                 if (response.status >= 200 && response.status < 300) {
diff --git a/Web/ClientApp/src/components/Movies.js b/Web/ClientApp/src/components/Movies.js
index 97ca0c6..65608a5 100644
--- a/Web/ClientApp/src/components/Movies.js
+++ b/Web/ClientApp/src/components/Movies.js
@@ -127,7 +127,7 @@ export class Movies extends Component {
                             <td>{mediaFile.videoCodec}</td>
                             <td>{mediaFile.videoResolution}</td>
                             <td>{this.humanizeByteSize(mediaFile.totalBytes)}</td>
-                            <td><DownloadButton mediaType='movie' mediaKey={movie.ratingKey}>Download</DownloadButton></td>
+                            <td><DownloadButton mediaType='movie' mediaKey={movie.ratingKey} mediaFileKey={mediaFile.downloadUri}>Download</DownloadButton></td>
                         </tr>)
                     
                 )}
diff --git a/Web/Controllers/DownloadController.cs b/Web/Controllers/DownloadController.cs
index 77756c7..3f78572 100644
--- a/Web/Controllers/DownloadController.cs
+++ b/Web/Controllers/DownloadController.cs
@@ -31,13 +31,13 @@ public async Task<IEnumerable<DownloadElementResource>> GetPendingDownloads()
     }
 
     [HttpPost("movie/{key}")]
-    public async Task DownloadMovie( string key, [FromQuery] string mediaFileKey)
+    public async Task DownloadMovie( string key, [FromQuery] string? mediaFileKey)
     {
         await _downloadService.DownloadMovie(key, mediaFileKey);
     }
     
     [HttpPost("episode/{key}")]
-    public async Task DownloadEpisode( string key, [FromQuery] string mediaFileKey)
+    public async Task DownloadEpisode( string key, [FromQuery] string? mediaFileKey)
     {
         await _downloadService.DownloadEpisode(key, mediaFileKey);
     }
diff --git a/Web/Services/DownloadService.cs b/Web/Services/DownloadService.cs
index 992e834..1b7a673 100644
--- a/Web/Services/DownloadService.cs
+++ b/Web/Services/DownloadService.cs
@@ -51,7 +51,7 @@ public IReadOnlyCollection<DownloadElement> GetAll()
             return returnList;
         }
 
-        private async Task<DownloadElement> CreateDownloadElement(string key, string mediaFileKey, ElementType elementType)
+        private async Task<DownloadElement> CreateDownloadElement(string key, string? mediaFileKey, ElementType elementType)
         {
             using (var scope = _scopeFactory.CreateScope())
             {
@@ -60,7 +60,11 @@ private async Task<DownloadElement> CreateDownloadElement(string key, string med
                 IMediaElement? mediaElement = await GetMediaElement(unitOfWork, elementType, key);
                 if (mediaElement == null)
                     throw new ArgumentException();
-                MediaFile? mediaFile = mediaElement.MediaFiles.FirstOrDefault(x => x.DownloadUri == mediaFileKey);
+                MediaFile? mediaFile;
+                if (mediaFileKey == null)
+                    mediaFile = await SelectMediaFile(mediaElement.MediaFiles, settingsService);
+                else 
+                    mediaFile = mediaElement.MediaFiles.FirstOrDefault(x => x.DownloadUri == mediaFileKey);
                 if (mediaElement == null)
                 {
                     _logger.LogError("Could not prepare download of {0} due to missing media file.", mediaElement!.Title);
@@ -265,7 +269,7 @@ private void StartDownloaderIfNotActive()
             }
         }
 
-        private async Task<MediaFile> SelectMediaFile(IEnumerable<MediaFile> mediaFiles, SettingsService settingsService)
+        private async Task<MediaFile> SelectMediaFile(IEnumerable<MediaFile> mediaFiles, ISettingsService settingsService)
         {
             var preferredResolution = await settingsService.GetPreferredResolution();
             var preferredVideoCodec = await settingsService.GetPreferredVideoCodec();