enso/lib/rust/parser/tests/parse.rs

771 lines
25 KiB
Rust
Raw Normal View History

//! Parse expressions and compare their results to expected values.
// === Non-Standard Linter Configuration ===
#![allow(clippy::option_map_unit_fn)]
#![allow(clippy::precedence)]
#![allow(dead_code)]
#![deny(non_ascii_idents)]
#![deny(unconditional_recursion)]
#![warn(unsafe_code)]
#![warn(missing_copy_implementations)]
#![warn(missing_debug_implementations)]
#![warn(missing_docs)]
#![warn(trivial_casts)]
#![warn(trivial_numeric_casts)]
#![warn(unused_import_braces)]
#![warn(unused_qualifications)]
mod metadata;
use lexpr::sexp;
2022-07-20 17:53:20 +03:00
use lexpr::Value;
// ===========================
// === Test support macros ===
// ===========================
2022-07-20 17:53:20 +03:00
/// Parses input as a sequence of S-expressions, and wraps it in a `BodyBlock`.
macro_rules! block {
2022-07-20 17:53:20 +03:00
( $($statements:tt)* ) => {
sexp![(BodyBlock #( $( $statements )* ) )]
}
}
// =============
// === Tests ===
// =============
2022-07-20 17:53:20 +03:00
#[test]
fn nothing() {
test("", block![()]);
}
#[test]
fn application() {
test("a b c", block![(App (App (Ident a) (Ident b)) (Ident c))]);
}
#[test]
2022-07-20 17:53:20 +03:00
fn parentheses_simple() {
let expected = block![(Group "(" (App (Ident a) (Ident b)) ")")];
2022-07-20 17:53:20 +03:00
test("(a b)", expected);
}
#[test]
fn section_simple() {
let expected_lhs = block![(OprSectionBoundary (OprApp () (Ok "+") (Ident a)))];
test("+ a", expected_lhs);
let expected_rhs = block![(OprSectionBoundary (OprApp (Ident a) (Ok "+") ()))];
test("a +", expected_rhs);
}
#[test]
fn parentheses_nested() {
#[rustfmt::skip]
let expected = block![
(Group
"("
(App (Group "(" (App (Ident a) (Ident b)) ")")
(Ident c))
")")];
2022-07-20 17:53:20 +03:00
test("((a b) c)", expected);
}
#[test]
fn comments() {
test("# a b c", block![() ()]);
}
// === Type Definitions ===
#[test]
fn type_definition_no_body() {
test("type Bool", block![(TypeDef (Ident type) (Ident Bool) #() #() #())]);
test("type Option a", block![(TypeDef (Ident type) (Ident Option) #((Ident a)) #() #())]);
}
#[test]
fn type_constructors() {
let code = [
"type Geo",
" Circle",
" radius",
" 4",
" Rectangle width height",
" Point",
];
#[rustfmt::skip]
let expected = block![
(TypeDef (Ident type) (Ident Geo) #()
#(((Circle #() #((Ident radius) (Number 4))))
((Rectangle #((Ident width) (Ident height)) #()))
((Point #() #())))
#())
];
test(&code.join("\n"), expected);
}
#[test]
fn type_methods() {
let code = ["type Geo", " number =", " 23", " area self = 1 + 1"];
#[rustfmt::skip]
let expected = block![
(TypeDef (Ident type) (Ident Geo) #() #()
#((Function number #() "=" (BodyBlock #((Number 23))))
(Function area #((Ident self)) "=" (OprApp (Number 1) (Ok "+") (Number 1)))))
];
test(&code.join("\n"), expected);
}
#[test]
fn type_def_full() {
let code = [
"type Geo",
" Circle",
" radius : float",
" 4",
" Rectangle width height",
" Point",
"",
" number =",
" 23",
" area self = 1 + 1",
];
#[rustfmt::skip]
let expected = block![
(TypeDef (Ident type) (Ident Geo) #()
#(((Circle #() #((TypeAnnotated (Ident radius) ":" (Ident float)) (Number 4))))
((Rectangle #((Ident width) (Ident height)) #()))
((Point #() #()))
(()))
#((Function number #() "=" (BodyBlock #((Number 23))))
(Function area #((Ident self)) "=" (OprApp (Number 1) (Ok "+") (Number 1)))))
];
test(&code.join("\n"), expected);
}
#[test]
fn type_def_nested() {
#[rustfmt::skip]
let code = [
"type Foo",
" type Bar",
" type Baz",
];
#[rustfmt::skip]
let expected = block![
(TypeDef (Ident type) (Ident Foo) #() #()
#((TypeDef (Ident type) (Ident Bar) #() #() #())
(TypeDef (Ident type) (Ident Baz) #() #() #())))
];
test(&code.join("\n"), expected);
}
// === Variable Assignment ===
#[test]
fn assignment_simple() {
test("foo = 23", block![(Assignment (Ident foo) "=" (Number 23))]);
}
// === Functions ===
#[test]
fn function_inline_simple_args() {
test("foo a = 23", block![(Function foo #((Ident a)) "=" (Number 23))]);
test("foo a b = 23", block![(Function foo #((Ident a) (Ident b)) "=" (Number 23))]);
test("foo a b c = 23", block![(Function foo #((Ident a) (Ident b) (Ident c)) "=" (Number 23))]);
}
#[test]
fn function_block_noargs() {
test("foo =", block![(Function foo #() "=" ())]);
}
#[test]
fn function_block_simple_args() {
test("foo a =", block![(Function foo #((Ident a)) "=" ())]);
test("foo a b =", block![(Function foo #((Ident a) (Ident b)) "=" ())]);
test("foo a b c =", block![(Function foo #((Ident a) (Ident b) (Ident c)) "=" ())]);
}
// === Code Blocks ===
2022-07-20 17:53:20 +03:00
#[test]
fn code_block_body() {
let code = ["main =", " 4"];
test(&code.join("\n"), block![(Function main #() "=" (BodyBlock #((Number 4))))]);
let code = ["main =", " ", " 4"];
test(&code.join("\n"), block![(Function main #() "=" (BodyBlock #(() (Number 4))))]);
let code = ["main =", " ", " 4"];
test(&code.join("\n"), block![(Function main #() "=" (BodyBlock #(() (Number 4))))]);
let code = ["main =", " ", " 4"];
test(&code.join("\n"), block![(Function main #() "=" (BodyBlock #(() (Number 4))))]);
let code = ["main =", "", " 4"];
test(&code.join("\n"), block![(Function main #() "=" (BodyBlock #(() (Number 4))))]);
#[rustfmt::skip]
let code = [
"main =",
" +4",
" print 23",
];
#[rustfmt::skip]
let expect = block![
(Function main #() "=" (BodyBlock #(
(OprSectionBoundary (OprApp () (Ok "+") (Number 4)))
(App (Ident print) (Number 23)))))
];
test(&code.join("\n"), expect);
}
#[test]
fn code_block_operator() {
let code = ["value = nums", " * each random", " + constant"];
let expect = block![
(Assignment (Ident value) "="
(OperatorBlockApplication (Ident nums)
#(((Ok "*") (App (Ident each) (Ident random)))
((Ok "+") (Ident constant)))
#()))
];
test(&code.join("\n"), expect);
}
#[test]
fn code_block_argument_list() {
#[rustfmt::skip]
let code = [
"value = foo",
" bar",
];
let expect = block![
(Assignment (Ident value) "=" (ArgumentBlockApplication (Ident foo) #((Ident bar))))
];
test(&code.join("\n"), expect);
#[rustfmt::skip]
let code = [
"value = foo",
" +1",
" bar",
];
#[rustfmt::skip]
let expect = block![
(Assignment (Ident value) "="
(ArgumentBlockApplication (Ident foo) #(
(OprSectionBoundary (OprApp () (Ok "+") (Number 1)))
(Ident bar))))
];
test(&code.join("\n"), expect);
}
#[test]
fn code_block_empty() {
// The first line here should parse as a function with no body expression (which is an error).
// No input would parse as an empty `ArgumentBlock` or `OperatorBlock`, because those types are
// distinguished from a body continuation by the presence of non-empty indented lines.
let code = ["foo =", "bar"];
test(&code.join("\n"), block![(Function foo #() "=" ()) (Ident bar)]);
// This parses similarly to above; a line with no non-whitespace content does not create a code
// block.
let code = ["foo =", " ", "bar"];
test(&code.join("\n"), block![(Function foo #() "=" ()) () (Ident bar)]);
}
#[test]
fn code_block_bad_indents1() {
let code = ["main =", " foo", " bar", " baz"];
let expected = block![
(Function main #() "=" (BodyBlock #((Ident foo) (Ident bar) (Ident baz))))
];
test(&code.join("\n"), expected);
}
#[test]
fn code_block_bad_indents2() {
let code = ["main =", " foo", " bar", "baz"];
let expected = block![
(Function main #() "=" (BodyBlock #((Ident foo) (Ident bar))))
(Ident baz)
];
test(&code.join("\n"), expected);
}
#[test]
fn code_block_with_following_statement() {
let code = ["main =", " foo", "bar"];
let expected = block![
(Function main #() "=" (BodyBlock #((Ident foo))))
(Ident bar)
];
test(&code.join("\n"), expected);
}
// === Binary Operators ===
#[test]
fn multiple_operator_error() {
let code = ["4 + + 1"];
let expected = block![
(OprApp (Number 4) (Err (#("+" "+"))) (Number 1))
];
test(&code.join("\n"), expected);
let code = ["4 + + + 1"];
let expected = block![
(OprApp (Number 4) (Err (#("+" "+" "+"))) (Number 1))
];
test(&code.join("\n"), expected);
}
#[test]
fn precedence() {
let code = ["1 * 2 + 3"];
let expected = block![
(OprApp (OprApp (Number 1) (Ok "*") (Number 2)) (Ok "+") (Number 3))
];
test(&code.join("\n"), expected);
}
// === Unary Operators ===
#[test]
fn unevaluated_argument() {
let code = ["main ~foo = 4"];
let expected = block![
(Function main #((UnaryOprApp "~" (Ident foo))) "=" (Number 4))
];
test(&code.join("\n"), expected);
}
#[test]
fn unary_operator_missing_operand() {
let code = ["main ~ = 4"];
let expected = block![
(Function main #((UnaryOprApp "~" ())) "=" (Number 4))
];
test(&code.join("\n"), expected);
}
#[test]
fn unary_operator_at_end_of_expression() {
let code = ["foo ~"];
let expected = block![
(App (Ident foo) (UnaryOprApp "~" ()))
];
test(&code.join("\n"), expected);
}
#[test]
fn plus_negative() {
let code = ["x = 4+-1"];
let expected = block![
(Assignment (Ident x) "=" (OprApp (Number 4) (Ok "+") (UnaryOprApp "-" (Number 1))))
];
test(&code.join("\n"), expected);
}
#[test]
fn minus_binary() {
let cases = [
("x - 5", block![(OprApp (Ident x) (Ok "-") (Number 5))]),
("x-5", block![(OprApp (Ident x) (Ok "-") (Number 5))]),
("x.-y", block![(OprApp (Ident x) (Ok ".") (UnaryOprApp "-" (Ident y)))]),
("x.~y", block![(OprApp (Ident x) (Ok ".") (UnaryOprApp "~" (Ident y)))]),
];
cases.into_iter().for_each(|(code, expected)| test(code, expected));
}
#[test]
fn minus_section() {
#[rustfmt::skip]
let cases = [
("- x", block![(OprSectionBoundary (OprApp () (Ok "-") (Ident x)))]),
("(- x)", block![(Group "(" (OprSectionBoundary (OprApp () (Ok "-") (Ident x))) ")")]),
("- (x * 2)", block![
(OprSectionBoundary (OprApp () (Ok "-")
(Group "(" (OprApp (Ident x) (Ok "*") (Number 2)) ")")))]),
];
cases.into_iter().for_each(|(code, expected)| test(code, expected));
}
#[test]
fn minus_unary() {
#[rustfmt::skip]
let cases = [
("f -5", block![(App (Ident f) (UnaryOprApp "-" (Number 5)))]),
("-5", block![(UnaryOprApp "-" (Number 5))]),
("(-5)", block![(Group "(" (UnaryOprApp "-" (Number 5)) ")")]),
("-(x * 2)", block![
(UnaryOprApp "-" (Group "(" (OprApp (Ident x) (Ok "*") (Number 2)) ")"))]),
("x=-1", block![(Assignment (Ident x) "=" (UnaryOprApp "-" (Number 1)))]),
("-1+2", block![(OprApp (UnaryOprApp "-" (Number 1)) (Ok "+") (Number 2))]),
("-1*2", block![(OprApp (UnaryOprApp "-" (Number 1)) (Ok "*") (Number 2))]),
("-1.x", block![(OprApp (UnaryOprApp "-" (Number 1)) (Ok ".") (Ident x))]),
];
cases.into_iter().for_each(|(code, expected)| test(code, expected));
}
// === Import/Export ===
#[test]
fn import() {
#[rustfmt::skip]
let cases = [
("import project.IO", block![
(Import () () () ((Ident import) (OprApp (Ident project) (Ok ".") (Ident IO))) () ())]),
("import Standard.Base as Enso_List", block![
(Import () () ()
((Ident import) (OprApp (Ident Standard) (Ok ".") (Ident Base)))
((Ident as) (Ident Enso_List))
())]),
("from Standard.Base import all", block![
(Import ()
((Ident from) (OprApp (Ident Standard) (Ok ".") (Ident Base)))
()
((Ident import) (Ident all))
() ())]),
("from Standard.Base import all hiding Number, Boolean", block![
(Import ()
((Ident from) (OprApp (Ident Standard) (Ok ".") (Ident Base)))
()
((Ident import) (Ident all))
()
((Ident hiding) (OprApp (Ident Number) (Ok ",") (Ident Boolean))))]),
("from Standard.Table as Column_Module import Column", block![
(Import ()
((Ident from) (OprApp (Ident Standard) (Ok ".") (Ident Table)))
((Ident as) (Ident Column_Module))
((Ident import) (Ident Column))
() ())]),
("polyglot java import java.lang.Float", block![
(Import
((Ident polyglot) (Ident java))
()
()
((Ident import)
(OprApp (OprApp (Ident java) (Ok ".") (Ident lang)) (Ok ".") (Ident Float)))
() ())]),
("polyglot java import java.net.URI as Java_URI", block![
(Import
((Ident polyglot) (Ident java))
()
()
((Ident import)
(OprApp (OprApp (Ident java) (Ok ".") (Ident net)) (Ok ".") (Ident URI)))
((Ident as) (Ident Java_URI))
())]),
];
cases.into_iter().for_each(|(code, expected)| test(code, expected));
}
#[test]
fn export() {
#[rustfmt::skip]
let cases = [
("export Foo", block![(Export () () ((Ident export) (Ident Foo)) () ())]),
("export Foo as Bar", block![
(Export () () ((Ident export) (Ident Foo)) ((Ident as) (Ident Bar)) ())]),
("from Foo export Bar, Baz", block![
(Export
((Ident from) (Ident Foo))
()
((Ident export) (OprApp (Ident Bar) (Ok ",") (Ident Baz)))
() ())]),
("from Foo export all hiding Bar, Baz", block![
(Export
((Ident from) (Ident Foo))
()
((Ident export) (Ident all))
()
((Ident hiding) (OprApp (Ident Bar) (Ok ",") (Ident Baz))))]),
("from Foo as Bar export Baz, Quux", block![
(Export
((Ident from) (Ident Foo))
((Ident as) (Ident Bar))
((Ident export) (OprApp (Ident Baz) (Ok ",") (Ident Quux)))
() ())
]),
];
cases.into_iter().for_each(|(code, expected)| test(code, expected));
}
// === Metadata ===
#[test]
fn metadata_raw() {
let code = [
"4",
"#### METADATA ####",
r#"[[{"index":{"value":7},"size":{"value":8}},"5bad897e-099b-4b00-9348-64092636746d"]]"#,
];
let code = code.join("\n");
let (_meta, code) = enso_parser::metadata::parse(&code).unwrap();
let expected = block![
(Number 4)
()
];
test(code, expected);
}
#[test]
fn metadata_parsing() {
let code = metadata::ORDERS_WITH_METADATA;
let (meta, code) = enso_parser::metadata::parse(code).unwrap();
let _ast = enso_parser::Parser::new().run(code);
let _meta: enso_parser::metadata::Metadata = meta.unwrap();
}
// === Type annotations and signatures ===
#[test]
fn type_signatures() {
let cases = [
("val : Bool", block![(TypeSignature val ":" (Ident Bool))]),
("val : List Int", block![(TypeSignature val ":" (App (Ident List) (Ident Int)))]),
];
cases.into_iter().for_each(|(code, expected)| test(code, expected));
}
#[test]
fn type_annotations() {
#[rustfmt::skip]
let cases = [
("val = 123 : Int", block![
(Assignment (Ident val) "=" (TypeAnnotated (Number 123) ":" (Ident Int)))]),
("val = foo (123 : Int)", block![
(Assignment (Ident val) "="
(App (Ident foo)
(Group "(" (TypeAnnotated (Number 123) ":" (Ident Int)) ")")))]),
];
cases.into_iter().for_each(|(code, expected)| test(code, expected));
}
// === Text Literals ===
#[test]
fn inline_text_literals() {
#[rustfmt::skip]
let cases = [
(r#""I'm an inline raw text!""#, block![
(TextLiteral "\"" #((Section "I'm an inline raw text!")) "\"" 0)]),
(r#"zero_length = """#, block![
(Assignment (Ident zero_length) "=" (TextLiteral "\"" #() "\"" 0))]),
(r#"unclosed = ""#, block![(Assignment (Ident unclosed) "=" (TextLiteral "\"" #() () 0))]),
(r#"unclosed = "a"#, block![
(Assignment (Ident unclosed) "=" (TextLiteral "\"" #((Section "a")) () 0))]),
(r#"'Other quote type'"#, block![(TextLiteral "'" #((Section "Other quote type")) "'" 0)]),
(r#""Non-escape: \n""#, block![(TextLiteral "\"" #((Section "Non-escape: \\n")) "\"" 0)]),
(r#""String with \" escape""#, block![
(TextLiteral
"\""
#((Section "String with ") (Escape "\\") (Section "\" escape"))
"\"" 0)]),
];
cases.into_iter().for_each(|(code, expected)| test(code, expected));
}
#[test]
fn multiline_text_literals() {
test("'''", block![(TextLiteral "'''" #() () 0)]);
const CODE: &str = r#"'''
part of the string
3-spaces indented line, part of the Text Block
this does not end the string -> '''
also part of the string
3"#;
#[rustfmt::skip]
let expected = block![
(TextLiteral
"'''"
#((Section "\n") (Section "part of the string")
(Section "\n") (Section "3-spaces indented line, part of the Text Block")
(Section "\n") (Section "this does not end the string -> '''")
(Section "\n") (Section "")
(Section "\n") (Section "also part of the string")
(Section "\n") (Section ""))
() 4)
(Number 3)
];
test(CODE, expected);
}
// ====================
// === Test Support ===
// ====================
use enso_metamodel_lexpr::ToSExpr;
use enso_reflect::Reflect;
use std::collections::HashSet;
/// Given a block of input Enso code, test that:
/// - The given code parses to the AST represented by the given S-expression.
/// - The AST pretty-prints back to the original code.
/// - Rust's deserialization is compatible with Rust's serialization for the type. (The Java format
/// tests check Java's deserialization against Rust's deserialization).
///
/// The S-expression format is as documented for [`enso_metamodel_lexpr`], with some
/// postprocessing:
/// - For concision, field names are stripped (as if all structs were tuple structs).
/// - Most token types are represented as their contents, rather than as a token struct. For
/// example, a `token::Number` may be represented like: `sexp![10]`, and a `token::Ident` may look
/// like `sexp![foo]`.
2022-07-20 17:53:20 +03:00
fn test(code: &str, expect: Value) {
let ast = enso_parser::Parser::new().run(code);
let ast_s_expr = to_s_expr(&ast, code);
2022-07-20 17:53:20 +03:00
assert_eq!(ast_s_expr.to_string(), expect.to_string(), "{:?}", &ast);
assert_eq!(ast.code(), code, "{:?}", &ast);
let serialized = enso_parser::serialization::serialize_tree(&ast).unwrap();
let deserialized = enso_parser::serialization::deserialize_tree(&serialized);
deserialized.unwrap();
}
// =====================
// === S-expressions ===
// =====================
/// Produce an S-expression representation of the input AST type.
2022-07-20 17:53:20 +03:00
pub fn to_s_expr<T>(value: &T, code: &str) -> Value
where T: serde::Serialize + Reflect {
2022-07-20 17:53:20 +03:00
use enso_parser::syntax::token;
use enso_parser::syntax::tree;
let (graph, rust_to_meta) = enso_metamodel::rust::to_meta(value.reflect_type());
let ast_ty = rust_to_meta[&value.reflect_type().id];
let base = code.as_bytes().as_ptr() as usize;
let code: Box<str> = Box::from(code);
let mut to_s_expr = ToSExpr::new(&graph);
to_s_expr.mapper(ast_ty, strip_hidden_fields);
2022-07-20 17:53:20 +03:00
let ident_token = rust_to_meta[&token::variant::Ident::reflect().id];
let operator_token = rust_to_meta[&token::variant::Operator::reflect().id];
let symbol_token = rust_to_meta[&token::variant::Symbol::reflect().id];
let number_token = rust_to_meta[&token::variant::Number::reflect().id];
let newline_token = rust_to_meta[&token::variant::Newline::reflect().id];
let text_start_token = rust_to_meta[&token::variant::TextStart::reflect().id];
let text_end_token = rust_to_meta[&token::variant::TextEnd::reflect().id];
let text_escape_token = rust_to_meta[&token::variant::TextEscape::reflect().id];
let text_section_token = rust_to_meta[&token::variant::TextSection::reflect().id];
2022-07-20 17:53:20 +03:00
// TODO: Implement `#[reflect(flag = "enso::concrete")]`, which just attaches user data to the
// type info; then filter by flag here instead of hard-coding these simplifications.
let line = rust_to_meta[&tree::block::Line::reflect().id];
let operator_line = rust_to_meta[&tree::block::OperatorLine::reflect().id];
let token_to_str = move |token: Value| {
let range = token_code_range(&token, base);
code[range].to_owned().into_boxed_str()
};
let token_to_str_ = token_to_str.clone();
2022-07-20 17:53:20 +03:00
to_s_expr.mapper(ident_token, move |token| Value::symbol(token_to_str_(token)));
let token_to_str_ = token_to_str.clone();
2022-07-20 17:53:20 +03:00
to_s_expr.mapper(operator_token, move |token| Value::string(token_to_str_(token)));
let token_to_str_ = token_to_str.clone();
to_s_expr.mapper(symbol_token, move |token| Value::string(token_to_str_(token)));
let token_to_str_ = token_to_str.clone();
to_s_expr.mapper(text_start_token, move |token| Value::string(token_to_str_(token)));
let token_to_str_ = token_to_str.clone();
to_s_expr.mapper(text_end_token, move |token| Value::string(token_to_str_(token)));
let token_to_str_ = token_to_str.clone();
to_s_expr.mapper(text_escape_token, move |token| Value::string(token_to_str_(token)));
let token_to_str_ = token_to_str.clone();
to_s_expr.mapper(text_section_token, move |token| Value::string(token_to_str_(token)));
let token_to_str_ = token_to_str;
to_s_expr.mapper(number_token, move |token| {
2022-07-20 17:53:20 +03:00
Value::Number(token_to_str_(token).parse::<u64>().unwrap().into())
});
2022-07-20 17:53:20 +03:00
let into_car = |cons| match cons {
Value::Cons(cons) => cons.into_pair().0,
_ => panic!(),
};
to_s_expr.mapper(line, into_car);
to_s_expr.mapper(operator_line, into_car);
to_s_expr.skip(newline_token);
tuplify(to_s_expr.value(ast_ty, &value))
}
/// Strip certain fields that should be excluded from output.
2022-07-20 17:53:20 +03:00
fn strip_hidden_fields(tree: Value) -> Value {
let hidden_tree_fields = [
":spanLeftOffsetVisible",
":spanLeftOffsetCodeReprBegin",
":spanLeftOffsetCodeReprLen",
":spanCodeLength",
];
let hidden_tree_fields: HashSet<_> = hidden_tree_fields.into_iter().collect();
2022-07-20 17:53:20 +03:00
Value::list(tree.to_vec().unwrap().into_iter().filter(|val| match val {
Value::Cons(cons) => match cons.car() {
Value::Symbol(symbol) => !hidden_tree_fields.contains(symbol.as_ref()),
_ => panic!(),
},
_ => true,
}))
}
/// Given an S-expression representation of a [`Token`] and the base address for `Code` `Cow`s,
/// return the range of the input code the token references.
2022-07-20 17:53:20 +03:00
fn token_code_range(token: &Value, base: usize) -> std::ops::Range<usize> {
let get_u32 =
|field| fields(token).find(|(name, _)| *name == field).unwrap().1.as_u64().unwrap() as u32;
let begin = get_u32(":codeReprBegin");
let len = get_u32(":codeReprLen");
let begin = (begin as u64) | (base as u64 & !0xFFFF_FFFF);
let begin = if begin < (base as u64) { begin + 0x1_0000_0000 } else { begin };
let begin = begin as usize - base;
2022-07-20 17:53:20 +03:00
let len = len as usize;
begin..(begin + len)
}
/// Iterate the field `(name, value)` pairs of the S-expression of a struct with named fields.
2022-07-20 17:53:20 +03:00
fn fields(value: &'_ Value) -> impl Iterator<Item = (&'_ str, &'_ Value)> {
value.list_iter().unwrap().filter_map(|value| match value {
2022-07-20 17:53:20 +03:00
Value::Cons(cons) => match cons.car() {
Value::Symbol(symbol) => Some((&symbol[..], cons.cdr())),
_ => None,
},
_ => None,
})
}
/// Strip field names from struct representations, so that they are printed more concisely, as if
/// they were tuple-structs.
2022-07-20 17:53:20 +03:00
fn tuplify(value: Value) -> Value {
let (car, cdr) = match value {
2022-07-20 17:53:20 +03:00
Value::Cons(cons) => cons.into_pair(),
Value::Vector(mut vector) => {
for value in vector.iter_mut() {
2022-07-20 17:53:20 +03:00
let original = std::mem::replace(value, Value::Nil);
*value = tuplify(original);
}
2022-07-20 17:53:20 +03:00
return Value::Vector(vector);
}
value => return value,
};
2022-07-20 17:53:20 +03:00
if let Value::Symbol(symbol) = &car {
if let Some(':') = symbol.chars().next() {
return tuplify(cdr);
}
}
let car = tuplify(car);
let cdr = tuplify(cdr);
2022-07-20 17:53:20 +03:00
Value::Cons(lexpr::Cons::new(car, cdr))
}