icydb_paths/
lib.rs

1use proc_macro2::TokenStream;
2use quote::quote;
3use syn::Path;
4
5const INTERNAL_CRATES: &[&str] = &[
6    "icydb",
7    "icydb-base",
8    "icydb-build",
9    "icydb-core",
10    "icydb-error",
11    "icydb-macros",
12    "icydb-paths",
13    "icydb-schema",
14];
15
16fn env_path(name: &str) -> Option<TokenStream> {
17    std::env::var(name)
18        .ok()
19        .map(|value| value.trim().to_string())
20        .and_then(|value| syn::parse_str::<Path>(&value).ok())
21        .map(|path| quote!(#path))
22}
23
24///
25/// CratePaths
26///
27/// Resolves crate roots for generated code. Internal icydb crates default to
28/// direct crate names to avoid meta-crate cycles; other crates prefer the
29/// public `icydb::` facade. Env vars allow overrides:
30/// `ICYDB_CORE_CRATE`, `ICYDB_SCHEMA_CRATE`, `ICYDB_ERROR_CRATE`.
31///
32
33#[derive(Clone, Debug, Default)]
34pub struct CratePaths {
35    pub core: TokenStream,
36    pub schema: TokenStream,
37    pub error: TokenStream,
38}
39
40impl CratePaths {
41    #[must_use]
42    /// Resolve crate paths for generated code, honoring environment overrides.
43    pub fn new() -> Self {
44        let pkg = std::env::var("CARGO_PKG_NAME").unwrap_or_default();
45        let use_meta_paths = !INTERNAL_CRATES.contains(&pkg.as_str());
46
47        let core = if use_meta_paths {
48            quote!(icydb::core)
49        } else {
50            quote!(icydb_core)
51        };
52
53        let schema = if use_meta_paths {
54            quote!(icydb::schema)
55        } else {
56            quote!(icydb_schema)
57        };
58
59        let error = if use_meta_paths {
60            quote!(icydb::error)
61        } else {
62            quote!(icydb_error)
63        };
64
65        Self {
66            core: env_path("ICYDB_CORE_CRATE").unwrap_or(core),
67            schema: env_path("ICYDB_SCHEMA_CRATE").unwrap_or(schema),
68            error: env_path("ICYDB_ERROR_CRATE").unwrap_or(error),
69        }
70    }
71}
72
73/// Singleton accessor for proc-macro contexts.
74#[must_use]
75pub fn paths() -> CratePaths {
76    CratePaths::new()
77}
78
79///
80/// TESTS
81///
82
83#[cfg(test)]
84mod tests {
85    use super::*;
86    use quote::quote;
87    use std::env;
88
89    struct TempEnv {
90        key: &'static str,
91        prev: Option<String>,
92    }
93
94    impl TempEnv {
95        fn set(key: &'static str, value: Option<&str>) -> Self {
96            let prev = env::var(key).ok();
97            unsafe {
98                match value {
99                    Some(v) => env::set_var(key, v),
100                    None => env::remove_var(key),
101                }
102            }
103            Self { key, prev }
104        }
105    }
106
107    impl Drop for TempEnv {
108        fn drop(&mut self) {
109            unsafe {
110                match &self.prev {
111                    Some(value) => env::set_var(self.key, value),
112                    None => env::remove_var(self.key),
113                }
114            }
115        }
116    }
117
118    #[test]
119    fn uses_internal_crate_names_inside_workspace() {
120        let _pkg = TempEnv::set("CARGO_PKG_NAME", Some("icydb-paths"));
121        let _core = TempEnv::set("ICYDB_CORE_CRATE", None);
122        let _schema = TempEnv::set("ICYDB_SCHEMA_CRATE", None);
123        let _error = TempEnv::set("ICYDB_ERROR_CRATE", None);
124
125        let paths = CratePaths::new();
126
127        assert_eq!(paths.core.to_string(), quote!(icydb_core).to_string());
128        assert_eq!(paths.schema.to_string(), quote!(icydb_schema).to_string());
129        assert_eq!(paths.error.to_string(), quote!(icydb_error).to_string());
130    }
131
132    #[test]
133    fn honors_env_overrides_for_external_consumers() {
134        let _pkg = TempEnv::set("CARGO_PKG_NAME", Some("external-app"));
135        let _core = TempEnv::set("ICYDB_CORE_CRATE", Some("custom::core"));
136        let _schema = TempEnv::set("ICYDB_SCHEMA_CRATE", Some("custom::schema"));
137        let _error = TempEnv::set("ICYDB_ERROR_CRATE", Some("custom::error"));
138
139        let paths = CratePaths::new();
140
141        assert_eq!(paths.core.to_string(), quote!(custom::core).to_string());
142        assert_eq!(paths.schema.to_string(), quote!(custom::schema).to_string());
143        assert_eq!(paths.error.to_string(), quote!(custom::error).to_string());
144    }
145}
icydb_paths/lib.rs

icydb_paths/
lib.rs
