Pārlūkot izejas kodu

drk: restrict aliases to 5 characters

aggstam 3 gadi atpakaļ
vecāks
revīzija
cd8c619503
1 mainītis faili ar 6 papildinājumiem un 1 dzēšanām
  1. 6 1
      bin/drk/src/main.rs

+ 6 - 1
bin/drk/src/main.rs

@@ -981,7 +981,7 @@ async fn main() -> Result<()> {
                     tx
                     tx
                 } else {
                 } else {
                     eprintln!("Tx not found");
                     eprintln!("Tx not found");
-                    exit(1);
+                    return Ok(())
                 };
                 };
                 // Make sure the tx is correct
                 // Make sure the tx is correct
                 assert_eq!(tx.hash(), tx_hash);
                 assert_eq!(tx.hash(), tx_hash);
@@ -1017,6 +1017,11 @@ async fn main() -> Result<()> {
 
 
         Subcmd::Alias(cmd) => match cmd {
         Subcmd::Alias(cmd) => match cmd {
             AliasSubcmd::Add { alias, token } => {
             AliasSubcmd::Add { alias, token } => {
+                if alias.chars().count() > 5 {
+                    eprintln!("Error: Alias exceeds 5 characters");
+                    return Ok(())
+                }
+
                 let token_id =
                 let token_id =
                     TokenId::try_from(token.as_str()).with_context(|| "Invalid Token ID")?;
                     TokenId::try_from(token.as_str()).with_context(|| "Invalid Token ID")?;
                 let drk = Drk::new(args.endpoint).await?;
                 let drk = Drk::new(args.endpoint).await?;