initial constructor callback commit

include/quic/context.hpp

@@ -35,6 +35,7 @@ namespace oxen::quic
         stream_data_callback stream_data_cb;
         stream_open_callback stream_open_cb;
         stream_close_callback stream_close_cb;
+        stream_constructor_callback stream_construct_cb;
         user_config config{};
 
         template <typename... Opt>
@@ -60,6 +61,7 @@ namespace oxen::quic
         void handle_ioctx_opt(stream_data_callback func);
         void handle_ioctx_opt(stream_open_callback func);
         void handle_ioctx_opt(stream_close_callback func);
+        void handle_ioctx_opt(stream_constructor_callback func);
     };
 
 }  // namespace oxen::quic

include/quic/stream.hpp

@@ -29,6 +29,7 @@ namespace oxen::quic
     // Stream callbacks
     using stream_data_callback = std::function<void(Stream&, bstring_view)>;
     using stream_close_callback = std::function<void(Stream&, uint64_t error_code)>;
+    using stream_constructor_callback = std::function<std::shared_ptr<Stream>(int64_t)>;
     // returns 0 on success
     using stream_open_callback = std::function<uint64_t(Stream&)>;
     using stream_unblocked_callback = std::function<bool(Stream&)>;

src/connection.cpp

@@ -362,7 +362,9 @@ namespace oxen::quic
             if (!data_cb)
                 data_cb = context->stream_data_cb;
 
-            auto stream = std::make_shared<Stream>(*this, _endpoint, std::move(data_cb), std::move(close_cb));
+            auto stream = (context->stream_construct_cb)
+                                ? context->stream_construct_cb(0)
+                                : std::make_shared<Stream>(*this, _endpoint, std::move(data_cb), std::move(close_cb));
 
             if (int rv = ngtcp2_conn_open_bidi_stream(conn.get(), &stream->_stream_id, stream.get()); rv != 0)
             {
@@ -773,7 +775,10 @@ namespace oxen::quic
         log::trace(log_cat, "{} called", __PRETTY_FUNCTION__);
         log::info(log_cat, "New stream ID:{}", id);
 
-        auto stream = std::make_shared<Stream>(*this, _endpoint, context->stream_data_cb, context->stream_close_cb, id);
+        auto stream =
+                (context->stream_construct_cb)
+                        ? context->stream_construct_cb(id)
+                        : std::make_shared<Stream>(*this, _endpoint, context->stream_data_cb, context->stream_close_cb, id);
         stream->set_ready();
 
         log::debug(log_cat, "Local endpoint creating stream to match remote");

src/context.cpp

@@ -32,4 +32,10 @@ namespace oxen::quic
         log::trace(log_cat, "IO context stored stream open callback");
         stream_close_cb = std::move(func);
     }
+
+    void IOContext::handle_ioctx_opt(stream_constructor_callback func)
+    {
+        log::trace(log_cat, "IO context stored stream constructor callback");
+        stream_construct_cb = std::move(func);
+    }
 }  // namespace oxen::quic