macroquad-lua-macros 0.0.1

do not use
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

use quote::quote;
use std::path::Path;
use std::path::PathBuf;
use syn::{parse_macro_input, LitStr};
// use quote::ToTokens;

#[proc_macro]
pub fn embed_lua_files(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let base_path_str = parse_macro_input!(input as LitStr).value();
    let base_path = Path::new(&base_path_str);
    let mut parts: Vec<proc_macro2::TokenStream> = vec![];

    for path in &list_all_paths(&base_path) {
        let mut name = path
            .iter()
            .skip(base_path.iter().count())
            .take(path.iter().count() - base_path.iter().count())
            .map(|p| p.to_str().unwrap())
            .collect::<Vec<&str>>()
            .join(".");
        if !name.ends_with(".lua") {
            continue;
        }
        name.pop();
        name.pop();
        name.pop();
        name.pop();
        let absolute_path = path.canonicalize().unwrap();
        let path_str = absolute_path.to_str().unwrap();
        parts.push(quote! {
            map.insert(#name.to_string(), include_str!(#path_str).to_string());
        });
    }
    quote! {
        {
            let mut map = std::collections::HashMap::new();
            #(#parts)*
            map
        }
    }
    .into()
}

fn list_all_paths(path: &Path) -> Vec<PathBuf> {
    let mut paths = vec![];
    for entry in path.read_dir().unwrap() {
        let entry = entry.unwrap();
        let path = entry.path();
        if path.is_dir() {
            paths.extend(list_all_paths(&path));
        } else {
            paths.push(path);
        }
    }
    paths
}