linked_data_schema_derive/
lib.rs

1use linked_data_core::{RdfEnum, RdfField, RdfStruct, RdfType, RdfVariant, TokenGenerator};
2use proc_macro_error::proc_macro_error;
3use proc_macro2::{Literal, TokenStream};
4use quote::ToTokens;
5use syn::DeriveInput;
6use uuid::Uuid;
7
8#[proc_macro_error]
9#[proc_macro_derive(LinkedDataSchema, attributes(ld))]
10pub fn derive_serialize(item: proc_macro::TokenStream) -> proc_macro::TokenStream {
11  let raw_input = syn::parse_macro_input!(item as DeriveInput);
12  let linked_data_type: RdfType<Schema> = RdfType::from_derive(raw_input);
13
14  let mut output = TokenStream::new();
15  linked_data_type.to_tokens(&mut output);
16  output.into()
17}
18
19#[derive(Debug)]
20struct Schema;
21
22impl TokenGenerator for Schema {
23  fn generate_type_tokens(linked_data_type: &RdfType<Self>, tokens: &mut TokenStream) {
24    let implementations = match linked_data_type {
25      RdfType::Enum(rdf_enum) => quote::quote! {#rdf_enum},
26      RdfType::Struct(rdf_struct) => quote::quote! {#rdf_struct},
27    };
28
29    tokens.extend(implementations)
30  }
31
32  fn generate_struct_tokens(rdf_struct: &RdfStruct<Self>, tokens: &mut TokenStream) {
33    let type_iri = rdf_struct.type_iri().unwrap();
34
35    let type_iri_shape = Literal::string(&format!("{}Shape", type_iri.as_str()));
36    let type_iri = Literal::string(type_iri.as_str());
37
38    let prefix_mapping = rdf_struct.prefix_mappings().clone();
39
40    let insert_all_prefix_mapping = prefix_mapping
41      .into_iter()
42      .map(|(prefix, iri)| {
43        let prefix = Literal::string(&prefix.to_string());
44        let iri = Literal::string(&iri);
45
46        quote::quote! {
47          prefix_map.insert(#prefix, &iri!(#iri)).unwrap();
48        }
49      })
50      .collect::<TokenStream>();
51
52    let ident = &rdf_struct.ident;
53    let fields = &rdf_struct.fields;
54
55    let property_shapes_iris = fields
56      .iter()
57      .map(|field| {
58        if let Some(predicate) = field.predicate() {
59          let identifier = Literal::string(&format!("{}Field", predicate.as_str()));
60
61          quote::quote! {
62            RDFNode::iri(IriS::from_str(#identifier).unwrap()),
63          }
64        } else {
65          quote::quote! {}
66        }
67      })
68      .collect::<TokenStream>();
69
70    let struct_blank_node = Literal::string(Uuid::new_v4().to_string().as_str());
71
72    tokens.extend(quote::quote! {
73      impl ::linked_data_schema::LinkedDataSchemaFieldVisitor for #ident {
74        fn field_components() -> Vec<::linked_data_schema::reexports::shacl_ast::ast::component::Component> {
75          Self::components()
76        }
77
78        fn type_iri_ref() -> Option<::linked_data_schema::reexports::prefixmap::IriRef> {
79          use ::linked_data_schema::reexports::prefixmap::IriRef;
80          use ::linked_data_schema::reexports::iri_s::iri;
81
82          Some(IriRef::iri(iri!(#type_iri_shape)))
83        }
84      }
85
86      impl ::linked_data_schema::LinkedDataSchema for #ident {
87        fn shacl() -> ::linked_data_schema::reexports::shacl_ast::Schema<::linked_data_schema::reexports::srdf::SRDFGraph> {
88          use ::linked_data_schema::{
89            reexports::{
90              iri_s::{iris::IriS, iri},
91              prefixmap::{PrefixMap, IriRef},
92              shacl_ast::{
93                ast::{
94                  component::Component,
95                  shape::Shape,
96                  node_shape::NodeShape,
97                  property_shape::PropertyShape,
98                  target::Target,
99                },
100                Schema,
101              },
102              srdf::{
103                RDFNode,
104                SHACLPath,
105              },
106            },
107            LinkedDataSchemaFieldVisitor,
108          };
109          use std::str::FromStr;
110          use std::collections::HashMap;
111
112          let mut prefix_map = PrefixMap::new();
113          #insert_all_prefix_mapping
114
115          let mut shapes = HashMap::default();
116
117          let rdf_node_type_iri = RDFNode::iri(IriS::from_str(#type_iri_shape).unwrap());
118
119          let property_shapes = vec![
120            #property_shapes_iris
121          ];
122
123          let node_shape = NodeShape::new(rdf_node_type_iri.clone())
124            .with_targets(vec![Target::TargetClass(RDFNode::iri(IriS::from_str(#type_iri).unwrap()))])
125            .with_property_shapes(property_shapes);
126
127          let _ = shapes.insert(RDFNode::BlankNode(#struct_blank_node.to_string()), Shape::NodeShape(Box::new(node_shape)));
128
129          #(#fields)*
130
131          Schema::default()
132            .with_prefixmap(prefix_map)
133            .with_shapes(shapes)
134        }
135
136        fn components() -> Vec<::linked_data_schema::reexports::shacl_ast::ast::component::Component> {
137          use ::linked_data_schema::{
138            reexports::{
139              iri_s::iri,
140              prefixmap::IriRef,
141              shacl_ast::ast::component::Component,
142            }
143          };
144
145          vec![
146            Component::Datatype(IriRef::iri(iri!(#type_iri_shape))),
147          ]
148        }
149      }
150    })
151  }
152
153  fn generate_enum_tokens(r#enum: &RdfEnum<Self>, tokens: &mut TokenStream) {
154    let _variants = &r#enum.variants;
155    let ident = &r#enum.ident;
156
157    tokens.extend(quote::quote! {
158      impl ::linked_data_schema::LinkedDataSchema for #ident {
159        fn shacl() -> ::linked_data_schema::reexports::shacl_ast::Schema<::linked_data_schema::reexports::srdf::SRDFGraph> {
160          use ::linked_data_schema::reexports::{
161            prefixmap::PrefixMap,
162            shacl_ast::{
163              ast::shape::Shape,
164              Schema,
165            },
166            srdf::RDFNode,
167          };
168          use std::collections::HashMap;
169
170          let prefix_map = PrefixMap::new();
171          let shapes = HashMap::default();
172
173          Schema::default()
174            .with_prefixmap(prefix_map)
175            .with_shapes(shapes)
176        }
177      }
178    })
179  }
180
181  fn generate_variant_tokens(_variant: &RdfVariant<Self>, _tokens: &mut TokenStream) {
182    todo!()
183  }
184
185  fn generate_field_tokens(field: &RdfField<Self>, tokens: &mut TokenStream) {
186    if field.is_ignored() {
187      return;
188    }
189
190    //if field.is_flattened() {
191
192    //}
193
194    if let Some(predicate) = field.predicate() {
195      let identifier = Literal::string(&format!("{}Field", predicate.as_str()));
196      let predicate = Literal::string(predicate.as_str());
197
198      let field_type = &field.ty;
199
200      tokens.extend(quote::quote! {
201        let node = RDFNode::BlankNode(::linked_data_schema::reexports::uuid::Uuid::new_v4().to_string());
202
203        let rdf_node_type_iri = RDFNode::iri(IriS::from_str(#identifier).unwrap());
204
205        let property_shape = PropertyShape::new(
206          rdf_node_type_iri,
207          SHACLPath::iri(IriS::from_str(#predicate).unwrap()),
208        ).with_components(<#field_type>::field_components());
209
210        let _ = shapes.insert(node, Shape::PropertyShape(Box::new(property_shape)));
211      })
212    }
213  }
214}
linked_data_schema_derive/lib.rs

linked_data_schema_derive/
lib.rs
