on new connection event subscription (#93)

* on new connection event subscription

Signed-off-by: iceseer <[email protected]>

* const callback

Signed-off-by: iceseer <[email protected]>

Author: iceseer

include/libp2p/host/basic_host/basic_host.hpp

@@ -75,6 +75,8 @@ namespace libp2p::host {
 
     event::Bus &getBus() override;
 
+    event::Handle setOnNewConnectionHandler(const NewConnectionHandler &h) const override;
+
    private:
     std::shared_ptr<peer::IdentityManager> idmgr_;
     std::unique_ptr<network::Network> network_;

include/libp2p/host/host.hpp

@@ -39,11 +39,19 @@ namespace libp2p {
     using StreamResultHandler = std::function<void(
         outcome::result<std::shared_ptr<connection::Stream>>)>;
 
+    using NewConnectionHandler = std::function<void(peer::PeerInfo &&)>;
+
     /**
      * @brief Get a version of Libp2p, supported by this Host
      */
     virtual std::string_view getLibp2pVersion() const = 0;
 
+    /**
+     * @brief Stores OnNewConnectionHandler.
+     * @param h handler function to store
+     */
+    virtual event::Handle setOnNewConnectionHandler(const NewConnectionHandler &h) const = 0;
+
     /**
      * @brief Get a version of this Libp2p client
      */

src/host/basic_host/basic_host.cpp

@@ -109,6 +109,26 @@ namespace libp2p::host {
     network_->getListener().start();
   }
 
+  event::Handle BasicHost::setOnNewConnectionHandler(const NewConnectionHandler &h) const {
+    return bus_->getChannel<network::event::OnNewConnectionChannel>().subscribe(
+        [h{std::move(h)}](auto &&conn) {
+          if (auto connection = conn.lock()) {
+            auto remote_peer_res = connection->remotePeer();
+            if (!remote_peer_res)
+              return;
+
+            auto remote_peer_addr_res = connection->remoteMultiaddr();
+            if (!remote_peer_addr_res)
+              return;
+
+            if (h != nullptr)
+              h(peer::PeerInfo{std::move(remote_peer_res.value()),
+                               std::vector<multi::Multiaddress>{
+                                   std::move(remote_peer_addr_res.value())}});
+          }
+        });
+  }
+
   void BasicHost::stop() {
     network_->getListener().stop();
   }

test/mock/libp2p/host/host_mock.hpp

@@ -16,6 +16,7 @@ namespace libp2p {
    public:
     ~HostMock() override = default;
 
+    MOCK_CONST_METHOD1(setOnNewConnectionHandler, event::Handle(const NewConnectionHandler &));
     MOCK_CONST_METHOD0(getLibp2pVersion, std::string_view());
     MOCK_CONST_METHOD0(getLibp2pClientVersion, std::string_view());
     MOCK_CONST_METHOD0(getId, peer::PeerId());