feat: route publishes to jetstream with puback

src/NATS.Server/NatsServer.cs

@@ -12,6 +12,7 @@ using NATS.Server.Configuration;
 using NATS.Server.Gateways;
 using NATS.Server.JetStream;
 using NATS.Server.JetStream.Api;
+using NATS.Server.JetStream.Publish;
 using NATS.Server.LeafNodes;
 using NATS.Server.Monitoring;
 using NATS.Server.Protocol;
@@ -49,6 +50,8 @@ public sealed class NatsServer : IMessageRouter, ISubListAccess, IDisposable
     private readonly LeafNodeManager? _leafNodeManager;
     private readonly JetStreamService? _jetStreamService;
     private readonly JetStreamApiRouter? _jetStreamApiRouter;
+    private readonly StreamManager? _jetStreamStreamManager;
+    private readonly JetStreamPublisher? _jetStreamPublisher;
     private Socket? _listener;
     private Socket? _wsListener;
     private readonly TaskCompletionSource _wsAcceptLoopExited = new(TaskCreationOptions.RunContinuationsAsynchronously);
@@ -94,6 +97,14 @@ public sealed class NatsServer : IMessageRouter, ISubListAccess, IDisposable
 
     public IEnumerable<Auth.Account> GetAccounts() => _accounts.Values;
     public bool HasRemoteInterest(string subject) => _globalAccount.SubList.HasRemoteInterest(subject);
+    public bool TryCaptureJetStreamPublish(string subject, ReadOnlyMemory<byte> payload, out PubAck ack)
+    {
+        if (_jetStreamPublisher != null)
+            return _jetStreamPublisher.TryCapture(subject, payload, out ack);
+
+        ack = new PubAck();
+        return false;
+    }
 
     public Task WaitForReadyAsync() => _listeningStarted.Task;
 
@@ -329,8 +340,10 @@ public sealed class NatsServer : IMessageRouter, ISubListAccess, IDisposable
 
         if (options.JetStream != null)
         {
+            _jetStreamStreamManager = new StreamManager();
             _jetStreamService = new JetStreamService(options.JetStream);
-            _jetStreamApiRouter = new JetStreamApiRouter();
+            _jetStreamApiRouter = new JetStreamApiRouter(_jetStreamStreamManager);
+            _jetStreamPublisher = new JetStreamPublisher(_jetStreamStreamManager);
         }
 
         if (options.HasTls)
@@ -746,6 +759,9 @@ public sealed class NatsServer : IMessageRouter, ISubListAccess, IDisposable
     public void ProcessMessage(string subject, string? replyTo, ReadOnlyMemory<byte> headers,
         ReadOnlyMemory<byte> payload, NatsClient sender)
     {
+        if (TryCaptureJetStreamPublish(subject, payload, out var pubAck))
+            sender.RecordJetStreamPubAck(pubAck);
+
         // Apply subject transforms
         if (_subjectTransforms.Length > 0)
         {