anydir-macro 0.1.0

A crate for working with both compile-time embedded directories and runtime directories using an abstraction.
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
29
30
31
32
33
34
35
36
37
38
39

use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, LitStr};

/// Embed a directory at compile time
#[proc_macro]
pub fn embed_dir(input: TokenStream) -> TokenStream {
    // Parse the input as a string literal
    let input = parse_macro_input!(input as LitStr);
    let dir_path = input.value();

    // Generate a unique identifier for the static variable
    let var_name = sanitize_identifier(&dir_path);

    let var_ident = syn::Ident::new(&var_name, proc_macro2::Span::call_site());

    // Generate the code
    let expanded = quote! {
        {
            use include_dir::include_dir;
            static #var_ident: include_dir::Dir = include_dir!(#dir_path);
            &#var_ident
        }
    };

    TokenStream::from(expanded)
}

fn sanitize_identifier(s: &str) -> String {
    let mut result = String::from("DIR_");
    for c in s.chars() {
        if c.is_ascii_alphanumeric() {
            result.push(c.to_ascii_uppercase());
        } else {
            result.push('_');
        }
    }
    result
}