Implement 'Download All' feature for search results

DJDoubleD · Oct 20, 2024 · 1da3ac7 · 1da3ac7
1 parent e7b3986
commit 1da3ac7
Show file tree

Hide file tree

Showing 4 changed files with 200 additions and 125 deletions.
diff --git a/QobuzDownloaderX/View/MainForm.cs b/QobuzDownloaderX/View/MainForm.cs
@@ -28,7 +28,7 @@ public QobuzDownloaderX()
             this.AutoSize = true;
             InitializeComponent();
             logger = new DownloadLogger(output, "MainForm");
-            RequestHandler.MainRequestHandlers[0].StaticDegreeOfParallelism = 1;
+            RequestHandler.MainRequestHandlers[0].StaticDegreeOfParallelism = 2;
             // Remove previous download error log
             logger.RemovePreviousErrorLog();
             _requests.SpeedReporter.Timeout = 1000;

diff --git a/QobuzDownloaderX/View/SearchForm.Designer.cs b/QobuzDownloaderX/View/SearchForm.Designer.cs
diff --git a/QobuzDownloaderX/View/SearchForm.cs b/QobuzDownloaderX/View/SearchForm.cs
@@ -8,6 +8,7 @@
 using System.Diagnostics;
 using System.Drawing;
 using System.IO;
+using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
 using System.Windows.Forms;
@@ -34,6 +35,38 @@ public SearchForm()
             searchTypeSelect.SelectedItem = "Album";
         }
 
+        private void DownloadAllButton_Click(object sender, EventArgs e)
+        {
+            try
+            {
+                downloadAllButton.Enabled = false;
+                List<string> downloadLinks = [];
+                for (int row = 0; row < resultsTableLayoutPanel.RowCount; row++)
+                {
+                    if (resultsTableLayoutPanel.GetControlFromPosition(3, row) is Button downloadButton)
+                    {
+                        // Get the download link from the button's Tag property
+                        string downloadLink = downloadButton.Tag?.ToString();
+                        if (!string.IsNullOrEmpty(downloadLink))
+                            downloadLinks.Add(downloadLink);
+                    }
+                }
+                Globals.QbdlxForm.downloadUrl.Invoke(new Action(() => Globals.QbdlxForm.downloadUrl.Text = downloadLinks.FirstOrDefault()));
+                this.Close();
+                Globals.QbdlxForm.StartLinkItemDownload([.. downloadLinks]);
+            }
+            catch (Exception ex)
+            {
+                List<string> errorLines = ["Error starting download for all items", ex.ToString(), ""];
+                System.IO.File.AppendAllLines(errorLog, errorLines);
+                MessageBox.Show("Error starting download for all items\nlog saved to " + errorLog, "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
+            }
+            finally
+            {
+                downloadAllButton.Enabled = true;
+            }
+        }
+
         private void ResetResultsTableLayoutPanel()
         {
             if (resultsTableLayoutPanel != null)
@@ -61,6 +94,8 @@ private void ResetResultsTableLayoutPanel()
             resultsTableLayoutPanel.CellPaint += ResultsTableLayoutPanel_CellPaint;
 
             containerScrollPanel.Controls.Add(resultsTableLayoutPanel);
+            downloadAllButton.Visible = false;
+            downloadAllButton.Enabled = false;
         }
 
         private void ShowAndLogSearchResultError(Exception ex)
@@ -133,6 +168,12 @@ private void Fill_AlbumResultsTablePanel(SearchResult searchResult)
                 HiRes = album.Hires.GetValueOrDefault(),
                 TrackCount = album.TracksCount.GetValueOrDefault()
             });
+            // Show and enable the "Download All" button if there are rows
+            if (resultsTableLayoutPanel.RowCount > 0)
+            {
+                downloadAllButton.Visible = true;
+                downloadAllButton.Enabled = true;
+            }
         }
 
         private void Fill_TrackResultsTablePanel(SearchResult searchResult)
@@ -150,6 +191,12 @@ private void Fill_TrackResultsTablePanel(SearchResult searchResult)
                 ReleaseDate = track.Album.ReleaseDateOriginal.GetValueOrDefault().ToString("yyyy-MM-dd"),
                 HiRes = track.Hires.GetValueOrDefault()
             });
+            // Show and enable the "Download All" button if there are rows
+            if (resultsTableLayoutPanel.RowCount > 0)
+            {
+                downloadAllButton.Visible = true;
+                downloadAllButton.Enabled = true;
+            }
         }
 
         private void FillResultsTablePanel<T>(IEnumerable<T> items, Func<T, SearchResultRow> createSearchResultRow)
@@ -365,6 +412,7 @@ private Button CreateDownloadButton(string webPlayerUrl)
                 UseVisualStyleBackColor = false,
                 Height = 23,
                 Anchor = AnchorStyles.None,
+                Tag = webPlayerUrl,
                 TextAlign = ContentAlignment.MiddleCenter
             };
             downloadButton.FlatAppearance.BorderSize = 0;