vite_static_derive/
lib.rs1#![warn(clippy::pedantic)]
2#![allow(clippy::ref_option)]
3
4use proc_macro::TokenStream;
5use quote::quote;
6use syn::{DeriveInput, parse_macro_input};
7
8mod embedding;
9mod parse;
10
11use parse::parse_manifest_input;
12
13fn internal_manifest_derive(input: DeriveInput) -> syn::Result<TokenStream> {
14 let settings = parse_manifest_input(&input)?;
15
16 let ident = input.ident;
17
18 let manifest = embedding::get_manifest(&settings.vite_dist.0)
19 .map_err(|err| syn::Error::new(settings.vite_dist.1, format!("{err:#}")))?;
20
21 let (key_output_pairs, chunks) = embedding::generate_manifest_data(&manifest);
22
23 let expanded = quote! {
24 const KEY_OUTPUT_PAIRS: &[(&str, &str)] = &[ #(#key_output_pairs),* ];
25 const CHUNKS: &[(&str, &::vite_static::StaticManifestChunk)] = &[ #(#chunks),* ];
26
27 impl ::vite_static::Manifest<'static> for #ident {
28 fn manifest_key_output_pairs(&self) -> &'static [(&'static str, &'static str)] {
29 KEY_OUTPUT_PAIRS
30 }
31 fn manifest_chunks(&self) -> &'static [(&'static str, &'static ::vite_static::StaticManifestChunk)] {
32 CHUNKS
33 }
34 fn boxed(&self) -> Box<dyn ::vite_static::Manifest<'static>> { Box::new(#ident) }
35 }
36 };
37 Ok(TokenStream::from(expanded))
38}
39
40#[proc_macro_derive(Manifest, attributes(vite_dist))]
67pub fn manifest_derive(input: TokenStream) -> TokenStream {
68 let input = parse_macro_input!(input as DeriveInput);
69
70 match internal_manifest_derive(input) {
71 Ok(stream) => stream,
72 Err(err) => TokenStream::from(err.to_compile_error()),
73 }
74}