dotenvy/dotenv_codegen/src/lib.rs at b571a03c9a43c8eb1721801b01e653d270dd707b · allan2/dotenvy · GitHub

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
#![forbid(unsafe_code)]

use quote::quote;

#[proc_macro]
pub fn dotenv(_input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    if let Err(err) = dotenvy::dotenv() {
        let msg = format!("Error loading .env file: {}", err);
        return quote! {
            compile_error!(#msg);
        }
        .into();
    }

    quote! {}.into()
}

#[proc_macro]
pub fn dotenv_override(_input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    if let Err(err) = dotenvy::dotenv_override() {
        let msg = format!("Error loading .env file: {}", err);
        return quote! {
            compile_error!(#msg);
        }
        .into();
    }

    quote! {}.into()
}