geam 0.1.1

Experimental Rust-embedded execution runtime for typed Gleam programs
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167

mod function;
mod schema;
mod storage;

pub(crate) use schema::{StringTree, StringTreeSchema};
pub(super) use storage::Stores;
pub(crate) use storage::{
    StringTree as StoredStringTree, StringTreeExternalStorage, StringTreePayload,
};

use self::function::{
    StringTreeProvider, append_tree, byte_size, concat, do_to_graphemes, erl_split, from_string,
    from_strings, is_empty, is_equal, lowercase, replace, to_string, uppercase,
};
use self::schema::{Direction, SplitConstructions, StringList, StringTreeList};
use super::GleamStdlibHostProfile;
use crate::{HostProviderModule, HostRegistrationError};
use ecow::EcoString;
use num_bigint::BigInt;

pub(super) fn host_provider<Profile>() -> Result<HostProviderModule<Profile>, HostRegistrationError>
where
    Profile: GleamStdlibHostProfile,
{
    HostProviderModule::new("gleam_stdlib", "gleam/string_tree")
        .and_then(
            HostProviderModule::with_external_type::<StringTreeProvider<Profile>, StringTreeSchema>,
        )
        .and_then(|provider| {
            provider.with_scoped_function::<
                StringTreeProvider<Profile>,
                (StringTree, StringTree),
                StringTree,
                _,
            >("append_tree", append_tree::<Profile>)
        })
        .and_then(|provider| {
            provider
                .with_scoped_function::<StringTreeProvider<Profile>, (StringList,), StringTree, _>(
                    "from_strings",
                    from_strings::<Profile>,
                )
        })
        .and_then(|provider| {
            provider.with_scoped_function::<
                StringTreeProvider<Profile>,
                (StringTreeList,),
                StringTree,
                _,
            >("concat", concat::<Profile>)
        })
        .and_then(|provider| {
            provider
                .with_scoped_function::<StringTreeProvider<Profile>, (EcoString,), StringTree, _>(
                    "from_string",
                    from_string::<Profile>,
                )
        })
        .and_then(|provider| {
            provider
                .with_scoped_function::<StringTreeProvider<Profile>, (StringTree,), EcoString, _>(
                    "to_string",
                    to_string::<Profile>,
                )
        })
        .and_then(|provider| {
            provider.with_scoped_function::<StringTreeProvider<Profile>, (StringTree,), BigInt, _>(
                "byte_size",
                byte_size::<Profile>,
            )
        })
        .and_then(|provider| {
            provider
                .with_scoped_function::<StringTreeProvider<Profile>, (StringTree,), StringTree, _>(
                    "lowercase",
                    lowercase::<Profile>,
                )
        })
        .and_then(|provider| {
            provider
                .with_scoped_function::<StringTreeProvider<Profile>, (StringTree,), StringTree, _>(
                    "uppercase",
                    uppercase::<Profile>,
                )
        })
        .and_then(|provider| {
            provider
                .with_scoped_function::<StringTreeProvider<Profile>, (EcoString,), StringList, _>(
                    "do_to_graphemes",
                    do_to_graphemes::<Profile>,
                )
        })
        .and_then(|provider| {
            provider.with_scoped_function_and_constructions::<
                StringTreeProvider<Profile>,
                (StringTree, EcoString, Direction),
                StringTreeList,
                SplitConstructions,
                _,
            >("erl_split", erl_split::<Profile>)
        })
        .and_then(|provider| {
            provider.with_scoped_function::<
                StringTreeProvider<Profile>,
                (StringTree, EcoString, EcoString),
                StringTree,
                _,
            >("replace", replace::<Profile>)
        })
        .and_then(|provider| {
            provider.with_scoped_function::<
                StringTreeProvider<Profile>,
                (StringTree, StringTree),
                bool,
                _,
            >("is_equal", is_equal::<Profile>)
        })
        .and_then(|provider| {
            provider.with_scoped_function::<StringTreeProvider<Profile>, (StringTree,), bool, _>(
                "is_empty",
                is_empty::<Profile>,
            )
        })
}

#[cfg(test)]
mod tests {
    use super::host_provider;
    use crate::gleam_stdlib::GleamStdlibProfile;

    #[test]
    fn registers_the_exact_official_string_tree_provider_inventory() {
        let provider = host_provider::<GleamStdlibProfile>()
            .expect("official string tree provider should register");

        assert_eq!(provider.package(), "gleam_stdlib");
        assert_eq!(provider.module(), "gleam/string_tree");
        assert_eq!(
            provider
                .external_types()
                .map(|schema| (schema.name().as_str(), schema.parameter_count()))
                .collect::<Vec<_>>(),
            [("StringTree", 0)],
        );
        assert_eq!(
            provider
                .functions()
                .map(|function| function.name().as_str())
                .collect::<Vec<_>>(),
            [
                "append_tree",
                "from_strings",
                "concat",
                "from_string",
                "to_string",
                "byte_size",
                "lowercase",
                "uppercase",
                "do_to_graphemes",
                "erl_split",
                "replace",
                "is_equal",
                "is_empty",
            ],
        );
    }
}