68 lines
2.1 KiB
Rust
68 lines
2.1 KiB
Rust
use proc_macro2::{Ident, Literal, Span, TokenStream};
|
|
use quote::ToTokens;
|
|
|
|
pub fn is_keyword(txt: &str) -> bool {
|
|
match txt {
|
|
"abstract" | "alignof" | "as" | "become" | "box" | "break" | "const" | "continue"
|
|
| "crate" | "do" | "else" | "enum" | "extern" | "false" | "final" | "fn" | "for" | "if"
|
|
| "impl" | "in" | "let" | "loop" | "macro" | "match" | "mod" | "move" | "mut"
|
|
| "offsetof" | "override" | "priv" | "proc" | "pub" | "pure" | "ref" | "return"
|
|
| "Self" | "self" | "sizeof" | "static" | "struct" | "super" | "trait" | "true"
|
|
| "type" | "typeof" | "unsafe" | "unsized" | "use" | "virtual" | "where" | "while"
|
|
| "yield" | "__handler" | "__object" => true,
|
|
_ => false,
|
|
}
|
|
}
|
|
|
|
pub fn is_camel_keyword(txt: &str) -> bool {
|
|
match txt {
|
|
"Self" => true,
|
|
_ => false,
|
|
}
|
|
}
|
|
|
|
pub fn snake_to_camel(input: &str) -> String {
|
|
let result = input
|
|
.split('_')
|
|
.flat_map(|s| {
|
|
let mut first = true;
|
|
s.chars().map(move |c| {
|
|
if first {
|
|
first = false;
|
|
c.to_ascii_uppercase()
|
|
} else {
|
|
c
|
|
}
|
|
})
|
|
})
|
|
.collect::<String>();
|
|
|
|
if is_camel_keyword(&result) {
|
|
format!("_{}", &result)
|
|
} else {
|
|
result
|
|
}
|
|
}
|
|
|
|
pub fn dotted_to_relname(input: &str) -> TokenStream {
|
|
let mut it = input.split('.');
|
|
match (it.next(), it.next()) {
|
|
(Some(module), Some(name)) => {
|
|
let module = Ident::new(module, Span::call_site());
|
|
let ident = Ident::new(&snake_to_camel(name), Span::call_site());
|
|
quote::quote!(super::#module::#ident)
|
|
}
|
|
(Some(name), None) => {
|
|
Ident::new(&snake_to_camel(name), Span::call_site()).into_token_stream()
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
pub fn null_terminated_byte_string_literal(string: &str) -> Literal {
|
|
let mut val = Vec::with_capacity(string.len() + 1);
|
|
val.extend_from_slice(string.as_bytes());
|
|
val.push(0);
|
|
|
|
Literal::byte_string(&val)
|
|
}
|