1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
//! The `\mathchardef` primitive.
use texlang::prelude as txl;
use texlang::traits::*;
use texlang::*;

/// Get the `\mathchardef` command.
pub fn get_mathchardef<S: TexlangState>() -> command::BuiltIn<S> {
    command::BuiltIn::new_execution(mathchardef_primitive_fn)
}

fn mathchardef_primitive_fn<S: TexlangState>(
    _: token::Token,
    input: &mut vm::ExecutionInput<S>,
) -> txl::Result<()> {
    let scope = TexlangState::variable_assignment_scope_hook(input.state_mut());
    let (cmd_ref_or, _, c) = <(
        Option<token::CommandRef>,
        parse::OptionalEquals,
        types::MathCode,
    )>::parse(input)?;
    if let Some(cmd_ref) = cmd_ref_or {
        input
            .commands_map_mut()
            .insert(cmd_ref, command::Command::MathCharacter(c), scope);
    }
    Ok(())
}

#[cfg(test)]
mod test {
    use super::*;
    use crate::the;
    use std::collections::HashMap;
    use texlang_testing::*;

    fn built_in_commands() -> HashMap<&'static str, command::BuiltIn<State>> {
        HashMap::from([
            ("mathchardef", get_mathchardef()),
            ("the", the::get_the()),
            ("i", TestingComponent::get_integer()),
        ])
    }

    test_suite![
        expansion_equality_tests(
            (
                basic_case,
                r"\mathchardef\Hello = `\+ \Hello",
                "MathCode(43)"
            ),
            (
                basic_case_with_the,
                r"\mathchardef\Hello = 123 \the\Hello",
                "123"
            ),
            (
                parsable_as_number,
                r"\mathchardef\Hello = 13 \i=\Hello x\the\i",
                "x13"
            ),
            (
                parsable_as_number_negative,
                r"\mathchardef\Hello = 13 \i=-\Hello x\the\i",
                "x-13"
            ),
        ),
        serde_tests((basic_case, r"\mathchardef\Hello = `\+ ", r"\Hello"),),
    ];
}