diff options
Diffstat (limited to 'vendor/wasm-bindgen-backend/src/util.rs')
-rw-r--r-- | vendor/wasm-bindgen-backend/src/util.rs | 161 |
1 files changed, 161 insertions, 0 deletions
diff --git a/vendor/wasm-bindgen-backend/src/util.rs b/vendor/wasm-bindgen-backend/src/util.rs new file mode 100644 index 000000000..1a2b07dce --- /dev/null +++ b/vendor/wasm-bindgen-backend/src/util.rs @@ -0,0 +1,161 @@ +//! Common utility function for manipulating syn types and +//! handling parsed values + +use std::collections::hash_map::DefaultHasher; +use std::env; +use std::fmt; +use std::hash::{Hash, Hasher}; +use std::iter::FromIterator; +use std::sync::atomic::AtomicBool; +use std::sync::atomic::AtomicUsize; +use std::sync::atomic::Ordering::SeqCst; + +use crate::ast; +use proc_macro2::{self, Ident}; + +/// Check whether a given `&str` is a Rust keyword +fn is_rust_keyword(name: &str) -> bool { + match name { + "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" | "bool" | "_" => true, + _ => false, + } +} + +/// Create an `Ident`, possibly mangling it if it conflicts with a Rust keyword. +pub fn rust_ident(name: &str) -> Ident { + if name == "" { + panic!("tried to create empty Ident (from \"\")"); + } else if is_rust_keyword(name) { + Ident::new(&format!("{}_", name), proc_macro2::Span::call_site()) + + // we didn't historically have `async` in the `is_rust_keyword` list above, + // so for backwards compatibility reasons we need to generate an `async` + // identifier as well, but we'll be sure to use a raw identifier to ease + // compatibility with the 2018 edition. + // + // Note, though, that `proc-macro` doesn't support a normal way to create a + // raw identifier. To get around that we do some wonky parsing to + // roundaboutly create one. + } else if name == "async" { + let ident = "r#async" + .parse::<proc_macro2::TokenStream>() + .unwrap() + .into_iter() + .next() + .unwrap(); + match ident { + proc_macro2::TokenTree::Ident(i) => i, + _ => unreachable!(), + } + } else if name.chars().next().unwrap().is_ascii_digit() { + Ident::new(&format!("N{}", name), proc_macro2::Span::call_site()) + } else { + raw_ident(name) + } +} + +/// Create an `Ident` without checking to see if it conflicts with a Rust +/// keyword. +pub fn raw_ident(name: &str) -> Ident { + Ident::new(name, proc_macro2::Span::call_site()) +} + +/// Create a path type from the given segments. For example an iterator yielding +/// the idents `[foo, bar, baz]` will result in the path type `foo::bar::baz`. +pub fn simple_path_ty<I>(segments: I) -> syn::Type +where + I: IntoIterator<Item = Ident>, +{ + path_ty(false, segments) +} + +/// Create a global path type from the given segments. For example an iterator +/// yielding the idents `[foo, bar, baz]` will result in the path type +/// `::foo::bar::baz`. +pub fn leading_colon_path_ty<I>(segments: I) -> syn::Type +where + I: IntoIterator<Item = Ident>, +{ + path_ty(true, segments) +} + +fn path_ty<I>(leading_colon: bool, segments: I) -> syn::Type +where + I: IntoIterator<Item = Ident>, +{ + let segments: Vec<_> = segments + .into_iter() + .map(|i| syn::PathSegment { + ident: i, + arguments: syn::PathArguments::None, + }) + .collect(); + + syn::TypePath { + qself: None, + path: syn::Path { + leading_colon: if leading_colon { + Some(Default::default()) + } else { + None + }, + segments: syn::punctuated::Punctuated::from_iter(segments), + }, + } + .into() +} + +/// Create a path type with a single segment from a given Identifier +pub fn ident_ty(ident: Ident) -> syn::Type { + simple_path_ty(Some(ident)) +} + +/// Convert an ImportFunction into the more generic Import type, wrapping the provided function +pub fn wrap_import_function(function: ast::ImportFunction) -> ast::Import { + ast::Import { + module: None, + js_namespace: None, + kind: ast::ImportKind::Function(function), + } +} + +/// Small utility used when generating symbol names. +/// +/// Hashes the public field here along with a few cargo-set env vars to +/// distinguish between runs of the procedural macro. +#[derive(Debug)] +pub struct ShortHash<T>(pub T); + +impl<T: Hash> fmt::Display for ShortHash<T> { + fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { + static HASHED: AtomicBool = AtomicBool::new(false); + static HASH: AtomicUsize = AtomicUsize::new(0); + + // Try to amortize the cost of loading env vars a lot as we're gonna be + // hashing for a lot of symbols. + if !HASHED.load(SeqCst) { + let mut h = DefaultHasher::new(); + env::var("CARGO_PKG_NAME") + .expect("should have CARGO_PKG_NAME env var") + .hash(&mut h); + env::var("CARGO_PKG_VERSION") + .expect("should have CARGO_PKG_VERSION env var") + .hash(&mut h); + // This may chop off 32 bits on 32-bit platforms, but that's ok, we + // just want something to mix in below anyway. + HASH.store(h.finish() as usize, SeqCst); + HASHED.store(true, SeqCst); + } + + let mut h = DefaultHasher::new(); + HASH.load(SeqCst).hash(&mut h); + self.0.hash(&mut h); + write!(f, "{:016x}", h.finish()) + } +} |