feat(dmq): update aggregator command for DMQ messages

Author: jpraynaud

mithril-relay/src/commands/aggregator.rs

@@ -1,11 +1,15 @@
+use std::path::PathBuf;
+
 use clap::Parser;
 use libp2p::Multiaddr;
-use mithril_common::StdResult;
 use slog::error;
 
-use super::CommandContext;
+use mithril_common::{CardanoNetwork, StdResult};
+
 use crate::AggregatorRelay;
 
+use super::CommandContext;
+
 #[derive(Parser, Debug, Clone)]
 pub struct AggregatorCommand {
     /// Peer listening port
@@ -16,6 +20,24 @@ pub struct AggregatorCommand {
     #[clap(long, env = "DIAL_TO")]
     dial_to: Option<Multiaddr>,
 
+    /// Path to the DMQ socket file
+    #[clap(
+        long,
+        env = "DMQ_NODE_SOCKET_PATH",
+        value_name = "PATH",
+        default_value = "./dmq.socket"
+    )]
+    dmq_node_socket_path: PathBuf,
+
+    /// Cardano network
+    #[clap(long, env = "NETWORK")]
+    pub network: String,
+
+    /// Cardano Network Magic number
+    /// useful for TestNet & DevNet
+    #[clap(long, env = "NETWORK_MAGIC")]
+    pub network_magic: Option<u64>,
+
     /// Aggregator endpoint URL.
     #[clap(long, env = "AGGREGATOR_ENDPOINT")]
     aggregator_endpoint: String,
@@ -24,12 +46,21 @@ pub struct AggregatorCommand {
 impl AggregatorCommand {
     /// Main command execution
     pub async fn execute(&self, context: CommandContext) -> StdResult<()> {
+        let logger = context.logger();
         let dial_to = self.dial_to.to_owned();
         let addr: Multiaddr = format!("/ip4/0.0.0.0/tcp/{}", self.listen_port).parse()?;
         let aggregator_endpoint = self.aggregator_endpoint.to_owned();
-        let logger = context.logger();
+        let cardano_network =
+            CardanoNetwork::from_code(self.network.to_owned(), self.network_magic)?;
 
-        let mut relay = AggregatorRelay::start(&addr, &aggregator_endpoint, logger).await?;
+        let mut relay = AggregatorRelay::start(
+            &addr,
+            &self.dmq_node_socket_path,
+            &cardano_network,
+            &aggregator_endpoint,
+            logger,
+        )
+        .await?;
         if let Some(dial_to_address) = dial_to {
             relay.dial_peer(dial_to_address.clone())?;
         }