sarsoo/Selector
1
0
Fork 0
Code Issues 21 Pull Requests Actions Packages Projects Releases Wiki Activity

adding migrate command

Browse Source
This commit is contained in:
andy 2022-02-18 22:13:16 +00:00
parent 8a14f2a1db
commit 526b06775e
4 changed files with 59 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
Selector.CLI/Command/MigrateCommand.cs Normal file
View File

@ -0,0 +1,55 @@
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using Selector.CLI.Extensions;
using Selector.Model;
using System;
using System.CommandLine;
using System.CommandLine.Invocation;
namespace Selector.CLI
{
public class MigrateCommand : Command
{
public MigrateCommand(string name, string description = null) : base(name, description)
{
var connectionString = new Option<string>("--connection", "database to migrate");
connectionString.AddAlias("-c");
AddOption(connectionString);
Handler = CommandHandler.Create((string connectionString) => Execute(connectionString));
}
public static int Execute(string connectionString)
{
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>());
logger.LogInformation("Preparing to migrate ({})", db.Database.GetConnectionString());
Console.WriteLine("Migrate database? (y/n) ");
var input = Console.ReadLine();
if (input.Trim().Equals("y", StringComparison.OrdinalIgnoreCase))
{
logger.LogInformation("Migrating database");
db.Database.Migrate();
}
else
{
logger.LogInformation("Exiting");
}
}
catch (Exception ex)
{
Console.WriteLine(ex);
return 1;
}
return 0;
}
}
}

2
Selector.CLI/Command/Scrobble/ScrobbleSave.cs
View File

@ -52,7 +52,7 @@ namespace Selector.CLI
var context = new CommandContext().WithLogger().WithDb().WithLastfmApi();
var logger = context.Logger.CreateLogger("Scrobble");
var db = new ApplicationDbContext(context.DatabaseConfig.Options, context.Logger.CreateLogger<ApplicationDbContext>());
using var db = new ApplicationDbContext(context.DatabaseConfig.Options, context.Logger.CreateLogger<ApplicationDbContext>());
logger.LogInformation("Running from {0} to {1}", from, to);

4
Selector.CLI/Extensions/CommandContextExtensions.cs
View File

@ -34,7 +34,7 @@ namespace Selector.CLI.Extensions
return context;
}
public static CommandContext WithDb(this CommandContext context)
public static CommandContext WithDb(this CommandContext context, string connectionString = null)
{
if (context.Config is null)
{
@ -42,7 +42,7 @@ namespace Selector.CLI.Extensions
}
context.DatabaseConfig = new DbContextOptionsBuilder<ApplicationDbContext>();
context.DatabaseConfig.UseNpgsql(context.Config.DatabaseOptions.ConnectionString);
context.DatabaseConfig.UseNpgsql(string.IsNullOrWhiteSpace(connectionString) ? context.Config.DatabaseOptions.ConnectionString : connectionString);
return context;
}

1
Selector.CLI/Program.cs
View File

@ -8,6 +8,7 @@ namespace Selector.CLI
{
var cmd = new HostRootCommand();
cmd.AddCommand(new ScrobbleCommand("scrobble", "Manipulate scrobbles"));
cmd.AddCommand(new MigrateCommand("migrate", "Migrate database"));
cmd.Invoke(args);
}