rune/compile/
prelude.rs

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
use core::ops::Deref;

use crate::alloc::{self, Box, HashMap};
use crate::item::{IntoComponent, Item, ItemBuf};

/// The contents of a prelude.
#[derive(Default)]
pub struct Prelude {
    /// Prelude imports.
    prelude: HashMap<Box<str>, ItemBuf>,
}

impl Prelude {
    /// Construct a new unit with the default prelude.
    pub(crate) fn with_default_prelude() -> alloc::Result<Self> {
        let mut this = Self::default();

        this.add_prelude("Type", ["any", "Type"])?;
        this.add_prelude("assert_eq", ["test", "assert_eq"])?;
        this.add_prelude("assert_ne", ["test", "assert_ne"])?;
        this.add_prelude("assert", ["test", "assert"])?;
        this.add_prelude("bool", ["bool"])?;
        this.add_prelude("u8", ["u8"])?;
        this.add_prelude("f64", ["f64"])?;
        this.add_prelude("i64", ["i64"])?;
        this.add_prelude("u64", ["u64"])?;
        this.add_prelude("char", ["char"])?;
        this.add_prelude("dbg", ["io", "dbg"])?;
        this.add_prelude("drop", ["mem", "drop"])?;
        this.add_prelude("clone", ["clone", "clone"])?;
        this.add_prelude("Err", ["result", "Result", "Err"])?;
        this.add_prelude("file", ["macros", "builtin", "file"])?;
        this.add_prelude("format", ["fmt", "format"])?;
        this.add_prelude("is_readable", ["is_readable"])?;
        this.add_prelude("is_writable", ["is_writable"])?;
        this.add_prelude("line", ["macros", "builtin", "line"])?;
        this.add_prelude("None", ["option", "Option", "None"])?;
        this.add_prelude("Tuple", ["tuple", "Tuple"])?;
        this.add_prelude("Object", ["object", "Object"])?;
        this.add_prelude("Ok", ["result", "Result", "Ok"])?;
        this.add_prelude("Option", ["option", "Option"])?;
        this.add_prelude("panic", ["panic"])?;
        this.add_prelude("print", ["io", "print"])?;
        this.add_prelude("println", ["io", "println"])?;
        this.add_prelude("Result", ["result", "Result"])?;
        this.add_prelude("Some", ["option", "Option", "Some"])?;
        this.add_prelude("String", ["string", "String"])?;
        this.add_prelude("stringify", ["stringify"])?;
        this.add_prelude("Vec", ["vec", "Vec"])?;
        this.add_prelude("Bytes", ["bytes", "Bytes"])?;

        Ok(this)
    }

    /// Access a value from the prelude.
    pub(crate) fn get<'a>(&'a self, name: &str) -> Option<&'a Item> {
        Some(self.prelude.get(name)?)
    }

    /// Return the local name of an item
    #[allow(dead_code)]
    pub(crate) fn get_local<'a>(&'a self, item: &ItemBuf) -> Option<&'a str> {
        self.prelude
            .iter()
            .find(|(_, i)| i == &item)
            .map(|(s, _)| s.deref())
    }

    /// Define a prelude item.
    fn add_prelude<I>(&mut self, local: &str, path: I) -> alloc::Result<()>
    where
        I: IntoIterator,
        I::Item: IntoComponent,
    {
        self.prelude
            .try_insert(local.try_into()?, ItemBuf::with_crate_item("std", path)?)?;
        Ok(())
    }
}