Update

Author: JamesNK

src/Aspire.Hosting/Dashboard/CommandsConfigurationExtensions.cs renamed to src/Aspire.Hosting/ApplicationModel/CommandsConfigurationExtensions.cs

@@ -1,18 +1,21 @@
 // Licensed to the .NET Foundation under one or more agreements.
 // The .NET Foundation licenses this file to you under the MIT license.
 
-using Aspire.Hosting.ApplicationModel;
 using Aspire.Hosting.Dcp;
 using Microsoft.Extensions.DependencyInjection;
 
-namespace Aspire.Hosting.Dashboard;
+namespace Aspire.Hosting.ApplicationModel;
 
 internal static class CommandsConfigurationExtensions
 {
+    internal const string StartType = "start";
+    internal const string StopType = "stop";
+    internal const string RestartType = "restart";
+
     internal static IResourceBuilder<T> WithLifeCycleCommands<T>(this IResourceBuilder<T> builder) where T : IResource
     {
         builder.WithCommand(
-            "start",
+            StartType,
             "Start",
             context =>
             {
@@ -39,7 +42,7 @@ internal static IResourceBuilder<T> WithLifeCycleCommands<T>(this IResourceBuild
             isHighlighted: true);
 
         builder.WithCommand(
-            "stop",
+            StopType,
             "Stop",
             context =>
             {
@@ -66,7 +69,7 @@ internal static IResourceBuilder<T> WithLifeCycleCommands<T>(this IResourceBuild
             isHighlighted: true);
 
         builder.WithCommand(
-            "restart",
+            RestartType,
             "Restart",
             context =>
             {

src/Aspire.Hosting/ContainerResourceBuilderExtensions.cs

@@ -3,7 +3,6 @@
 
 using System.Diagnostics.CodeAnalysis;
 using Aspire.Hosting.ApplicationModel;
-using Aspire.Hosting.Dashboard;
 using Aspire.Hosting.Utils;
 
 namespace Aspire.Hosting;

src/Aspire.Hosting/ExecutableResourceBuilderExtensions.cs

@@ -2,7 +2,6 @@
 // The .NET Foundation licenses this file to you under the MIT license.
 
 using Aspire.Hosting.ApplicationModel;
-using Aspire.Hosting.Dashboard;
 using Aspire.Hosting.Publishing;
 using Aspire.Hosting.Utils;
 

tests/Aspire.Hosting.Tests/ResourceCommandAnnotationTests.cs

@@ -0,0 +1,79 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+using Xunit;
+
+namespace Aspire.Hosting.Tests;
+
+public class ResourceCommandAnnotationTests
+{
+    [Fact]
+    public void AddContainer_HasKnownCommandAnnotations()
+    {
+        HasKnownCommandAnnotationsCore(builder => builder.AddContainer("name", "image"));
+    }
+
+    [Fact]
+    public void AddProject_HasKnownCommandAnnotations()
+    {
+        HasKnownCommandAnnotationsCore(builder => builder.AddProject("name", "path", o => o.ExcludeLaunchProfile = true));
+    }
+
+    [Fact]
+    public void AddExecutable_HasKnownCommandAnnotations()
+    {
+        HasKnownCommandAnnotationsCore(builder => builder.AddExecutable("name", "command", "workingDirectory"));
+    }
+
+    [Theory]
+    [InlineData(CommandsConfigurationExtensions.StartType, "Starting", ResourceCommandState.Disabled)]
+    [InlineData(CommandsConfigurationExtensions.StartType, "Stopping", ResourceCommandState.Hidden)]
+    [InlineData(CommandsConfigurationExtensions.StartType, "Running", ResourceCommandState.Hidden)]
+    [InlineData(CommandsConfigurationExtensions.StartType, "Exited", ResourceCommandState.Enabled)]
+    [InlineData(CommandsConfigurationExtensions.StopType, "Starting", ResourceCommandState.Hidden)]
+    [InlineData(CommandsConfigurationExtensions.StopType, "Stopping", ResourceCommandState.Disabled)]
+    [InlineData(CommandsConfigurationExtensions.StopType, "Running", ResourceCommandState.Enabled)]
+    [InlineData(CommandsConfigurationExtensions.StopType, "Exited", ResourceCommandState.Hidden)]
+    [InlineData(CommandsConfigurationExtensions.RestartType, "Starting", ResourceCommandState.Disabled)]
+    [InlineData(CommandsConfigurationExtensions.RestartType, "Stopping", ResourceCommandState.Disabled)]
+    [InlineData(CommandsConfigurationExtensions.RestartType, "Running", ResourceCommandState.Enabled)]
+    [InlineData(CommandsConfigurationExtensions.RestartType, "Exited", ResourceCommandState.Disabled)]
+    public void LifeCycleCommands_CommandState(string commandType, string resourceState, ResourceCommandState commandState)
+    {
+        // Arrange
+        var builder = DistributedApplication.CreateBuilder();
+        var resourceBuilder = builder.AddContainer("name", "image");
+
+        var startCommand = resourceBuilder.Resource.Annotations.OfType<ResourceCommandAnnotation>().Single(a => a.Type == commandType);
+
+        // Act
+        var state = startCommand.UpdateState(new UpdateCommandStateContext
+        {
+            ResourceSnapshot = new CustomResourceSnapshot
+            {
+                Properties = [],
+                ResourceType = "test",
+                State = resourceState
+            }
+        });
+
+        // Assert
+        Assert.Equal(commandState, state);
+    }
+
+    private static void HasKnownCommandAnnotationsCore<T>(Func<IDistributedApplicationBuilder, IResourceBuilder<T>> createResourceBuilder) where T : IResource
+    {
+        // Arrange
+        var builder = DistributedApplication.CreateBuilder();
+
+        // Act
+        var resourceBuilder = createResourceBuilder(builder);
+
+        // Assert
+        var commandAnnotations = resourceBuilder.Resource.Annotations.OfType<ResourceCommandAnnotation>().ToList();
+        Assert.Collection(commandAnnotations,
+            a => Assert.Equal(CommandsConfigurationExtensions.StartType, a.Type),
+            a => Assert.Equal(CommandsConfigurationExtensions.StopType, a.Type),
+            a => Assert.Equal(CommandsConfigurationExtensions.RestartType, a.Type));
+    }
+}

tests/Aspire.Hosting.Tests/ResourceExtensionsTests.cs

@@ -4,6 +4,7 @@
 using Xunit;
 
 namespace Aspire.Hosting.Tests;
+
 public class ResourceExtensionsTests
 {
     [Fact]