Add: detection of download limit and removed beatmaps

Piotrekol · Jun 16, 2019 · 195cd68 · 195cd68
1 parent 1ba546e
commit 195cd68
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/CollectionManagerExtensionsDll/Modules/DownloadManager/API/DownloadManager.cs b/CollectionManagerExtensionsDll/Modules/DownloadManager/API/DownloadManager.cs
@@ -5,6 +5,7 @@
 using System.IO;
 using System.Net;
 using System.Threading;
+using System.Threading.Tasks;
 
 namespace CollectionManagerExtensionsDll.Modules.DownloadManager.API
 {
@@ -172,14 +173,46 @@ private void Completed(object sender, AsyncCompletedEventArgs e)
                     url.DownloadAborted = true;//Progress = "download cancelled";
                     error = true;
                     lock (_urlsToDownload)
+                    {
                         _urlsToDownload.AddFirst(url);
+                    }
                 }
                 else if (e.Error != null)
                 {
                     url.OtherError = true;
                     url.Error = "Error: " + e.Error.ToString();
                     error = true;
                 }
+                else if (url.BytesRecived<10000)
+                {
+                    //check if that's a disabled dl or download cap
+                    var tempFileLocation = GetFullTempLocation(url.FileName);
+                    var tempFileContents = File.ReadAllText(tempFileLocation);
+                    if (tempFileContents.Contains("disabled")) // "This download has been disabled (<some url>)"
+                    {
+                        error = true;
+                        url.Error = tempFileContents;
+                    }
+                    else if(tempFileContents.Contains("slow down")) // "slow down, play more."
+                    {
+                        //We've hit download cap, lets chill a little 
+                        if (!_stopDownloads)
+                        {
+                            StopDownloads();
+                            Task.Run(async () =>
+                            {
+                                //Stop downloads for 6 minutes - user can try to resume these earlier, but it'll just get paused again if limit is still in place.
+                                await Task.Delay(60 * 1000 * 6);
+                                ResumeNewDownloads();
+                            });
+                        }
+
+                        lock (_urlsToDownload)
+                        {
+                            _urlsToDownload.AddFirst(url);
+                        }
+                    }
+                }
                 if (error)
                 {
                     string tempFileLocation = GetFullTempLocation(url.FileName);