From 9478c255680d50b1046b81c5438b693e9270601e Mon Sep 17 00:00:00 2001
From: Norm Johanson <normj@amazon.com>
Date: Wed, 29 Jan 2025 00:16:18 -0800
Subject: [PATCH] Hide documentation banner for configuring Lambda functions
 when tool is hosted by Aspire

---
 .../Components/Pages/Home.razor                    | 14 +++++++++-----
 .../src/Amazon.Lambda.TestTool/Utilities/Utils.cs  | 10 +++++++++-
 2 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Components/Pages/Home.razor b/Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Components/Pages/Home.razor
index d1ab9ea3d..684c6fc76 100644
--- a/Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Components/Pages/Home.razor
+++ b/Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Components/Pages/Home.razor
@@ -3,6 +3,7 @@
 @using Amazon.Lambda.TestTool.Models
 @using Amazon.Lambda.TestTool.SampleRequests
 @using Amazon.Lambda.TestTool.Services
+@using Amazon.Lambda.TestTool.Utilities
 
 <PageTitle>Lambda Function Tester</PageTitle>
 
@@ -26,11 +27,14 @@
     </div>
 </div>
 
-<div class="alert alert-secondary" role="alert">
-    For Lambda functions written as executable assemblies, i.e. custom runtimes functions and top level statement functions, this page can be used for testing the functions locally.
-    Set the <b>AWS_LAMBDA_RUNTIME_API</b> environment variable to <b>@HttpContextAccessor.HttpContext?.Request.Host</b> while debugging executable assembly
-    Lambda function. More information can be found in the <a href="/documentation">documentation</a>.
-</div>
+@if (!Utils.IsAspireHosted)
+{
+    <div class="alert alert-secondary" role="alert">
+        For Lambda functions written as executable assemblies, i.e. custom runtimes functions and top level statement functions, this page can be used for testing the functions locally.
+        Set the <b>AWS_LAMBDA_RUNTIME_API</b> environment variable to <b>@HttpContextAccessor.HttpContext?.Request.Host</b> while debugging executable assembly
+        Lambda function. More information can be found in the <a href="/documentation">documentation</a>.
+    </div>
+}
 
 @if (DataStore == null)
 {
diff --git a/Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Utilities/Utils.cs b/Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Utilities/Utils.cs
index 700588009..ead444dd3 100644
--- a/Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Utilities/Utils.cs
+++ b/Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Utilities/Utils.cs
@@ -1,4 +1,4 @@
-// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
 
 using System.Reflection;
@@ -42,6 +42,14 @@ public static string DetermineToolVersion()
         return version ?? unknownVersion;
     }
 
+    /// <summary>
+    /// If true it means the test tool was launched via an Aspire AppHost.
+    /// </summary>
+    internal static bool IsAspireHosted
+    {
+        get { return string.Equals(Environment.GetEnvironmentVariable("ASPIRE_HOSTED"), "true", StringComparison.OrdinalIgnoreCase); }
+    }
+
     /// <summary>
     /// Attempt to pretty print the input string. If pretty print fails return back the input string in its original form.
     /// </summary>