chore: Update user agent to include CLI command

aws · Oct 27, 2021 · 659d16a · 659d16a
1 parent c4fad95
commit 659d16a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/AWS.Deploy.CLI/App.cs b/src/AWS.Deploy.CLI/App.cs
@@ -26,7 +26,7 @@ public async Task<int> Run(string[] args)
         {
             Console.OutputEncoding = Encoding.UTF8;
 
-            SetExecutionEnvironment();
+            SetExecutionEnvironment(args);
 
             _toolInteractiveService.WriteLine("AWS .NET deployment tool for deploying .NET Core applications to AWS.");
             _toolInteractiveService.WriteLine("Project Home: https://github.com/aws/aws-dotnet-deploy");
@@ -45,7 +45,7 @@ public async Task<int> Run(string[] args)
         /// Set up the execution environment variable picked up by the AWS .NET SDK. This can be useful for identify calls
         /// made by this tool in AWS CloudTrail.
         /// </summary>
-        private static void SetExecutionEnvironment()
+        private static void SetExecutionEnvironment(string[] args)
         {
             const string envName = "AWS_EXECUTION_ENV";
             const string awsDotnetDeployCLI = "aws-dotnet-deploy-cli";
@@ -65,6 +65,11 @@ private static void SetExecutionEnvironment()
 
             envValue.Append($"{awsDotnetDeployCLI}_{assemblyVersion?.InformationalVersion}");
 
+            if (args?.Length > 0)
+            {
+                envValue.Append($"_{args[0]}");
+            }
+
             Environment.SetEnvironmentVariable(envName, envValue.ToString());
         }
     }