Update TidalCommand --ignore-artist-album-amount

Author: MusicMoveArr

MiniMediaScanner/Commands/UpdateTidalCommand.cs

@@ -74,6 +74,12 @@ public class UpdateTidalCommand : ICommand
         IsRequired = false,
         EnvironmentVariable = "UPDATETIDAL_NONPULLED_ARTISTS")]
     public bool UpdateNonpulledArtists { get; set; }
+
+    [CommandOption("ignore-artist-album-amount",
+        Description = "Ignore artists that have over a certain amount of albums, >500 albums is not normal",
+        IsRequired = false,
+        EnvironmentVariable = "UPDATETIDAL_IGNORE_ARTIST_ALBUM_AMOUNT")]
+    public int IgnoreArtistAlbumAmount { get; set; } = 500;
 
     public async ValueTask ExecuteAsync(IConsole console)
     {
@@ -96,7 +102,8 @@ public async ValueTask ExecuteAsync(IConsole console)
             ProxyFile, 
             Proxy, 
             ProxyMode, 
-            PreventUpdateWithinDays);
+            PreventUpdateWithinDays,
+            IgnoreArtistAlbumAmount);
 
         if (!string.IsNullOrWhiteSpace(ArtistFilePath) && File.Exists(ArtistFilePath))
         {

MiniMediaScanner/Commands/UpdateTidalCommandHandler.cs

@@ -18,10 +18,11 @@ public UpdateTidalCommandHandler(string connectionString,
         string proxyFile, 
         string singleProxy, 
         string proxyMode,
-        int preventUpdateWithinDays)
+        int preventUpdateWithinDays,
+        int ignoreArtistAlbumAmount)
     {
         _tidalRepository = new TidalRepository(connectionString);
-        _tidalService = new TidalService(connectionString, secretTokens, countryCode, proxyFile, singleProxy, proxyMode, preventUpdateWithinDays);
+        _tidalService = new TidalService(connectionString, secretTokens, countryCode, proxyFile, singleProxy, proxyMode, preventUpdateWithinDays, ignoreArtistAlbumAmount);
     }
 
     public async Task UpdateTidalArtistsByNameAsync(string artistName)

MiniMediaScanner/Services/TidalService.cs

@@ -3,27 +3,32 @@
 using MiniMediaScanner.Callbacks;
 using MiniMediaScanner.Callbacks.Status;
 using MiniMediaScanner.Enums;
+using MiniMediaScanner.Helpers;
 using MiniMediaScanner.Models.Tidal;
 using MiniMediaScanner.Repositories;
+using Spectre.Console;
 
 namespace MiniMediaScanner.Services;
 
 public class TidalService
 {
     public readonly int PreventUpdateWithinDays; 
     private readonly TidalAPICacheLayerService _tidalAPIService;
-    private readonly UpdateTidalRepository _updateTidalRepository;
+    private readonly string _connectionString;
+    private readonly int _ignoreArtistAlbumAmount;
 
     public TidalService(string connectionString, 
         List<TidalTokenClientSecret> secretTokens,
         string countryCode, 
         string proxyFile, 
         string singleProxy, 
         string proxyMode,
-        int preventUpdateWithinDays)
+        int preventUpdateWithinDays,
+        int ignoreArtistAlbumAmount)
     {
+        _ignoreArtistAlbumAmount = ignoreArtistAlbumAmount;
+        _connectionString =  connectionString;
         this.PreventUpdateWithinDays = preventUpdateWithinDays;
-        _updateTidalRepository = new UpdateTidalRepository(connectionString);
         _tidalAPIService = new TidalAPICacheLayerService(secretTokens, countryCode, proxyFile, singleProxy, proxyMode);
     }
 
@@ -65,6 +70,7 @@ public async Task UpdateArtistByNameAsync(string artistName,
     public async Task UpdateArtistByIdAsync(int artistId,
         Action<UpdateTidalCallback>? callback = null)
     {
+        UpdateTidalRepository _updateTidalRepository = new UpdateTidalRepository(_connectionString);
         await _updateTidalRepository.SetConnectionAsync();
 
         try
@@ -115,6 +121,12 @@ public async Task UpdateArtistByIdAsync(int artistId,
                 .Where(x => x.Type == "albums")
                 .ToList();
 
+            if (albums.Count >= _ignoreArtistAlbumAmount)
+            {
+                await _updateTidalRepository.CommitAsync();
+                return;
+            }
+
             int progress = 1;
             foreach (var album in albums)
             {
@@ -371,11 +383,14 @@ await _updateTidalRepository.UpsertTrackAsync(int.Parse(track.Id),
 
     private async Task<TidalSearchResponse?> InsertArtistInfoAsync(int artistId, bool ignorePeventCheck = false)
     {
+        UpdateTidalRepository _updateTidalRepository = new UpdateTidalRepository(_connectionString);
+        await _updateTidalRepository.SetConnectionAsync();
         if (!ignorePeventCheck)
         {
             DateTime? lastSyncTime = await _updateTidalRepository.GetArtistLastSyncTimeAsync(artistId);
             if (lastSyncTime?.Year > 2000 && DateTime.Now.Subtract(lastSyncTime.Value).TotalDays < PreventUpdateWithinDays)
             {
+                await _updateTidalRepository.CommitAsync();
                 return null;
             }
         }
@@ -386,6 +401,7 @@ await _updateTidalRepository.UpsertTrackAsync(int.Parse(track.Id),
             artistInfo?.Data == null || 
             artistInfo?.Included == null)
         {
+            await _updateTidalRepository.CommitAsync();
             return null;
         }
 
@@ -406,6 +422,7 @@ await _updateTidalRepository.UpsertArtistImageLinkAsync(artistId,
                 imageLink.Meta.Height);
         }
 
+        await _updateTidalRepository.CommitAsync();
         return artistInfo;
     }
 }