Support encoding streams in the cli (#387)

* Support encoding streams in the cli * remove unnecessary clone * upd hash * add todos for other streaming types
stellar · Aug 29, 2024 · 1746499 · 1746499
1 parent 550743b
commit 1746499
Show file tree

Hide file tree

Showing 4 changed files with 2,978 additions and 11 deletions.
diff --git a/Makefile b/Makefile
@@ -4,7 +4,7 @@ CARGO_HACK_ARGS=--feature-powerset --exclude-features default --group-features b
 
 CARGO_DOC_ARGS?=--open
 
-XDRGEN_VERSION=dda3c5ecea32847b7ab2333cfec602e892ae1478
+XDRGEN_VERSION=b7bc57ecdd277c9575930d3e17c12dfaa76655fc
 # XDRGEN_LOCAL=1
 XDRGEN_TYPES_CUSTOM_STR_IMPL_CURR=PublicKey,AccountId,MuxedAccount,MuxedAccountMed25519,SignerKey,SignerKeyEd25519SignedPayload,NodeId,ScAddress,AssetCode,AssetCode4,AssetCode12,ClaimableBalanceId
 XDRGEN_TYPES_CUSTOM_STR_IMPL_NEXT=PublicKey,AccountId,MuxedAccount,MuxedAccountMed25519,SignerKey,SignerKeyEd25519SignedPayload,NodeId,ScAddress,AssetCode,AssetCode4,AssetCode12,ClaimableBalanceId

diff --git a/src/cli/encode.rs b/src/cli/encode.rs
@@ -96,6 +96,9 @@ impl Default for InputFormat {
 pub enum OutputFormat {
     Single,
     SingleBase64,
+    Stream,
+    // TODO: StreamBase64,
+    // TODO: StreamFramed,
 }
 
 impl Default for OutputFormat {
@@ -114,15 +117,33 @@ macro_rules! run_x {
             })?;
             for f in &mut files {
                 match self.input {
-                    InputFormat::Json => {
-                        let t = crate::$m::Type::read_json(r#type, f)?;
-                        let l = crate::$m::Limits::none();
-
-                        match self.output {
-                            OutputFormat::Single => stdout().write_all(&t.to_xdr(l)?)?,
-                            OutputFormat::SingleBase64 => println!("{}", t.to_xdr_base64(l)?),
+                    InputFormat::Json => match self.output {
+                        OutputFormat::Single => {
+                            let t = crate::$m::Type::from_json(r#type, f)?;
+                            let l = crate::$m::Limits::none();
+                            stdout().write_all(&t.to_xdr(l)?)?
+                        }
+                        OutputFormat::SingleBase64 => {
+                            let t = crate::$m::Type::from_json(r#type, f)?;
+                            let l = crate::$m::Limits::none();
+                            println!("{}", t.to_xdr_base64(l)?)
+                        }
+                        OutputFormat::Stream => {
+                            let mut de =
+                                serde_json::Deserializer::new(serde_json::de::IoRead::new(f));
+                            loop {
+                                let t = match crate::$m::Type::deserialize_json(r#type, &mut de) {
+                                    Ok(t) => t,
+                                    Err(crate::$m::Error::Json(ref inner)) if inner.is_eof() => {
+                                        break;
+                                    }
+                                    Err(e) => Err(e)?,
+                                };
+                                let l = crate::$m::Limits::none();
+                                stdout().write_all(&t.to_xdr(l)?)?
+                            }
                         }
-                    }
+                    },
                 };
             }
             Ok(())