Command line option to specify the check in memo

Author: yanliu38

cli/src/argparse.rs

@@ -115,6 +115,12 @@ pub fn parse_args() -> clap::ArgMatches {
                         .action(ArgAction::Set)
                         .num_args(1),
                 )
+                .arg(
+                    Arg::new("check-in-memo")
+                        .long("check-in-memo")
+                        .help("Provide the given memo value for the check-in")
+                        .action(ArgAction::Set)
+                )
                 .arg(
                     Arg::new("check-in-nonce")
                         .long("check-in-nonce")

cli/src/ledger_canister_client.rs

@@ -96,8 +96,13 @@ impl LedgerCanister {
         Decode!(response.as_slice(), ResultString).map_err(|e| e.to_string())?
     }
 
-    pub async fn node_provider_check_in(&self, key: &[u8], value: &[u8]) -> Result<String, String> {
-        let args = Encode!(&key, &value).map_err(|e| e.to_string())?;
+    pub async fn node_provider_check_in(
+        &self,
+        key: &[u8],
+        memo: &String,
+        nonce_crypto_sig: &[u8],
+    ) -> Result<String, String> {
+        let args = Encode!(&key, &memo, &nonce_crypto_sig).map_err(|e| e.to_string())?;
         let response = self.call_update("node_provider_check_in", &args).await?;
         Decode!(response.as_slice(), ResultString).map_err(|e| e.to_string())?
     }

cli/src/main.rs

@@ -228,11 +228,23 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                         nonce_string,
                         nonce_bytes.len()
                     );
+                    let check_in_memo = match arg_matches
+                        .get_one::<String>("check-in-memo")
+                        .cloned()
+                    {
+                        Some(memo) => memo,
+                        None => {
+                            println!("No memo specified, did you know that you can specify one? Try out --check-in-memo");
+                            String::new()
+                        }
+                    };
+                    let nonce_crypto_signature = dcc_ident.sign(nonce_bytes.as_ref())?;
                     let result = ledger_canister(ic_auth)
                         .await?
                         .node_provider_check_in(
                             &dcc_ident.to_bytes_verifying(),
-                            &dcc_ident.sign(&nonce_bytes)?.to_bytes(),
+                            &check_in_memo,
+                            &nonce_crypto_signature.to_bytes(),
                         )
                         .await
                         .map_err(|e| format!("Check-in failed: {}", e))?;