entity-derive-impl 0.3.0

Internal proc-macro implementation for entity-derive. Use entity-derive instead.
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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

// SPDX-FileCopyrightText: 2025-2026 RAprogramm <andrey.rozanov.vl@gmail.com>
// SPDX-License-Identifier: MIT

//! Projection struct generation.
//!
//! This module generates projection structs for partial entity views.
//! Each projection defines a subset of fields for specific use cases:
//!
//! | Projection | Use Case |
//! |------------|----------|
//! | `UserPublic` | Public profile (id, name, avatar) |
//! | `UserAdmin` | Admin view (id, name, email, role) |
//! | `PostSummary` | List view (id, title, created_at) |
//!
//! # Definition
//!
//! Projections are defined at entity level:
//!
//! ```rust,ignore
//! #[derive(Entity)]
//! #[entity(table = "users")]
//! #[projection(Public: id, name, avatar)]
//! #[projection(Admin: id, name, email, role)]
//! pub struct User { ... }
//! ```
//!
//! # Generated Code
//!
//! For each projection, generates:
//! - `{Entity}{Projection}` struct with specified fields
//! - `From<{Entity}> for {Entity}{Projection}` implementation

use proc_macro2::TokenStream;
use quote::{format_ident, quote};

use super::parse::EntityDef;
use crate::utils::marker;

/// Generates all projection structs for the entity.
///
/// Returns a combined `TokenStream` containing projection struct definitions
/// and their From implementations.
pub fn generate(entity: &EntityDef) -> TokenStream {
    let projections: Vec<TokenStream> = entity
        .projections
        .iter()
        .map(|proj| generate_projection(entity, proj))
        .collect();

    quote! { #(#projections)* }
}

/// Generate a single projection struct and its From impl.
fn generate_projection(entity: &EntityDef, proj: &super::parse::ProjectionDef) -> TokenStream {
    let vis = &entity.vis;
    let entity_name = entity.name();
    let proj_name = format_ident!("{}{}", entity_name, proj.name);

    let field_defs: Vec<TokenStream> = proj
        .fields
        .iter()
        .filter_map(|field_name| {
            entity
                .fields
                .iter()
                .find(|f| f.name() == field_name)
                .map(|f| {
                    let n = f.name();
                    let t = f.ty();
                    quote! { pub #n: #t }
                })
        })
        .collect();

    if field_defs.is_empty() {
        return TokenStream::new();
    }

    let field_mappings: Vec<TokenStream> = proj
        .fields
        .iter()
        .filter_map(|field_name| {
            entity
                .fields
                .iter()
                .find(|f| f.name() == field_name)
                .map(|f| {
                    let n = f.name();
                    quote! { #n: value.#n.clone() }
                })
        })
        .collect();

    let marker = marker::generated();

    quote! {
        #marker
        #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
        #[cfg_attr(feature = "api", derive(utoipa::ToSchema))]
        #[cfg_attr(feature = "postgres", derive(sqlx::FromRow))]
        #vis struct #proj_name {
            #(#field_defs),*
        }

        #marker
        impl From<#entity_name> for #proj_name {
            fn from(value: #entity_name) -> Self {
                Self {
                    #(#field_mappings),*
                }
            }
        }

        #marker
        impl From<&#entity_name> for #proj_name {
            fn from(value: &#entity_name) -> Self {
                Self {
                    #(#field_mappings),*
                }
            }
        }
    }
}