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

Add optins parsing nuget for msSqlMigratorUtility

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

75
util/MsSqlMigratorUtility/Program.cs
View File

@ -1,5 +1,6 @@
using Bit.Migrator;
using Microsoft.Extensions.Logging;
using CommandDotNet;
internal class Program
{
@ -7,56 +8,20 @@ internal class Program
private static int Main(string[] args)
{
if (args.Length == 0)
{
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;
return new AppRunner<Program>().Run(args);
}
[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()
{
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 = "")
{
Console.WriteLine($"rerunable: {rerunable}");
Console.WriteLine($"folderName: {folderName}");
var logger = CreateLogger(verbose);
var migrator = new DbMigrator(databaseConnectionString, logger);
bool success = false;
if (string.IsNullOrWhiteSpace(folderName))
if (!string.IsNullOrWhiteSpace(folderName))
{
success = migrator.MigrateMsSqlDatabaseWithRetries(verbose, rerunable, folderName);
}
@ -114,8 +81,14 @@ internal class Program
{
continue;
}
Parameters.Add(args[i].Substring(1), args[i + 1]);
if (i + 1 <= args.Length && !args[i + 1].StartsWith("-"))
{
Parameters.Add(args[i].Substring(1), args[i + 1]);
}
else
{
Parameters.Add(args[i].Substring(1), null);
}
}
}
}