pass the indent string and the depth recursively

This commit calls the main "to nuon" function with
- no argument when `--raw` is used
- " " repeated $i$ times with the `--indent <i>`
- "\t" repeated $t$ times with the `--tabs <t>`
This commit is contained in:
amtoine 2023-03-08 20:09:11 +01:00
parent ac2f6b79e6
commit e504bbc12d
No known key found for this signature in database
GPG Key ID: 37AAE9B486CFF1AB
2 changed files with 53 additions and 23 deletions

View File

@ -244,7 +244,7 @@ fn sort_attributes(val: Value) -> Value {
fn generate_key(item: &ValueCounter) -> Result<String, ShellError> { fn generate_key(item: &ValueCounter) -> Result<String, ShellError> {
let value = sort_attributes(item.val_to_compare.clone()); //otherwise, keys could be different for Records let value = sort_attributes(item.val_to_compare.clone()); //otherwise, keys could be different for Records
value_to_string(&value, Span::unknown()) value_to_string(&value, Span::unknown(), 0, &None)
} }
fn generate_results_with_count(head: Span, uniq_values: Vec<ValueCounter>) -> Vec<Value> { fn generate_results_with_count(head: Span, uniq_values: Vec<ValueCounter>) -> Vec<Value> {

View File

@ -1,11 +1,13 @@
use core::fmt::Write; use core::fmt::Write;
use fancy_regex::Regex; use fancy_regex::Regex;
use nu_engine::get_columns; use nu_engine::get_columns;
use nu_engine::CallExt;
use nu_parser::escape_quote_string; use nu_parser::escape_quote_string;
use nu_protocol::ast::{Call, RangeInclusion}; use nu_protocol::ast::{Call, RangeInclusion};
use nu_protocol::engine::{Command, EngineState, Stack}; use nu_protocol::engine::{Command, EngineState, Stack};
use nu_protocol::{ use nu_protocol::{
Category, Example, IntoPipelineData, PipelineData, ShellError, Signature, SyntaxShape, Span, Type, Value, Category, Example, IntoPipelineData, PipelineData, ShellError, Signature, Span, SyntaxShape,
Type, Value,
}; };
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
@ -42,8 +44,8 @@ impl Command for ToNuon {
fn run( fn run(
&self, &self,
_engine_state: &EngineState, engine_state: &EngineState,
_stack: &mut Stack, stack: &mut Stack,
call: &Call, call: &Call,
input: PipelineData, input: PipelineData,
) -> Result<PipelineData, ShellError> { ) -> Result<PipelineData, ShellError> {
@ -51,9 +53,19 @@ impl Command for ToNuon {
let use_tabs = call.has_flag("tabs"); let use_tabs = call.has_flag("tabs");
let span = call.head; let span = call.head;
let nuon_value = to_nuon(call, input)?; // allow ranges to expand and turn into array
let input = input.try_expand_range()?;
let value = input.into_value(span);
let nuon_result: Result<String, ()> = Ok(nuon_value); let nuon_result = if raw {
value_to_string(&value, span, 0, &None)
} else if use_tabs {
let tab_count: usize = call.get_flag(engine_state, stack, "tabs")?.unwrap_or(1);
value_to_string(&value, span, 0, &Some("\t".repeat(tab_count)))
} else {
let indent: usize = call.get_flag(engine_state, stack, "indent")?.unwrap_or(2);
value_to_string(&value, span, 0, &Some(" ".repeat(indent)))
};
match nuon_result { match nuon_result {
Ok(serde_nuon_string) => Ok(Value::String { Ok(serde_nuon_string) => Ok(Value::String {
@ -64,7 +76,7 @@ impl Command for ToNuon {
_ => Ok(Value::Error { _ => Ok(Value::Error {
error: ShellError::CantConvert { error: ShellError::CantConvert {
to_type: "NUON".into(), to_type: "NUON".into(),
from_type: "TYPE".to_string(), from_type: value.get_type().to_string(),
span, span,
help: None, help: None,
}, },
@ -82,7 +94,12 @@ impl Command for ToNuon {
} }
} }
pub fn value_to_string(v: &Value, span: Span) -> Result<String, ShellError> { pub fn value_to_string(
v: &Value,
span: Span,
depth: usize,
indent: &Option<String>,
) -> Result<String, ShellError> {
match v { match v {
Value::Binary { val, .. } => { Value::Binary { val, .. } => {
let mut s = String::with_capacity(2 * val.len()); let mut s = String::with_capacity(2 * val.len());
@ -171,7 +188,12 @@ pub fn value_to_string(v: &Value, span: Span) -> Result<String, ShellError> {
if let Value::Record { vals, .. } = val { if let Value::Record { vals, .. } = val {
for val in vals { for val in vals {
row.push(value_to_string_without_quotes(val, span)?); row.push(value_to_string_without_quotes(
val,
span,
depth + 1,
indent,
)?);
} }
} }
@ -186,7 +208,12 @@ pub fn value_to_string(v: &Value, span: Span) -> Result<String, ShellError> {
} else { } else {
let mut collection = vec![]; let mut collection = vec![];
for val in vals { for val in vals {
collection.push(value_to_string_without_quotes(val, span)?); collection.push(value_to_string_without_quotes(
val,
span,
depth + 1,
indent,
)?);
} }
Ok(format!("[{}]", collection.join(", "))) Ok(format!("[{}]", collection.join(", ")))
} }
@ -194,13 +221,13 @@ pub fn value_to_string(v: &Value, span: Span) -> Result<String, ShellError> {
Value::Nothing { .. } => Ok("null".to_string()), Value::Nothing { .. } => Ok("null".to_string()),
Value::Range { val, .. } => Ok(format!( Value::Range { val, .. } => Ok(format!(
"{}..{}{}", "{}..{}{}",
value_to_string(&val.from, span)?, value_to_string(&val.from, span, depth + 1, indent)?,
if val.inclusion == RangeInclusion::RightExclusive { if val.inclusion == RangeInclusion::RightExclusive {
"<" "<"
} else { } else {
"" ""
}, },
value_to_string(&val.to, span)? value_to_string(&val.to, span, depth + 1, indent)?
)), )),
Value::Record { cols, vals, .. } => { Value::Record { cols, vals, .. } => {
let mut collection = vec![]; let mut collection = vec![];
@ -209,17 +236,21 @@ pub fn value_to_string(v: &Value, span: Span) -> Result<String, ShellError> {
format!( format!(
"\"{}\": {}", "\"{}\": {}",
col, col,
value_to_string_without_quotes(val, span)? value_to_string_without_quotes(val, span, depth + 1, indent)?
) )
} else { } else {
format!("{}: {}", col, value_to_string_without_quotes(val, span)?) format!(
"{}: {}",
col,
value_to_string_without_quotes(val, span, depth + 1, indent)?
)
}); });
} }
Ok(format!("{{{}}}", collection.join(", "))) Ok(format!("{{{}}}", collection.join(", ")))
} }
Value::LazyRecord { val, .. } => { Value::LazyRecord { val, .. } => {
let collected = val.collect()?; let collected = val.collect()?;
value_to_string(&collected, span) value_to_string(&collected, span, depth + 1, indent)
} }
// All strings outside data structures are quoted because they are in 'command position' // All strings outside data structures are quoted because they are in 'command position'
// (could be mistaken for commands by the Nu parser) // (could be mistaken for commands by the Nu parser)
@ -227,7 +258,12 @@ pub fn value_to_string(v: &Value, span: Span) -> Result<String, ShellError> {
} }
} }
fn value_to_string_without_quotes(v: &Value, span: Span) -> Result<String, ShellError> { fn value_to_string_without_quotes(
v: &Value,
span: Span,
depth: usize,
indent: &Option<String>,
) -> Result<String, ShellError> {
match v { match v {
Value::String { val, .. } => Ok({ Value::String { val, .. } => Ok({
if needs_quotes(val) { if needs_quotes(val) {
@ -236,16 +272,10 @@ fn value_to_string_without_quotes(v: &Value, span: Span) -> Result<String, Shell
val.clone() val.clone()
} }
}), }),
_ => value_to_string(v, span), _ => value_to_string(v, span, depth + 1, indent),
} }
} }
fn to_nuon(call: &Call, input: PipelineData) -> Result<String, ShellError> {
let v = input.into_value(call.head);
value_to_string(&v, call.head)
}
// This hits, in order: // This hits, in order:
// • Any character of []:`{}#'";()|$, // • Any character of []:`{}#'";()|$,
// • Any digit (\d) // • Any digit (\d)