macroforge_ts 0.1.80

TypeScript macro expansion engine - write compile-time macros in Rust
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

//! # Import Registry (re-exports + foreign types)
//!
//! The core [`ImportRegistry`] lives in `macroforge_ts_syn`. This module re-exports
//! it and adds the `foreign_types` thread-local, which depends on [`ForeignTypeConfig`]
//! from this crate.

use std::cell::RefCell;

use super::ForeignTypeConfig;

// Re-export everything from macroforge_ts_syn's import_registry
pub use macroforge_ts_syn::import_registry::{
    GeneratedImport, ImportRegistry, SourceImport, clear_registry, install_registry, take_registry,
    with_registry, with_registry_mut,
};

// ============================================================================
// Foreign types thread-local (depends on ForeignTypeConfig from this crate)
// ============================================================================

thread_local! {
    static FOREIGN_TYPES: RefCell<Vec<ForeignTypeConfig>> = const { RefCell::new(Vec::new()) };
}

/// Read foreign types (immutable access).
pub fn with_foreign_types<R>(f: impl FnOnce(&[ForeignTypeConfig]) -> R) -> R {
    FOREIGN_TYPES.with(|ft| f(&ft.borrow()))
}

/// Write foreign types (mutable access).
pub fn with_foreign_types_mut<R>(f: impl FnOnce(&mut Vec<ForeignTypeConfig>) -> R) -> R {
    FOREIGN_TYPES.with(|ft| f(&mut ft.borrow_mut()))
}

/// Set foreign types.
pub fn set_foreign_types(types: Vec<ForeignTypeConfig>) {
    FOREIGN_TYPES.with(|ft| {
        *ft.borrow_mut() = types;
    });
}

/// Clear foreign types.
pub fn clear_foreign_types() {
    FOREIGN_TYPES.with(|ft| {
        ft.borrow_mut().clear();
    });
}