using Microsoft.EntityFrameworkCore; using Microsoft.Extensions.Logging; using Selector.CLI.Extensions; using Selector.Model; using Selector.Model.Extensions; using System; using System.CommandLine; using System.CommandLine.Invocation; using System.Linq; using System.Threading; using System.Threading.Tasks; namespace Selector.CLI { public class ScrobbleSaveCommand : Command { public ScrobbleSaveCommand(string name, string description = null) : base(name, description) { var fromOption = new Option("--from", getDefaultValue: () => DateTime.UtcNow.AddDays(-7), "Date from which to pull scrobbles"); AddOption(fromOption); var toOption = new Option( "--to", getDefaultValue: () => DateTime.UtcNow.AddHours(1), "Last date for which to pull scrobbles"); AddOption(toOption); var pageOption = new Option("--page", getDefaultValue: () => 100, "number of scrobbles per page"); pageOption.AddAlias("-p"); AddOption(pageOption); var delayOption = new Option("--delay", getDefaultValue: () => 200, "milliseconds to delay"); delayOption.AddAlias("-d"); AddOption(delayOption); var username = new Option("--username", "user to pulls scrobbles for"); username.AddAlias("-u"); AddOption(username); var dontAdd = new Option("--no-add", "don't add any scrobbles to the database"); dontAdd.AddAlias("-na"); AddOption(dontAdd); var dontRemove = new Option("--no-remove", "don't remove any scrobbles from the database"); dontRemove.AddAlias("-nr"); AddOption(dontRemove); Handler = CommandHandler.Create(async (DateTime from, DateTime to, int page, int delay, string username, bool noAdd, bool noRemove, CancellationToken token) => await Execute(from, to, page, delay, username, noAdd, noRemove, token)); } public static async Task Execute(DateTime from, DateTime to, int page, int delay, string username, bool noAdd, bool noRemove, CancellationToken token) { try { var context = new CommandContext().WithLogger().WithDb().WithLastfmApi(); var logger = context.Logger.CreateLogger("Scrobble"); var db = new ApplicationDbContext(context.DatabaseConfig.Options, context.Logger.CreateLogger()); logger.LogInformation("Running from {0} to {1}", from, to); logger.LogInformation("Searching for {0}", username); var user = db.Users.AsNoTracking().FirstOrDefault(u => u.UserName == username); if (user is not null) { if (user.LastFmConnected()) { logger.LogInformation("Last.fm username found ({0}), starting...", user.LastFmUsername); await new ScrobbleSaver( context.LastFmClient.User, new ScrobbleSaverConfig() { User = user, InterRequestDelay = new TimeSpan(0, 0, 0, 0, delay), From = from, To = to, PageSize = page, DontAdd = noAdd, DontRemove = noRemove }, db, context.Logger.CreateLogger()) .Execute(token); } else { logger.LogError("{0} doesn't have a Last.fm username", username); } } else { logger.LogError("{0} not found", username); } } catch (Exception ex) { Console.WriteLine(ex); return 1; } return 0; } } }