vespera_macro 0.1.5

Procedural macros for Vespera - automatic route discovery and OpenAPI generation
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99

//! Collector for routes and structs

use crate::file_utils::{collect_files, file_to_segments};
use crate::metadata::{CollectedMetadata, RouteMetadata, StructMetadata};
use crate::route::extract_route_info;
use std::path::Path;
use syn::Item;

/// Collect routes and structs from a folder
pub fn collect_metadata(folder_path: &Path, folder_name: &str) -> CollectedMetadata {
    let mut metadata = CollectedMetadata::new(folder_name.to_string());

    let files = match collect_files(folder_path) {
        Ok(files) => files,
        Err(_) => return metadata,
    };

    for file in files {
        if !file.extension().map(|e| e == "rs").unwrap_or(false) {
            continue;
        }

        let content = match std::fs::read_to_string(&file) {
            Ok(content) => content,
            Err(_) => continue,
        };

        let file_ast = match syn::parse_file(&content) {
            Ok(ast) => ast,
            Err(_) => continue,
        };

        // Get module path
        let segments = if let Ok(file_stem) = file.strip_prefix(folder_path) {
            file_to_segments(file_stem, folder_path)
        } else {
            continue;
        };

        let module_path = if folder_name.is_empty() {
            segments.join("::")
        } else {
            format!("{}::{}", folder_name, segments.join("::"))
        };

        let file_path = file.display().to_string();

        // Collect routes
        for item in &file_ast.items {
            if let Item::Fn(fn_item) = item
                && let Some(route_info) = extract_route_info(&fn_item.attrs)
            {
                let route_path = if let Some(custom_path) = &route_info.path {
                    let base = format!("/{}", segments.join("/"));
                    let trimmed_base = base.trim_end_matches('/');
                    format!("{}/{}", trimmed_base, custom_path.trim_start_matches('/'))
                } else {
                    format!("/{}", segments.join("/"))
                };

                metadata.routes.push(RouteMetadata {
                    method: route_info.method,
                    path: route_path,
                    function_name: fn_item.sig.ident.to_string(),
                    module_path: module_path.clone(),
                    file_path: file_path.clone(),
                    signature: quote::quote!(#fn_item).to_string(),
                    error_status: route_info.error_status.clone(),
                });
            }

            // Collect structs with Schema derive
            if let Item::Struct(struct_item) = item {
                // Check if struct has Schema derive by checking attribute tokens
                let has_schema = struct_item.attrs.iter().any(|attr| {
                    if attr.path().is_ident("derive") {
                        // Convert attribute to tokens and check for Schema
                        let tokens = quote::quote!(#attr).to_string();
                        tokens.contains("Schema")
                    } else {
                        false
                    }
                });

                if has_schema {
                    metadata.structs.push(StructMetadata {
                        name: struct_item.ident.to_string(),
                        module_path: module_path.clone(),
                        file_path: file_path.clone(),
                        definition: quote::quote!(#struct_item).to_string(),
                    });
                }
            }
        }
    }

    metadata
}