Skip to main content

cljrs_runtime/env/
env.rs

1//! Lexical environment: local frames, global namespace table, and current Env.
2
3use std::collections::{HashMap, HashSet};
4use std::sync::atomic::{AtomicBool, AtomicU8, Ordering};
5use std::sync::{Arc, Condvar, Mutex, RwLock};
6
7use crate::env::async_hook::AsyncRuntime;
8
9use crate::env::error::EvalResult;
10use crate::mode::{ExecutionMode, TierState};
11use cljrs_gc::{GcConfig, GcPtr};
12use cljrs_reader::Form;
13use cljrs_value::{CljxFn, Namespace, ReferClojureFilter, Value, Var};
14// ── RequireSpec / RequireRefer ─────────────────────────────────────────────────
15
16/// How symbols should be referred into the requiring namespace.
17#[derive(Debug, Clone)]
18pub enum RequireRefer {
19    None,
20    All,
21    Named(Vec<Arc<str>>),
22}
23
24/// A parsed `require` specification.
25#[derive(Debug, Clone)]
26pub struct RequireSpec {
27    pub ns: Arc<str>,
28    /// Present when the namespace symbol carried a `@<hash>` version suffix.
29    pub version: Option<Arc<str>>,
30    pub alias: Option<Arc<str>>,
31    pub refer: RequireRefer,
32}
33
34// ── Frame ─────────────────────────────────────────────────────────────────────
35
36/// One stack frame of local bindings (a single `let*`, `fn`, or `loop*` scope).
37pub struct Frame {
38    pub bindings: Vec<(Arc<str>, Value)>,
39}
40
41impl Default for Frame {
42    fn default() -> Self {
43        Self::new()
44    }
45}
46
47impl Frame {
48    pub fn new() -> Self {
49        Self {
50            bindings: Vec::new(),
51        }
52    }
53
54    pub fn bind(&mut self, name: Arc<str>, val: Value) {
55        // Shadow: push new binding; lookup searches from the end.
56        self.bindings.push((name, val));
57    }
58
59    pub fn lookup(&self, name: &str) -> Option<&Value> {
60        // Search in reverse order so later bindings shadow earlier ones.
61        tracing::trace!(target: "env", "lookup {}", name);
62        for (n, v) in self.bindings.iter().rev() {
63            if n.as_ref() == name {
64                return Some(v);
65            }
66        }
67        None
68    }
69}
70
71// ── GlobalEnv ─────────────────────────────────────────────────────────────────
72
73/// The global mutable store of all namespaces.
74pub struct GlobalEnv {
75    /// Process-unique identity of this runtime instance.
76    ///
77    /// Allocated from a counter, not derived from the `Arc`'s address: an
78    /// address is only unique while the allocation is live, so a dropped
79    /// runtime could hand its key to the next one and let it inherit stale
80    /// cross-defn IR.  Used to scope per-instance registries.
81    id: u64,
82    pub namespaces: RwLock<HashMap<Arc<str>, GcPtr<Namespace>>>,
83    /// Directories to search when resolving namespace names to files.
84    pub source_paths: RwLock<Vec<std::path::PathBuf>>,
85    /// Namespaces that have been fully loaded from a file (idempotent guard).
86    pub loaded: Mutex<std::collections::HashSet<Arc<str>>>,
87    /// Namespaces currently being loaded, mapped to the thread loading them.
88    /// Used to detect true circular requires (same thread) vs concurrent loads
89    /// (different thread — those wait on `loading_done` instead of erroring).
90    pub loading: Mutex<HashMap<Arc<str>, std::thread::ThreadId>>,
91    /// Signalled whenever a namespace finishes loading (or fails).
92    pub loading_done: Condvar,
93    /// Built-in namespace sources embedded in the binary.
94    /// Checked by `load_ns` before falling back to source-path search.
95    pub builtin_sources: RwLock<HashMap<Arc<str>, &'static str>>,
96    /// GC configuration for automatic collection based on memory pressure.
97    pub gc_config: RwLock<Option<Arc<GcConfig>>>,
98    /// How this runtime executes function calls.  Fixed when the runtime is
99    /// built; see [`crate::RuntimeBuilder::execution_mode`].
100    execution_mode: ExecutionMode,
101    /// Which tiers are live right now (see [`TierState`]).  Starts at
102    /// [`TierState::TreeWalk`] — nothing can be lowered until `clojure.core`
103    /// exists — and is raised once to `execution_mode.target_tier()` when the
104    /// builder finishes bootstrapping.
105    tier_state: AtomicU8,
106    /// This runtime's Tier-1 and Tier-2 state: the lowered-IR cache, the JIT
107    /// counters and native-code tables, and the JIT backend attached to this
108    /// runtime.  Instance state: two runtimes in one process never read,
109    /// evict, or invalidate each other's entries, and everything dies with
110    /// the runtime.
111    tiers: Arc<crate::tiered::tiers::Tiers>,
112    /// Optional async runtime registered by `cljrs-async`.
113    /// `None` when the library is not linked; `Some` after `cljrs_async::init`.
114    pub async_rt: RwLock<Option<Arc<dyn AsyncRuntime>>>,
115    /// Cache of values resolved at a specific commit.
116    /// Key format: `"<ns>/<name>@<commit>"` for individual vars,
117    /// or `"<ns>@<commit>"` for whole versioned namespaces.
118    pub version_cache: Mutex<HashMap<Arc<str>, Value>>,
119    /// Parsed `cljrs.edn` config, loaded once at startup.
120    pub deps_config: RwLock<Option<Arc<cljrs_project::config::DepsConfig>>>,
121    /// When true, every versioned-symbol or versioned-namespace resolution must
122    /// carry a valid commit signature (verified natively against `trusted_keys`)
123    /// before the historical code is executed.  Off by default; enabled via
124    /// `--verify-commit-signatures` CLI flag or `:verify-commit-signatures true`
125    /// in `cljrs.edn`.
126    pub verify_commit_signatures: AtomicBool,
127    /// The git backend used by versioned resolution and signature checking, or
128    /// `None` in builds that carry no VCS implementation (wasm, or
129    /// `cljrs-runtime` without its default `deps` feature).  With no provider,
130    /// source files are treated as living outside any repository and versioned
131    /// resolution can only use embedded (AOT) sources.  See [`crate::env::vcs`].
132    vcs: RwLock<Option<Arc<dyn crate::env::vcs::VcsProvider>>>,
133    /// Session-scoped cache of commits that have already passed signature
134    /// verification this run, keyed by `(repo_root, commit_hash)`.
135    pub sig_verify_cache: Mutex<HashSet<(Arc<str>, Arc<str>)>>,
136    /// Pinned source texts fetched from git this session, keyed by
137    /// `"<ns>@<commit>"`.  The AOT compiler embeds these in the produced
138    /// binary so versioned namespaces resolve without git at runtime.
139    pub versioned_sources: RwLock<HashMap<Arc<str>, Arc<str>>>,
140    /// When true (set by AOT harness main), versioned namespaces resolve
141    /// only from embedded builtin sources — never from git.  A versioned
142    /// namespace that was not embedded at compile time fails with a clear
143    /// error instead of attempting a fetch.
144    pub versioned_offline: AtomicBool,
145    /// Provenance of native (Rust-backed) packages recorded at registration:
146    /// namespace → the git commit the package was built from.  Consulted by
147    /// the versioned resolver's native HEAD fallback to detect pinned-commit
148    /// mismatches.
149    pub native_provenance: RwLock<HashMap<Arc<str>, Arc<str>>>,
150    /// When true, a pinned lookup of a native function whose recorded
151    /// provenance does not match the requested commit is an error instead of
152    /// a once-per-pin warning.  CLI: `--enforce-native-versions`; cljrs.edn:
153    /// `:enforce-native-versions true`.
154    pub enforce_native_versions: AtomicBool,
155    /// Pinned-native mismatches already warned about this session
156    /// (key: `"<ns>@<commit>"`), so each pin warns at most once.
157    pub provenance_warned: Mutex<HashSet<Arc<str>>>,
158    /// Optional loader for **pinned native packages** (`:rust/load :dylib`),
159    /// installed by the CLI.  Called by the versioned resolver with
160    /// `(globals, base_ns, commit)` before falling back to the HEAD native
161    /// binding; returns `Ok(true)` when it registered the package's pinned
162    /// implementations into the `"<base_ns>@<commit>"` namespace.
163    #[allow(clippy::type_complexity)]
164    pub pinned_native_loader: RwLock<Option<PinnedNativeLoader>>,
165    /// Optional loader for **native dependencies on the plain `require` path**
166    /// (`:rust/load :dylib`), installed by the CLI.  Called by the
167    /// unversioned namespace loader with `(globals, ns)` when a `require`d
168    /// namespace has no Clojure source on the source path; returns `Ok(true)`
169    /// when it built the dep's crate at the pinned `:git/sha` and registered
170    /// the package's exports into the **unversioned** namespace, so a plain
171    /// `(require '[my.native.lib :as lib])` brings the native code in.
172    #[allow(clippy::type_complexity)]
173    pub native_require_loader: RwLock<Option<NativeRequireLoader>>,
174    /// Loaders for **AOT-compiled namespaces**, installed by the binary
175    /// produced by `cljrs compile`.  Keyed by namespace name.  When a plain
176    /// `require` resolves a namespace that has a registered loader, `load_ns`
177    /// invokes the loader instead of interpreting Clojure source: the loader
178    /// evaluates the namespace's small interpreted preamble (its `ns`/`require`
179    /// and macro definitions) and then calls the namespace's natively compiled
180    /// initializer, so the bulk of the namespace runs as machine code rather
181    /// than being tree-walked at startup.
182    #[allow(clippy::type_complexity)]
183    pub compiled_ns_loaders: RwLock<HashMap<Arc<str>, CompiledNsLoader>>,
184}
185
186/// Loader callback for an AOT-compiled namespace (see
187/// `GlobalEnv::compiled_ns_loaders`).  Given the global env, it loads the
188/// namespace by running its interpreted preamble and its compiled initializer.
189pub type CompiledNsLoader = Arc<dyn Fn(&Arc<GlobalEnv>) -> EvalResult<()> + Send + Sync>;
190
191/// Loader callback for pinned native packages (see
192/// `GlobalEnv::pinned_native_loader`).
193pub type PinnedNativeLoader =
194    Arc<dyn Fn(&Arc<GlobalEnv>, &str, &str) -> EvalResult<bool> + Send + Sync>;
195
196/// Loader callback for native dependencies reached through a plain `require`
197/// (see `GlobalEnv::native_require_loader`).
198pub type NativeRequireLoader = Arc<dyn Fn(&Arc<GlobalEnv>, &str) -> EvalResult<bool> + Send + Sync>;
199
200/// Source of [`GlobalEnv::id`] values.
201static NEXT_GLOBAL_ENV_ID: std::sync::atomic::AtomicU64 = std::sync::atomic::AtomicU64::new(1);
202
203impl std::fmt::Debug for GlobalEnv {
204    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
205        write!(f, "GlobalEnv {{ ... }}")
206    }
207}
208
209impl GlobalEnv {
210    /// Create an empty global environment for the given execution mode.
211    ///
212    /// This is the *raw* constructor: no builtins, no bootstrap, no source
213    /// paths.  Use [`crate::Runtime::builder`] unless you are the builder.
214    pub fn new(execution_mode: ExecutionMode) -> Arc<Self> {
215        let id = NEXT_GLOBAL_ENV_ID.fetch_add(1, Ordering::Relaxed);
216        Arc::new(Self {
217            id,
218            namespaces: RwLock::new(HashMap::new()),
219            source_paths: RwLock::new(Vec::new()),
220            loaded: Mutex::new(std::collections::HashSet::new()),
221            loading: Mutex::new(HashMap::new()),
222            loading_done: Condvar::new(),
223            builtin_sources: RwLock::new(HashMap::new()),
224            gc_config: RwLock::new(None),
225            execution_mode,
226            tier_state: AtomicU8::new(TierState::TreeWalk as u8),
227            tiers: crate::tiered::tiers::Tiers::new(id),
228            async_rt: RwLock::new(None),
229            version_cache: Mutex::new(HashMap::new()),
230            deps_config: RwLock::new(None),
231            verify_commit_signatures: AtomicBool::new(false),
232            vcs: RwLock::new(crate::env::vcs::default_provider()),
233            sig_verify_cache: Mutex::new(HashSet::new()),
234            versioned_sources: RwLock::new(HashMap::new()),
235            versioned_offline: AtomicBool::new(false),
236            native_provenance: RwLock::new(HashMap::new()),
237            enforce_native_versions: AtomicBool::new(false),
238            provenance_warned: Mutex::new(HashSet::new()),
239            pinned_native_loader: RwLock::new(None),
240            native_require_loader: RwLock::new(None),
241            compiled_ns_loaders: RwLock::new(HashMap::new()),
242        })
243    }
244
245    /// Replace the source path list.
246    pub fn set_source_paths(&self, paths: Vec<std::path::PathBuf>) {
247        *self.source_paths.write().unwrap() = paths;
248    }
249
250    /// Register an embedded namespace source (called by cljrs-stdlib at startup).
251    pub fn register_builtin_source(&self, ns: &str, src: &'static str) {
252        self.builtin_sources
253            .write()
254            .unwrap()
255            .insert(Arc::from(ns), src);
256    }
257
258    /// Look up an embedded source for a namespace, if one has been registered.
259    pub fn builtin_source(&self, ns: &str) -> Option<&'static str> {
260        self.builtin_sources.read().unwrap().get(ns).copied()
261    }
262
263    /// Register a loader for an AOT-compiled namespace (called by the harness
264    /// `main` of a binary produced by `cljrs compile`).
265    pub fn register_compiled_ns_loader(&self, ns: &str, loader: CompiledNsLoader) {
266        self.compiled_ns_loaders
267            .write()
268            .unwrap()
269            .insert(Arc::from(ns), loader);
270    }
271
272    /// Look up the loader for an AOT-compiled namespace, if one is registered.
273    pub fn compiled_ns_loader(&self, ns: &str) -> Option<CompiledNsLoader> {
274        self.compiled_ns_loaders.read().unwrap().get(ns).cloned()
275    }
276
277    /// Mark a namespace as fully loaded from a file.
278    pub fn mark_loaded(&self, ns: &str) {
279        self.loaded.lock().unwrap().insert(Arc::from(ns));
280    }
281
282    /// True if the namespace has already been loaded from a file.
283    pub fn is_loaded(&self, ns: &str) -> bool {
284        self.loaded.lock().unwrap().contains(ns)
285    }
286
287    /// Set the GC configuration for automatic memory pressure management.
288    pub fn set_gc_config(&self, config: Arc<GcConfig>) {
289        *self.gc_config.write().unwrap() = Some(config);
290    }
291
292    /// Get the GC configuration, if one has been set.
293    pub fn gc_config(&self) -> Option<Arc<GcConfig>> {
294        self.gc_config.read().unwrap().clone()
295    }
296
297    /// Resolve a short alias to a full namespace name in `current_ns`.
298    pub fn resolve_alias(&self, current_ns: &str, alias: &str) -> Option<Arc<str>> {
299        let map = self.namespaces.read().unwrap();
300        let ns = map.get(current_ns)?;
301        let aliases = ns.get().aliases.lock().unwrap();
302        aliases.get(alias).cloned()
303    }
304
305    /// Resolve an auto-resolved keyword name (the text after `::`) to its
306    /// fully-qualified `ns/name` form.
307    ///
308    /// `::kw` qualifies with `current_ns` directly; `::alias/kw` looks
309    /// `alias` up in `current_ns`'s alias table (populated by `(require
310    /// '[... :as alias])`) and qualifies with the resolved namespace.
311    pub fn resolve_auto_keyword(&self, current_ns: &str, name: &str) -> Result<String, String> {
312        match name.split_once('/') {
313            Some((alias, kw_name)) => match self.resolve_alias(current_ns, alias) {
314                Some(ns) => Ok(format!("{ns}/{kw_name}")),
315                None => Err(format!(
316                    "invalid token: ::{name} (no such namespace alias: {alias})"
317                )),
318            },
319            None => Ok(format!("{current_ns}/{name}")),
320        }
321    }
322
323    /// Return the namespace with this name, creating it if it doesn't exist.
324    pub fn get_or_create_ns(&self, name: &str) -> GcPtr<Namespace> {
325        // Fast path: already exists.
326        {
327            let map = self.namespaces.read().unwrap();
328            if let Some(ns) = map.get(name) {
329                return ns.clone();
330            }
331        }
332        // Slow path: insert.
333        let mut map = self.namespaces.write().unwrap();
334        // Re-check after acquiring write lock.
335        if let Some(ns) = map.get(name) {
336            return ns.clone();
337        }
338        let ns = GcPtr::new(Namespace::new(name));
339        map.insert(Arc::from(name), ns.clone());
340        ns
341    }
342
343    /// Intern `name` with `val` in the given namespace, returning the Var.
344    pub fn intern(&self, ns_name: &str, name: Arc<str>, val: Value) -> GcPtr<Var> {
345        let ns = self.get_or_create_ns(ns_name);
346        let mut interns = ns.get().interns.lock().unwrap();
347        if let Some(var) = interns.get(&name) {
348            // Update existing var.
349            var.get().bind(val);
350            return var.clone();
351        }
352        let var = GcPtr::new(Var::new(ns_name, name.as_ref()));
353        var.get().bind(val);
354        interns.insert(name, var.clone());
355        var
356    }
357
358    /// Look up a Var in the named namespace (interns only).
359    pub fn lookup_var(&self, ns_name: &str, sym_name: &str) -> Option<GcPtr<Var>> {
360        let map = self.namespaces.read().unwrap();
361        let ns = map.get(ns_name)?;
362        let interns = ns.get().interns.lock().unwrap();
363        interns.get(sym_name).cloned()
364    }
365
366    /// Look up a value in `ns_name`: checks interns then refers.
367    /// Routes through the dynamic binding stack so `binding` overrides work.
368    pub fn lookup_in_ns(&self, ns_name: &str, sym_name: &str) -> Option<Value> {
369        let map = self.namespaces.read().unwrap();
370        let ns = map.get(ns_name)?;
371        let ns_ref = ns.get();
372        // Check interns first.
373        {
374            let interns = ns_ref.interns.lock().unwrap();
375            if let Some(var) = interns.get(sym_name) {
376                return crate::env::dynamics::deref_var(var);
377            }
378        }
379        // Then refers.
380        {
381            let refers = ns_ref.refers.lock().unwrap();
382            if let Some(var) = refers.get(sym_name) {
383                return crate::env::dynamics::deref_var(var);
384            }
385        }
386        None
387    }
388
389    /// Look up the raw Var (not its value) in `ns_name`: interns then refers.
390    pub fn lookup_var_in_ns(&self, ns_name: &str, sym_name: &str) -> Option<GcPtr<Var>> {
391        let map = self.namespaces.read().unwrap();
392        let ns = map.get(ns_name)?;
393        let ns_ref = ns.get();
394        {
395            let interns = ns_ref.interns.lock().unwrap();
396            if let Some(var) = interns.get(sym_name) {
397                return Some(var.clone());
398            }
399        }
400        {
401            let refers = ns_ref.refers.lock().unwrap();
402            if let Some(var) = refers.get(sym_name) {
403                return Some(var.clone());
404            }
405        }
406        None
407    }
408
409    /// Copy all interns from `src_ns` into `dst_ns` as refers.
410    ///
411    /// This is the *explicit* refer — `(:require [x :refer :all])` — so it is
412    /// never narrowed by `dst_ns`'s `(:refer-clojure ...)` filter, matching
413    /// `clojure.core/refer`: naming a namespace explicitly re-maps even names
414    /// an earlier `refer-clojure` left out.  The automatic core refer every
415    /// namespace starts with goes through [`GlobalEnv::refer_core`] instead.
416    pub fn refer_all(&self, dst_ns: &str, src_ns: &str) {
417        let map = self.namespaces.read().unwrap();
418        let src = match map.get(src_ns) {
419            Some(ns) => ns.clone(),
420            None => return,
421        };
422        let dst = match map.get(dst_ns) {
423            Some(ns) => ns.clone(),
424            None => return,
425        };
426        let src_interns = src.get().interns.lock().unwrap();
427        let mut dst_refers = dst.get().refers.lock().unwrap();
428        for (name, var) in src_interns.iter() {
429            dst_refers.insert(name.clone(), var.clone());
430        }
431    }
432
433    /// Apply the automatic `clojure.core` refer that every namespace starts
434    /// with, narrowed by `dst_ns`'s `(:refer-clojure ...)` filter.
435    pub fn refer_core(&self, dst_ns: &str) {
436        self.refer_core_impl(dst_ns, false);
437    }
438
439    /// `replace`: drop the refers `dst_ns` already inherited from
440    /// `clojure.core` before re-referring, under the same lock — so installing
441    /// a filter after the namespace was pre-referred neither leaves stale
442    /// names behind nor exposes a window where core is only half-referred.
443    fn refer_core_impl(&self, dst_ns: &str, replace: bool) {
444        let map = self.namespaces.read().unwrap();
445        let src = match map.get("clojure.core") {
446            Some(ns) => ns.clone(),
447            None => return,
448        };
449        let dst = match map.get(dst_ns) {
450            Some(ns) => ns.clone(),
451            None => return,
452        };
453        // Lock order is filter → src interns → dst refers throughout.
454        let filter = dst.get().refer_clojure_filter.lock().unwrap();
455        let src_interns = src.get().interns.lock().unwrap();
456        let mut dst_refers = dst.get().refers.lock().unwrap();
457        if replace {
458            dst_refers.retain(|_, var| var.get().namespace.as_ref() != "clojure.core");
459        }
460        match filter.as_ref() {
461            Some(f) => {
462                for (name, var) in src_interns.iter() {
463                    if let Some(local) = f.local_name(name) {
464                        dst_refers.insert(local, var.clone());
465                    }
466                }
467            }
468            None => {
469                for (name, var) in src_interns.iter() {
470                    dst_refers.insert(name.clone(), var.clone());
471                }
472            }
473        }
474    }
475
476    /// Install `dst_ns`'s `(:refer-clojure ...)` filter (`None` removes any
477    /// previous one) and re-apply the automatic `clojure.core` refer under it.
478    ///
479    /// Refers already inherited from `clojure.core` are dropped first, so a
480    /// filter set after the namespace was pre-referred — the loader refers core
481    /// before it reads the file, and `ns` itself refers core before it reaches
482    /// the clause — still takes effect.
483    pub fn set_refer_clojure_filter(
484        &self,
485        dst_ns: &str,
486        filter: Option<ReferClojureFilter>,
487    ) -> Result<(), String> {
488        if let Some(f) = &filter {
489            self.validate_refer_clojure_filter(f)?;
490        }
491        let dst = self.get_or_create_ns(dst_ns);
492        {
493            let mut slot = dst.get().refer_clojure_filter.lock().unwrap();
494            // Nothing to install and nothing to undo: leave the refers alone.
495            if slot.is_none() && filter.is_none() {
496                return Ok(());
497            }
498            *slot = filter;
499        }
500        self.refer_core_impl(dst_ns, true);
501        Ok(())
502    }
503
504    /// Check a `(:refer-clojure ...)` filter against the names `clojure.core`
505    /// actually publishes, so a typo fails at the `ns` form rather than as an
506    /// unbound symbol somewhere further down the file.
507    ///
508    /// `:only` and `:rename` name specific vars and must resolve; `:exclude` is
509    /// subtractive and stays permissive (excluding a name core does not have is
510    /// harmless, and lets a file stay portable across core versions).  Clojure
511    /// validates `:only` the same way but ignores an unresolvable `:rename`
512    /// key, since it only consults the rename map for names already in its
513    /// to-do list.
514    ///
515    /// Two names landing on the same local name is an error rather than a coin
516    /// flip: with `:rename {inc str}` both `inc` and core's own `str` want the
517    /// name `str`.  Clojure warns and lets whichever one its intern table
518    /// yields last win; picking a winner by hash order here would make the
519    /// choice unstable from run to run.
520    fn validate_refer_clojure_filter(&self, filter: &ReferClojureFilter) -> Result<(), String> {
521        let map = self.namespaces.read().unwrap();
522        let Some(core) = map.get("clojure.core").cloned() else {
523            return Ok(());
524        };
525        drop(map);
526        let interns = core.get().interns.lock().unwrap();
527        // A runtime built without the core bootstrap has nothing to check
528        // against; do not fail every name.
529        if interns.is_empty() {
530            return Ok(());
531        }
532
533        for (opt, names) in [
534            ("only", filter.only.iter().flatten().collect::<Vec<_>>()),
535            ("rename", filter.rename.keys().collect::<Vec<_>>()),
536        ] {
537            let mut unknown: Vec<&str> = names
538                .into_iter()
539                .filter(|n| !interns.contains_key(*n))
540                .map(|n| n.as_ref())
541                .collect();
542            if !unknown.is_empty() {
543                unknown.sort_unstable();
544                return Err(format!(
545                    ":refer-clojure :{opt} names {}, which clojure.core does not define",
546                    unknown.join(", ")
547                ));
548            }
549        }
550
551        // local name → the core name referred under it.
552        let mut taken: HashMap<Arc<str>, Arc<str>> = HashMap::new();
553        let mut conflicts: Vec<(Arc<str>, Arc<str>, Arc<str>)> = Vec::new();
554        for name in interns.keys() {
555            let Some(local) = filter.local_name(name) else {
556                continue;
557            };
558            if let Some(prev) = taken.insert(local.clone(), name.clone()) {
559                let (a, b) = if prev.as_ref() <= name.as_ref() {
560                    (prev, name.clone())
561                } else {
562                    (name.clone(), prev)
563                };
564                conflicts.push((local, a, b));
565            }
566        }
567        if !conflicts.is_empty() {
568            conflicts.sort_unstable();
569            let (local, a, b) = &conflicts[0];
570            return Err(format!(
571                ":refer-clojure would refer both {a} and {b} as {local}; \
572                 rename or exclude one of them"
573            ));
574        }
575        Ok(())
576    }
577
578    /// Names `ns_name` resolves to something other than `clojure.core`'s var
579    /// of the same name — what the IR lowerer must not inline as a builtin.
580    ///
581    /// Three things put a name in here: a `def` in `ns_name` itself, a refer
582    /// of that name from another namespace (or of a *different* core name
583    /// under it, via `:rename`), and a `(:refer-clojure ...)` filter that
584    /// leaves the name out of the automatic core refer.  In each case an
585    /// unqualified call resolves to something that is not `clojure.core/name`
586    /// — or to nothing at all — and the tree-walking interpreter calls that,
587    /// so the lowered call site has to as well (issue #337).
588    ///
589    /// A name absent from the namespace's tables is *not* reported: with no
590    /// evidence to the contrary the lowerer keeps assuming core, which is what
591    /// the automatic core refer makes true for the overwhelming majority of
592    /// call sites (and keeps the lowerer's synthetic names, e.g. `case=`,
593    /// working in namespaces whose refers are not populated yet).
594    pub fn core_shadowed_names(&self, ns_name: &str) -> HashSet<Arc<str>> {
595        let mut out: HashSet<Arc<str>> = HashSet::new();
596        let (ns, core) = {
597            let map = self.namespaces.read().unwrap();
598            match map.get(ns_name) {
599                Some(ns) => (ns.clone(), map.get("clojure.core").cloned()),
600                None => return out,
601            }
602        };
603        let ns_ref = ns.get();
604
605        // A var bound here is core's only if it *is* core's var under its own
606        // name; core's own namespace passes this trivially for its interns.
607        let shadows = |name: &Arc<str>, var: &GcPtr<Var>| {
608            let v = var.get();
609            v.namespace.as_ref() != "clojure.core" || v.name != *name
610        };
611        for table in [&ns_ref.interns, &ns_ref.refers] {
612            let entries = table.lock().unwrap();
613            for (name, var) in entries.iter() {
614                if shadows(name, var) {
615                    out.insert(name.clone());
616                }
617            }
618        }
619
620        // Names the `(:refer-clojure ...)` filter kept out of the automatic
621        // refer: nothing binds them here, so they are not core's either.
622        // Lock order is filter → core interns, as in `refer_core_impl`.
623        let filter = ns_ref.refer_clojure_filter.lock().unwrap();
624        if let (Some(filter), Some(core)) = (filter.as_ref(), core) {
625            let interns = core.get().interns.lock().unwrap();
626            for name in interns.keys() {
627                if filter.local_name(name).as_ref() != Some(name) {
628                    out.insert(name.clone());
629                }
630            }
631        }
632        out
633    }
634
635    /// Copy selected interns from `src_ns` into `dst_ns` as refers.
636    pub fn refer_named(&self, dst_ns: &str, src_ns: &str, names: &[Arc<str>]) {
637        let map = self.namespaces.read().unwrap();
638        let src = match map.get(src_ns) {
639            Some(ns) => ns.clone(),
640            None => return,
641        };
642        let dst = match map.get(dst_ns) {
643            Some(ns) => ns.clone(),
644            None => return,
645        };
646        let src_interns = src.get().interns.lock().unwrap();
647        let mut dst_refers = dst.get().refers.lock().unwrap();
648        for name in names {
649            if let Some(var) = src_interns.get(name) {
650                // Use insert (not or_insert_with) so that an explicit
651                // `require :refer [name]` always overrides a previous refer
652                // (e.g. one inherited from clojure.core via refer-all).
653                // clojure.core.async's `into` intentionally shadows clojure.core/into;
654                // or_insert_with would silently drop the override.
655                dst_refers.insert(name.clone(), var.clone());
656            }
657        }
658    }
659
660    /// Register `alias` → `full_ns` in `current_ns`'s alias table.
661    pub fn add_alias(&self, current_ns: &str, alias: &str, full_ns: &str) {
662        let ns_ptr = self.get_or_create_ns(current_ns);
663        let mut aliases = ns_ptr.get().aliases.lock().unwrap();
664        aliases.insert(Arc::from(alias), Arc::from(full_ns));
665    }
666
667    /// Process-unique identity of this runtime instance.
668    #[inline(always)]
669    pub fn id(&self) -> u64 {
670        self.id
671    }
672
673    /// This runtime's Tier-1/Tier-2 state.
674    #[inline(always)]
675    pub fn tiers(&self) -> &Arc<crate::tiered::tiers::Tiers> {
676        &self.tiers
677    }
678
679    /// This runtime's cache of lowered IR.
680    #[inline(always)]
681    pub fn ir_cache(&self) -> &crate::tiered::ir_cache::IrCache {
682        self.tiers.ir_cache()
683    }
684
685    /// This runtime's JIT counters, profiles, and native-code tables.
686    #[inline(always)]
687    pub fn jit(&self) -> &crate::tiered::jit_state::JitState {
688        self.tiers.jit()
689    }
690
691    /// The JIT compiler attached to this runtime, if any.
692    ///
693    /// `None` when no JIT is linked or installed; callers then keep to the
694    /// interpreter tiers.  Installed by `cljrs_compiler::jit::install`.
695    #[inline(always)]
696    pub fn jit_backend(&self) -> Option<Arc<dyn crate::tiered::backend::JitBackend>> {
697        self.tiers.jit().backend().cloned()
698    }
699
700    // ── Execution mode and tier state ────────────────────────────────────
701
702    /// How this runtime executes function calls.
703    #[inline(always)]
704    pub fn execution_mode(&self) -> ExecutionMode {
705        self.execution_mode
706    }
707
708    /// Which tiers are live right now.
709    #[inline(always)]
710    pub fn tier_state(&self) -> TierState {
711        TierState::from_u8(self.tier_state.load(Ordering::Acquire))
712    }
713
714    /// Raise the live tier state.  Called once by the runtime builder after
715    /// the bootstrap completes; lowering the tier is not supported, so a
716    /// request below the current state is ignored.
717    pub fn set_tier_state(&self, tier: TierState) {
718        let _ = self.tier_state.fetch_max(tier as u8, Ordering::AcqRel);
719    }
720
721    /// True when IR may be lowered, cached, and interpreted.  This is the
722    /// gate the old `compiler_ready` flag served.
723    #[inline(always)]
724    pub fn ir_enabled(&self) -> bool {
725        self.tier_state().ir_enabled()
726    }
727
728    // ── Evaluation entry points ──────────────────────────────────────────
729
730    /// Evaluate `form` in `env`.
731    #[inline(always)]
732    pub fn eval(&self, form: &Form, env: &mut Env) -> EvalResult {
733        crate::interp::eval::eval(form, env)
734    }
735
736    /// Call a Clojure function, taking the path this runtime's
737    /// [`ExecutionMode`] selects.
738    ///
739    /// This is the single function-call dispatch point: tree walk, tier-1 IR,
740    /// and JIT-native execution are all reached from here.
741    #[inline(always)]
742    pub fn call_cljrs_fn(&self, func: &CljxFn, args: &[Value], env: &mut Env) -> EvalResult {
743        match self.execution_mode {
744            ExecutionMode::TreeWalk => crate::interp::apply::call_cljrs_fn(func, args, env),
745            ExecutionMode::Tiered | ExecutionMode::TieredNoJit => {
746                crate::tiered::apply::call_cljrs_fn(func, args, env)
747            }
748            ExecutionMode::NoGcTransaction => crate::env::depth::call_cljrs_fn(func, args, env),
749        }
750    }
751
752    /// Notify the active tier that a new `fn*` was defined.
753    ///
754    /// In a tiered runtime with IR enabled this eagerly lowers the function
755    /// (when eager lowering is on); in every other mode it does nothing.
756    #[inline(always)]
757    pub fn on_fn_defined(&self, f: &CljxFn, env: &mut Env) {
758        if self.execution_mode.is_tiered() && self.ir_enabled() {
759            crate::tiered::ir_interp::eager_lower_fn(f, env);
760        }
761    }
762
763    /// Install an async runtime. Called once by `cljrs_async::init`.
764    /// Subsequent calls are silently ignored (first writer wins).
765    pub fn set_async_runtime(&self, rt: Arc<dyn AsyncRuntime>) {
766        let mut guard = self.async_rt.write().unwrap();
767        if guard.is_none() {
768            *guard = Some(rt);
769        }
770    }
771
772    /// Return the async runtime, if one has been registered.
773    pub fn async_runtime(&self) -> Option<Arc<dyn AsyncRuntime>> {
774        self.async_rt.read().unwrap().clone()
775    }
776
777    /// Return `(source_file, git_repo_root)` for the named namespace, if
778    /// both have been populated by the loader.
779    pub fn get_ns_git_context(&self, ns_name: &str) -> Option<(Arc<str>, Arc<str>)> {
780        let map = self.namespaces.read().unwrap();
781        let ns = map.get(ns_name)?;
782        let ns_ref = ns.get();
783        let file = ns_ref.source_file.lock().unwrap().clone()?;
784        let repo = ns_ref.git_repo_root.lock().unwrap().clone()?;
785        Some((file, repo))
786    }
787
788    /// Store a resolved versioned value in the cache.
789    /// Key: `"<ns>/<name>@<commit>"`.
790    pub fn cache_versioned(&self, ns: &str, name: &str, commit: &str, val: Value) {
791        let key: Arc<str> = Arc::from(format!("{ns}/{name}@{commit}"));
792        self.version_cache.lock().unwrap().insert(key, val);
793    }
794
795    /// Retrieve a previously resolved versioned value, if cached.
796    pub fn get_cached_versioned(&self, ns: &str, name: &str, commit: &str) -> Option<Value> {
797        let key = format!("{ns}/{name}@{commit}");
798        self.version_cache
799            .lock()
800            .unwrap()
801            .get(key.as_str())
802            .cloned()
803    }
804
805    /// Mark namespace `name@commit` as loaded in the standard loaded set.
806    pub fn cache_versioned_ns(&self, ns: &str, commit: &str) {
807        let key: Arc<str> = Arc::from(format!("{ns}@{commit}"));
808        self.version_cache.lock().unwrap().insert(key, Value::Nil);
809    }
810
811    /// Record the source text of a versioned namespace fetched from git.
812    /// Key: `"<ns>@<commit>"`.  Consumed by the AOT compiler for embedding.
813    pub fn record_versioned_source(&self, versioned_ns: &str, src: &str) {
814        self.versioned_sources
815            .write()
816            .unwrap()
817            .insert(Arc::from(versioned_ns), Arc::from(src));
818    }
819
820    /// Snapshot of all versioned sources fetched this session, sorted by key.
821    pub fn versioned_sources_snapshot(&self) -> Vec<(Arc<str>, Arc<str>)> {
822        let map = self.versioned_sources.read().unwrap();
823        let mut entries: Vec<_> = map.iter().map(|(k, v)| (k.clone(), v.clone())).collect();
824        entries.sort_by(|a, b| a.0.cmp(&b.0));
825        entries
826    }
827
828    /// Restrict versioned-namespace resolution to embedded builtin sources
829    /// (no git).  Called by AOT harness binaries, which embed every pinned
830    /// source discovered at compile time.
831    pub fn set_versioned_offline(&self, offline: bool) {
832        self.versioned_offline.store(offline, Ordering::Relaxed);
833    }
834
835    /// True when versioned namespaces may only come from embedded sources.
836    pub fn versioned_offline(&self) -> bool {
837        self.versioned_offline.load(Ordering::Relaxed)
838    }
839
840    /// Record the git commit a native (Rust-backed) package was built from.
841    /// Called at registration time (`Registry::set_provenance` or the
842    /// `register_provenance!` inventory entry in cljrs-interop).
843    pub fn set_native_provenance(&self, ns: &str, commit: &str) {
844        self.native_provenance
845            .write()
846            .unwrap()
847            .insert(Arc::from(ns), Arc::from(commit));
848    }
849
850    /// The recorded provenance commit for a native package's namespace.
851    pub fn native_provenance_for(&self, ns: &str) -> Option<Arc<str>> {
852        self.native_provenance.read().unwrap().get(ns).cloned()
853    }
854
855    /// Make pinned-native provenance mismatches hard errors.
856    pub fn set_enforce_native_versions(&self, enforce: bool) {
857        self.enforce_native_versions
858            .store(enforce, Ordering::Relaxed);
859    }
860
861    /// True when pinned-native provenance mismatches are errors.
862    pub fn enforce_native_versions(&self) -> bool {
863        self.enforce_native_versions.load(Ordering::Relaxed)
864    }
865
866    /// Install the pinned-native package loader (called once by
867    /// `cljrs::native::pinned::install`; first writer wins).
868    pub fn set_pinned_native_loader(&self, loader: PinnedNativeLoader) {
869        let mut guard = self.pinned_native_loader.write().unwrap();
870        if guard.is_none() {
871            *guard = Some(loader);
872        }
873    }
874
875    /// Install the native-dependency `require` loader (called once by
876    /// `cljrs::native::pinned::install`; first writer wins).
877    pub fn set_native_require_loader(&self, loader: NativeRequireLoader) {
878        let mut guard = self.native_require_loader.write().unwrap();
879        if guard.is_none() {
880            *guard = Some(loader);
881        }
882    }
883
884    /// The installed VCS backend, or `None` when this build has none (see
885    /// [`crate::env::vcs`]).  Callers must degrade gracefully: "no provider"
886    /// means "this source file is not in a git repository".
887    pub fn vcs(&self) -> Option<Arc<dyn crate::env::vcs::VcsProvider>> {
888        self.vcs.read().unwrap().clone()
889    }
890
891    /// Replace the VCS backend.  Lets an embedder that built without the
892    /// `deps` feature supply its own git implementation, or a sandboxed host
893    /// remove the default one (`None`) so no versioned resolution can reach
894    /// the filesystem's git history.
895    ///
896    /// Drops every cached signature verdict: those were reached by the
897    /// outgoing provider, against its trust set and its view of the
898    /// repository, and say nothing about what the incoming one would decide
899    /// for the same `(repo, commit)`.  Keeping them would let a permissive
900    /// provider launder an approval for source a later provider serves.
901    pub fn set_vcs_provider(&self, provider: Option<Arc<dyn crate::env::vcs::VcsProvider>>) {
902        // Neither this nor `check_commit_signature` ever holds the `vcs` and
903        // `sig_verify_cache` locks at the same time, and the two reach for
904        // them in opposite orders — keep it that way, or the pair becomes a
905        // lock-order inversion.
906        *self.vcs.write().unwrap() = provider;
907        self.invalidate_signature_cache();
908    }
909
910    /// Forget every cached signature verdict, so the next
911    /// [`check_commit_signature`](Self::check_commit_signature) re-asks the
912    /// current provider.  Called whenever the thing that produced those
913    /// verdicts changes: the provider itself, or its trusted-key set.
914    pub fn invalidate_signature_cache(&self) {
915        self.sig_verify_cache.lock().unwrap().clear();
916    }
917
918    /// If `:verify-commit-signatures` is enabled, verify that `commit` inside
919    /// `repo_root` carries a valid GPG or SSH signature.
920    ///
921    /// Returns `Ok(())` immediately when the feature is off.  On the happy
922    /// path the result is cached per `(repo_root, commit)` so each commit is
923    /// only verified once per session.  On failure returns
924    /// `EvalError::CommitSignatureVerificationFailed`.
925    ///
926    /// If verification is demanded but this build has no VCS provider, the
927    /// check fails: silently accepting an unverifiable commit would defeat the
928    /// flag the user explicitly turned on.
929    pub fn check_commit_signature(&self, repo_root: &str, commit: &str) -> EvalResult<()> {
930        if !self.verify_commit_signatures.load(Ordering::Relaxed) {
931            return Ok(());
932        }
933        let key = (Arc::<str>::from(repo_root), Arc::<str>::from(commit));
934        if self.sig_verify_cache.lock().unwrap().contains(&key) {
935            return Ok(());
936        }
937        let Some(vcs) = self.vcs() else {
938            return Err(crate::env::error::EvalError::Runtime(format!(
939                "commit-signature verification is enabled, but this build has no VCS \
940                 provider to verify commit {commit} with (cljrs-runtime built without \
941                 the `deps` feature)"
942            )));
943        };
944        vcs.verify_commit_signature(std::path::Path::new(repo_root), commit)
945            .map_err(|e| match e {
946                crate::env::vcs::SignatureFailure::Untrusted { commit, reason } => {
947                    crate::env::error::EvalError::CommitSignatureVerificationFailed {
948                        commit,
949                        reason,
950                    }
951                }
952                crate::env::vcs::SignatureFailure::Error(msg) => {
953                    crate::env::error::EvalError::Runtime(msg)
954                }
955            })?;
956        self.sig_verify_cache.lock().unwrap().insert(key);
957        Ok(())
958    }
959
960    /// Build the trusted-signer key set from a parsed `cljrs.edn` config and
961    /// install it, so subsequent `check_commit_signature` calls verify against
962    /// it.  Inline keys are parsed directly; `File` entries are read from disk.
963    /// Returns the number of keys loaded; warns (to stderr) on any key that
964    /// fails to load rather than aborting.  Returns 0 when this build has no
965    /// VCS provider, since there is nothing that could consume the keys.
966    ///
967    /// Replacing the trust set invalidates the signature cache for the same
968    /// reason replacing the provider does: a verdict reached under the old
969    /// keys is not a verdict under the new ones.  (In the normal flow this
970    /// runs at session start, before anything has been verified.)
971    pub fn load_trusted_signers(&self, config: &cljrs_project::config::DepsConfig) -> usize {
972        let Some(vcs) = self.vcs() else {
973            return 0;
974        };
975        let loaded = vcs.load_trusted_signers(&config.trusted_signers);
976        self.invalidate_signature_cache();
977        loaded
978    }
979}
980
981// ── Env ───────────────────────────────────────────────────────────────────────
982
983/// The full execution environment: a stack of local frames plus the global env.
984pub struct Env {
985    pub frames: Vec<Frame>,
986    pub current_ns: Arc<str>,
987    pub globals: Arc<GlobalEnv>,
988    /// When set, unversioned same-namespace symbol lookups implicitly resolve
989    /// at this commit hash instead of HEAD.  Set by the versioned resolver when
990    /// evaluating a function body fetched from git history.
991    pub versioned_eval_commit: Option<Arc<str>>,
992    /// True when evaluating the body of an `^:async` function.
993    /// Set by `cljrs-async`; allows the `await` special form to know whether
994    /// to yield (async context) or block the OS thread (sync context).
995    pub is_async: bool,
996}
997
998impl Env {
999    pub fn new(globals: Arc<GlobalEnv>, ns: &str) -> Self {
1000        Self {
1001            frames: Vec::new(),
1002            current_ns: Arc::from(ns),
1003            globals,
1004            versioned_eval_commit: None,
1005            is_async: false,
1006        }
1007    }
1008
1009    /// Create an Env for evaluating source at a specific commit.
1010    pub fn new_versioned(globals: Arc<GlobalEnv>, ns: &str, commit: &str) -> Self {
1011        Self {
1012            versioned_eval_commit: Some(Arc::from(commit)),
1013            ..Self::new(globals, ns)
1014        }
1015    }
1016
1017    /// Create an Env pre-loaded with a function's closed-over bindings.
1018    pub fn with_closure(globals: Arc<GlobalEnv>, ns: &str, f: &CljxFn) -> Self {
1019        let mut env = Self::new(globals, ns);
1020        if !f.closed_over_names.is_empty() {
1021            env.push_frame();
1022            for (name, val) in f.closed_over_names.iter().zip(f.closed_over_vals.iter()) {
1023                env.bind(name.clone(), val.clone());
1024            }
1025        }
1026        env
1027    }
1028
1029    pub fn push_frame(&mut self) {
1030        self.frames.push(Frame::new());
1031    }
1032
1033    pub fn pop_frame(&mut self) {
1034        self.frames.pop();
1035    }
1036
1037    /// Bind `name` to `val` in the top frame.
1038    pub fn bind(&mut self, name: Arc<str>, val: Value) {
1039        if let Some(frame) = self.frames.last_mut() {
1040            frame.bind(name, val);
1041        }
1042        // If there are no frames, the binding is silently dropped.
1043        // Callers must push a frame first.
1044    }
1045
1046    /// Look up `name`: local frames (innermost first), then the current namespace.
1047    pub fn lookup(&self, name: &str) -> Option<Value> {
1048        tracing::trace!(target: "env", "lookup {} in {} frames", name, self.frames.len());
1049        for frame in self.frames.iter().rev() {
1050            if let Some(v) = frame.lookup(name) {
1051                return Some(v.clone());
1052            }
1053        }
1054        self.globals.lookup_in_ns(&self.current_ns, name)
1055    }
1056
1057    /// Look up `name` in local frames only — does **not** fall back to the
1058    /// global namespace.  Used by the versioned resolver to check for local
1059    /// bindings before applying commit inheritance.
1060    pub fn lookup_local_frames(&self, name: &str) -> Option<Value> {
1061        for frame in self.frames.iter().rev() {
1062            if let Some(v) = frame.lookup(name) {
1063                return Some(v.clone());
1064            }
1065        }
1066        None
1067    }
1068
1069    /// Look up the Var object for `name` in the current namespace.
1070    pub fn lookup_var(&self, name: &str) -> Option<GcPtr<Var>> {
1071        self.globals.lookup_var_in_ns(&self.current_ns, name)
1072    }
1073
1074    /// Collect all current local bindings (all frames, innermost last).
1075    /// Used for closure capture.
1076    pub fn all_local_bindings(&self) -> (Vec<Arc<str>>, Vec<Value>) {
1077        let mut names = Vec::new();
1078        let mut vals = Vec::new();
1079        // Outermost first so inner frames override on lookup.
1080        for frame in &self.frames {
1081            for (n, v) in &frame.bindings {
1082                names.push(n.clone());
1083                vals.push(v.clone());
1084            }
1085        }
1086        (names, vals)
1087    }
1088
1089    /// Create a child Env for closure capture (same globals, same ns, captures locals).
1090    pub fn child(&self) -> Self {
1091        let (names, vals) = self.all_local_bindings();
1092        let mut child = Self::new(self.globals.clone(), &self.current_ns);
1093        child.is_async = self.is_async;
1094        if !names.is_empty() {
1095            child.push_frame();
1096            for (n, v) in names.into_iter().zip(vals) {
1097                child.bind(n, v);
1098            }
1099        }
1100        child
1101    }
1102
1103    #[inline(always)]
1104    pub fn eval(&mut self, form: &Form) -> EvalResult {
1105        let globals = self.globals.clone();
1106        globals.eval(form, self)
1107    }
1108
1109    #[inline(always)]
1110    pub fn call_cljrs_fn(&mut self, func: &CljxFn, args: &[Value]) -> EvalResult {
1111        let globals = self.globals.clone();
1112        globals.call_cljrs_fn(func, args, self)
1113    }
1114
1115    #[inline(always)]
1116    pub fn on_fn_defined(&mut self, func: &CljxFn) {
1117        let globals = self.globals.clone();
1118        globals.on_fn_defined(func, self);
1119    }
1120}