Public/nushell
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add view ir --json for use in external tooling

Browse Source
This commit is contained in:
Devyn Cairns 2024-06-27 17:31:59 -07:00
parent 9e19f15cb5
commit 06b7e5c30a
No known key found for this signature in database
1 changed files with 24 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
crates/nu-command/src/debug/view_ir.rs
View File

@ -12,11 +12,13 @@ impl Command for ViewIr {
}
fn signature(&self) -> Signature {
Signature::new(self.name()).required(
"closure",
SyntaxShape::Closure(None),
"the closure to see compiled code for",
)
Signature::new(self.name())
.required(
"closure",
SyntaxShape::Closure(None),
"the closure to see compiled code for",
)
.switch("json", "Dump the raw block data as JSON", Some('j'))
}
fn usage(&self) -> &str {
@ -31,6 +33,7 @@ impl Command for ViewIr {
_input: PipelineData,
) -> Result<PipelineData, ShellError> {
let closure: Closure = call.req(engine_state, stack, 0)?;
let json = call.has_flag(engine_state, stack, "json")?;
let block = engine_state.get_block(closure.block_id);
// Use the pre-compiled block if available, otherwise try to compile it
@ -43,7 +46,22 @@ impl Command for ViewIr {
),
};
let formatted = format!("{}", ir_block.display(engine_state));
let formatted = if json {
serde_json::to_string_pretty(&serde_json::json!({
"block_id": closure.block_id,
"span": block.span,
"ir_block": ir_block,
}))
.map_err(|err| ShellError::GenericError {
error: "JSON serialization failed".into(),
msg: err.to_string(),
span: Some(call.head),
help: None,
inner: vec![],
})?
} else {
format!("{}", ir_block.display(engine_state))
};
Ok(Value::string(formatted, call.head).into_pipeline_data())
}
}