up

src/Excititor/StellaOps.Excititor.Worker/Program.cs

@@ -1,25 +1,27 @@
-using System.IO;
-using System.Linq;
-using Microsoft.Extensions.DependencyInjection;
-using Microsoft.Extensions.Hosting;
+using System.IO;
+using System.Linq;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Hosting;
 using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Options;
 using StellaOps.Plugin;
 using StellaOps.Excititor.Connectors.RedHat.CSAF.DependencyInjection;
 using StellaOps.Excititor.Core;
 using StellaOps.Excititor.Core.Aoc;
+using StellaOps.Excititor.Core.Orchestration;
 using StellaOps.Excititor.Formats.CSAF;
 using StellaOps.Excititor.Formats.CycloneDX;
 using StellaOps.Excititor.Formats.OpenVEX;
 using StellaOps.Excititor.Storage.Mongo;
 using StellaOps.Excititor.Worker.Auth;
 using StellaOps.Excititor.Worker.Options;
+using StellaOps.Excititor.Worker.Orchestration;
 using StellaOps.Excititor.Worker.Scheduling;
 using StellaOps.Excititor.Worker.Signature;
 using StellaOps.Excititor.Attestation.Extensions;
 using StellaOps.Excititor.Attestation.Verification;
 using StellaOps.IssuerDirectory.Client;
-
+
 var builder = Host.CreateApplicationBuilder(args);
 var services = builder.Services;
 var configuration = builder.Configuration;
@@ -40,11 +42,11 @@ services.PostConfigure<VexWorkerOptions>(options =>
     }
 });
 services.AddRedHatCsafConnector();
-
-services.AddOptions<VexMongoStorageOptions>()
-    .Bind(configuration.GetSection("Excititor:Storage:Mongo"))
-    .ValidateOnStart();
-
+
+services.AddOptions<VexMongoStorageOptions>()
+    .Bind(configuration.GetSection("Excititor:Storage:Mongo"))
+    .ValidateOnStart();
+
 services.AddExcititorMongoStorage();
 services.AddCsafNormalizer();
 services.AddCycloneDxNormalizer();
@@ -71,38 +73,45 @@ services.PostConfigure<VexAttestationVerificationOptions>(options =>
     }
 });
 services.AddExcititorAocGuards();
-
-services.AddSingleton<IValidateOptions<VexWorkerOptions>, VexWorkerOptionsValidator>();
-services.AddSingleton(TimeProvider.System);
-services.PostConfigure<VexWorkerOptions>(options =>
-{
-    if (!options.Providers.Any(provider => string.Equals(provider.ProviderId, "excititor:redhat", StringComparison.OrdinalIgnoreCase)))
-    {
-        options.Providers.Add(new VexWorkerProviderOptions
-        {
-            ProviderId = "excititor:redhat",
-        });
-    }
-});
+
+services.AddSingleton<IValidateOptions<VexWorkerOptions>, VexWorkerOptionsValidator>();
+services.AddSingleton(TimeProvider.System);
+services.PostConfigure<VexWorkerOptions>(options =>
+{
+    if (!options.Providers.Any(provider => string.Equals(provider.ProviderId, "excititor:redhat", StringComparison.OrdinalIgnoreCase)))
+    {
+        options.Providers.Add(new VexWorkerProviderOptions
+        {
+            ProviderId = "excititor:redhat",
+        });
+    }
+});
 services.AddSingleton<PluginCatalog>(provider =>
 {
     var pluginOptions = provider.GetRequiredService<IOptions<VexWorkerPluginOptions>>().Value;
     var catalog = new PluginCatalog();
 
-    var directory = pluginOptions.ResolveDirectory();
-    if (Directory.Exists(directory))
-    {
-        catalog.AddFromDirectory(directory, pluginOptions.ResolveSearchPattern());
-    }
-    else
-    {
-        var logger = provider.GetRequiredService<ILogger<Program>>();
-        logger.LogWarning("Excititor worker plugin directory '{Directory}' does not exist; proceeding without external connectors.", directory);
-    }
-
-    return catalog;
+    var directory = pluginOptions.ResolveDirectory();
+    if (Directory.Exists(directory))
+    {
+        catalog.AddFromDirectory(directory, pluginOptions.ResolveSearchPattern());
+    }
+    else
+    {
+        var logger = provider.GetRequiredService<ILogger<Program>>();
+        logger.LogWarning("Excititor worker plugin directory '{Directory}' does not exist; proceeding without external connectors.", directory);
+    }
+
+    return catalog;
 });
 
+// Orchestrator worker SDK integration
+services.AddOptions<VexWorkerOrchestratorOptions>()
+    .Bind(configuration.GetSection("Excititor:Worker:Orchestrator"))
+    .ValidateOnStart();
+services.AddSingleton<IVexWorkerOrchestratorClient, VexWorkerOrchestratorClient>();
+services.AddSingleton<VexWorkerHeartbeatService>();
+
 services.AddSingleton<IVexProviderRunner, DefaultVexProviderRunner>();
 services.AddHostedService<VexWorkerHostedService>();
 if (!workerConfigSnapshot.DisableConsensus)
@@ -115,5 +124,5 @@ services.AddSingleton<ITenantAuthorityClientFactory, TenantAuthorityClientFactor
 
 var host = builder.Build();
 await host.RunAsync();
-
-public partial class Program;
+
+public partial class Program;