1
0
mirror of https://github.com/bitwarden/server.git synced 2025-07-01 16:12:49 -05:00

Add optins parsing nuget for msSqlMigratorUtility

This commit is contained in:
Michal Checinski
2023-09-01 17:20:36 +02:00
parent 50145fb06f
commit 3bbf36e69f
3 changed files with 34 additions and 51 deletions

View File

@ -10,6 +10,7 @@
</ItemGroup> </ItemGroup>
<ItemGroup> <ItemGroup>
<PackageReference Include="CommandDotNet" Version="7.0.2" />
<PackageReference Include="Microsoft.Extensions.Logging" Version="6.0.0" /> <PackageReference Include="Microsoft.Extensions.Logging" Version="6.0.0" />
<PackageReference Include="Microsoft.Extensions.Logging.Console" Version="6.0.0" /> <PackageReference Include="Microsoft.Extensions.Logging.Console" Version="6.0.0" />
</ItemGroup> </ItemGroup>

View File

@ -1,5 +1,6 @@
using Bit.Migrator; using Bit.Migrator;
using Microsoft.Extensions.Logging; using Microsoft.Extensions.Logging;
using CommandDotNet;
internal class Program internal class Program
{ {
@ -7,56 +8,20 @@ internal class Program
private static int Main(string[] args) private static int Main(string[] args)
{ {
if (args.Length == 0) return new AppRunner<Program>().Run(args);
{
Console.WriteLine("Please enter a database connection string argument.");
WriteUsageToConsole();
return 1;
}
if (args.Length == 1 && (args[0] == "--verbose" || args[0] == "-v"))
{
Console.WriteLine($"Please enter a database connection string argument before {args[0]} option.");
WriteUsageToConsole();
return 1;
}
var databaseConnectionString = args[0];
ParseParameters(args);
var verbose = false;
if (Parameters.ContainsKey("--verbose") || Parameters.ContainsKey("-v"))
{
verbose = true;
}
var rerunable = false;
if (Parameters.ContainsKey("--rerunable") || Parameters.ContainsKey("-r"))
{
rerunable = true;
}
var folderName = "";
if (Parameters.ContainsKey("--folder") || Parameters.ContainsKey("-f"))
{
folderName = Parameters["--folder"] ?? Parameters["-f"];
}
var success = MigrateDatabase(databaseConnectionString, verbose, rerunable, folderName);
if (!success)
{
return -1;
}
return 0;
} }
[DefaultCommand]
public void Execute(
[Operand(Description = "Database connection string")]
string databaseConnectionString,
[Option('v', "verbose", Description = "Enable verbose output of migrator logs")]
bool verbose = false,
[Option('r', "rerunable", Description = "Mark scripts as rerunable")]
bool rerunable = false,
[Option('f', "folder", Description = "Folder name of database scripts")]
string folderName = "DbScripts") => MigrateDatabase(databaseConnectionString, verbose, rerunable, folderName);
private static void WriteUsageToConsole() private static void WriteUsageToConsole()
{ {
Console.WriteLine("Usage: MsSqlMigratorUtility <database-connection-string>"); Console.WriteLine("Usage: MsSqlMigratorUtility <database-connection-string>");
@ -67,11 +32,13 @@ internal class Program
private static bool MigrateDatabase(string databaseConnectionString, bool verbose = false, bool rerunable = false, string folderName = "") private static bool MigrateDatabase(string databaseConnectionString, bool verbose = false, bool rerunable = false, string folderName = "")
{ {
Console.WriteLine($"rerunable: {rerunable}");
Console.WriteLine($"folderName: {folderName}");
var logger = CreateLogger(verbose); var logger = CreateLogger(verbose);
var migrator = new DbMigrator(databaseConnectionString, logger); var migrator = new DbMigrator(databaseConnectionString, logger);
bool success = false; bool success = false;
if (string.IsNullOrWhiteSpace(folderName)) if (!string.IsNullOrWhiteSpace(folderName))
{ {
success = migrator.MigrateMsSqlDatabaseWithRetries(verbose, rerunable, folderName); success = migrator.MigrateMsSqlDatabaseWithRetries(verbose, rerunable, folderName);
} }
@ -114,8 +81,14 @@ internal class Program
{ {
continue; continue;
} }
if (i + 1 <= args.Length && !args[i + 1].StartsWith("-"))
Parameters.Add(args[i].Substring(1), args[i + 1]); {
Parameters.Add(args[i].Substring(1), args[i + 1]);
}
else
{
Parameters.Add(args[i].Substring(1), null);
}
} }
} }
} }

View File

@ -2,6 +2,15 @@
"version": 1, "version": 1,
"dependencies": { "dependencies": {
"net6.0": { "net6.0": {
"CommandDotNet": {
"type": "Direct",
"requested": "[7.0.2, )",
"resolved": "7.0.2",
"contentHash": "sFfqn4T6Ux4AbGnhS+BZvf9iVeP6b9p9bwaMT8nsefVcYH2tC9MHj3AoY+GG0nnBPmFawRqdgud08cBjBdPByQ==",
"dependencies": {
"Microsoft.CSharp": "4.7.0"
}
},
"Microsoft.Extensions.Logging": { "Microsoft.Extensions.Logging": {
"type": "Direct", "type": "Direct",
"requested": "[6.0.0, )", "requested": "[6.0.0, )",