feat: Add default p2p node type

eigerco · Jan 7, 2025 · 64987d6 · 64987d6
1 parent 1647ce3
commit 64987d6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/storage-provider/server/src/main.rs b/storage-provider/server/src/main.rs
@@ -251,7 +251,7 @@ pub struct ServerArguments {
     parallel_prove_commits: NonZero<usize>,
 
     /// P2P Node type, can be either a bootstrap node or a registration node.
-    #[arg(long)]
+    #[arg(long, default_value = "bootstrap")]
     node_type: NodeType,
 
     /// Path to P2P config file

diff --git a/storage-provider/server/src/p2p.rs b/storage-provider/server/src/p2p.rs
@@ -1,4 +1,5 @@
 use std::{
+    fmt::Display,
     fs::read_to_string,
     path::{Path, PathBuf},
 };
@@ -23,6 +24,12 @@ pub enum NodeType {
     Register,
 }
 
+impl Display for NodeType {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "{:?}", self)
+    }
+}
+
 #[derive(Debug, thiserror::Error)]
 pub enum P2PError {
     #[error(transparent)]