using System; using System.Collections.Generic; using System.Linq; using System.Text; using System.Threading; using System.Threading.Tasks; using Microsoft.EntityFrameworkCore; using Microsoft.Extensions.Configuration; using Microsoft.Extensions.DependencyInjection; using Microsoft.Extensions.Hosting; using Microsoft.Extensions.Logging; using Microsoft.Extensions.Options; using Selector.Cache; using Selector.Model; using IF.Lastfm.Core.Api; using StackExchange.Redis; namespace Selector.CLI { class DbWatcherService : IHostedService { private const int PollPeriod = 1000; private readonly ILogger Logger; private readonly IServiceProvider ServiceProvider; private readonly IWatcherFactory WatcherFactory; private readonly IWatcherCollectionFactory WatcherCollectionFactory; private readonly IRefreshTokenFactoryProvider SpotifyFactory; private readonly IAudioFeatureInjectorFactory AudioFeatureInjectorFactory; private readonly IPlayCounterFactory PlayCounterFactory; private readonly IPublisherFactory PublisherFactory; private readonly ICacheWriterFactory CacheWriterFactory; private Dictionary Watchers { get; set; } = new(); public DbWatcherService( IWatcherFactory watcherFactory, IWatcherCollectionFactory watcherCollectionFactory, IRefreshTokenFactoryProvider spotifyFactory, IAudioFeatureInjectorFactory audioFeatureInjectorFactory, IPlayCounterFactory playCounterFactory, ILogger logger, IServiceProvider serviceProvider, IPublisherFactory publisherFactory = null, ICacheWriterFactory cacheWriterFactory = null ) { Logger = logger; ServiceProvider = serviceProvider; WatcherFactory = watcherFactory; WatcherCollectionFactory = watcherCollectionFactory; SpotifyFactory = spotifyFactory; AudioFeatureInjectorFactory = audioFeatureInjectorFactory; PlayCounterFactory = playCounterFactory; PublisherFactory = publisherFactory; CacheWriterFactory = cacheWriterFactory; } public async Task StartAsync(CancellationToken cancellationToken) { Logger.LogInformation("Starting database watcher service..."); var watcherIndices = await InitInstances(); Logger.LogInformation($"Starting {watcherIndices.Count()} affected watcher collection(s)..."); StartWatcherCollections(watcherIndices); } private async Task> InitInstances() { using var scope = ServiceProvider.CreateScope(); var db = scope.ServiceProvider.GetService(); var indices = new HashSet(); foreach (var dbWatcher in db.Watcher.Include(w => w.User)) { Logger.LogInformation($"Creating new [{dbWatcher.Type}] watcher"); var watcherCollectionIdx = dbWatcher.UserId; indices.Add(watcherCollectionIdx); if (!Watchers.ContainsKey(watcherCollectionIdx)) Watchers[watcherCollectionIdx] = WatcherCollectionFactory.Get(); var watcherCollection = Watchers[watcherCollectionIdx]; Logger.LogDebug("Getting Spotify factory"); var spotifyFactory = await SpotifyFactory.GetFactory(dbWatcher.User.SpotifyRefreshToken); IWatcher watcher = null; List consumers = new(); switch (dbWatcher.Type) { case WatcherType.Player: watcher = await WatcherFactory.Get(spotifyFactory, id: dbWatcher.UserId, pollPeriod: PollPeriod); consumers.Add(await AudioFeatureInjectorFactory.Get(spotifyFactory)); if (CacheWriterFactory is not null) consumers.Add(await CacheWriterFactory.Get()); if (PublisherFactory is not null) consumers.Add(await PublisherFactory.Get()); if (!string.IsNullOrWhiteSpace(dbWatcher.User.LastFmUsername)) { consumers.Add(await PlayCounterFactory.Get(creds: new() { Username = dbWatcher.User.LastFmUsername })); } else { Logger.LogDebug($"[{dbWatcher.User.UserName}] No Last.fm username, skipping play counter"); } break; case WatcherType.Playlist: throw new NotImplementedException("Playlist watchers not implemented"); // break; } watcherCollection.Add(watcher, consumers); } return indices; } private void StartWatcherCollections(IEnumerable indices) { foreach (var index in indices) { try { Logger.LogInformation($"Starting watcher collection [{index}]"); Watchers[index].Start(); } catch (KeyNotFoundException) { Logger.LogError($"Unable to retrieve watcher collection [{index}] when starting"); } } } public Task StopAsync(CancellationToken cancellationToken) { Logger.LogInformation("Shutting down"); foreach((var key, var watcher) in Watchers) { Logger.LogInformation($"Stopping watcher collection [{key}]"); watcher.Stop(); } return Task.CompletedTask; } } }