From 8732f4f8c9ba257858b8e7097e672666dcec8253 Mon Sep 17 00:00:00 2001
From: Donovan Dall <awesomeibex@proton.me>
Date: Tue, 14 May 2024 16:54:54 +0100
Subject: [PATCH] feat: expose mode to the cli

---
 bin/cli/src/main.rs          |  5 ++++-
 crates/primitives/src/lib.rs | 11 +++++++++++
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/bin/cli/src/main.rs b/bin/cli/src/main.rs
index 9fafd3c..19b8a23 100644
--- a/bin/cli/src/main.rs
+++ b/bin/cli/src/main.rs
@@ -1,6 +1,7 @@
 use clap;
 use clap::{command, Parser};
 use near_da_http_api_data::ConfigureClientRequest;
+use near_da_primitives::Mode;
 use near_da_rpc::near::config::Config;
 use near_da_rpc::near::Client;
 use near_da_rpc::{CryptoHash, DataAvailability};
@@ -20,6 +21,8 @@ struct Args {
     config: Option<String>,
     #[command(subcommand)]
     command: Commands,
+    #[clap(short, long)]
+    mode: Option<Mode>,
 }
 struct AppState {
     client: Option<Client>,
@@ -37,7 +40,7 @@ fn config_request_to_config(request: ConfigureClientRequest) -> Result<Config, a
         namespace: request
             .namespace
             .map(|ns| near_da_primitives::Namespace::new(ns.version, ns.id)),
-        mode: Default::default(),
+        mode: request.mode.unwrap_or_default(),
     })
 }
 
diff --git a/crates/primitives/src/lib.rs b/crates/primitives/src/lib.rs
index b510984..4437b91 100644
--- a/crates/primitives/src/lib.rs
+++ b/crates/primitives/src/lib.rs
@@ -137,6 +137,17 @@ pub enum Mode {
     Pessimistic,
 }
 
+impl From<&str> for Mode {
+    fn from(s: &str) -> Self {
+        match s.to_lowercase().as_str() {
+            "optimistic" => Mode::Optimistic,
+            "standard" => Mode::Standard,
+            "pessimistic" => Mode::Pessimistic,
+            _ => Mode::Pessimistic,
+        }
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;