Fix Azure ServiceBus persistent container support

playground/AzureServiceBus/ServiceBus.AppHost/Program.cs

@@ -39,7 +39,7 @@
 serviceBus.RunAsEmulator(configure => configure.ConfigureEmulator(document =>
 {
     document["UserConfig"]!["Logging"] = new JsonObject { ["Type"] = "Console" };
-}));
+}).WithLifetime(ContainerLifetime.Persistent));
 
 builder.AddProject<Projects.ServiceBusWorker>("worker")
     .WithReference(serviceBus).WaitFor(serviceBus);

src/Aspire.Hosting.AppHost/build/Aspire.Hosting.AppHost.targets

@@ -241,6 +241,15 @@ namespace Projects%3B
     </ItemGroup>
   </Target>
 
+  <Target Name="EmbedAppHostIntermediateOutputPath" BeforeTargets="GetAssemblyAttributes" Condition=" '$(IsAspireHost)' == 'true' ">
+    <ItemGroup>
+      <AssemblyAttribute Include="System.Reflection.AssemblyMetadataAttribute">
+        <_Parameter1>apphostprojectbaseintermediateoutputpath</_Parameter1>
+        <_Parameter2>$(BaseIntermediateOutputPath)</_Parameter2>
+      </AssemblyAttribute>
+    </ItemGroup>
+  </Target>
+
   <PropertyGroup>
     <AspirePublisher Condition="'$(AspirePublisher)' == ''">manifest</AspirePublisher>
     <AspireManifestPublishOutputPath Condition="'$(AspireManifestPublishOutputPath)' == ''">$(_AspireIntermediatePath)</AspireManifestPublishOutputPath>

src/Aspire.Hosting.Testing/DistributedApplicationTestingBuilder.cs

@@ -274,16 +274,13 @@ public async Task StopAsync(CancellationToken cancellationToken)
         }
     }
 
