Add CLI flag to dump header

Author: matt-cornell

cobalt-cli/Cargo.toml

@@ -29,10 +29,11 @@ object = { version = "0.32.1", features = ["write"] }
 ar = "0.9.0"
 os_str_bytes = { version = "6.6.1", features = ["conversions"] }
 clio = { version = "0.3.4", features = ["clap-parse"] }
+serde_json = "1.0.111"
 
 [[bin]]
 name = "co"
 path = "src/main.rs"
 
 [features]
-http = ["clio/http-ureq"]
+http = ["clio/http-ureq"]

cobalt-cli/src/lib.rs

@@ -186,6 +186,9 @@ pub enum Cli {
         /// print timings
         #[arg(long)]
         timings: bool,
+        /// Print header as JSON
+        #[arg(long)]
+        dump_header: bool,
     },
     /// multi-file utilities
     #[command(subcommand)]
@@ -219,12 +222,6 @@ pub enum DbgSubcommand {
         #[arg(long)]
         timings: bool,
     },
-    /// Parse a Cobalt header
-    ParseHeader {
-        /// header files to parse
-        #[arg(required = true)]
-        inputs: Vec<Input>,
-    },
 }
 #[derive(Debug, Clone, Subcommand)]
 pub enum MultiSubcommand {
@@ -314,6 +311,9 @@ pub enum MultiSubcommand {
         /// print timings
         #[arg(long)]
         timings: bool,
+        /// Print header as JSON
+        #[arg(long)]
+        dump_header: bool,
     },
 }
 #[derive(Debug, Clone, Subcommand)]
@@ -602,18 +602,6 @@ pub fn driver(cli: Cli) -> anyhow::Result<()> {
                 print!("{}", ctx.module.to_string());
                 reporter.finish();
             }
-            #[cfg(debug_assertions)]
-            DbgSubcommand::ParseHeader { inputs } => {
-                for mut input in inputs {
-                    let ink_ctx = inkwell::context::Context::create();
-                    let ctx = CompCtx::new(&ink_ctx, "<anon>");
-                    let mut file = BufReader::new(&mut input);
-                    match ctx.load(&mut file) {
-                        Ok(_) => ctx.with_vars(|v| v.dump()),
-                        Err(e) => eprintln!("error loading {}: {e}", input.path().display()),
-                    }
-                }
-            }
         },
         Cli::Aot {
             mut input,
@@ -1268,6 +1256,7 @@ pub fn driver(cli: Cli) -> anyhow::Result<()> {
             headers,
             no_default_link,
             timings,
+            dump_header,
         } => {
             struct Reporter {
                 timings: bool,
@@ -1453,6 +1442,9 @@ pub fn driver(cli: Cli) -> anyhow::Result<()> {
             if fail {
                 anyhow::bail!(CompileErrors(ec))
             }
+            if dump_header {
+                serde_json::to_writer_pretty(std::io::stdout(), &ctx)?;
+            }
             reporter.finish();
         }
         Cli::Multi(cmd) => match cmd {
@@ -2191,6 +2183,7 @@ pub fn driver(cli: Cli) -> anyhow::Result<()> {
                 headers,
                 no_default_link,
                 timings,
+                dump_header,
             } => {
                 struct Reporter {
                     timings: bool,
@@ -2431,6 +2424,9 @@ pub fn driver(cli: Cli) -> anyhow::Result<()> {
                 if fail {
                     anyhow::bail!(CompileErrors(ec))
                 }
+                if dump_header {
+                    serde_json::to_writer_pretty(std::io::stdout(), &ctx)?;
+                }
                 reporter.finish();
             }
         },