91 lines
3.1 KiB
C#
91 lines
3.1 KiB
C#
using Microsoft.EntityFrameworkCore;
|
|
using Microsoft.Extensions.Logging;
|
|
using Selector.CLI.Extensions;
|
|
using Selector.Data;
|
|
using Selector.Model;
|
|
using System;
|
|
using System.IO;
|
|
using System.CommandLine;
|
|
using System.CommandLine.Invocation;
|
|
using System.Linq;
|
|
using System.Collections.Generic;
|
|
|
|
namespace Selector.CLI
|
|
{
|
|
public class SpotifyHistoryCommand : Command
|
|
{
|
|
public SpotifyHistoryCommand(string name, string description = null) : base(name, description)
|
|
{
|
|
var connectionString = new Option<string>("--connection", "database to migrate");
|
|
connectionString.AddAlias("-c");
|
|
AddOption(connectionString);
|
|
|
|
var pathString = new Option<string>("--path", "path to find data");
|
|
pathString.AddAlias("-i");
|
|
AddOption(pathString);
|
|
|
|
var username = new Option<string>("--username", "user to pulls scrobbles for");
|
|
username.AddAlias("-u");
|
|
AddOption(username);
|
|
|
|
Handler = CommandHandler.Create((string connectionString, string path, string username) => Execute(connectionString, path, username));
|
|
}
|
|
|
|
public static int Execute(string connectionString, string path, string username)
|
|
{
|
|
var streams = new List<FileStream>();
|
|
|
|
try
|
|
{
|
|
var context = new CommandContext().WithLogger().WithDb(connectionString).WithLastfmApi();
|
|
var logger = context.Logger.CreateLogger("Scrobble");
|
|
|
|
using var db = new ApplicationDbContext(context.DatabaseConfig.Options, context.Logger.CreateLogger<ApplicationDbContext>());
|
|
|
|
var historyPersister = new HistoryPersister(db, new DataJsonContext(), new()
|
|
{
|
|
Username = username
|
|
}, context.Logger.CreateLogger<HistoryPersister>());
|
|
|
|
logger.LogInformation("Preparing to parse from {} for {}", path, username);
|
|
|
|
var directoryContents = Directory.EnumerateFiles(path);
|
|
var endSongs = directoryContents.Where(f => f.Contains("endsong_")).ToArray();
|
|
|
|
foreach(var file in endSongs)
|
|
{
|
|
streams.Add(File.OpenRead(file));
|
|
}
|
|
|
|
Console.WriteLine("Parse {0} historical data files? (y/n) ", endSongs.Length);
|
|
var input = Console.ReadLine();
|
|
|
|
if (input.Trim().Equals("y", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
logger.LogInformation("Parsing files");
|
|
|
|
historyPersister.Process(streams).Wait();
|
|
}
|
|
else
|
|
{
|
|
logger.LogInformation("Exiting");
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Console.WriteLine(ex);
|
|
return 1;
|
|
}
|
|
finally
|
|
{
|
|
foreach(var stream in streams)
|
|
{
|
|
stream.Dispose();
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
}
|
|
}
|