vrl 0.32.0

Vector Remap Language
Documentation
use super::{super::ObjectMap, Kind, Value};

impl Kind {
    /// Returns a tree representation of `Kind`, in a more human readable format.
    /// This is for debugging / development purposes only.
    #[must_use]
    pub fn debug_info(&self) -> ObjectMap {
        let mut output = ObjectMap::new();
        insert_kind(&mut output, self, true);
        output
    }
}

fn insert_kind(tree: &mut ObjectMap, kind: &Kind, show_unknown: bool) {
    if kind.is_never() {
        insert_if_true(tree, "never", true);
    } else if kind.is_any() {
        insert_if_true(tree, "any", true);
    } else if kind.is_json() {
        insert_if_true(tree, "json", true);
    } else {
        insert_if_true(tree, "bytes", kind.contains_bytes());
        insert_if_true(tree, "integer", kind.contains_integer());
        insert_if_true(tree, "float", kind.contains_float());
        insert_if_true(tree, "boolean", kind.contains_boolean());
        insert_if_true(tree, "timestamp", kind.contains_timestamp());
        insert_if_true(tree, "regex", kind.contains_regex());
        insert_if_true(tree, "null", kind.contains_null());
        insert_if_true(tree, "undefined", kind.contains_undefined());

        if let Some(fields) = &kind.object {
            let mut object_tree = ObjectMap::new();
            for (field, field_kind) in fields.known() {
                let mut field_tree = ObjectMap::new();
                insert_kind(&mut field_tree, field_kind, show_unknown);
                object_tree.insert(field.to_string().into(), Value::Object(field_tree));
            }
            tree.insert("object".into(), Value::Object(object_tree));
            if show_unknown {
                insert_unknown(
                    tree,
                    &fields.unknown_kind(),
                    fields.is_unknown_exact(),
                    "object",
                );
            }
        }

        if let Some(indices) = &kind.array {
            let mut array_tree = ObjectMap::new();
            for (index, index_kind) in indices.known() {
                let mut index_tree = ObjectMap::new();
                insert_kind(&mut index_tree, index_kind, show_unknown);
                array_tree.insert(index.to_string().into(), Value::Object(index_tree));
            }
            tree.insert("array".to_owned().into(), Value::Object(array_tree));
            if show_unknown {
                insert_unknown(
                    tree,
                    &indices.unknown_kind(),
                    indices.is_unknown_exact(),
                    "array",
                );
            }
        }
    }
}

fn insert_unknown(tree: &mut ObjectMap, unknown: &Kind, unknown_exact: bool, prefix: &str) {
    if unknown.is_undefined() {
        return;
    }
    let mut unknown_tree = ObjectMap::new();
    insert_kind(&mut unknown_tree, unknown, unknown_exact);
    if unknown.is_exact() {
        tree.insert(
            format!("{prefix}_unknown_exact").into(),
            Value::Object(unknown_tree),
        );
    } else {
        tree.insert(
            format!("{prefix}_unknown_infinite").into(),
            Value::Object(unknown_tree),
        );
    }
}

fn insert_if_true(tree: &mut ObjectMap, key: &str, value: bool) {
    if value {
        tree.insert(key.to_owned().into(), Value::Boolean(true));
    }
}