-    private sealed class TestingBuilder(
-        string[] args,
-        Action<DistributedApplicationOptions, HostApplicationBuilderSettings> configureBuilder) : IDistributedApplicationTestingBuilder
+    private sealed class TestingBuilder : IDistributedApplicationTestingBuilder
     {
-        private readonly DistributedApplicationBuilder _innerBuilder = CreateInnerBuilder(args, configureBuilder);
+        private readonly DistributedApplicationBuilder _innerBuilder;
         private DistributedApplication? _app;
+        private readonly string? _tempAspireStorePath;
 
-        private static DistributedApplicationBuilder CreateInnerBuilder(
-            string[] args,
-            Action<DistributedApplicationOptions, HostApplicationBuilderSettings> configureBuilder)
+        public TestingBuilder(string[] args, Action<DistributedApplicationOptions, HostApplicationBuilderSettings> configureBuilder)
         {
             var builder = TestingBuilderFactory.CreateBuilder(args, onConstructing: (applicationOptions, hostBuilderOptions) =>
             {
@@ -297,7 +294,12 @@ private static DistributedApplicationBuilder CreateInnerBuilder(
                 builder.Services.ConfigureHttpClientDefaults(http => http.AddStandardResilienceHandler());
             }
 
-            return builder;
+            if (!builder.Configuration.GetValue("Aspire:Store:Path", false))
+            {
+                builder.Configuration["Aspire:Store:Path"] = _tempAspireStorePath = Path.GetTempPath();
+            }
+
+            _innerBuilder = builder;
 
             static Assembly FindApplicationAssembly()
             {
@@ -373,6 +375,18 @@ public void Dispose()
             {
                 app.Dispose();
             }
+
+            if (_tempAspireStorePath is { } path)
+            {
+                try
+                {
+                    Directory.Delete(path, recursive: true);
+                }
+                catch
+                {
+                    // Suppress.
+                }
+            }
         }
 
         public async ValueTask DisposeAsync()
@@ -393,6 +407,18 @@ public async ValueTask DisposeAsync()
             {
                 await asyncDisposable.DisposeAsync().ConfigureAwait(false);
             }
+
+            if (_tempAspireStorePath is { } path)
+            {
+                try
+                {
+                    Directory.Delete(path, recursive: true);
+                }
+                catch
+                {
+                    // Suppress.
+                }
+            }
         }
     }
 }

src/Aspire.Hosting/AspireStore.cs

@@ -0,0 +1,108 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+using System.Security.Cryptography;
+
+namespace Aspire.Hosting;
+
+internal sealed class AspireStore : IAspireStore
+{
+    private readonly string _basePath;
+
+    /// <summary>
+    /// Initializes a new instance of the <see cref="AspireStore"/> class with the specified base path.
+    /// </summary>
+    /// <param name="basePath">The base path for the store.</param>
+    /// <returns>A new instance of <see cref="AspireStore"/>.</returns>
+    public AspireStore(string basePath)
+    {
+        ArgumentNullException.ThrowIfNull(basePath);
+
+        _basePath = basePath;
+        EnsureDirectory();
+    }
+
+    public string BasePath => _basePath;
+
+    public string GetFileNameWithContent(string filenameTemplate, string sourceFilename)
+    {
+        ArgumentNullException.ThrowIfNullOrWhiteSpace(filenameTemplate);
+        ArgumentNullException.ThrowIfNullOrWhiteSpace(sourceFilename);
+
+        if (!File.Exists(sourceFilename))
+        {
+            throw new FileNotFoundException("The source file '{0}' does not exist.", sourceFilename);
+        }
+
+        EnsureDirectory();
+
+        // Strip any folder information from the filename.
+        filenameTemplate = Path.GetFileName(filenameTemplate);
+
+        var hashStream = File.OpenRead(sourceFilename);
+
+        // Compute the hash of the content.
+        var hash = SHA256.HashData(hashStream);
+
+        hashStream.Dispose();
+
+        var name = Path.GetFileNameWithoutExtension(filenameTemplate);
+        var ext = Path.GetExtension(filenameTemplate);
+        var finalFilePath = Path.Combine(_basePath, $"{name}.{Convert.ToHexString(hash)[..12].ToLowerInvariant()}{ext}");
+
+        if (!File.Exists(finalFilePath))
+        {
+            File.Copy(sourceFilename, finalFilePath, overwrite: true);
+        }
+
+        return finalFilePath;
+    }
+
+    public string GetFileNameWithContent(string filenameTemplate, Stream contentStream)
+    {
+        ArgumentNullException.ThrowIfNullOrWhiteSpace(filenameTemplate);
+        ArgumentNullException.ThrowIfNull(contentStream);
+
+        // Create a temporary file to write the content to.
+        var tempFileName = Path.GetTempFileName();
+
+        // Write the content to the temporary file.
+        using (var fileStream = File.OpenWrite(tempFileName))
+        {
+            contentStream.CopyTo(fileStream);
+        }
+
+        var finalFilePath = GetFileNameWithContent(filenameTemplate, tempFileName);
+
+        try
+        {
+            File.Delete(tempFileName);
+        }
+        catch
+        {
+        }
+
+        return finalFilePath;
+    }
+
+    public string GetFileName(string filename)
+    {
+        EnsureDirectory();
+
+        // Strip any folder information from the filename.
+        filename = Path.GetFileName(filename);
+
+        return Path.Combine(_basePath, filename);
+    }
+
+    /// <summary>
+    /// Ensures that the directory for the store exists.
+    /// </summary>
+    private void EnsureDirectory()
+    {
+        if (!string.IsNullOrEmpty(_basePath))
+        {
+            Directory.CreateDirectory(_basePath);
+        }
+    }
+}

src/Aspire.Hosting/AspireStoreExtensions.cs

@@ -0,0 +1,49 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+using System.Reflection;
+
+namespace Aspire.Hosting;
+
+/// <summary>
+/// Provides extension methods for <see cref="IDistributedApplicationBuilder"/> to create an <see cref="IAspireStore"/> instance.
+/// </summary>
+public static class AspireStoreExtensions
+{
+    internal const string AspireStorePathKeyName = "Aspire:Store:Path";
+
+    /// <summary>
+    /// Creates a new App Host store using the provided <paramref name="builder"/>.
+    /// </summary>
+    /// <param name="builder">The <see cref="IDistributedApplicationBuilder"/>.</param>
+    /// <returns>The <see cref="IAspireStore"/>.</returns>
+    public static IAspireStore CreateStore(this IDistributedApplicationBuilder builder)
+    {
+        ArgumentNullException.ThrowIfNull(builder);
+
+        var aspireDir = builder.Configuration[AspireStorePathKeyName];
+
+        if (string.IsNullOrWhiteSpace(aspireDir))
+        {
+            var assemblyMetadata = builder.AppHostAssembly?.GetCustomAttributes<AssemblyMetadataAttribute>();
+            aspireDir = GetMetadataValue(assemblyMetadata, "AppHostProjectBaseIntermediateOutputPath");
+
+            if (string.IsNullOrWhiteSpace(aspireDir))
+            {
+                throw new InvalidOperationException($"Could not determine an appropriate location for local storage. Set the {AspireStorePathKeyName} setting to a folder where the App Host content should be stored.");
+            }
+        }
+
+        return new AspireStore(Path.Combine(aspireDir, ".aspire"));
+    }
+
+    /// <summary>
+    /// Gets the metadata value for the specified key from the assembly metadata.
+    /// </summary>
+    /// <param name="assemblyMetadata">The assembly metadata.</param>
+    /// <param name="key">The key to look for.</param>
+    /// <returns>The metadata value if found; otherwise, null.</returns>
+    private static string? GetMetadataValue(IEnumerable<AssemblyMetadataAttribute>? assemblyMetadata, string key) =>
+        assemblyMetadata?.FirstOrDefault(a => string.Equals(a.Key, key, StringComparison.OrdinalIgnoreCase))?.Value;
+
+}

src/Aspire.Hosting/IAspireStore.cs

@@ -0,0 +1,49 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+namespace Aspire.Hosting;
+
+/// <summary>
+/// Represents a store for managing files in the Aspire hosting environment that can be reused across runs.
+/// </summary>
+/// <remarks>
+/// The store is created in the ./obj folder of the Application Host.
+/// If the ASPIRE_STORE_DIR environment variable is set this will be used instead.
+///
+/// The store is specific to a <see cref="IDistributedApplicationBuilder"/> instance such that each application can't
+/// conflict with others. A <em>.aspire</em> prefix is also used to ensure that the folder can be delete without impacting
+/// unrelated files.
+/// </remarks>
+public interface IAspireStore
+{
+    /// <summary>
+    /// Gets the base path of this store.
+    /// </summary>
+    string BasePath { get; }
+
+    /// <summary>
+    /// Creates an absolute file name for the provided <paramref name="filename"/> in the store.
+    /// </summary>
+    /// <param name="filename">The file name to use in the store.</param>
+    /// <returns>The absolute file name in the store.</returns>
+    string GetFileName(string filename);
+
+    /// <summary>
+    /// Gets a deterministic file path that is a copy of the content from the provided stream.
+    /// The resulting file name will depend on the content of the stream.
+    /// </summary>
+    /// <param name="filenameTemplate">A file name to base the result on.</param>
+    /// <param name="contentStream">A stream containing the content.</param>
+    /// <returns>A deterministic file path with the same content as the provided stream.</returns>
+    string GetFileNameWithContent(string filenameTemplate, Stream contentStream);
+
+    /// <summary>
+    /// Gets a deterministic file path that is a copy of the <paramref name="sourceFilename"/>.
+    /// The resulting file name will depend on the content of the file.
+    /// </summary>
+    /// <param name="filenameTemplate">A file name to base the result on.</param>
+    /// <param name="sourceFilename">An existing file.</param>
+    /// <returns>A deterministic file path with the same content as <paramref name="sourceFilename"/>.</returns>
+    /// <exception cref="FileNotFoundException">Thrown when the source file does not exist.</exception>
+    string GetFileNameWithContent(string filenameTemplate, string sourceFilename);
+}

src/Shared/SecretsStore.cs

@@ -1,7 +1,6 @@
 // Licensed to the .NET Foundation under one or more agreements.
 // The .NET Foundation licenses this file to you under the MIT license.
 
-using System.Runtime.InteropServices;
 using System.Text;
 using System.Text.Json.Nodes;
 using Microsoft.Extensions.Configuration;