sqlite-loadable-rs/sqlite-loadable-macros/src/lib.rs at e2c4585bb35da0e97b4070f10b526ae8878b567c · asg017/sqlite-loadable-rs · 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
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
use proc_macro2::Ident;

use syn::{parse_macro_input, spanned::Spanned, Item};

use proc_macro::TokenStream;
use quote::quote_spanned;

/// Wraps an entrypoint function to expose an unsafe extern "C" function of the same name.
#[proc_macro_attribute]
pub fn sqlite_entrypoint(_attr: TokenStream, item: TokenStream) -> TokenStream {
    let ast = parse_macro_input!(item as syn::Item);
    match ast {
        Item::Fn(mut func) => {
            let c_entrypoint = func.sig.ident.clone();

            let original_funcname = func.sig.ident.to_string();
            func.sig.ident = Ident::new(
                format!("_{}", original_funcname).as_str(),
                func.sig.ident.span(),
            );

            let prefixed_original_function = func.sig.ident.clone();

            quote_spanned! {func.span()=>
                #func

                /// # Safety
                ///
                /// Should only be called by underlying SQLite C APIs,
                /// like sqlite3_auto_extension and sqlite3_cancel_auto_extension.
                #[unsafe(no_mangle)]
                pub unsafe extern "C" fn #c_entrypoint(
                    db: *mut sqlite3,
                    pz_err_msg: *mut *mut c_char,
                    p_api: *mut sqlite3_api_routines,
                ) -> c_uint {
                    register_entrypoint(db, pz_err_msg, p_api, #prefixed_original_function)
                }


            }
            .into()
        }
        _ => panic!("Only function items are allowed on sqlite_entrypoint"),
    }
}

/// Wraps an entrypoint function to expose an unsafe extern "C" function of the same name.
#[proc_macro_attribute]
pub fn sqlite_entrypoint_permanent(_attr: TokenStream, item: TokenStream) -> TokenStream {
    let ast = parse_macro_input!(item as syn::Item);
    match ast {
        Item::Fn(mut func) => {
            let c_entrypoint = func.sig.ident.clone();

            let original_funcname = func.sig.ident.to_string();
            func.sig.ident = Ident::new(
                format!("_{}", original_funcname).as_str(),
                func.sig.ident.span(),
            );

            let prefixed_original_function = func.sig.ident.clone();

            quote_spanned! {func.span()=>
                #func

                /// # Safety
                ///
                /// Should only be called by underlying SQLite C APIs,
                /// like sqlite3_auto_extension and sqlite3_cancel_auto_extension.
                #[unsafe(no_mangle)]
                pub unsafe extern "C" fn #c_entrypoint(
                    db: *mut sqlite3,
                    pz_err_msg: *mut *mut c_char,
                    p_api: *mut sqlite3_api_routines,
                ) -> c_uint {
                    register_entrypoint_load_permanently(db, pz_err_msg, p_api, #prefixed_original_function)
                }


            }
            .into()
        }
        _ => panic!("Only function items are allowed on sqlite_entrypoint"),
    }
}