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
use rune::ast;
use rune::macros;
use rune::{quote, Parser, TokenStream};
pub fn module(_stdio: bool) -> Result<runestick::Module, runestick::ContextError> {
let mut module = runestick::Module::new(&["std", "core"]);
module.macro_(&["stringify"], stringify_macro)?;
module.macro_(&["panic"], panic_macro)?;
Ok(module)
}
pub(crate) fn stringify_macro(
stream: &TokenStream,
) -> runestick::Result<TokenStream> {
let lit = macros::stringify(stream);
let lit = ast::Lit::new(lit);
Ok(quote!(#lit).into_token_stream())
}
pub(crate) fn panic_macro(
stream: &TokenStream,
) -> runestick::Result<TokenStream> {
let mut p = Parser::from_token_stream(stream);
let args = p.parse_all::<macros::FormatArgs>()?;
let expanded = args.expand()?;
Ok(quote!(std::core::panic(#expanded)).into_token_stream())
}