rune/macros/
into_lit.rs

Help
1use crate::alloc::{self, String};
2use crate::ast;
3use crate::macros::MacroContext;
4
5/// Helper trait used for things that can be converted into tokens.
6pub trait IntoLit {
7    /// Convert the current thing into a token.
8    fn into_lit(self, cx: &mut MacroContext<'_, '_, '_>) -> alloc::Result<ast::Lit>;
9}
10
11impl<T> IntoLit for T
12where
13    ast::Number: From<T>,
14{
15    fn into_lit(self, cx: &mut MacroContext<'_, '_, '_>) -> alloc::Result<ast::Lit> {
16        let span = cx.macro_span();
17        let id = cx.idx.q.storage.insert_number(self)?;
18        let source = ast::NumberSource::Synthetic(id);
19        Ok(ast::Lit::Number(ast::LitNumber { span, source }))
20    }
21}
22
23impl IntoLit for char {
24    fn into_lit(self, cx: &mut MacroContext<'_, '_, '_>) -> alloc::Result<ast::Lit> {
25        let span = cx.macro_span();
26        let source = ast::CopySource::Inline(self);
27        Ok(ast::Lit::Char(ast::LitChar { span, source }))
28    }
29}
30
31impl IntoLit for &str {
32    fn into_lit(self, cx: &mut MacroContext<'_, '_, '_>) -> alloc::Result<ast::Lit> {
33        let span = cx.macro_span();
34        let id = cx.idx.q.storage.insert_str(self)?;
35        let source = ast::StrSource::Synthetic(id);
36        Ok(ast::Lit::Str(ast::LitStr { span, source }))
37    }
38}
39
40impl IntoLit for &String {
41    fn into_lit(self, cx: &mut MacroContext<'_, '_, '_>) -> alloc::Result<ast::Lit> {
42        <&str>::into_lit(self, cx)
43    }
44}
45
46impl IntoLit for String {
47    fn into_lit(self, cx: &mut MacroContext<'_, '_, '_>) -> alloc::Result<ast::Lit> {
48        let span = cx.macro_span();
49        let id = cx.idx.q.storage.insert_string(self)?;
50        let source = ast::StrSource::Synthetic(id);
51        Ok(ast::Lit::Str(ast::LitStr { span, source }))
52    }
53}
54
55impl IntoLit for &[u8] {
56    fn into_lit(self, cx: &mut MacroContext<'_, '_, '_>) -> alloc::Result<ast::Lit> {
57        let span = cx.macro_span();
58        let id = cx.idx.q.storage.insert_byte_string(self)?;
59        let source = ast::StrSource::Synthetic(id);
60        Ok(ast::Lit::ByteStr(ast::LitByteStr { span, source }))
61    }
62}
63
64impl<const N: usize> IntoLit for [u8; N] {
65    #[inline]
66    fn into_lit(self, cx: &mut MacroContext<'_, '_, '_>) -> alloc::Result<ast::Lit> {
67        <&[u8]>::into_lit(&self[..], cx)
68    }
69}
70
71impl<const N: usize> IntoLit for &[u8; N] {
72    #[inline]
73    fn into_lit(self, cx: &mut MacroContext<'_, '_, '_>) -> alloc::Result<ast::Lit> {
74        <&[u8]>::into_lit(self, cx)
75    }
76}
rune/macros/into_lit.rs

rune/macros/
into_lit.rs
