From 0381a48ec9f85550db3d70462b8ae376ca5bc280 Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Tue, 27 Mar 2018 15:12:28 -0400
Subject: [PATCH] re-attempt db migrate on update mode failure

---
 util/Setup/Program.cs | 82 ++++++++++++++++++++++++++-----------------
 1 file changed, 49 insertions(+), 33 deletions(-)

diff --git a/util/Setup/Program.cs b/util/Setup/Program.cs
index 79e9384c44..b3646a7ca8 100644
--- a/util/Setup/Program.cs
+++ b/util/Setup/Program.cs
@@ -227,43 +227,59 @@ namespace Bit.Setup
             Console.WriteLine("\n");
         }
 
-        private static void MigrateDatabase()
+        private static void MigrateDatabase(int attempt = 1)
         {
-            Console.WriteLine("Migrating database.");
-
-            var dbPass = Helpers.GetValueFronEnvFile("mssql", "SA_PASSWORD");
-            var masterConnectionString = Helpers.MakeSqlConnectionString(
-                "mssql", "master", "sa", dbPass ?? string.Empty);
-            var vaultConnectionString = Helpers.MakeSqlConnectionString(
-                "mssql", "vault", "sa", dbPass ?? string.Empty);
-
-            using(var connection = new SqlConnection(masterConnectionString))
+            try
             {
-                var command = new SqlCommand(
-                    "IF ((SELECT COUNT(1) FROM sys.databases WHERE [name] = 'vault') = 0) CREATE DATABASE [vault];",
-                    connection);
-                command.Connection.Open();
-                command.ExecuteNonQuery();
+                Console.WriteLine("Migrating database.");
+
+                var dbPass = Helpers.GetValueFronEnvFile("mssql", "SA_PASSWORD");
+                var masterConnectionString = Helpers.MakeSqlConnectionString(
+                    "mssql", "master", "sa", dbPass ?? string.Empty);
+                var vaultConnectionString = Helpers.MakeSqlConnectionString(
+                    "mssql", "vault", "sa", dbPass ?? string.Empty);
+
+                using(var connection = new SqlConnection(masterConnectionString))
+                {
+                    var command = new SqlCommand(
+                        "IF ((SELECT COUNT(1) FROM sys.databases WHERE [name] = 'vault') = 0) " +
+                        "CREATE DATABASE [vault];", connection);
+                    command.Connection.Open();
+                    command.ExecuteNonQuery();
+                }
+
+                var upgrader = DeployChanges.To
+                    .SqlDatabase(vaultConnectionString)
+                    .JournalToSqlTable("dbo", "Migration")
+                    .WithScriptsAndCodeEmbeddedInAssembly(Assembly.GetExecutingAssembly(),
+                        s => s.Contains($".DbScripts.") && !s.Contains(".Archive."))
+                    .WithTransaction()
+                    .WithExecutionTimeout(new TimeSpan(0, 5, 0))
+                    .LogToConsole()
+                    .Build();
+
+                var result = upgrader.PerformUpgrade();
+                if(result.Successful)
+                {
+                    Console.WriteLine("Migration successful.");
+                }
+                else
+                {
+                    Console.WriteLine("Migration failed.");
+                }
             }
-
-            var upgrader = DeployChanges.To
-                .SqlDatabase(vaultConnectionString)
-                .JournalToSqlTable("dbo", "Migration")
-                .WithScriptsAndCodeEmbeddedInAssembly(Assembly.GetExecutingAssembly(),
-                    s => s.Contains($".DbScripts.") && !s.Contains(".Archive."))
-                .WithTransaction()
-                .WithExecutionTimeout(new TimeSpan(0, 5, 0))
-                .LogToConsole()
-                .Build();
-
-            var result = upgrader.PerformUpgrade();
-            if(result.Successful)
+            catch(SqlException e)
             {
-                Console.WriteLine("Migration successful.");
-            }
-            else
-            {
-                Console.WriteLine("Migration failed.");
+                if(e.Message.Contains("Server is in script upgrade mode") && attempt < 3)
+                {
+                    var nextAttempt = attempt + 1;
+                    Console.WriteLine("Database is in script upgrade mode. " +
+                        "Trying again (attempt #{0})...", nextAttempt);
+                    MigrateDatabase(nextAttempt);
+                    return;
+                }
+
+                throw e;
             }
         }