unic_locale_macros_impl/
lib.rs

1extern crate proc_macro;
2
3use proc_macro::TokenStream;
4
5use proc_macro_hack::proc_macro_hack;
6use quote::quote;
7use syn::{parse_macro_input, LitStr};
8
9use unic_locale_impl::Locale;
10
11#[proc_macro_hack]
12pub fn locale(input: TokenStream) -> TokenStream {
13    let id = parse_macro_input!(input as LitStr);
14    let parsed: Locale = id.value().parse().expect("Malformed Locale Identifier");
15
16    let (lang, script, region, variants, extensions) = parsed.into_parts();
17
18    let lang: Option<u64> = lang.into();
19    let lang = if let Some(lang) = lang {
20        quote!(unsafe { $crate::subtags::Language::from_raw_unchecked(#lang) })
21    } else {
22        quote!($crate::subtags::Language::default())
23    };
24    let script = if let Some(script) = script {
25        let script: u32 = script.into();
26        quote!(Some(unsafe { $crate::subtags::Script::from_raw_unchecked(#script) }))
27    } else {
28        quote!(None)
29    };
30    let region = if let Some(region) = region {
31        let region: u32 = region.into();
32        quote!(Some(unsafe { $crate::subtags::Region::from_raw_unchecked(#region) }))
33    } else {
34        quote!(None)
35    };
36    let variants = if !variants.is_empty() {
37        let v: Vec<_> = variants
38            .iter()
39            .map(|v| {
40                let variant: u64 = v.into();
41                quote!(unsafe { $crate::subtags::Variant::from_raw_unchecked(#variant) })
42            })
43            .collect();
44        quote!(Some(Box::new([#(#v,)*])))
45    } else {
46        quote!(None)
47    };
48
49    TokenStream::from(quote! {
50        unsafe { $crate::Locale::from_raw_parts_unchecked(
51            #lang,
52            #script,
53            #region,
54            #variants,
55            #extensions.parse().expect("must parse")
56        ) }
57    })
58}