Update AzureEventGrid to support IMessageBus and composite mode

Co-authored-by: dandrejvv <[email protected]>

Author: Copilot

Modules/Intent.Modules.Eventing.AzureEventGrid/Intent.Modules.Eventing.AzureEventGrid.csproj

@@ -7,7 +7,7 @@
 
   <ItemGroup>
     <PackageReference Include="Intent.Modules.Common" Version="3.8.0" />
-    <PackageReference Include="Intent.Modules.Common.CSharp" Version="3.8.3" />
+    <PackageReference Include="Intent.Modules.Common.CSharp" Version="3.9.0" />
     <PackageReference Include="Intent.Modules.Eventing.Contracts" Version="5.2.0" />
     <PackageReference Include="Intent.Modules.Modelers.Eventing" Version="6.0.1" />
     <PackageReference Include="Intent.Modules.Modelers.Services" Version="3.10.0" />
@@ -22,6 +22,7 @@
 
   <ItemGroup>
     <ProjectReference Include="..\Intent.Modules.Constants\Intent.Modules.Constants.csproj" PrivateAssets="All" />
+    <ProjectReference Include="..\Intent.Modules.Eventing.Contracts\Intent.Modules.Eventing.Contracts.csproj" PrivateAssets="All" />
   </ItemGroup>
 
   <Import Project="..\Intent.Modules.Eventing.Shared\Intent.Modules.Eventing.Shared.projitems" Label="Shared" />

Modules/Intent.Modules.Eventing.AzureEventGrid/Templates/AzureEventGridConfiguration/AzureEventGridConfigurationTemplatePartial.cs

@@ -42,8 +42,26 @@ public AzureEventGridConfigurationTemplate(IOutputTarget outputTarget, object mo
                         method.Static()
                             .AddParameter("IServiceCollection", "services", param => param.WithThisModifier())
                             .AddParameter("IConfiguration", "configuration");
+                        
+                        var compositeTemplate = GetTemplate<ICSharpFileBuilderTemplate>(TemplateDependency.OnTemplate("Intent.Eventing.Contracts.CompositeMessageBusConfiguration"));
+                        var isCompositeMode = compositeTemplate != null;
+                        
+                        if (isCompositeMode)
+                        {
+                            method.AddParameter(this.GetMessageBrokerRegistryName(), "registry");
+                        }
 
-                        method.AddStatement($"services.AddScoped<{this.GetEventBusInterfaceName()}, {this.GetAzureEventGridEventBusName()}>();");
+                        if (isCompositeMode)
+                        {
+                            method.AddStatement($"// Register as concrete type for composite message bus");
+                            method.AddStatement($"services.AddScoped<{this.GetAzureEventGridEventBusName()}>();");
+                        }
+                        else
+                        {
+                            method.AddStatement($"// Register as IMessageBus and IEventBus for standalone mode");
+                            method.AddStatement($"services.AddScoped<{this.GetMessageBusInterfaceName()}, {this.GetAzureEventGridEventBusName()}>();");
+                            method.AddStatement($"services.AddScoped<{this.GetEventBusInterfaceName()}, {this.GetAzureEventGridEventBusName()}>();");
+                        }
                         method.AddStatement($"services.AddSingleton<{this.GetAzureEventGridMessageDispatcherName()}>();");
                         method.AddStatement($"services.AddSingleton<{this.GetAzureEventGridMessageDispatcherInterfaceName()}, {this.GetAzureEventGridMessageDispatcherName()}>();");
 
@@ -58,6 +76,16 @@ public AzureEventGridConfigurationTemplate(IOutputTarget outputTarget, object mo
                         if (publishMessages.Count != 0)
                         {
                             ConfigurePublisherOptions(method, publishMessages);
+                            
+                            if (isCompositeMode)
+                            {
+                                method.AddStatement("", s => s.SeparatedFromPrevious());
+                                method.AddStatement("// Register message types with the composite message bus registry");
+                                foreach (var publishMessage in publishMessages)
+                                {
+                                    method.AddStatement($"registry.Register<{publishMessage.GetModelTypeName(this)}, {this.GetAzureEventGridEventBusName()}>();");
+                                }
+                            }
                         }
 
                         var eventHandlers = IntegrationManager.Instance.GetSubscribedAzureEventGridMessages(ExecutionContext.GetApplicationConfig().Id);

Modules/Intent.Modules.Eventing.AzureEventGrid/Templates/AzureEventGridEventBus/AzureEventGridEventBusTemplatePartial.cs

@@ -37,7 +37,7 @@ public AzureEventGridEventBusTemplate(IOutputTarget outputTarget, object model =
                 .AddUsing("Microsoft.Extensions.Options")
                 .AddClass($"AzureEventGridEventBus", @class =>
                 {
-                    @class.ImplementsInterface(this.GetEventBusInterfaceName());
+                    @class.ImplementsInterface(this.GetMessageBusInterfaceName());
                     @class.AddField("List<MessageEntry>", "_messageQueue", field => field.PrivateReadOnly().WithAssignment(new CSharpStatement("[]")));
                     @class.AddField("Dictionary<string, PublisherEntry>", "_lookup", field => field.PrivateReadOnly());
 
@@ -78,6 +78,25 @@ public AzureEventGridEventBusTemplate(IOutputTarget outputTarget, object model =
                         method.AddStatement("_messageQueue.Add(new MessageEntry(message, additionalData));");
                     });
 
+                    @class.AddMethod("void", "Send", method =>
+                    {
+                        method.AddGenericParameter("T", out var T);
+                        method.AddGenericTypeConstraint(T, c => c.AddType("class"));
+                        method.AddParameter(T, "message");
+
+                        method.AddStatement("Publish(message);");
+                    });
+                    
+                    @class.AddMethod("void", "Send", method =>
+                    {
+                        method.AddGenericParameter("T", out var T);
+                        method.AddGenericTypeConstraint(T, c => c.AddType("class"));
+                        method.AddParameter(T, "message");
+                        method.AddParameter("IDictionary<string, object>", "additionalData");
+
+                        method.AddStatement("Publish(message, additionalData);");
+                    });
+
                     @class.AddMethod("Task", "FlushAllAsync", method =>
                     {
                         method.Async();