Add MPS Secrets as Env Variables in Game Servers (#182)

Co-authored-by: Jose Valenciano <[email protected]>
PlayFab · Apr 10, 2024 · fd5f361 · fd5f361
1 parent 194f41a
commit fd5f361
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 0 deletions.
diff --git a/VmAgent.Core.UnitTests/VmConfigurationTests.cs b/VmAgent.Core.UnitTests/VmConfigurationTests.cs
@@ -129,6 +129,30 @@ public void EnvironmentVariablesWithMultipleIpAddresses()
             envVariables.Should().Contain("PF_PUBLIC_IP_ADDRESSES_ROUTING_TYPE_1", "Microsoft");
         }
 
+        [TestMethod]
+        [TestCategory("BVT")]
+        public void EnvironmentVariablesWithMpsSecrets()
+        {
+            SessionHostsStartInfo sessionHostsStartInfo = CreateSessionHostStartInfo(new Dictionary<string, string>());
+            sessionHostsStartInfo.GameSecrets = new SecretDetail[]
+            {
+                new SecretDetail()
+                {
+                    Name = "secret1",
+                    Value = "value1"
+                },
+                new SecretDetail()
+                {
+                    Name = "secret2",
+                    Value = "value2"
+                }
+            };
+
+            IDictionary<string, string> envVariables = VmConfiguration.GetCommonEnvironmentVariables(sessionHostsStartInfo, VmConfiguration);
+            envVariables.Should().Contain("PF_MPS_SECRET_secret1", "value1");
+            envVariables.Should().Contain("PF_MPS_SECRET_secret2", "value2");
+        }
+
 
         private SessionHostsStartInfo CreateSessionHostStartInfo(IDictionary<string, string> buildMetadata = null, SessionHostType sessionHostType = SessionHostType.Process)
         {

diff --git a/VmAgent.Core/VmConfiguration.cs b/VmAgent.Core/VmConfiguration.cs
@@ -54,6 +54,9 @@ public class VmConfiguration
         /// Prefix used for all the Public Ip Addresses environment variables (count, ip addresss, fqdn, routing type)
         public const string PublicIpAddressesEnvVariablePrefix = "PF_PUBLIC_IP_ADDRESSES";
 
+        // Prefix for each secret configured in PlayFab MPS and saved as environment variable
+        public const string PlayfabMpsSecretEnvVariablePrefix = "PF_MPS_SECRET";
+
         private static readonly byte[] PlayFabTitleIdPrefix = BitConverter.GetBytes(0xFFFFFFFFFFFFFFFF);
 
         public int ListeningPort { get; }
@@ -117,6 +120,8 @@ public static IDictionary<string, string> GetCommonEnvironmentVariables(SessionH
 
             environmentVariables.AddRange(GetPublicIpAddressesEnvironmentVariables(sessionHostsStartInfo));
 
+            environmentVariables.AddRange(GetEnvironmentVariablesWithPlayFabMpsSecrets(sessionHostsStartInfo));
+
             return environmentVariables;
         }
 
@@ -148,6 +153,12 @@ private static IDictionary<string, string> GetPublicIpAddressesEnvironmentVariab
             return publicIPAddressesEnvironmentVariables;
         }
 
+        private static IDictionary<string, string> GetEnvironmentVariablesWithPlayFabMpsSecrets(SessionHostsStartInfo sessionHostsStartInfo)
+        {
+            return sessionHostsStartInfo.GameSecrets?.ToDictionary(
+                secretDetail => $"{PlayfabMpsSecretEnvVariablePrefix}_{secretDetail.Name}", secretDetail => secretDetail.Value) ?? new Dictionary<string, string>();
+        }
+
         private static string GetSharedContentFolderPath(SessionHostsStartInfo sessionHostsStartInfo, VmConfiguration vmConfiguration)
         {
             switch (sessionHostsStartInfo.SessionHostType)