feat: Corset code can be provided from STDIN

Consensys · Aug 28, 2023 · 024dc11 · 024dc11
1 parent e0939e4
commit 024dc11
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -14,6 +14,7 @@ path = "src/main.rs"
 
 [dependencies]
 anyhow = "1"
+atty = "0.2.14"
 buche = "0.7"
 buildstructor = "0.5.2"
 cached = { version = "0.43", default-features = false }

diff --git a/src/main.rs b/src/main.rs
@@ -421,6 +421,8 @@ impl ConstraintSetBuilder {
 
 #[cfg(feature = "cli")]
 fn main() -> Result<()> {
+    use std::io::{self, Read};
+
     let args = Args::parse();
     buche::new()
         .verbosity(args.verbose.log_level_filter())
@@ -434,7 +436,13 @@ fn main() -> Result<()> {
         .unwrap();
 
     let mut builder = if cfg!(feature = "parser") {
-        if matches!(args.command, Commands::Format { .. }) {
+        if !atty::is(atty::Stream::Stdin) {
+            let mut r = ConstraintSetBuilder::from_sources(args.no_stdlib, args.debug);
+            let mut input = String::new();
+            io::stdin().read_to_string(&mut input)?;
+            r.add_source(&input)?;
+            r
+        } else if matches!(args.command, Commands::Format { .. }) {
             if args.source.len() != 1 {
                 bail!("can only format one file at a time")
             } else if args.source.len() == 1