[PM-294] Add Database to MSSQL Environment File (#2962)

bitwarden · May 25, 2023 · b55d086 · b55d086
1 parent 0bdc4c3
commit b55d086
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/util/MsSql/entrypoint.sh b/util/MsSql/entrypoint.sh
@@ -43,8 +43,8 @@ fi
 # Replace database name in backup-db.sql
 if [ ! -z "$DATABASE" ]
 then
-  sed -i -e "/@DatabaseName /s/vault/$DATABASE/" backup-db.sql
-  sed -i -e "/@DatabaseNameSafe /s/vault/${DATABASE// /-}/" backup-db.sql
+  sed -i "/^SET @DatabaseName =/s/'[^']*'/'$DATABASE'/" backup-db.sql
+  sed -i "/^SET @DatabaseNameSafe =/s/'[^']*'/'${DATABASE// /-}'/" backup-db.sql
 fi
 
 # The rest...

diff --git a/util/Setup/EnvironmentFileBuilder.cs b/util/Setup/EnvironmentFileBuilder.cs
@@ -104,6 +104,7 @@ private void Init()
         _mssqlOverrideValues = new Dictionary<string, string>
         {
             ["SA_PASSWORD"] = dbPassword,
+            ["DATABASE"] = _context.Install?.Database ?? "vault"
         };
 
         _keyConnectorOverrideValues = new Dictionary<string, string>