channels-macros 0.1.0

Procedural macros for channels.
Documentation 
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

use std::collections::HashMap;

use proc_macro::{Ident, TokenStream, TokenTree};

#[derive(Debug)]
pub struct Translate {
	table: HashMap<String, TokenStream>,
}

impl Translate {
	pub fn new(table: HashMap<String, TokenStream>) -> Self {
		Self { table }
	}

	pub fn translate(&self, src: Ident) -> TokenStream {
		match self.table.get(&src.to_string()) {
			Some(x) => x
				.clone()
				.into_iter()
				.map(|mut x| {
					x.set_span(src.span());
					x
				})
				.collect(),
			None => TokenStream::from(TokenTree::Ident(src)),
		}
	}
}