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

//! The core library for the Koto language

pub mod io;
pub mod iterator;
pub mod koto;
pub mod list;
pub mod map;
pub mod number;
pub mod os;
pub mod range;
pub mod string;
pub mod test;
pub mod tuple;
mod value_sort;

use crate::KMap;

#[derive(Clone)]
#[allow(missing_docs)]
pub struct CoreLib {
    pub io: KMap,
    pub iterator: KMap,
    pub koto: KMap,
    pub list: KMap,
    pub map: KMap,
    pub os: KMap,
    pub number: KMap,
    pub range: KMap,
    pub string: KMap,
    pub test: KMap,
    pub tuple: KMap,
}

impl CoreLib {
    /// The core lib items made available in each Koto script
    pub fn prelude(&self) -> KMap {
        let result = KMap::default();
        result.add_map("io", self.io.clone());
        result.add_map("iterator", self.iterator.clone());
        result.add_map("koto", self.koto.clone());
        result.add_map("list", self.list.clone());
        result.add_map("map", self.map.clone());
        result.add_map("os", self.os.clone());
        result.add_map("number", self.number.clone());
        result.add_map("range", self.range.clone());
        result.add_map("string", self.string.clone());
        result.add_map("test", self.test.clone());
        result.add_map("tuple", self.tuple.clone());

        macro_rules! default_import {
            ($name:expr, $module:ident) => {{
                result.add_value($name, self.$module.data().get($name).unwrap().clone());
            }};
        }

        default_import!("assert", test);
        default_import!("assert_eq", test);
        default_import!("assert_ne", test);
        default_import!("assert_near", test);
        default_import!("print", io);
        default_import!("copy", koto);
        default_import!("deep_copy", koto);
        default_import!("type", koto);

        result
    }
}

impl Default for CoreLib {
    fn default() -> Self {
        Self {
            io: io::make_module(),
            iterator: iterator::make_module(),
            koto: koto::make_module(),
            list: list::make_module(),
            map: map::make_module(),
            os: os::make_module(),
            number: number::make_module(),
            range: range::make_module(),
            string: string::make_module(),
            test: test::make_module(),
            tuple: tuple::make_module(),
        }
    }
}