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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

extern crate proc_macro;

use proc_macro::TokenStream;

use proc_macro_hack::proc_macro_hack;
use quote::quote;
use syn::{parse_macro_input, LitStr};

use unic_locale_impl::Locale;

#[proc_macro_hack]
pub fn locale(input: TokenStream) -> TokenStream {
    let id = parse_macro_input!(input as LitStr);
    let parsed: Locale = id.value().parse().expect("Malformed Locale Identifier");

    let lang = parsed.get_language();
    let lang = if lang.is_empty() {
        quote!(None)
    } else {
        quote!(Some(#lang))
    };
    let script = parsed.get_script();
    let script = if let Some(script) = script {
        quote!(Some(#script))
    } else {
        quote!(None)
    };
    let region = parsed.get_region();
    let region = if let Some(region) = region {
        quote!(Some(#region))
    } else {
        quote!(None)
    };
    let variants = parsed.get_variants();
    let variants = if variants.is_empty() {
        quote!(None)
    } else {
        quote!(Some(&[#(#variants,)*]))
    };

    // We're reparsing the extensions here. It would be nice to just
    // pass the parsed ones.
    let extensions = parsed.get_extensions().to_string();
    let extensions = if extensions.is_empty() {
        quote!(None)
    } else {
        quote!(Some(#extensions))
    };

    TokenStream::from(quote! {
        ::unic_locale_impl::Locale::from_parts_unchecked(
            #lang,
            #script,
            #region,
            #variants,
            #extensions.map(|e: &str| e.parse().expect("must parse"))
        )
    })
}