Skip to main content

sui_eval/
value.rs

1//! Nix value types and environments.
2//!
3//! The evaluator is single-threaded: `Env` and `NixAttrs` contain
4//! `Rc<UnsafeCell<ThunkRepr>>` thunks.  All shared pointers use `Rc`
5//! (not `Arc`) because the values are never sent across threads.
6
7use std::cell::{Cell, OnceCell, RefCell, UnsafeCell};
8
9use std::fmt;
10pub use std::rc::Rc;
11
12use rustc_hash::FxBuildHasher;
13use smallvec::SmallVec;
14pub use smol_str::SmolStr;
15
16use rowan::ast::AstNode;
17
18use sui_intern::Symbol;
19
20/// Type alias for the persistent hash map used by `NixAttrs` and `Env`.
21///
22/// Uses `FxBuildHasher` (fast multiplication-based hash) instead of the
23/// default `RandomState`. This is optimal for `Symbol(u32)` keys where
24/// the hash is a single multiply-shift — no SipHash overhead.
25pub type FxHashMap<K, V> = im_rc::HashMap<K, V, FxBuildHasher>;
26
27/// Compact attrset map — a real `hashbrown` (std) `HashMap` with `FxBuildHasher`.
28///
29/// Used ONLY for `NixAttrs` (attribute sets), which are immutable-after-
30/// construction. Unlike `FxHashMap` (the persistent `im_rc` HAMT, retained for
31/// `Env` where `child()`/scope-push relies on O(1) structural sharing), this is a
32/// flat open-addressing table with ~0.875 load factor and NO branch-node
33/// allocations — a symbolicated dhat profile proved the `im_rc` HAMT branch nodes
34/// dominate eval heap, and the attrset slice is the safe one to compact.
35///
36/// BYTE-NEUTRAL: attrset observation order (which feeds drvPath hashing) comes
37/// from `NixAttrs::sorted_entries()` — it resolves each `Symbol` to its `String`
38/// and string-sorts on observation — NOT from this map's internal iteration
39/// order. Both `im_rc::HashMap` and `std::HashMap` are unordered, so swapping the
40/// implementation cannot change any observed order → drvPaths are unchanged.
41pub type AttrsMap<K, V> = std::collections::HashMap<K, V, FxBuildHasher>;
42
43/// Env-gated LIVE-OBJECT CENSUS.
44///
45/// A permanent, zero-cost-when-off diagnostic answering the question:
46/// when sui's eval peak is ~2× nix's, is the overhead (a) cyclic/lingering
47/// producer garbage sui retains, or (b) a uniform per-object representation
48/// overhead? These need different fixes, so we MEASURE.
49///
50/// Gated behind `SUI_LIVE_CENSUS=1`. The atomics are always compiled but the
51/// `_MADE`/`_LIVE` bookkeeping and the RSS/dump thread only run when enabled.
52/// All counters use `Relaxed` — we want a cheap high-water snapshot, not a
53/// linearizable total.
54///
55/// `_MADE` + `_LIVE` are incremented in the INNER heap type's constructor;
56/// `_LIVE` is decremented in the inner type's `Drop` so it fires exactly once
57/// when the last `Rc` drops. Counters live on the inner heap types
58/// (`NixAttrs`, `ThunkInner`, `EnvInner`, `NixString`, the list `Vec`) so we
59/// count distinct heap allocations, not `Rc` clones.
60pub mod census {
61    use std::sync::atomic::{AtomicI64, Ordering::Relaxed};
62    use std::sync::OnceLock;
63
64    pub static ATTRS_LIVE: AtomicI64 = AtomicI64::new(0);
65    pub static ATTRS_MADE: AtomicI64 = AtomicI64::new(0);
66    pub static THUNK_LIVE: AtomicI64 = AtomicI64::new(0);
67    pub static THUNK_MADE: AtomicI64 = AtomicI64::new(0);
68    pub static THUNK_EVALUATED: AtomicI64 = AtomicI64::new(0);
69    pub static ENV_LIVE: AtomicI64 = AtomicI64::new(0);
70    pub static ENV_MADE: AtomicI64 = AtomicI64::new(0);
71    pub static NIXSTR_LIVE: AtomicI64 = AtomicI64::new(0);
72    pub static NIXSTR_MADE: AtomicI64 = AtomicI64::new(0);
73    pub static LIST_LIVE: AtomicI64 = AtomicI64::new(0);
74    pub static LIST_MADE: AtomicI64 = AtomicI64::new(0);
75
76    /// True iff `SUI_LIVE_CENSUS=1`. Cached — read once.
77    #[inline]
78    pub fn enabled() -> bool {
79        static ON: OnceLock<bool> = OnceLock::new();
80        *ON.get_or_init(|| std::env::var("SUI_LIVE_CENSUS").as_deref() == Ok("1"))
81    }
82
83    #[inline(always)]
84    pub fn made(made: &AtomicI64, live: &AtomicI64) {
85        if enabled() {
86            made.fetch_add(1, Relaxed);
87            live.fetch_add(1, Relaxed);
88        }
89    }
90
91    #[inline(always)]
92    pub fn dropped(live: &AtomicI64) {
93        if enabled() {
94            live.fetch_sub(1, Relaxed);
95        }
96    }
97
98    #[inline(always)]
99    pub fn evaluated() {
100        if enabled() {
101            THUNK_EVALUATED.fetch_add(1, Relaxed);
102        }
103    }
104
105    /// Resident set size of this process, in bytes (macOS + Linux).
106    pub fn rss_bytes() -> u64 {
107        #[cfg(target_os = "macos")]
108        unsafe {
109            let mut info: libc::mach_task_basic_info = std::mem::zeroed();
110            let mut count = (std::mem::size_of::<libc::mach_task_basic_info>()
111                / std::mem::size_of::<libc::natural_t>()) as libc::mach_msg_type_number_t;
112            let kr = libc::task_info(
113                libc::mach_task_self(),
114                libc::MACH_TASK_BASIC_INFO,
115                std::ptr::addr_of_mut!(info).cast(),
116                &mut count,
117            );
118            if kr == libc::KERN_SUCCESS {
119                return info.resident_size;
120            }
121            0
122        }
123        #[cfg(not(target_os = "macos"))]
124        {
125            std::fs::read_to_string("/proc/self/statm")
126                .ok()
127                .and_then(|s| s.split_whitespace().nth(1).map(String::from))
128                .and_then(|pages| pages.parse::<u64>().ok())
129                .map(|pages| pages * 4096)
130                .unwrap_or(0)
131        }
132    }
133
134    /// Print all live/made counts + RSS to stderr, tagged.
135    ///
136    /// No-op unless `SUI_LIVE_CENSUS=1`. The counters only accumulate when the
137    /// census is enabled, so dumping while disabled emits an all-zeros
138    /// `[census exit] …` line to stderr — pure noise that pollutes any tool
139    /// parsing sui's stderr. Concretely it regressed the `derivation show→add`
140    /// ATerm round-trip parity row: that probe collects every non-`#` stderr
141    /// line from `derivation add` as the round-tripped ATerm, and the
142    /// exit-guard's unconditional dump appended the census line to it. Gating
143    /// here makes census-pollution-when-disabled unrepresentable at EVERY call
144    /// site (the process-exit guard AND the periodic poller), not just the one
145    /// that regressed.
146    pub fn dump(tag: &str) {
147        if !enabled() {
148            return;
149        }
150        let rss = rss_bytes();
151        eprintln!(
152            "[census {tag}] rss={rss_mb:.1}MB \
153attrs_live={al} attrs_made={am} \
154thunk_live={tl} thunk_made={tm} thunk_eval={te} \
155env_live={el} env_made={em} \
156nixstr_live={sl} nixstr_made={sm} \
157list_live={ll} list_made={lm}",
158            rss_mb = rss as f64 / (1024.0 * 1024.0),
159            al = ATTRS_LIVE.load(Relaxed),
160            am = ATTRS_MADE.load(Relaxed),
161            tl = THUNK_LIVE.load(Relaxed),
162            tm = THUNK_MADE.load(Relaxed),
163            te = THUNK_EVALUATED.load(Relaxed),
164            el = ENV_LIVE.load(Relaxed),
165            em = ENV_MADE.load(Relaxed),
166            sl = NIXSTR_LIVE.load(Relaxed),
167            sm = NIXSTR_MADE.load(Relaxed),
168            ll = LIST_LIVE.load(Relaxed),
169            lm = LIST_MADE.load(Relaxed),
170        );
171    }
172
173    /// Spawn the periodic-dump thread (only when enabled). Dumps every 2s so a
174    /// 30s+ eval captures the high-water region. Also usable as an at-exit
175    /// hook via the returned guard.
176    pub fn spawn_poller() {
177        if !enabled() {
178            return;
179        }
180        std::thread::spawn(|| loop {
181            std::thread::sleep(std::time::Duration::from_millis(2000));
182            dump("periodic");
183        });
184    }
185}
186
187// -- String interner (shared with sui-bytecode via sui-intern's thread-local) --
188//
189// Previously this module owned its own `thread_local! INTERNER`. That
190// diverged from `sui-bytecode`'s `sui_intern::*` thread-local — Symbols
191// were NOT portable across the tree-walker ↔ VM fallback boundary,
192// which only worked because both paths happened to re-intern strings
193// from the same source text. Delegating both to the same thread-local
194// closes the gap and makes `sui_intern::prewarm()` affect this crate
195// too.
196
197/// Intern a string key, returning a Symbol handle.
198/// Used for NixAttrs keys and Env binding names.
199pub fn intern(s: &str) -> Symbol {
200    sui_intern::intern(s)
201}
202
203/// Resolve a Symbol back to its string content. Allocates a fresh
204/// `String`. For hot paths prefer [`resolve_rc`] or [`with_resolved`]
205/// — `Rc::clone` is ~20x cheaper than `String::from` for identifier-
206/// sized inputs.
207pub fn resolve(sym: Symbol) -> String {
208    sui_intern::resolve(sym)
209}
210
211/// Resolve a Symbol to a shared `Rc<str>`. Zero-copy.
212pub fn resolve_rc(sym: Symbol) -> std::rc::Rc<str> {
213    sui_intern::resolve_rc(sym)
214}
215
216/// Borrow the resolved string inside a closure without allocating.
217pub fn with_resolved<F, R>(sym: Symbol, f: F) -> R
218where
219    F: FnOnce(&str) -> R,
220{
221    sui_intern::with_resolved(sym, f)
222}
223
224// -- Identifier symbol cache --
225//
226// Caches the interned Symbol for each AST identifier by (source_id, text_offset).
227// Avoids re-hashing identifier strings on repeated evaluations of the
228// same expression (common in loops, recursion, overlay fixpoints).
229//
230// The source_id discriminates different parse trees (main file vs imports)
231// so that identifiers at the same byte offset in different files don't
232// collide in the cache.
233
234thread_local! {
235    /// Monotonically increasing counter — bumped on each `rnix::Root::parse`.
236    // STARTS AT 1, NOT 0 — 0 is the reserved "untagged env" sentinel.
237    //
238    // `Env::new()` defaults `source_id: 0`. While the generator also started at
239    // 0, the FIRST file parsed shared key-space with every untagged env, so an
240    // identifier in that file could collide with one from an untagged context at
241    // the same byte offset. That is the same aliasing class as the
242    // CURRENT_SOURCE_ID bug fixed alongside this (see eval.rs's Ident arms) —
243    // reserving 0 costs nothing and removes the overlap by construction.
244    static SOURCE_GEN: Cell<u32> = const { Cell::new(1) };
245
246    /// Maps `(source_id, text_offset)` → interned `Symbol`.
247    static IDENT_CACHE: RefCell<rustc_hash::FxHashMap<u64, Symbol>> =
248        RefCell::new(rustc_hash::FxHashMap::default());
249}
250
251/// Allocate a new source ID for a freshly parsed AST tree.
252///
253/// Call once per `rnix::Root::parse` invocation. The returned ID is
254/// used as the high 32 bits of the `IDENT_CACHE` key, ensuring that
255/// identifiers from different source texts never collide.
256pub fn next_source_id() -> u32 {
257    SOURCE_GEN.with(|g| {
258        let id = g.get();
259        g.set(id.wrapping_add(1));
260        id
261    })
262}
263
264/// Intern a string with caching by source ID and AST text offset.
265///
266/// First call for a given `(source_id, text_offset)`: hash + intern
267/// (same cost as [`intern`]).
268/// Subsequent calls: `FxHashMap` u64 lookup (~5 ns) — no string hashing.
269pub fn intern_cached(name: &str, source_id: u32, text_offset: u32) -> Symbol {
270    intern_cached_with(source_id, text_offset, || intern(name))
271}
272
273/// Cache an interned `Symbol` by `(source_id, text_offset)`, computing it
274/// lazily via `cold` only on a cache miss.
275///
276/// Steady-state hit: `FxHashMap` u64 lookup — no string materialization, no
277/// string hashing. This lets the identifier-eval hot path avoid the
278/// per-lookup `ident_text().to_string()` heap allocation entirely, since the
279/// `&str` is only needed to intern on the (once-per-offset) cold miss.
280pub fn intern_cached_with<F>(source_id: u32, text_offset: u32, cold: F) -> Symbol
281where
282    F: FnOnce() -> Symbol,
283{
284    let key = (u64::from(source_id) << 32) | u64::from(text_offset);
285    IDENT_CACHE.with(|c| {
286        let mut cache = c.borrow_mut();
287        *cache.entry(key).or_insert_with(cold)
288    })
289}
290
291/// Clear the identifier symbol cache.
292///
293/// Call between independent top-level evaluations to reclaim memory.
294/// The cache grows unboundedly during a single evaluation pass.
295pub fn clear_ident_cache() {
296    IDENT_CACHE.with(|c| c.borrow_mut().clear());
297}
298
299// ── Nix string context ─────────────────────────────────────────
300
301/// An element of a Nix string's context set.
302#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
303pub enum ContextElement {
304    /// Store path reference (e.g., "/nix/store/abc-hello").
305    Plain(SmolStr),
306    /// Derivation output reference.
307    Output { drv: SmolStr, output: SmolStr },
308    /// Entire derivation closure.
309    DrvDeep(SmolStr),
310}
311
312impl fmt::Display for ContextElement {
313    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
314        match self {
315            ContextElement::Plain(p) => write!(f, "{p}"),
316            ContextElement::Output { drv, output } => write!(f, "{drv}!{output}"),
317            ContextElement::DrvDeep(d) => write!(f, "={d}"),
318        }
319    }
320}
321
322/// The context attached to a Nix string: a set of store-path references that
323/// the string depends on. Plain string literals have an empty context.
324///
325/// Uses a `Vec` with linear deduplication instead of `BTreeSet`.  Most strings
326/// have 0-2 context elements where linear search is faster than tree overhead,
327/// and `Vec` has the same size as `BTreeSet` (3 words) without per-node heap
328/// allocations for small sets.
329#[derive(Debug, Clone, PartialEq, Eq, Default)]
330pub struct StringContext(SmallVec<[ContextElement; 2]>);
331
332impl StringContext {
333    /// Create an empty context.
334    pub fn new() -> Self {
335        Self(SmallVec::new())
336    }
337
338    /// Merge another context into this one.
339    pub fn merge(&mut self, other: &StringContext) {
340        for elem in &other.0 {
341            if !self.0.contains(elem) {
342                self.0.push(elem.clone());
343            }
344        }
345    }
346
347    /// Add a plain store-path reference.
348    pub fn add_plain(&mut self, path: impl Into<SmolStr>) {
349        let elem = ContextElement::Plain(path.into());
350        if !self.0.contains(&elem) {
351            self.0.push(elem);
352        }
353    }
354
355    /// Add a derivation output reference.
356    pub fn add_output(&mut self, drv: impl Into<SmolStr>, output: impl Into<SmolStr>) {
357        let elem = ContextElement::Output { drv: drv.into(), output: output.into() };
358        if !self.0.contains(&elem) {
359            self.0.push(elem);
360        }
361    }
362
363    /// Add a derivation-deep reference.
364    pub fn add_drv_deep(&mut self, drv: impl Into<SmolStr>) {
365        let elem = ContextElement::DrvDeep(drv.into());
366        if !self.0.contains(&elem) {
367            self.0.push(elem);
368        }
369    }
370
371    /// Whether this context set is empty.
372    #[must_use]
373    pub fn is_empty(&self) -> bool {
374        self.0.is_empty()
375    }
376
377    /// Return the number of context elements.
378    #[must_use]
379    pub fn len(&self) -> usize {
380        self.0.len()
381    }
382
383    /// Iterate over all context elements.
384    pub fn iter(&self) -> impl Iterator<Item = &ContextElement> {
385        self.0.iter()
386    }
387
388    /// Insert a raw context element (deduplicating).
389    pub fn insert(&mut self, elem: ContextElement) {
390        if !self.0.contains(&elem) {
391            self.0.push(elem);
392        }
393    }
394
395    /// Return the elements as a slice.
396    pub fn elements(&self) -> &[ContextElement] {
397        &self.0
398    }
399}
400
401/// A Nix string value with associated context (store-path references).
402#[derive(Debug, PartialEq, Eq)]
403pub struct NixString {
404    /// The character data.
405    pub chars: SmolStr,
406    /// The context set (empty for plain string literals).
407    pub context: StringContext,
408}
409
410// `Clone` is hand-written so the census counts every NixString that comes
411// into existence (a clone is a fresh heap object once Rc-wrapped), keeping
412// `NIXSTR_MADE`/`NIXSTR_LIVE` consistent with the `Drop` below.
413impl Clone for NixString {
414    fn clone(&self) -> Self {
415        census::made(&census::NIXSTR_MADE, &census::NIXSTR_LIVE);
416        Self {
417            chars: self.chars.clone(),
418            context: self.context.clone(),
419        }
420    }
421}
422
423impl Drop for NixString {
424    fn drop(&mut self) {
425        census::dropped(&census::NIXSTR_LIVE);
426    }
427}
428
429impl NixString {
430    /// Create a context-free string.
431    pub fn plain(s: impl Into<SmolStr>) -> Self {
432        census::made(&census::NIXSTR_MADE, &census::NIXSTR_LIVE);
433        Self {
434            chars: s.into(),
435            context: StringContext::default(),
436        }
437    }
438
439    /// Create a string with an explicit context.
440    pub fn with_context(s: impl Into<SmolStr>, ctx: StringContext) -> Self {
441        census::made(&census::NIXSTR_MADE, &census::NIXSTR_LIVE);
442        Self {
443            chars: s.into(),
444            context: ctx,
445        }
446    }
447
448    /// Borrow the string content.
449    #[must_use]
450    pub fn as_str(&self) -> &str {
451        &self.chars
452    }
453
454    /// Whether this string carries any context (store path references).
455    #[must_use]
456    pub fn has_context(&self) -> bool {
457        !self.context.is_empty()
458    }
459}
460
461impl AsRef<str> for NixString {
462    fn as_ref(&self) -> &str {
463        &self.chars
464    }
465}
466
467/// Census wrapper around a list's backing `Vec<Value>`.
468///
469/// `#[repr(transparent)]` + `Deref`/`DerefMut` to `Vec<Value>` so nearly every
470/// existing call site (`.len()`, `.iter()`, indexing, `.as_slice()`, `.clone()`
471/// → produces a `NixList`) works unchanged. Its sole job is to carry the census
472/// hooks (`LIST_MADE`/`LIST_LIVE`) on the inner heap allocation.
473#[repr(transparent)]
474#[derive(Debug, PartialEq)]
475pub struct NixList(pub Vec<Value>);
476
477impl NixList {
478    #[inline]
479    pub fn new(v: Vec<Value>) -> Self {
480        census::made(&census::LIST_MADE, &census::LIST_LIVE);
481        NixList(v)
482    }
483
484    /// Consume into the backing `Vec<Value>`. `mem::take` because `NixList`
485    /// has a `Drop` impl (can't move the field out); the emptied husk's Drop
486    /// still fires, decrementing LIVE — correct, the list is consumed.
487    #[inline]
488    pub fn into_vec(mut self) -> Vec<Value> {
489        std::mem::take(&mut self.0)
490    }
491}
492
493impl From<Vec<Value>> for NixList {
494    #[inline]
495    fn from(v: Vec<Value>) -> Self {
496        NixList::new(v)
497    }
498}
499
500// Slice/array comparison so `assert_eq!(nixlist, [..])` in tests keeps working.
501impl<T: AsRef<[Value]>> PartialEq<T> for NixList {
502    #[inline]
503    fn eq(&self, other: &T) -> bool {
504        self.0.as_slice() == other.as_ref()
505    }
506}
507
508impl Clone for NixList {
509    fn clone(&self) -> Self {
510        census::made(&census::LIST_MADE, &census::LIST_LIVE);
511        NixList(self.0.clone())
512    }
513}
514
515impl Drop for NixList {
516    fn drop(&mut self) {
517        census::dropped(&census::LIST_LIVE);
518    }
519}
520
521impl FromIterator<Value> for NixList {
522    #[inline]
523    fn from_iter<I: IntoIterator<Item = Value>>(iter: I) -> Self {
524        NixList::new(iter.into_iter().collect())
525    }
526}
527
528impl std::ops::Deref for NixList {
529    type Target = Vec<Value>;
530    #[inline]
531    fn deref(&self) -> &Vec<Value> {
532        &self.0
533    }
534}
535
536impl std::ops::DerefMut for NixList {
537    #[inline]
538    fn deref_mut(&mut self) -> &mut Vec<Value> {
539        &mut self.0
540    }
541}
542
543impl<'a> IntoIterator for &'a NixList {
544    type Item = &'a Value;
545    type IntoIter = std::slice::Iter<'a, Value>;
546    #[inline]
547    fn into_iter(self) -> Self::IntoIter {
548        self.0.iter()
549    }
550}
551
552impl IntoIterator for NixList {
553    type Item = Value;
554    type IntoIter = std::vec::IntoIter<Value>;
555    #[inline]
556    fn into_iter(mut self) -> Self::IntoIter {
557        // Move the Vec out. `NixList`'s Drop still fires on the emptied husk,
558        // decrementing LIVE — correct, since the elements move to the iterator
559        // and the list allocation is consumed.
560        std::mem::take(&mut self.0).into_iter()
561    }
562}
563
564impl std::ops::Deref for NixString {
565    type Target = str;
566
567    fn deref(&self) -> &str {
568        &self.chars
569    }
570}
571
572impl fmt::Display for NixString {
573    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
574        write!(f, "{}", self.chars)
575    }
576}
577
578// ── Value enum ────────────────────────────────────────────────
579
580/// A Nix value — potentially lazy (may be a Thunk).
581///
582/// To get a guaranteed-concrete value, call `.demand()` which returns
583/// `Concrete`. The `Concrete` type has thunk-free accessors that the
584/// compiler enforces — you cannot accidentally skip forcing.
585#[derive(Debug, Clone)]
586#[derive(Default)]
587pub enum Value {
588    #[default]
589    Null,
590    Bool(bool),
591    Int(i64),
592    Float(f64),
593    String(Rc<NixString>),
594    Path(Box<SmolStr>),
595    List(Rc<NixList>),
596    Attrs(Rc<NixAttrs>),
597    Lambda(Rc<Closure>),
598    Builtin(Box<BuiltinFn>),
599    /// A lazy value (thunk) with memoization and blackhole detection.
600    Thunk(Thunk),
601}
602
603// ── Concrete: construction-guaranteed non-thunk ──────────────
604
605/// A demanded Nix value. Guaranteed NOT a Thunk at the TYPE level.
606///
607/// Unlike `Value` (which has a `Thunk` variant), `Concrete` is a separate
608/// enum that DOES NOT HAVE a Thunk variant. The compiler rejects any attempt
609/// to construct a `Concrete` from a thunk — the variant simply doesn't exist.
610///
611/// The ONLY way to obtain a `Concrete` is through `Value::demand()`.
612///
613/// ```rust,ignore
614/// let val: Value = eval_expr(expr, env)?;  // might be Thunk
615/// let c: Concrete = val.demand()?;          // NOW guaranteed concrete
616/// let n: i64 = c.as_int()?;                // type-safe, thunk-free
617/// ```
618#[derive(Debug, Clone)]
619pub enum Concrete {
620    Null,
621    Bool(bool),
622    Int(i64),
623    Float(f64),
624    String(Rc<NixString>),
625    Path(Box<SmolStr>),
626    List(Rc<NixList>),      // elements may be lazy (correct for Nix)
627    Attrs(Rc<NixAttrs>),       // values may be lazy (correct for Nix)
628    Lambda(Rc<Closure>),
629    Builtin(Box<BuiltinFn>),
630    // NO Thunk variant. The compiler enforces this.
631}
632
633impl Concrete {
634    /// Convert back to a Value (for APIs that still take Value).
635    #[inline]
636    pub fn into_value(self) -> Value {
637        match self {
638            Concrete::Null => Value::Null,
639            Concrete::Bool(b) => Value::Bool(b),
640            Concrete::Int(n) => Value::Int(n),
641            Concrete::Float(f) => Value::Float(f),
642            Concrete::String(s) => Value::String(s),
643            Concrete::Path(p) => Value::Path(p),
644            Concrete::List(l) => Value::List(l),
645            Concrete::Attrs(a) => Value::Attrs(a),
646            Concrete::Lambda(c) => Value::Lambda(c),
647            Concrete::Builtin(b) => Value::Builtin(b),
648        }
649    }
650
651    /// Borrow as a Value reference. Constructs a temporary Value.
652    /// Prefer specific accessors (as_bool, as_int, etc.) when possible.
653    pub fn to_value(&self) -> Value {
654        self.clone().into_value()
655    }
656
657    /// Extract bool — guaranteed no thunk.
658    pub fn as_bool(&self) -> Result<bool, EvalError> {
659        match self {
660            Concrete::Bool(b) => Ok(*b),
661            other => Err(EvalError::TypeMismatch { expected: "bool", got: other.type_name() }),
662        }
663    }
664
665    /// Extract int — guaranteed no thunk.
666    pub fn as_int(&self) -> Result<i64, EvalError> {
667        match self {
668            Concrete::Int(n) => Ok(*n),
669            other => Err(EvalError::TypeMismatch { expected: "int", got: other.type_name() }),
670        }
671    }
672
673    /// Extract string ref — guaranteed no thunk.
674    pub fn as_str(&self) -> Result<&str, EvalError> {
675        match self {
676            Concrete::String(s) => Ok(&s.chars),
677            other => Err(EvalError::TypeMismatch { expected: "string", got: other.type_name() }),
678        }
679    }
680
681    /// Extract NixString ref — guaranteed no thunk.
682    pub fn as_nix_string(&self) -> Result<&NixString, EvalError> {
683        match self {
684            Concrete::String(s) => Ok(s),
685            other => Err(EvalError::TypeMismatch { expected: "string", got: other.type_name() }),
686        }
687    }
688
689    /// Extract list ref — guaranteed no thunk at this level.
690    /// Note: list ELEMENTS may still be lazy (Value, not Concrete).
691    pub fn as_list(&self) -> Result<&[Value], EvalError> {
692        match self {
693            Concrete::List(l) => Ok(l.as_slice()),
694            other => Err(EvalError::TypeMismatch { expected: "list", got: other.type_name() }),
695        }
696    }
697
698    /// Extract attrs ref — guaranteed no thunk at this level.
699    /// Note: attr VALUES may still be lazy (Value, not Concrete).
700    pub fn as_attrs(&self) -> Result<&NixAttrs, EvalError> {
701        match self {
702            Concrete::Attrs(a) => Ok(a),
703            other => Err(EvalError::TypeMismatch { expected: "set", got: other.type_name() }),
704        }
705    }
706
707    /// Extract float — guaranteed no thunk.
708    pub fn as_float(&self) -> Result<f64, EvalError> {
709        match self {
710            Concrete::Float(f) => Ok(*f),
711            Concrete::Int(n) => Ok(*n as f64),
712            other => Err(EvalError::TypeMismatch { expected: "float", got: other.type_name() }),
713        }
714    }
715
716    /// Check the value type name.
717    pub fn type_name(&self) -> &'static str {
718        match self {
719            Concrete::Null => "null",
720            Concrete::Bool(_) => "bool",
721            Concrete::Int(_) => "int",
722            Concrete::Float(_) => "float",
723            Concrete::String(_) => "string",
724            Concrete::Path(_) => "path",
725            Concrete::List(_) => "list",
726            Concrete::Attrs(_) => "set",
727            Concrete::Lambda(_) | Concrete::Builtin(_) => "lambda",
728        }
729    }
730
731    /// Alias for `as_str()` — API parity with Value::as_string().
732    pub fn as_string(&self) -> Result<&str, EvalError> {
733        self.as_str()
734    }
735
736    /// Extract owned NixAttrs — guaranteed no thunk at this level.
737    pub fn to_attrs(&self) -> Result<NixAttrs, EvalError> {
738        match self {
739            Concrete::Attrs(a) => Ok((**a).clone()),
740            other => Err(EvalError::TypeMismatch { expected: "set", got: other.type_name() }),
741        }
742    }
743
744    /// Extract owned list — guaranteed no thunk at this level.
745    pub fn to_list(&self) -> Result<Vec<Value>, EvalError> {
746        match self {
747            Concrete::List(l) => Ok((**l).0.clone()),
748            other => Err(EvalError::TypeMismatch { expected: "list", got: other.type_name() }),
749        }
750    }
751
752    /// Extract a filesystem path from Path or String.
753    pub fn coerce_to_path(&self, context: &str) -> Result<String, EvalError> {
754        match self {
755            Concrete::Path(p) => Ok(p.to_string()),
756            Concrete::String(ns) => Ok(ns.chars.to_string()),
757            Concrete::Attrs(attrs) => {
758                if let Some(out_path) = attrs.get("outPath") {
759                    let forced = crate::eval::force_value(out_path)?;
760                    forced.coerce_to_path(context)
761                } else {
762                    Err(EvalError::type_error(format!(
763                        "{context}: expected path or string, got set without outPath"
764                    )))
765                }
766            }
767            other => Err(EvalError::type_error(format!(
768                "{context}: expected path or string, got {}", other.type_name()
769            ))),
770        }
771    }
772
773    /// Extract owned string.
774    pub fn to_str(&self) -> Result<String, EvalError> {
775        match self {
776            Concrete::String(s) => Ok(s.chars.to_string()),
777            other => Err(EvalError::TypeMismatch { expected: "string", got: other.type_name() }),
778        }
779    }
780
781    /// Extract owned NixString (with context).
782    pub fn to_nix_string(&self) -> Result<NixString, EvalError> {
783        match self {
784            Concrete::String(s) => Ok((**s).clone()),
785            other => Err(EvalError::TypeMismatch { expected: "string", got: other.type_name() }),
786        }
787    }
788
789    /// Check if value is a function (lambda or builtin).
790    pub fn is_function(&self) -> bool {
791        matches!(self, Concrete::Lambda(_) | Concrete::Builtin(_))
792    }
793}
794
795// Type-safe conversion: Concrete → Value (infallible)
796impl From<Concrete> for Value {
797    fn from(c: Concrete) -> Value {
798        c.into_value()
799    }
800}
801
802impl PartialEq for Concrete {
803    fn eq(&self, other: &Self) -> bool {
804        match (self, other) {
805            (Concrete::Null, Concrete::Null) => true,
806            (Concrete::Bool(a), Concrete::Bool(b)) => a == b,
807            (Concrete::Int(a), Concrete::Int(b)) => a == b,
808            (Concrete::Float(a), Concrete::Float(b)) => a == b,
809            (Concrete::Int(a), Concrete::Float(b)) | (Concrete::Float(b), Concrete::Int(a)) => (*a as f64) == *b,
810            (Concrete::String(a), Concrete::String(b)) => Rc::ptr_eq(a, b) || a.chars == b.chars,
811            (Concrete::Path(a), Concrete::Path(b)) => a == b,
812            (Concrete::List(a), Concrete::List(b)) => Rc::ptr_eq(a, b) || a == b,
813            (Concrete::Attrs(a), Concrete::Attrs(b)) => {
814                if Rc::ptr_eq(a, b) {
815                    return true;
816                }
817                // cppnix `EvalState::eqValues` derivation short-circuit:
818                // two attrsets that are BOTH derivations (each has
819                // `type == "derivation"`) AND each carry an `outPath`
820                // compare by their `outPath` string ONLY — never by deep
821                // structural equality.  This is load-bearing: derivations
822                // hold thunks/functions (`meta`, `override`, …) that never
823                // compare structurally-equal even when the two describe the
824                // same store output, and forcing every attr can throw.
825                // (Empirically characterized against the live nix oracle:
826                //  `hello == (hello // { x = 5; })` ⇒ true.)
827                if let (Some(pa), Some(pb)) =
828                    (derivation_out_path(a), derivation_out_path(b))
829                {
830                    return pa == pb;
831                }
832                // Structural compare by BORROW, not by clone. The prior
833                // `a.inner() == b.inner()` flattened AND cloned *both* backing
834                // `AttrsMap`s (`inner()` = `as_flat().clone()`) purely to feed
835                // `HashMap::eq` — the clone is dead work. `as_flat()` returns a
836                // borrow into the (memoized-if-overlay) map, so
837                // `a.as_flat() == b.as_flat()` runs the *identical*
838                // `HashMap::eq`: same keys, same per-value `Value::eq` calls.
839                // `HashMap::eq` is ORDER-INDEPENDENT by construction (it iterates
840                // one map and looks each key up in the other), so this holds
841                // regardless of the map's internal iteration order — true for the
842                // std `AttrsMap` exactly as it was for the old `im_rc` map.
843                // PROVABLY-NEUTRAL
844                // on the demand axis: cloning a `Value` is an `Rc`-bump that
845                // forces NOTHING; the only `.demand()` calls in this arm are (1)
846                // the derivation short-circuit above (unchanged) and (2) inside
847                // `Value::eq` (unchanged — same values, same order). Removing the
848                // clone cannot move which thunk forces, when, or whether a throw
849                // surfaces. See docs/PERF-ARSENAL.md C-A.
850                let (fa, fb) = (a.as_flat(), b.as_flat());
851                if crate::perf::enabled() {
852                    crate::perf::inc(crate::perf::Counter::AttrsEqStructuralCalls);
853                    // Combined entry count of the two maps the old `inner()`
854                    // path would have cloned before comparing.
855                    crate::perf::add(
856                        crate::perf::Counter::AttrsEqEntriesCloneElided,
857                        (fa.len() + fb.len()) as u64,
858                    );
859                }
860                fa == fb
861            }
862            (Concrete::Lambda(a), Concrete::Lambda(b)) => Rc::ptr_eq(a, b),
863            _ => false,
864        }
865    }
866}
867
868/// Concatenate two Nix lists: `left ++ right_elems`.
869///
870/// `left` must be a `Value::List`; `right_elems` is the right list's element
871/// slice. When `left`'s backing `Rc<Vec>` is uniquely owned (a fresh
872/// temporary, as in a left-associative `acc ++ [x]` fold), the right elements
873/// are appended IN PLACE — amortized O(1) instead of the O(n) full clone that
874/// `left.to_vec()` would cost. When the `Rc` is shared, the shared list is
875/// left untouched and a fresh clone-extended Vec is built (identical to the
876/// prior `to_vec()` + `extend_from_slice` path).
877///
878/// # Byte-neutrality
879/// PROVABLY-NEUTRAL. Both paths produce the identical ordered sequence of the
880/// same `Rc`-shared lazy `Value` thunks — no element is forced, reordered, or
881/// re-identified. The only observable difference is heap allocation reuse,
882/// which is not a Nix-observable property. See `docs/PERF-ARSENAL.md`.
883pub fn concat_lists(left: Value, right_elems: &[Value]) -> Result<Value, EvalError> {
884    // Take ownership of the left backing Vec, reusing its allocation when the
885    // Rc is unique. `Rc::try_unwrap` returns the inner Vec on refcount 1;
886    // otherwise it clones (identical bytes to the old `to_vec()`).
887    let mut la = match left {
888        Value::List(rc) => {
889            let reused = Rc::strong_count(&rc) == 1;
890            let vec: Vec<Value> = match Rc::try_unwrap(rc) {
891                Ok(v) => v.into_vec(), // uniquely owned: allocation reused
892                Err(rc) => (*rc).0.clone(), // shared: clone the left (unchanged)
893            };
894            if crate::perf::enabled() {
895                crate::perf::inc(crate::perf::Counter::ListConcatCalls);
896                if reused {
897                    // Left elements appended in place — copy elided.
898                    crate::perf::add(
899                        crate::perf::Counter::ListConcatElemsReused,
900                        vec.len() as u64,
901                    );
902                } else {
903                    // Left elements cloned into a fresh Vec (the storm).
904                    crate::perf::add(
905                        crate::perf::Counter::ListConcatElemsCopied,
906                        vec.len() as u64,
907                    );
908                }
909            }
910            vec
911        }
912        other => {
913            return Err(EvalError::TypeMismatch {
914                expected: "list",
915                got: other.type_name(),
916            });
917        }
918    };
919    // Right elements are always copied (appended); their thunks are Rc-shared.
920    la.extend_from_slice(right_elems);
921    Ok(Value::list(la))
922}
923
924/// If `attrs` is a derivation — an attrset whose `type` forces to the string
925/// `"derivation"` AND which carries a forceable `outPath` — return that
926/// `outPath` string.  Otherwise `None` (caller falls back to structural
927/// equality).  A force error on `type`/`outPath` yields `None`, so a broken
928/// derivation degrades to structural compare rather than a spurious match.
929fn derivation_out_path(attrs: &NixAttrs) -> Option<String> {
930    match attrs.get("type")?.demand().ok()? {
931        Concrete::String(s) if s.chars == "derivation" => {}
932        _ => return None,
933    }
934    match attrs.get("outPath")?.demand().ok()? {
935        Concrete::String(s) => Some(s.chars.to_string()),
936        _ => None,
937    }
938}
939
940/// If `attrs` is a **derivation** (`type` forces to `"derivation"`) carrying a
941/// forceable `drvPath` AND `outPath`, return `Ok(Some((drv_path, out_path)))`.
942///
943/// Returns `Ok(None)` when `attrs` is not a derivation (no `type ==
944/// "derivation"`, or missing `drvPath`) — e.g. a plain attrset that merely
945/// carries an `outPath` (a `{ outPath = "…"; }` path-like), which has nothing
946/// to realize. Returns `Err` only if forcing `drvPath`/`outPath` itself fails
947/// (a genuinely broken derivation), so the caller surfaces the eval error
948/// rather than silently treating a broken drv as "not a derivation".
949///
950/// This is the import-from-derivation sibling of [`derivation_out_path`]: that
951/// helper only needs `outPath` for equality; realize also needs `drvPath` to
952/// know *what* to build.
953fn derivation_drv_and_out(
954    attrs: &NixAttrs,
955) -> Result<Option<(String, String)>, EvalError> {
956    // Not a derivation unless `type` forces to exactly "derivation".
957    match attrs.get("type") {
958        Some(t) => match crate::eval::force_value(t)? {
959            Value::String(s) if s.chars == "derivation" => {}
960            _ => return Ok(None),
961        },
962        None => return Ok(None),
963    }
964    // A derivation without a drvPath cannot be realized — treat as non-drv so
965    // the caller falls back to plain coercion (the outPath arm).
966    let drv_path = match attrs.get("drvPath") {
967        Some(d) => crate::eval::force_value(d)?.coerce_to_path("drvPath")?,
968        None => return Ok(None),
969    };
970    let out_path = match attrs.get("outPath") {
971        Some(o) => crate::eval::force_value(o)?.coerce_to_path("outPath")?,
972        None => return Ok(None),
973    };
974    Ok(Some((drv_path, out_path)))
975}
976
977/// Given a store-path STRING (produced by interpolating a derivation) and its
978/// string context, return the producing `.drv` path IF this store path is a
979/// derivation output that should be realized on a filesystem read.
980///
981/// Returns `Some(drv_path)` only when the context carries a
982/// `ContextElement::Output { drv, output }` whose `output` store path matches
983/// `out_path` — i.e. this string IS the output of a derivation named by the
984/// context. `Plain`/`DrvDeep`-only contexts (a plain store-path reference, or a
985/// `.drv` self-reference) don't name an output to realize, and an
986/// empty-context string is a literal path with nothing to build.
987///
988/// This is how cppnix decides IFD across interpolation: the derivation-ness of
989/// `"${drv}"` survives as string context, not as a value shape.
990fn out_path_needs_realize(out_path: &str, ctx: &StringContext) -> Option<String> {
991    // Only store-path strings can be derivation outputs.
992    if !out_path.starts_with("/nix/store/") {
993        return None;
994    }
995    for elem in ctx.iter() {
996        if let ContextElement::Output { drv, output } = elem {
997            // The context stores the OUTPUT NAME (e.g. "out"/"dev"), while the
998            // string IS the output's store path. cppnix's `Output.outputName`
999            // matches the string it decorates; sui's tree-walker builds the
1000            // interpolated string FROM this output's store path, so a single
1001            // `Output` element on a store-path string is the producing drv.
1002            let _ = output; // output name is not needed to build the closure
1003            return Some(drv.to_string());
1004        }
1005    }
1006    None
1007}
1008
1009impl Value {
1010    /// Convert a known-concrete Value to Concrete. Panics if Thunk.
1011    /// Only use when the caller guarantees the value is not a thunk.
1012    pub(crate) fn demand_unchecked(self) -> Concrete {
1013        match self {
1014            Value::Null => Concrete::Null,
1015            Value::Bool(b) => Concrete::Bool(b),
1016            Value::Int(n) => Concrete::Int(n),
1017            Value::Float(f) => Concrete::Float(f),
1018            Value::String(s) => Concrete::String(s),
1019            Value::Path(p) => Concrete::Path(p),
1020            Value::List(l) => Concrete::List(l),
1021            Value::Attrs(a) => Concrete::Attrs(a),
1022            Value::Lambda(c) => Concrete::Lambda(c),
1023            Value::Builtin(b) => Concrete::Builtin(b),
1024            Value::Thunk(_) => panic!("demand_unchecked called on Thunk"),
1025        }
1026    }
1027}
1028
1029impl Value {
1030    /// Demand a concrete value. Forces if Thunk, returns as-is if concrete.
1031    ///
1032    /// This is the TYPED forcing API. The returned `Concrete` is guaranteed
1033    /// non-Thunk — enforced by the Concrete enum having NO Thunk variant.
1034    pub fn demand(&self) -> Result<Concrete, EvalError> {
1035        let v = match self {
1036            Value::Thunk(_) => crate::eval::force_value(self)?,
1037            other => other.clone(),
1038        };
1039        // Convert Value → Concrete. Thunk is impossible after force_value.
1040        match v {
1041            Value::Null => Ok(Concrete::Null),
1042            Value::Bool(b) => Ok(Concrete::Bool(b)),
1043            Value::Int(n) => Ok(Concrete::Int(n)),
1044            Value::Float(f) => Ok(Concrete::Float(f)),
1045            Value::String(s) => Ok(Concrete::String(s)),
1046            Value::Path(p) => Ok(Concrete::Path(p)),
1047            Value::List(l) => Ok(Concrete::List(l)),
1048            Value::Attrs(a) => Ok(Concrete::Attrs(a)),
1049            Value::Lambda(c) => Ok(Concrete::Lambda(c)),
1050            Value::Builtin(b) => Ok(Concrete::Builtin(b)),
1051            Value::Thunk(_) => {
1052                // force_value returned a Thunk — chase it.
1053                // This can happen when the transitive unwrap loop hits
1054                // a depth limit. Re-force to resolve.
1055                let re_forced = crate::eval::force_value(&v)?;
1056                match re_forced {
1057                    Value::Null => Ok(Concrete::Null),
1058                    Value::Bool(b) => Ok(Concrete::Bool(b)),
1059                    Value::Int(n) => Ok(Concrete::Int(n)),
1060                    Value::Float(f) => Ok(Concrete::Float(f)),
1061                    Value::String(s) => Ok(Concrete::String(s)),
1062                    Value::Path(p) => Ok(Concrete::Path(p)),
1063                    Value::List(l) => Ok(Concrete::List(l)),
1064                    Value::Attrs(a) => Ok(Concrete::Attrs(a)),
1065                    Value::Lambda(c) => Ok(Concrete::Lambda(c)),
1066                    Value::Builtin(b) => Ok(Concrete::Builtin(b)),
1067                    Value::Thunk(_) => Err(EvalError::InfiniteRecursion(
1068                        "demand: thunk chain could not be resolved".to_string(),
1069                    )),
1070                }
1071            }
1072        }
1073    }
1074}
1075
1076#[cfg(target_pointer_width = "64")]
1077const _: () = assert!(std::mem::size_of::<Value>() <= 16);
1078
1079/// Runaway backstop for overlay-fixpoint promotion.  A genuine fixpoint
1080/// re-entry converges in a bounded number of nested promotions (the
1081/// nixpkgs `libxcrypt`/`self:super:` overlay needs ≤18 concurrent
1082/// promotions).  A non-converging demand (e.g. a cross-system stdenv
1083/// fixpoint that keeps re-entering the empty partial) climbs the nesting
1084/// without bound.  When the active concurrent-promotion nesting
1085/// (`IN_PROMISE_EVAL`) reaches this cap we STOP promoting and fall through
1086/// to `InfiniteRecursion` — which `eval_select`'s `x.y or default` arm
1087/// recovers exactly like nix's lazy fall-through, converting a would-be
1088/// native stack overflow into the recoverable error nix itself raises.
1089///
1090/// This is the runaway half of the same discipline the release-build
1091/// `MAX_EVAL_DEPTH` guard provides (which is `usize::MAX` in release to
1092/// admit nixpkgs' legitimately-deep fixpoints); scoping the bound to
1093/// *promotions* keeps ordinary deep evaluation unbounded while still
1094/// catching a non-terminating fixpoint before the OS stack does.
1095const FIXPOINT_PROMOTE_NEST_CAP: u32 = 32;
1096
1097/// Force-stack-depth backstop that arms once a fixpoint promotion has fired
1098/// (`promotion_occurred()`).  A converging fixpoint (`libxcrypt`) bottoms
1099/// out at a force depth of a few dozen; a non-converging promoted partial
1100/// recurses without bound.  This cap (10× any observed real fixpoint's force
1101/// depth) converts a force-stack runaway into a recoverable
1102/// `InfiniteRecursion` before the native OS stack aborts, without touching
1103/// ordinary (non-promotion) deep evaluation.  Paired with the eval-depth
1104/// backstop (`eval::PROMOTION_RUNAWAY_EVAL_DEPTH`) for runaways that don't
1105/// climb the force stack.
1106const PROMOTION_RUNAWAY_FORCE_DEPTH: usize = 500;
1107
1108thread_local! {
1109    /// Depth counter for "currently evaluating the body of a Promise-state
1110    /// thunk".  Incremented before the body of a `ThunkRepr::Promise`
1111    /// runs, decremented after.  Used by `eval_select` to treat missing
1112    /// attribute lookups on the Promise's sentinel value as `null`
1113    /// instead of erroring with `AttrNotFound`.  Scoped to Promise
1114    /// evaluation so unrelated user code retains cppnix-strict semantics.
1115    pub(crate) static IN_PROMISE_EVAL: std::cell::Cell<u32> = const { std::cell::Cell::new(0) };
1116
1117    /// Set once a fixpoint promotion has occurred anywhere in the current
1118    /// top-level evaluation.  Arms the release-active force-depth runaway
1119    /// backstop for the REST of the eval (not just while `IN_PROMISE_EVAL`
1120    /// is non-zero) — a corrupted promoted partial can send a DOWNSTREAM
1121    /// fixpoint (`makeOverridable`/`commonAttrs`) into unbounded recursion
1122    /// AFTER the promoting force has already returned, so the backstop must
1123    /// outlive the promotion's own softening scope.
1124    pub(crate) static PROMOTION_OCCURRED: std::cell::Cell<bool> = const { std::cell::Cell::new(false) };
1125}
1126
1127/// `true` if any overlay-fixpoint promotion has fired in this eval.
1128#[inline(always)]
1129pub fn promotion_occurred() -> bool {
1130    PROMOTION_OCCURRED.with(|c| c.get())
1131}
1132
1133/// `true` if the evaluator is currently inside the body of a
1134/// `ThunkRepr::Promise` (used by `eval_select` to relax
1135/// `AttrNotFound` errors during fix-point construction).
1136#[inline(always)]
1137pub fn in_promise_eval() -> bool {
1138    IN_PROMISE_EVAL.with(|c| c.get() > 0)
1139}
1140
1141/// Internal representation of a thunk's state machine.
1142///
1143/// Transitions: `Suspended` → `Blackhole` → `Evaluated` (on success),
1144/// or `Suspended` → `Blackhole` → `Suspended` (on failure, to allow retry).
1145pub enum ThunkRepr {
1146    /// Not yet evaluated. Holds the AST expression and captured environment.
1147    Suspended {
1148        expr: rnix::ast::Expr,
1149        env: Env,
1150    },
1151    /// Pending `inherit (source) name` selection. When forced,
1152    /// forces the shared `source_thunk` and pulls out `name`.
1153    ///
1154    /// The `source_thunk` is created once per `inherit (source) a b c`
1155    /// clause and shared (via `Rc` clone) across all inherited names.
1156    /// This means N names share one source evaluation instead of N
1157    /// independent evaluations — the source thunk's own memoization
1158    /// ensures it is evaluated at most once.
1159    ///
1160    /// This is its own variant (rather than synthesizing a Select AST
1161    /// node) because rnix doesn't expose a public AST builder, and
1162    /// we want each inherited name to defer evaluation of the source
1163    /// expression so that `inherit (lib.trivial) ...` at the top of
1164    /// trivial.nix doesn't blackhole on the still-being-constructed
1165    /// `lib.trivial`.
1166    InheritSelect {
1167        source_thunk: Thunk,
1168        name: SmolStr,
1169    },
1170    /// A lazy value backed by a Rust closure.  Used for flake input
1171    /// evaluation: the closure calls `evaluate_flake` on first access
1172    /// instead of eagerly during flake setup, matching CppNix semantics
1173    /// where each input's outputs function is wrapped in a thunk.
1174    Native(Box<dyn FnOnce() -> Result<Value, EvalError>>),
1175    /// A deferred with-scope ident lookup.  Stores a direct reference to the
1176    /// with-scope's shared cache and the ident name.  When forced, checks the
1177    /// cache for the resolved attrset and looks up the name — O(1) hash lookup,
1178    /// no Env traversal, no fixpoint re-forcing.
1179    ///
1180    /// This is the construction-guarantee solution for the with-scope fixpoint
1181    /// problem: instead of creating 80K+ Env-capturing thunks (each doing a
1182    /// full lookup on force), we create 80K lightweight cache-referencing thunks
1183    /// that share the same resolved attrset.
1184    WithIdent {
1185        /// The ident name to look up
1186        name: SmolStr,
1187        /// Direct reference to the with-scope's cached attrset.
1188        /// Shared via Rc<RefCell> — all idents from the same `with` scope
1189        /// reference the same cache.  When ANY lookup forces the scope,
1190        /// the cache is populated and all subsequent WithIdent forces are O(1).
1191        scope_cache: Rc<RefCell<Option<NixAttrs>>>,
1192        /// The scope value (for initial force if cache is empty)
1193        scope_value: Value,
1194        /// Fallback: the full env for lexical+outer-scope lookup if the
1195        /// with-scope doesn't contain this name
1196        env: Env,
1197    },
1198    /// Currently being evaluated -- detects infinite recursion.
1199    Blackhole,
1200    /// Currently being evaluated, but the thunk is known to be
1201    /// self-recursive (its RHS references the bound name).  Inner
1202    /// re-entrance returns the partial value from the cell instead
1203    /// of erroring with `InfiniteRecursion` — matches cppnix's
1204    /// `let x = f x; in x` semantics where inner accesses to `x`
1205    /// see the not-yet-complete attrset under construction.
1206    ///
1207    /// The cell starts as `Value::Attrs(empty)` (the cheapest
1208    /// sentinel that propagates through `mapAttrs` / `attrNames` /
1209    /// `concatMap` without further type errors).  When the body
1210    /// completes, the cell is replaced with the final value and
1211    /// the repr transitions to `Evaluated`.
1212    Promise(Rc<RefCell<Value>>),
1213    /// A `Native` (`FnOnce`) thunk whose closure already ran and
1214    /// FAILED.  The closure is consumed and cannot be retried, so we
1215    /// memoize the error itself and re-raise it on every subsequent
1216    /// force.  This is the correctness-preserving replacement for the
1217    /// old `Evaluated(Null)` poisoning: a thunk that threw on its first
1218    /// force MUST NOT silently become `null` on a second read (which
1219    /// turned a swallowed transient flake-input error into a bogus
1220    /// `AttrNotFound`/`cannot select from set` far downstream — the
1221    /// stylix `darwinModules` marquee root).  A re-force re-throws the
1222    /// original error, exactly as cppnix re-throws a thunk that failed.
1223    Failed(EvalError),
1224    /// Already evaluated and memoized as a THUNK value.  The `cache`
1225    /// `OnceCell` is intentionally empty for this variant (caching a thunk
1226    /// would spin `force_value`), so the boxed `Value` is the sole store.
1227    Evaluated(Box<Value>),
1228    /// Already evaluated and memoized as a CONCRETE (non-thunk) value.
1229    /// The value lives ONLY in the `cache` `OnceCell` (`Box<Concrete>`);
1230    /// this variant is a valueless terminal marker that collapses the
1231    /// former double-store (a redundant `Evaluated(Box<Value>)` alongside
1232    /// the cache).  Any reader that finds this marker reconstructs the
1233    /// `Value` from `cache` via `Concrete::into_value()`, which is a
1234    /// byte-identical, lossless inverse of `demand_unchecked` (same enum
1235    /// shape, moves the inner `Rc`/`Box` — preserving string context and
1236    /// list/attrs `Rc` identity).  In practice the `cache` fast path in
1237    /// `force`/`force_inner` returns before this arm is ever matched.
1238    EvaluatedConcrete,
1239}
1240
1241/// Inner storage for a thunk: a fast-path `OnceCell` cache plus the
1242/// full `UnsafeCell` state machine.  Reads of already-evaluated thunks
1243/// hit the `OnceCell` and never touch the `UnsafeCell`, eliminating
1244/// all runtime overhead on the hot path (~150M+ cache hits per nixpkgs
1245/// eval).  The cold path (1.8M forces) uses `UnsafeCell` directly —
1246/// safe because the evaluator is single-threaded (`Rc`, not `Arc`) and
1247/// the state machine ensures no overlapping mutable access
1248/// (`Suspended` → `Blackhole` → `Evaluated` transitions are sequential).
1249struct ThunkInner {
1250    /// Fast-path cache for already-evaluated thunks.
1251    /// Set once when `Evaluated` is stored, never cleared.
1252    /// Reads bypass the `UnsafeCell` entirely.
1253    cache: OnceCell<Box<Concrete>>,
1254    /// Full state machine for the thunk lifecycle.
1255    repr: UnsafeCell<ThunkRepr>,
1256    /// `true` when the thunk's RHS references its own bound name
1257    /// (a fix-point pattern like `let x = f x; in x`).  On force,
1258    /// transitions to `ThunkRepr::Promise` instead of `Blackhole`
1259    /// so inner re-entrance sees the partial value rather than
1260    /// erroring.  Detected at thunk-construction time via AST
1261    /// text search.
1262    recursive: bool,
1263}
1264
1265impl Drop for ThunkInner {
1266    fn drop(&mut self) {
1267        census::dropped(&census::THUNK_LIVE);
1268    }
1269}
1270
1271/// A lazy value with memoization and blackhole detection.
1272#[derive(Clone)]
1273pub struct Thunk(pub(crate) Rc<ThunkInner>);
1274
1275impl Thunk {
1276    /// Create a thunk that will evaluate `expr` in `env` when forced.
1277    pub fn new_suspended(expr: rnix::ast::Expr, env: Env) -> Self {
1278        crate::trace::inc_thunks_created();
1279        census::made(&census::THUNK_MADE, &census::THUNK_LIVE);
1280        Self(Rc::new(ThunkInner {
1281            cache: OnceCell::new(),
1282            repr: UnsafeCell::new(ThunkRepr::Suspended { expr, env }),
1283            recursive: false,
1284        }))
1285    }
1286
1287    /// Like [`new_suspended`] but marks the thunk as self-recursive.
1288    /// On force, inner re-entrance returns the partial value from the
1289    /// promise cell instead of erroring with `InfiniteRecursion`,
1290    /// matching cppnix's `let x = f x; in x` semantics.  Use this for
1291    /// let-bindings whose RHS textually references the bound name
1292    /// (see `eval::is_self_recursive_binding`).
1293    pub fn new_suspended_recursive(expr: rnix::ast::Expr, env: Env) -> Self {
1294        crate::trace::inc_thunks_created();
1295        census::made(&census::THUNK_MADE, &census::THUNK_LIVE);
1296        crate::perf::inc(crate::perf::Counter::ThunkSiteLetForward);
1297        Self(Rc::new(ThunkInner {
1298            cache: OnceCell::new(),
1299            repr: UnsafeCell::new(ThunkRepr::Suspended { expr, env }),
1300            recursive: true,
1301        }))
1302    }
1303
1304    /// Create a thunk that, when forced, forces the shared
1305    /// `source_thunk` and pulls out the attribute named `name`.
1306    ///
1307    /// The caller creates ONE `Thunk::new_suspended(source_expr, env)`
1308    /// per `inherit (source)` clause and passes clones (Rc bump) to
1309    /// each inherited name.  This way the source is evaluated at most
1310    /// once regardless of how many names are inherited.
1311    pub fn new_inherit_select(source_thunk: Thunk, name: impl Into<SmolStr>) -> Self {
1312        crate::trace::inc_thunks_created();
1313        census::made(&census::THUNK_MADE, &census::THUNK_LIVE);
1314        crate::perf::inc(crate::perf::Counter::ThunkSiteInheritSrc);
1315        Self(Rc::new(ThunkInner {
1316            cache: OnceCell::new(),
1317            repr: UnsafeCell::new(ThunkRepr::InheritSelect {
1318                source_thunk,
1319                name: name.into(),
1320            }),
1321            recursive: false,
1322        }))
1323    }
1324
1325    /// Create a WithIdent thunk — a deferred with-scope ident lookup.
1326    /// Stores a direct reference to the shared with-scope cache.
1327    /// When forced: O(1) hash lookup in the cache, no Env traversal.
1328    pub fn new_with_ident(
1329        name: SmolStr,
1330        scope_cache: Rc<RefCell<Option<NixAttrs>>>,
1331        scope_value: Value,
1332        env: Env,
1333    ) -> Self {
1334        crate::trace::inc_thunks_created();
1335        census::made(&census::THUNK_MADE, &census::THUNK_LIVE);
1336        crate::perf::inc(crate::perf::Counter::ThunkSiteOther);
1337        Self(Rc::new(ThunkInner {
1338            cache: OnceCell::new(),
1339            repr: UnsafeCell::new(ThunkRepr::WithIdent {
1340                name,
1341                scope_cache,
1342                scope_value,
1343                env,
1344            }),
1345            recursive: false,
1346        }))
1347    }
1348
1349    /// Create a thunk backed by a Rust closure.  When forced, the
1350    /// closure is called exactly once and its result is memoized.
1351    /// This is used for lazy flake input evaluation.
1352    pub fn new_native(f: impl FnOnce() -> Result<Value, EvalError> + 'static) -> Self {
1353        crate::trace::inc_thunks_created();
1354        census::made(&census::THUNK_MADE, &census::THUNK_LIVE);
1355        crate::perf::inc(crate::perf::Counter::ThunkSiteNative);
1356        Self(Rc::new(ThunkInner {
1357            cache: OnceCell::new(),
1358            repr: UnsafeCell::new(ThunkRepr::Native(Box::new(f))),
1359            recursive: false,
1360        }))
1361    }
1362
1363    /// Create a thunk that is already evaluated (an optimization).
1364    /// Pre-populates the `OnceCell` cache so the fast path is
1365    /// immediately available.
1366    pub fn new_evaluated(value: Value) -> Self {
1367        crate::trace::inc_thunks_created();
1368        census::made(&census::THUNK_MADE, &census::THUNK_LIVE);
1369        crate::perf::inc(crate::perf::Counter::ThunkSiteEvaluated);
1370        let cache = OnceCell::new();
1371        // Collapse the double-store: a concrete value lives ONLY in the
1372        // cache with an `EvaluatedConcrete` marker repr; a thunk value
1373        // keeps the boxed `Evaluated` repr and an empty cache.
1374        let repr = if matches!(value, Value::Thunk(_)) {
1375            ThunkRepr::Evaluated(Box::new(value))
1376        } else {
1377            let _ = cache.set(Box::new(value.demand_unchecked()));
1378            ThunkRepr::EvaluatedConcrete
1379        };
1380        Self(Rc::new(ThunkInner {
1381            cache,
1382            repr: UnsafeCell::new(repr),
1383            recursive: false,
1384        }))
1385    }
1386
1387    /// Check whether this thunk has already been forced.
1388    /// Uses the `OnceCell` cache for a fast, borrow-free check.
1389    pub fn is_evaluated(&self) -> bool {
1390        self.0.cache.get().is_some()
1391    }
1392
1393    /// Check whether this thunk is a native (Rust closure) thunk.
1394    ///
1395    /// Native thunks are used for lazy flake input evaluation and can
1396    /// be very expensive to force (e.g., evaluating all of nixpkgs).
1397    /// This lets callers skip them in eager conversion paths.
1398    pub fn is_native(&self) -> bool {
1399        // SAFETY: Single-threaded evaluator (Rc, not Arc). Read-only access,
1400        // no mutable reference exists at this point.
1401        matches!(unsafe { &*self.0.repr.get() }, ThunkRepr::Native(_))
1402    }
1403
1404    /// Peek at the cached value WITHOUT forcing.
1405    /// Returns Some(&Value) if the thunk has been evaluated, None otherwise.
1406    /// This is used by with-scope lookup to check if the fixpoint thunk
1407    /// has already been resolved (by another evaluation path) without
1408    /// entering the force state machine.
1409    pub fn peek(&self) -> Option<&Concrete> {
1410        self.0.cache.get().map(|v| &**v)
1411    }
1412
1413    /// Replace the environment captured in a suspended thunk.
1414    /// For `InheritSelect`, delegates to the shared source thunk's
1415    /// `update_env` (which updates the source's captured env).
1416    /// No-op if the thunk is already evaluated or a blackhole.
1417    pub fn update_env(&self, new_env: &Env) {
1418        // SAFETY: Single-threaded evaluator. No other reference to repr
1419        // exists during env replacement.
1420        let repr = unsafe { &mut *self.0.repr.get() };
1421        match repr {
1422            ThunkRepr::Suspended { env, .. } => {
1423                *env = new_env.clone();
1424            }
1425            ThunkRepr::InheritSelect { source_thunk, .. } => {
1426                source_thunk.update_env(new_env);
1427            }
1428            _ => {}
1429        }
1430    }
1431
1432    /// Store a forced result into this thunk's terminal state, collapsing
1433    /// the former thunk double-store.
1434    ///
1435    /// - A CONCRETE (non-thunk) result is stored ONLY in the `cache`
1436    ///   `OnceCell` (`Box<Concrete>`), and `repr` becomes the valueless
1437    ///   `EvaluatedConcrete` marker — freeing the redundant
1438    ///   `Box<Value>` that `Evaluated` used to hold. Reconstruction via
1439    ///   `Concrete::into_value()` is a byte-identical inverse of the
1440    ///   `demand_unchecked()` used to fill the cache.
1441    /// - A THUNK result keeps `repr = Evaluated(Box<Value>)` and leaves
1442    ///   the cache empty (caching a thunk would spin `force_value`).
1443    ///
1444    /// SAFETY: single-threaded evaluator (`Rc`, not `Arc`); the caller
1445    /// must hold no other borrow of `repr` — every call site here is on
1446    /// the sequential `Suspended → Blackhole/Promise → Evaluated`
1447    /// transition, so no overlapping mutable access exists.
1448    ///
1449    /// Takes `&Value` and clones exactly as the former open-coded stores
1450    /// did (`Box::new(value.clone())` for the thunk repr,
1451    /// `Box::new(value.clone().demand_unchecked())` for the cache) — so the
1452    /// clone count is identical to the pre-collapse code and the change is
1453    /// byte-neutral by construction.
1454    #[inline]
1455    unsafe fn store_evaluated(&self, value: &Value) {
1456        census::evaluated();
1457        if matches!(value, Value::Thunk(_)) {
1458            *unsafe { &mut *self.0.repr.get() } = ThunkRepr::Evaluated(Box::new(value.clone()));
1459        } else {
1460            let _ = self.0.cache.set(Box::new(value.clone().demand_unchecked()));
1461            *unsafe { &mut *self.0.repr.get() } = ThunkRepr::EvaluatedConcrete;
1462        }
1463    }
1464
1465    /// Owned-value variant of the concrete branch of [`store_evaluated`],
1466    /// for the `force_inner` early-return that OWNS `value` and does not
1467    /// need it afterward.
1468    ///
1469    /// `store_evaluated(&value)` clones the whole `Value` to fill the
1470    /// cache (`Box::new(value.clone().demand_unchecked())`) and the caller
1471    /// then returns the owned `value` separately — an extra *outer*
1472    /// `Value` clone. Here we MOVE `value` into the cache (no outer clone)
1473    /// and clone the cheaper inner `Concrete` for the return, trading one
1474    /// `Value::clone` for one `Concrete::clone` (the same inner `Rc` bumps,
1475    /// one fewer throwaway `Value` temporary).
1476    ///
1477    /// Content-, order-, and census-neutral versus the
1478    /// `store_evaluated(&value); return Ok(value)` it replaces: the cache
1479    /// holds the identical `Box<Concrete>`, `repr` becomes the identical
1480    /// `EvaluatedConcrete` marker, `census::evaluated()` fires exactly
1481    /// once, and the returned `Value` is a byte-identical reconstruction
1482    /// of `value`.
1483    ///
1484    /// Panics (via `demand_unchecked`) if `value` is a `Thunk` — the sole
1485    /// call site only reaches it on the `!was_thunk_before_loop` branch,
1486    /// where `value` is guaranteed non-`Thunk`.
1487    ///
1488    /// SAFETY: same contract as [`store_evaluated`] — single-threaded,
1489    /// no overlapping `repr` borrow.
1490    #[inline]
1491    unsafe fn store_evaluated_owned(&self, value: Value) -> Value {
1492        census::evaluated();
1493        let concrete = value.demand_unchecked();
1494        let ret = concrete.clone().into_value();
1495        let _ = self.0.cache.set(Box::new(concrete));
1496        *unsafe { &mut *self.0.repr.get() } = ThunkRepr::EvaluatedConcrete;
1497        ret
1498    }
1499
1500    /// Force this thunk using the given evaluator function.
1501    ///
1502    /// On first force: transitions Suspended -> Blackhole -> Evaluated.
1503    /// Re-entering a Blackhole signals infinite recursion.
1504    /// If the evaluated result is itself a thunk, it is forced transitively.
1505    ///
1506    /// Uses `stacker::maybe_grow` to ensure sufficient stack space for
1507    /// deeply nested thunk chains (e.g., nixpkgs overlay fixpoints).
1508    pub fn force(
1509        &self,
1510        evaluator: &dyn Fn(&rnix::ast::Expr, &Env) -> Result<Value, EvalError>,
1511    ) -> Result<Value, EvalError> {
1512        // Ultra-fast path: if already evaluated, return cached value
1513        // WITHOUT entering stacker::maybe_grow. This avoids the stack
1514        // check overhead on ~150M cache hits during nixpkgs evaluation.
1515        if let Some(cached) = self.0.cache.get() {
1516            crate::perf::inc(crate::perf::Counter::ThunkHit);
1517            return Ok((**cached).clone().into_value());
1518        }
1519        // Cold path: evaluation may recurse deeply, so use stacker.
1520        stacker::maybe_grow(64 * 1024, 2 * 1024 * 1024, || {
1521            self.force_inner(evaluator)
1522        })
1523    }
1524
1525    /// Inner implementation of [`Thunk::force`] — called from the
1526    /// `stacker` trampoline.
1527    fn force_inner(
1528        &self,
1529        evaluator: &dyn Fn(&rnix::ast::Expr, &Env) -> Result<Value, EvalError>,
1530    ) -> Result<Value, EvalError> {
1531        // SAFETY (all `unsafe` blocks in this method): The evaluator is
1532        // single-threaded (`Rc`, not `Arc`).  `ThunkInner` is `!Send`/`!Sync`.
1533        // The `OnceCell` fast path handles all concurrent-safe reads (150M+
1534        // hits).  Only the cold path (1.8M forces) touches the `UnsafeCell`.
1535        // The state machine guarantees no overlapping mutable access:
1536        // Suspended → Blackhole → Evaluated transitions are sequential.
1537
1538        // Ultra-fast path: check OnceCell cache (no borrow).
1539        if let Some(cached) = self.0.cache.get() {
1540            crate::perf::inc(crate::perf::Counter::ThunkHit);
1541            return Ok((**cached).clone().into_value());
1542        }
1543
1544        let thunk_id = Rc::as_ptr(&self.0) as usize;
1545
1546        // Promise fast-path: if this thunk is currently in `Promise`
1547        // state (a self-recursive fix-point whose outer body is still
1548        // running, and *this* call is an inner re-entrance), return
1549        // the cell's current partial value without consuming the
1550        // repr.  Matches cppnix's `let x = f x; in x` semantics:
1551        // inner accesses to `x` during f's evaluation see the not-
1552        // yet-complete value instead of erroring with
1553        // `InfiniteRecursion`.
1554        //
1555        // SAFETY: Single-threaded evaluator. The immutable borrow
1556        // is scoped to this `if let` block; the early return exits
1557        // before any further access to `repr`.
1558        if let ThunkRepr::Promise(cell) = unsafe { &*self.0.repr.get() } {
1559            return Ok(cell.borrow().clone());
1560        }
1561
1562        // Take the current repr.  Replace with `Promise(cell)` if the
1563        // thunk is self-recursive (so inner re-entrance during body
1564        // evaluation hits the fast-path above), otherwise classic
1565        // `Blackhole` (so inner re-entrance errors with
1566        // `InfiniteRecursion`, which is the correct behaviour for
1567        // non-recursive bindings like `let r = r; in r`).
1568        // SAFETY: Single-threaded evaluator. State machine ensures no
1569        // overlapping mutable access: Suspended->Blackhole/Promise->Evaluated.
1570        let new_repr_on_force = if self.0.recursive {
1571            ThunkRepr::Promise(Rc::new(RefCell::new(
1572                Value::Attrs(Rc::new(NixAttrs::new())),
1573            )))
1574        } else {
1575            ThunkRepr::Blackhole
1576        };
1577        let is_promise = self.0.recursive;
1578        let repr = std::mem::replace(unsafe { &mut *self.0.repr.get() }, new_repr_on_force);
1579
1580        match repr {
1581            ThunkRepr::Suspended { expr, env } => {
1582                crate::perf::inc(crate::perf::Counter::ThunkForce);
1583                crate::trace::inc_thunks_forced_unique();
1584                let tracing = crate::trace::trace_enabled();
1585                // Always push a force frame — `pop_force` is matched in every
1586                // exit path below.  This keeps the cycle chain on
1587                // `EvalError::InfiniteRecursion` populated WITHOUT requiring
1588                // the operator to set `SUI_TRACE_EVAL=verbose` first.  In
1589                // tracing mode we also capture the (expensive) source-text
1590                // description; otherwise we keep the frame cheap (just the
1591                // file + thunk id) so the always-on overhead stays bounded.
1592                let desc: String = if tracing {
1593                    expr.syntax().text().to_string().chars().take(60).collect()
1594                } else {
1595                    String::new()
1596                };
1597                crate::trace::push_force(crate::trace::ForceFrame {
1598                    defined_in: env.eval_file().cloned(),
1599                    description: desc.clone(),
1600                    thunk_id,
1601                });
1602                // Runaway backstop #1 (force-stack depth) for overlay-fixpoint
1603                // promotion (release-active; belt-and-suspenders with the
1604                // eval-depth backstop in `eval::DepthGuard::enter`).
1605                //
1606                // A promoted empty-attrs partial is byte-correct for the
1607                // native-system stdenv fixpoint (`libxcrypt` — the actual
1608                // byte-parity root; its promotions bottom out at a force depth
1609                // ≤ ~50), but is the WRONG partial for a demand that indexes it
1610                // as a list / non-attrs (the cross-system Darwin `apple-sdk`
1611                // path `hello` hits when `builtins.currentSystem` is macOS).
1612                // There the empty partial feeds a downstream `makeOverridable`
1613                // fixpoint that recurses without bound.  Release disables the
1614                // general `MAX_EVAL_DEPTH` guard (`usize::MAX`) to admit
1615                // nixpkgs' legitimately-deep fixpoints, so nothing else stops
1616                // that recursion before the OS stack aborts.
1617                //
1618                // Armed only once a promotion has fired (`promotion_occurred()`)
1619                // and for the REST of the eval — a corrupted partial can send a
1620                // downstream fixpoint runaway AFTER the promoting force returns,
1621                // so the backstop must outlive the promotion's own softening
1622                // scope.  A runaway that climbs the force stack is caught here;
1623                // one that climbs `eval_expr` without pushing force frames is
1624                // caught by the eval-depth backstop.  Either converts the
1625                // would-be native abort into a recoverable `InfiniteRecursion`
1626                // (which `x.y or default` recovers exactly like nix).
1627                if crate::value::promotion_occurred()
1628                    && crate::trace::current_force_depth() as usize
1629                        > PROMOTION_RUNAWAY_FORCE_DEPTH
1630                {
1631                    crate::trace::pop_force();
1632                    *unsafe { &mut *self.0.repr.get() } =
1633                        ThunkRepr::Suspended { expr, env };
1634                    return Err(EvalError::InfiniteRecursion(
1635                        "overlay-fixpoint promotion runaway (force depth exceeded)".into(),
1636                    ));
1637                }
1638                if tracing {
1639                    crate::trace::trace_force_enter(
1640                        env.eval_file().map(|p| p.as_path()),
1641                        &desc,
1642                    );
1643                    if let Err(msg) = crate::trace::check_force_depth() {
1644                        crate::trace::dump_trace_on_error();
1645                        crate::trace::pop_force();
1646                        crate::trace::trace_force_exit();
1647                        *unsafe { &mut *self.0.repr.get() } = ThunkRepr::Suspended {
1648                            expr,
1649                            env,
1650                        };
1651                        return Err(EvalError::InfiniteRecursion(msg));
1652                    }
1653                }
1654                // Push the thunk's captured eval_file onto the thread-local
1655                // stack so PathRel literals and relative imports inside the
1656                // thunk body resolve against the file where the thunk was
1657                // *defined*, not where it is forced from. The RAII guard
1658                // pops on drop (including on error paths).
1659                let _file_guard = env.eval_file().cloned().map(crate::eval::push_eval_file);
1660                // Restore the thunk's DEFINING source_id in lockstep with
1661                // eval_file above, so idents evaluated in the thunk body key
1662                // the `(source_id, offset)` symbol cache against the file the
1663                // thunk was defined in — not the ambient source at force time.
1664                // Without this a cross-file force (a lazy thunk from an
1665                // imported file, forced after `eval_with_file` restored the
1666                // top-level source_id) collides on a reused offset and returns
1667                // a wrong Symbol (`parse.nix` `cannot select from null`).
1668                let _srcid_guard = crate::eval::push_source_id(env.source_id());
1669                // M2.6 Promise scope: bump the thread-local counter so
1670                // downstream `eval_select` can soften `AttrNotFound`
1671                // errors on the Promise's sentinel value to `null`.
1672                // Scoped strictly to Promise-thunk body evaluation;
1673                // non-recursive thunks retain cppnix-strict semantics.
1674                if is_promise {
1675                    IN_PROMISE_EVAL.with(|c| c.set(c.get() + 1));
1676                }
1677                let result = evaluator(&expr, &env);
1678                if is_promise {
1679                    IN_PROMISE_EVAL.with(|c| c.set(c.get().saturating_sub(1)));
1680                }
1681                // A `Blackhole` thunk (non-recursive at construction) may have
1682                // been PROMOTED to `Promise` mid-body by a same-thunk fixpoint
1683                // re-entry (the overlay-fixpoint path in the Blackhole arm
1684                // below).  That promotion bumped `IN_PROMISE_EVAL` once; balance
1685                // it here, and populate its cell exactly like a
1686                // recursive-at-construction Promise.  `is_promise` covers the
1687                // construction-time case; `became_promise` covers the mid-body
1688                // semantic-promotion case.  They're mutually exclusive (a
1689                // construction-time Promise never re-enters the Blackhole arm).
1690                let became_promise = !is_promise
1691                    && matches!(unsafe { &*self.0.repr.get() }, ThunkRepr::Promise(_));
1692                if became_promise {
1693                    IN_PROMISE_EVAL.with(|c| c.set(c.get().saturating_sub(1)));
1694                }
1695                match result {
1696                    Ok(mut value) => {
1697                        crate::perf::inc(crate::perf::Counter::ThunkStoreWrites);
1698                        // M2.6 Promise update: if this thunk transitioned
1699                        // through Promise(cell), populate the cell with the
1700                        // final value BEFORE setting Evaluated.  Any
1701                        // outstanding Rc clones of the cell (held by inner
1702                        // thunks whose bodies haven't yet run) will see the
1703                        // complete value when they later force.
1704                        if is_promise || became_promise {
1705                            if let ThunkRepr::Promise(cell) = unsafe { &*self.0.repr.get() } {
1706                                *cell.borrow_mut() = value.clone();
1707                            }
1708                        }
1709                        // Whether the body returned a Thunk decides the store
1710                        // shape.  Computed BEFORE the store so the non-thunk
1711                        // path can MOVE `value` into the cache (owned store)
1712                        // instead of cloning it (see `store_evaluated_owned`).
1713                        //
1714                        // C-store PROVABLY-NEUTRAL narrow win (M2, byte-verified):
1715                        // when `value` is NOT a Thunk, the collapse loop below
1716                        // does not execute (its guard is `while let Value::Thunk`),
1717                        // so the second store (in the thunk branch) would rewrite
1718                        // BYTE-IDENTICAL repr content and re-attempt a no-op
1719                        // OnceCell `cache.set`.  Skipping it is content-AND-order-
1720                        // neutral: the single store already established the
1721                        // terminal (cache=concrete, repr=EvaluatedConcrete);
1722                        // nothing between the stores observes `self.0.repr` (the
1723                        // body has returned — no re-entrant force of self is in
1724                        // flight; the loop only `peek()`s OTHER thunks' OnceCell
1725                        // caches, never self's repr), and no code observes the
1726                        // `Box`'s pointer identity (repr is only ever read by
1727                        // value — grep-confirmed). Only when `value` IS a Thunk
1728                        // (the loop may collapse it to a different concrete) do we
1729                        // re-store the unwrapped result.
1730                        let was_thunk_before_loop = matches!(value, Value::Thunk(_));
1731                        if !was_thunk_before_loop {
1732                            // Non-thunk: single owned store (no outer Value clone),
1733                            // return the reconstruction. Byte-, order-, and census-
1734                            // identical to `store_evaluated(&value); return Ok(value)`
1735                            // (Store#2 is pure redundant and skipped, as before).
1736                            crate::perf::inc(crate::perf::Counter::ThunkStoreRedundant);
1737                            let ret = unsafe { self.store_evaluated_owned(value) };
1738                            crate::trace::pop_force();
1739                            if tracing { crate::trace::trace_force_exit(); }
1740                            return Ok(ret);
1741                        }
1742                        // Thunk path (unchanged): Store#1, collapse loop, Store#2.
1743                        unsafe { self.store_evaluated(&value) };
1744                        // Transitively unwrap thunk-in-thunk chains, with a
1745                        // depth limit to catch `let x = x; in x` cycles.
1746                        // Chase already-resolved thunks only (peek).
1747                        // force_value handles full transitive resolution.
1748                        while let Value::Thunk(ref inner) = value {
1749                            match inner.peek() {
1750                                Some(cached) => value = cached.clone().into_value(),
1751                                None => break,
1752                            }
1753                        }
1754                        if !matches!(value, Value::Thunk(_)) {
1755                            crate::perf::inc(crate::perf::Counter::ThunkStoreLoopMutated);
1756                        }
1757                        unsafe { self.store_evaluated(&value) };
1758                        crate::trace::pop_force();
1759                        if tracing { crate::trace::trace_force_exit(); }
1760                        Ok(value)
1761                    }
1762                    Err(e) => {
1763                        *unsafe { &mut *self.0.repr.get() } = ThunkRepr::Suspended { expr, env };
1764                        if tracing { crate::trace::dump_trace_on_error(); }
1765                        crate::trace::pop_force();
1766                        if tracing { crate::trace::trace_force_exit(); }
1767                        Err(e)
1768                    }
1769                }
1770            }
1771            ThunkRepr::InheritSelect { source_thunk, name } => {
1772                let tracing = crate::trace::trace_enabled();
1773                let desc = if tracing { format!("inherit (..) {name}") } else { String::new() };
1774                crate::trace::push_force(crate::trace::ForceFrame {
1775                    defined_in: None,
1776                    description: desc.clone(),
1777                    thunk_id,
1778                });
1779                if tracing {
1780                    crate::trace::trace_force_enter(None, &desc);
1781                }
1782                crate::trace::inc_thunks_forced_unique();
1783                if tracing {
1784                    if let Err(msg) = crate::trace::check_force_depth() {
1785                        crate::trace::dump_trace_on_error();
1786                        crate::trace::pop_force();
1787                        crate::trace::trace_force_exit();
1788                        *unsafe { &mut *self.0.repr.get() } = ThunkRepr::InheritSelect {
1789                            source_thunk,
1790                            name,
1791                        };
1792                        return Err(EvalError::InfiniteRecursion(msg));
1793                    }
1794                }
1795                let attempt = (|| -> Result<Value, EvalError> {
1796                    let mut forced = source_thunk.force(evaluator)?;
1797                    while let Value::Thunk(inner) = forced {
1798                        forced = inner.force(evaluator)?;
1799                    }
1800                    let attrs = match &forced {
1801                        Value::Attrs(a) => a,
1802                        _ => {
1803                            return Err(EvalError::TypeError(format!(
1804                                "inherit (source) {name}: source is {}, not a set",
1805                                forced.type_name()
1806                            )))
1807                        }
1808                    };
1809                    attrs
1810                        .get(&name)
1811                        .cloned()
1812                        .ok_or_else(|| EvalError::AttrNotFound(name.to_string()))
1813                })();
1814                match attempt {
1815                    Ok(mut value) => {
1816                        *unsafe { &mut *self.0.repr.get() } = ThunkRepr::Evaluated(Box::new(value.clone()));
1817                        while let Value::Thunk(ref inner) = value {
1818                            match inner.peek() { Some(c) => value = c.clone().into_value(), None => break }
1819                        }
1820                        unsafe { self.store_evaluated(&value) };
1821                        crate::trace::pop_force();
1822                        if tracing { crate::trace::trace_force_exit(); }
1823                        Ok(value)
1824                    }
1825                    Err(e) => {
1826                        *unsafe { &mut *self.0.repr.get() } = ThunkRepr::InheritSelect { source_thunk, name };
1827                        if tracing { crate::trace::dump_trace_on_error(); }
1828                        crate::trace::pop_force();
1829                        if tracing { crate::trace::trace_force_exit(); }
1830                        Err(e)
1831                    }
1832                }
1833            }
1834            ThunkRepr::Native(f) => {
1835                let tracing = crate::trace::trace_enabled();
1836                crate::trace::push_force(crate::trace::ForceFrame {
1837                    defined_in: None,
1838                    description: if tracing { "<native-thunk>".into() } else { String::new() },
1839                    thunk_id,
1840                });
1841                if tracing {
1842                    crate::trace::trace_force_enter(None, "<native-thunk>");
1843                }
1844                crate::trace::inc_thunks_forced_unique();
1845                // The closure is consumed (FnOnce).  On success we
1846                // memoize the result.  On failure we leave Blackhole
1847                // — unlike Suspended thunks the closure cannot be
1848                // retried because it has been consumed.
1849                match f() {
1850                    Ok(mut value) => {
1851                        *unsafe { &mut *self.0.repr.get() } = ThunkRepr::Evaluated(Box::new(value.clone()));
1852                        while let Value::Thunk(ref inner) = value {
1853                            match inner.peek() { Some(c) => value = c.clone().into_value(), None => break }
1854                        }
1855                        unsafe { self.store_evaluated(&value) };
1856                        crate::trace::pop_force();
1857                        if tracing { crate::trace::trace_force_exit(); }
1858                        Ok(value)
1859                    }
1860                    Err(e) => {
1861                        // The `FnOnce` closure is consumed and cannot be
1862                        // retried.  Memoize the ERROR (not `Null`): a
1863                        // re-force must re-raise, never silently return a
1864                        // value the first force did not produce.  The old
1865                        // `Evaluated(Null)` here poisoned a flake-input
1866                        // thunk whose first force failed transiently
1867                        // (e.g. a not-yet-cached transitive source) so a
1868                        // later re-read saw `null` — surfacing as a bogus
1869                        // downstream `AttrNotFound` /
1870                        // `cannot select from set` (the stylix
1871                        // `darwinModules` marquee root).  Do NOT populate
1872                        // the OnceCell (there is no correct concrete value
1873                        // to cache); the `Failed` repr arm re-raises.
1874                        *unsafe { &mut *self.0.repr.get() } = ThunkRepr::Failed(e.clone());
1875                        if tracing { crate::trace::dump_trace_on_error(); }
1876                        crate::trace::pop_force();
1877                        if tracing { crate::trace::trace_force_exit(); }
1878                        Err(e)
1879                    }
1880                }
1881            }
1882            ThunkRepr::WithIdent { name, scope_cache, scope_value, env } => {
1883                crate::perf::inc(crate::perf::Counter::ThunkForce);
1884                crate::trace::inc_thunks_forced_unique();
1885                // Fast path: check the shared with-scope cache.
1886                // All WithIdent thunks from the same `with` scope share
1887                // this cache. Once ANY lookup populates it, all others
1888                // are O(1) hash lookups.
1889                {
1890                    let cache = scope_cache.borrow();
1891                    if let Some(ref attrs) = *cache {
1892                        if let Some(v) = attrs.get(&name) {
1893                            let value = v.clone();
1894                            unsafe { self.store_evaluated(&value) };
1895                            return Ok(value);
1896                        }
1897                        // Name not in cached attrset — fall through to env lookup
1898                    }
1899                }
1900                // Cache not populated yet — force the scope value to populate it
1901                if let Ok(forced) = crate::eval::force_value(&scope_value) {
1902                    if let Value::Attrs(ref attrs) = forced {
1903                        *scope_cache.borrow_mut() = Some((**attrs).clone());
1904                        if let Some(v) = attrs.get(&name) {
1905                            let value = v.clone();
1906                            unsafe { self.store_evaluated(&value) };
1907                            return Ok(value);
1908                        }
1909                    }
1910                }
1911                // Name not in with-scope — fall back to full env lookup.
1912                //
1913                // The cache-first with-scope search may have skipped a scope
1914                // whose CACHE is a stale mid-fixpoint PARTIAL — e.g. `f self`
1915                // cached BEFORE makeScope's `self = f self // { callPackage = …; }`
1916                // merged the scope infra in, so `callPackage` is absent from
1917                // the stale partial yet present in the COMPLETED `self`. On any
1918                // lexical-scope miss (both inside and outside a Promise body),
1919                // re-resolve by force_value-ing each with-scope FRESH (bypassing
1920                // the cache) via `lookup_fresh`. It catches errors, so a
1921                // genuinely mid-fixpoint / throwing scope simply skips and
1922                // returns None — leaving the Promise-body null softening (below)
1923                // for the case where the with-source really IS the empty-attrset
1924                // sentinel. A completed value always wins over the null sentinel.
1925                //
1926                // This is the SAME class as the neovim/python27
1927                // `with self; with super; callPackage` root, but reached through
1928                // the resholve `python27' = (…).override { self = python27'; }`
1929                // recursive-fixpoint hooks scope, where the miss lands inside a
1930                // Promise body (`in_promise_eval()` true) and was previously
1931                // softened to `null` BEFORE `lookup_fresh` ran — silently
1932                // dropping `pip = callPackage …` (→ empty `propagatedBuildInputs`
1933                // on `pip-install-hook.drv`). Trying the completed-`self`
1934                // resolution first restores the drop.
1935                //
1936                // Byte-neutral: `lookup_fresh` only ever returns a value nix's
1937                // single lazy `self` would ALSO expose; when it misses (genuine
1938                // empty-partial sentinel) the softening / error behavior below is
1939                // exactly as before.
1940                let result = match env.lookup(&name) {
1941                    Some(v) => v,
1942                    None => match env.lookup_fresh(&name) {
1943                        Some(v) => v,
1944                        None if in_promise_eval() => Value::Null,
1945                        None => return Err(EvalError::UndefinedVar(format!("'{name}'"))),
1946                    },
1947                };
1948                unsafe { self.store_evaluated(&result) };
1949                Ok(result)
1950            }
1951            ThunkRepr::Blackhole => {
1952                // M2.6 bridge: when an inner force re-enters a thunk
1953                // that's currently being evaluated, cppnix's effective
1954                // behavior is to expose the not-yet-complete value
1955                // (typically a partial attrset).  Without the proper
1956                // `Promise(NixAttrs)` thunk variant (see
1957                // docs/M2.6-MODULE-SYSTEM-FIXPOINT.md::Genuine fix),
1958                // we approximate by returning a sentinel of the
1959                // operator's choice:
1960                //
1961                //   SUI_BLACKHOLE_AS_NULL=1         → Value::Null
1962                //   SUI_BLACKHOLE_AS_EMPTY_ATTRS=1  → Value::Attrs({})
1963                //   SUI_BLACKHOLE_AS_EMPTY_LIST=1   → Value::List([])
1964                //
1965                // `EMPTY_ATTRS` is the closest approximation for the
1966                // NixOS module-system fix-point because the cppnix
1967                // partial is itself an attrset — downstream
1968                // `mapAttrs`/`attrNames`/`concatMap` on the sentinel
1969                // see "no keys to map" rather than a type error.
1970                //
1971                // Default-off for all variants because each silently
1972                // hides legitimate cycles in user code (`let r = r;
1973                // in r.x` would return missing-attr or 0 instead of
1974                // erroring).
1975                if std::env::var_os("SUI_BLACKHOLE_AS_NULL").is_some() {
1976                    return Ok(Value::Null);
1977                }
1978                if std::env::var_os("SUI_BLACKHOLE_AS_EMPTY_LIST").is_some() {
1979                    return Ok(Value::List(Rc::new(NixList::new(Vec::new()))));
1980                }
1981                if std::env::var_os("SUI_BLACKHOLE_AS_EMPTY_ATTRS").is_some() {
1982                    return Ok(Value::Attrs(Rc::new(NixAttrs::new())));
1983                }
1984                if std::env::var_os("SUI_DEBUG_CYCLE").is_some() {
1985                    let same = crate::trace::force_stack_contains(thunk_id);
1986                    eprintln!(
1987                        "[SUI_DEBUG_CYCLE] blackhole re-entry thunk_id={thunk_id:#x} same_thunk_on_stack={same} recursive_flag={}",
1988                        self.0.recursive
1989                    );
1990                    crate::trace::dump_force_stack_ids();
1991                }
1992                // OVERLAY-FIXPOINT SEMANTIC PROMOTION (2026-07-10, default-ON).
1993                //
1994                // When the re-entered thunk is the SAME thunk currently
1995                // mid-evaluation on the force stack, this is a genuine fixpoint
1996                // self-reference — the nixpkgs `self:super:` overlay / `lib.fix`
1997                // pattern threading through `callPackage`/`self`/`super` across
1998                // file boundaries.  `is_self_recursive_binding` (syntactic RHS
1999                // name search) MISSES this because the binding's RHS never
2000                // textually names itself, so the thunk was classified
2001                // `recursive=false` and installed a hard `Blackhole` where nix
2002                // exposes the not-yet-complete value.  That misclassification is
2003                // exactly the byte-parity defect (`sui-spec/src/laziness.rs`
2004                // `RecursionKind::Fixpoint` ⇒ MUST be recursive + Promise): the
2005                // dropped perl `nativeBuildInput` on `pkgs.libxcrypt` (sui
2006                // q9b9v7a9… vs nix jb9k6090…).
2007                //
2008                // The FIX is the Blackhole↔Promise machinery, not a sentinel:
2009                // retroactively PROMOTE this Blackhole to a real `Promise(cell)`
2010                // and return the cell's in-progress partial.  Unlike the earlier
2011                // blank-empty-attrs sentinel (which left the thunk in Blackhole
2012                // forever and stack-overflowed `hello`), the promoted cell is a
2013                // first-class fixpoint cell:
2014                //   * the outer body populates it on completion (the
2015                //     `is_promise || became_promise` branch below), so any inner
2016                //     Rc clones that already read the empty partial converge, and
2017                //     the repr transitions cleanly to `Evaluated`;
2018                //   * `IN_PROMISE_EVAL` is bumped so downstream `eval_select`
2019                //     softens `AttrNotFound`/`cannot-select` on the partial to
2020                //     `null` (the `x.y or default` fall-through nix relies on),
2021                //     which is what stops the `hello` overflow.
2022                //
2023                // Genuine NON-terminating cycles (`let r = r; in r`) remain
2024                // errors: the promoted partial cannot make progress, so the
2025                // force-depth backstop (`check_force_depth`, ~2048/100 in
2026                // test/release) still fires `InfiniteRecursion` — nix's own
2027                // behaviour.  This is the semantic (fixpoint) classification the
2028                // typed discipline demands, done in the demand-order engine
2029                // instead of at syntactic construction time.
2030                if crate::trace::force_stack_contains(thunk_id)
2031                    && IN_PROMISE_EVAL.with(|c| c.get()) < FIXPOINT_PROMOTE_NEST_CAP
2032                {
2033                    if std::env::var_os("SUI_DEBUG_CYCLE").is_some() {
2034                        let chain = crate::trace::capture_cycle(thunk_id);
2035                        let nest = IN_PROMISE_EVAL.with(|c| c.get());
2036                        let fdepth = crate::trace::current_force_depth();
2037                        eprintln!("[SUI_PROMOTE] thunk_id={thunk_id:#x} cycle_len={} nest={nest} fdepth={fdepth}", chain.0.len());
2038                    }
2039                    let cell = Rc::new(RefCell::new(
2040                        Value::Attrs(Rc::new(NixAttrs::new())),
2041                    ));
2042                    // SAFETY: single-threaded evaluator; we hold no other borrow
2043                    // of `repr` here (the outer match consumed it, we replace it).
2044                    *unsafe { &mut *self.0.repr.get() } =
2045                        ThunkRepr::Promise(cell.clone());
2046                    // Enable Promise-body softening for the remainder of the
2047                    // outer force.  Decremented once by the outer force's
2048                    // post-body reconciliation (`became_promise`).
2049                    IN_PROMISE_EVAL.with(|c| c.set(c.get() + 1));
2050                    // Arm the release runaway backstop for the rest of the eval.
2051                    PROMOTION_OCCURRED.with(|c| c.set(true));
2052                    return Ok(cell.borrow().clone());
2053                }
2054                let chain = crate::trace::capture_cycle(thunk_id);
2055                crate::trace::dump_trace_on_error();
2056                Err(EvalError::InfiniteRecursion(chain.to_string()))
2057            }
2058            ThunkRepr::Promise(cell) => {
2059                // Inner re-entrance into a self-recursive thunk that's
2060                // currently being evaluated.  Return the partial value
2061                // the body has constructed so far (the cell starts as
2062                // `Value::Attrs(empty)` and gets updated on body return).
2063                // This is sui's cppnix-equivalent for `let x = f x; in x`
2064                // — the inner reference to `x` during f's evaluation
2065                // sees a partial attrset instead of the original cycle's
2066                // `InfiniteRecursion`.
2067                Ok(cell.borrow().clone())
2068            }
2069            ThunkRepr::Evaluated(v) => {
2070                // Reached when OnceCell wasn't populated (value was a thunk
2071                // when first evaluated). Cache only concrete values — caching
2072                // a thunk would cause force_value's loop to spin.
2073                crate::perf::inc(crate::perf::Counter::ThunkHit);
2074                let cloned = (*v).clone();
2075                if !matches!(cloned, Value::Thunk(_)) {
2076                    if !matches!(cloned, Value::Thunk(_)) { let _ = self.0.cache.set(Box::new(cloned.clone().demand_unchecked())); }
2077                }
2078                *unsafe { &mut *self.0.repr.get() } = ThunkRepr::Evaluated(v);
2079                Ok(cloned)
2080            }
2081            ThunkRepr::EvaluatedConcrete => {
2082                // The concrete value lives in `cache`; the repr is a valueless
2083                // marker (the collapsed former double-store). In practice this
2084                // arm is unreachable: `force`/`force_inner` check the `cache`
2085                // fast path BEFORE the `mem::replace` that consumes the repr,
2086                // and `EvaluatedConcrete` always co-occurs with a populated
2087                // cache — so the fast path returns first. Handle it faithfully
2088                // anyway: reconstruct the `Value` from the cache (a byte-
2089                // identical inverse of `demand_unchecked`) and restore the
2090                // marker (the outer `mem::replace` swapped in Blackhole/Promise).
2091                crate::perf::inc(crate::perf::Counter::ThunkHit);
2092                let value = self
2093                    .0
2094                    .cache
2095                    .get()
2096                    .expect("EvaluatedConcrete implies a populated cache")
2097                    .as_ref()
2098                    .clone()
2099                    .into_value();
2100                *unsafe { &mut *self.0.repr.get() } = ThunkRepr::EvaluatedConcrete;
2101                Ok(value)
2102            }
2103            ThunkRepr::Failed(e) => {
2104                // A previously-forced `Native` thunk whose closure threw.
2105                // Re-raise the memoized error — never fall through to a
2106                // silent value.  Restore the repr (the outer
2107                // `mem::replace` swapped in Blackhole/Promise).
2108                let err = e.clone();
2109                *unsafe { &mut *self.0.repr.get() } = ThunkRepr::Failed(e);
2110                Err(err)
2111            }
2112        }
2113    }
2114}
2115
2116impl fmt::Debug for Thunk {
2117    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2118        // SAFETY: Single-threaded evaluator, read-only access during formatting.
2119        match unsafe { &*self.0.repr.get() } {
2120            ThunkRepr::Suspended { .. } => write!(f, "<thunk>"),
2121            ThunkRepr::InheritSelect { name, .. } => write!(f, "<inherit-select {name}>"),
2122            ThunkRepr::Native(_) => write!(f, "<native-thunk>"),
2123            ThunkRepr::WithIdent { name, .. } => write!(f, "<with-ident {name}>"),
2124            ThunkRepr::Blackhole => write!(f, "<blackhole>"),
2125            ThunkRepr::Promise(_) => write!(f, "<promise>"),
2126            ThunkRepr::Failed(e) => write!(f, "<failed-thunk: {e}>"),
2127            ThunkRepr::Evaluated(v) => write!(f, "{v:?}"),
2128            ThunkRepr::EvaluatedConcrete => match self.0.cache.get() {
2129                Some(c) => write!(f, "{:?}", c.as_ref().clone().into_value()),
2130                None => write!(f, "<evaluated-concrete>"),
2131            },
2132        }
2133    }
2134}
2135
2136/// A Nix attribute set with lazy overlay support.
2137///
2138/// Internally uses either a concrete compact `AttrsMap` or a lazy overlay chain.
2139/// The `//` operator creates O(1) overlay nodes instead of O(m log n) merges.
2140/// Attribute access walks the chain right-to-left in O(depth).
2141/// Full iteration (attrNames, attrValues) flattens on demand.
2142///
2143/// The second tuple field is an OPTIONAL source-position table (`None` for
2144/// the vast majority of attrsets — merges, overlays, builtin-built, dynamic
2145/// keys). `eval_attrset` attaches it for a literal with static keys so
2146/// `builtins.unsafeGetAttrPos` can report a key's file/line/column (the
2147/// `attrTag` `declarations` — options.json dock root). It is behind `Rc`, so
2148/// a clone is a refcount bump; `None` costs one pointer-sized word.
2149pub struct NixAttrs(AttrsInner, Option<Rc<crate::pos::AttrPositions>>);
2150
2151// Hand-written `Clone`/`Drop` so the census counts every NixAttrs value that
2152// comes into existence (a clone is a fresh heap object once Rc-wrapped),
2153// keeping `ATTRS_MADE`/`ATTRS_LIVE` consistent. Fresh (non-clone)
2154// constructions bump the counter at each `NixAttrs(...)` tuple-construct site.
2155impl Clone for NixAttrs {
2156    fn clone(&self) -> Self {
2157        census::made(&census::ATTRS_MADE, &census::ATTRS_LIVE);
2158        NixAttrs(self.0.clone(), self.1.clone())
2159    }
2160}
2161
2162impl Drop for NixAttrs {
2163    fn drop(&mut self) {
2164        census::dropped(&census::ATTRS_LIVE);
2165    }
2166}
2167
2168/// Internal representation: either a flat map or an overlay chain.
2169#[derive(Clone)]
2170enum AttrsInner {
2171    /// Concrete attribute set — compact flat `AttrsMap` (std hashbrown).
2172    Flat(AttrsMap<Symbol, Value>),
2173    /// Lazy overlay: right overrides left. O(1) construction.
2174    /// `cache` is populated on first full iteration (attrNames, etc.).
2175    ///
2176    /// `left`/`right` are interior-mutable so they can be RELEASED (swapped to an
2177    /// empty attrs) once `cache` is populated: after flatten the merged `cache`
2178    /// is the complete answer and every reader (`get_sym`/`contains_key`/
2179    /// `is_empty`) routes through `as_flat()` (the cache), so the un-merged
2180    /// parents are dead weight. Releasing them cascade-frees the intermediate
2181    /// overlay chain (the 50+-deep module-fixpoint retention — `EVAL-MEMORY.md`).
2182    /// Byte-neutral: the cache is the same map nix's flatten yields.
2183    Overlay {
2184        left: RefCell<Rc<NixAttrs>>,
2185        right: RefCell<Rc<NixAttrs>>,
2186        cache: Rc<OnceCell<AttrsMap<Symbol, Value>>>,
2187    },
2188}
2189
2190impl fmt::Debug for NixAttrs {
2191    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2192        write!(f, "NixAttrs({})", self.len())
2193    }
2194}
2195
2196impl Default for NixAttrs {
2197    fn default() -> Self {
2198        census::made(&census::ATTRS_MADE, &census::ATTRS_LIVE);
2199        Self(AttrsInner::Flat(AttrsMap::default()), None)
2200    }
2201}
2202
2203impl NixAttrs {
2204    pub fn new() -> Self {
2205        Self::default()
2206    }
2207
2208    pub fn with_capacity(_capacity: usize) -> Self {
2209        Self::default()
2210    }
2211
2212    /// Attach a source-position table (the static keys' byte offsets of the
2213    /// literal that built this attrset). Called by `eval_attrset`; consumed
2214    /// by `builtins.unsafeGetAttrPos`. Never affects any observed value.
2215    pub fn set_positions(&mut self, pos: Rc<crate::pos::AttrPositions>) {
2216        self.1 = Some(pos);
2217    }
2218
2219    /// The source-position table, if this attrset carries one (a literal with
2220    /// static keys). `None` for merges/overlays/builtin-built/dynamic-key
2221    /// attrsets.
2222    #[must_use]
2223    pub fn positions(&self) -> Option<&Rc<crate::pos::AttrPositions>> {
2224        self.1.as_ref()
2225    }
2226
2227    /// Resolve the source position of `key` in this attrset — the file/line/
2228    /// column `builtins.unsafeGetAttrPos` returns. `None` when the attrset
2229    /// has no position table, the key is absent from it, or the source has
2230    /// no file (a `<string>`-eval'd literal).
2231    #[must_use]
2232    pub fn pos_for(&self, key: &str) -> Option<crate::pos::ResolvedPos> {
2233        let table = self.1.as_ref()?;
2234        let sym = intern(key);
2235        let offset = *table.keys.get(&sym)?;
2236        crate::pos::resolve(table.file.as_deref(), offset)
2237    }
2238
2239    /// Borrow the underlying map. Flattens if overlay.
2240    #[must_use]
2241    pub fn inner(&self) -> AttrsMap<Symbol, Value> {
2242        self.as_flat().clone()
2243    }
2244
2245    /// Get a reference to a flat `AttrsMap`, populating cache if overlay.
2246    fn as_flat(&self) -> &AttrsMap<Symbol, Value> {
2247        match &self.0 {
2248            AttrsInner::Flat(m) => m,
2249            AttrsInner::Overlay { left, right, cache } => {
2250                crate::perf::inc(crate::perf::Counter::OverlayFlattenAttempt);
2251                let flat = cache.get_or_init(|| {
2252                    // Cache MISS: this Overlay node is being flattened for the
2253                    // first time — real O(left+right) merge work.
2254                    crate::perf::inc(crate::perf::Counter::OverlayFlattenBuild);
2255                    let timed = crate::perf::enabled();
2256                    let t0 = if timed { Some(std::time::Instant::now()) } else { None };
2257                    let mut result = left.borrow().as_flat().clone();
2258                    for (k, v) in right.borrow().as_flat().iter() {
2259                        result.insert(*k, v.clone());
2260                    }
2261                    crate::perf::add(
2262                        crate::perf::Counter::OverlayFlattenEntries,
2263                        result.len() as u64,
2264                    );
2265                    if let Some(t0) = t0 {
2266                        crate::trace::add_overlay_flatten_nanos(t0.elapsed().as_nanos());
2267                    }
2268                    result
2269                });
2270                // RELEASE the parents now that `cache` is the complete answer —
2271                // cascade-frees the intermediate overlay chain + their caches
2272                // (nothing else references them). Byte-neutral: every reader now
2273                // routes through this `cache`. Only swaps a still-held parent; a
2274                // second as_flat sees them already empty and skips. The closure's
2275                // borrows above are dropped by here, so these borrow_muts can't
2276                // conflict (single-threaded, sequential).
2277                {
2278                    let mut l = left.borrow_mut();
2279                    if !l.is_empty() { *l = Rc::new(NixAttrs::new()); }
2280                }
2281                {
2282                    let mut r = right.borrow_mut();
2283                    if !r.is_empty() { *r = Rc::new(NixAttrs::new()); }
2284                }
2285                flat
2286            }
2287        }
2288    }
2289
2290    fn sorted_entries(&self) -> Vec<(String, &Value)> {
2291        crate::perf::inc(crate::perf::Counter::SortedEntriesCalls);
2292        let m = self.as_flat();
2293        crate::perf::add(crate::perf::Counter::SortedEntriesRows, m.len() as u64);
2294        let timed = crate::perf::enabled();
2295        let t0 = if timed { Some(std::time::Instant::now()) } else { None };
2296        let mut pairs: Vec<(String, &Value)> = m.iter()
2297            .map(|(sym, v)| (resolve(*sym), v))
2298            .collect();
2299        pairs.sort_by(|(a, _), (b, _)| a.cmp(b));
2300        if let Some(t0) = t0 {
2301            crate::trace::add_sorted_entries_nanos(t0.elapsed().as_nanos());
2302        }
2303        pairs
2304    }
2305
2306    /// Look up an attribute by name. Walks overlay chain right-to-left.
2307    #[must_use]
2308    pub fn get(&self, key: &str) -> Option<&Value> {
2309        let sym = intern(key);
2310        self.get_sym(&sym)
2311    }
2312
2313    /// Look up by pre-interned Symbol.
2314    ///
2315    /// Fast path: if the overlay's flat cache has been populated (by any
2316    /// prior full iteration — `attrNames`, `attrValues`, `//` merge that
2317    /// needed key enumeration, etc.), read directly from it in O(1). This
2318    /// matters in real Nix workloads where an attrset is first iterated
2319    /// (module eval, `with` desugaring) and then hit many times by dotted
2320    /// access — CppNix has no such structure and pays O(1) always; we want
2321    /// to match that whenever the cache is warm.
2322    ///
2323    /// Slow path: walk the overlay chain right-to-left in O(depth). Not
2324    /// populating the cache on cold lookups is deliberate — the cache
2325    /// costs O(n) to build and the chain is usually short (1–3 overlays).
2326    #[must_use]
2327    pub fn get_sym(&self, sym: &Symbol) -> Option<&Value> {
2328        match &self.0 {
2329            AttrsInner::Flat(m) => m.get(sym),
2330            // Route through `as_flat()` (the memoized cache) rather than borrowing
2331            // into `left`/`right` — this is what lets the parents be released
2332            // post-flatten. `as_flat` returns the cached map in O(1) when warm and
2333            // flattens+caches on the first cold lookup; the returned `&Value`
2334            // borrows the stable `cache`, never a `RefCell`.
2335            AttrsInner::Overlay { .. } => self.as_flat().get(sym),
2336        }
2337    }
2338
2339    /// Insert or overwrite an attribute. Flattens overlay if needed.
2340    pub fn insert(&mut self, key: String, value: Value) {
2341        self.ensure_flat();
2342        if let AttrsInner::Flat(ref mut m) = self.0 {
2343            m.insert(intern(&key), value);
2344        }
2345    }
2346
2347    /// Ensure the inner representation is Flat (for mutation).
2348    fn ensure_flat(&mut self) {
2349        if matches!(self.0, AttrsInner::Overlay { .. }) {
2350            self.0 = AttrsInner::Flat(self.as_flat().clone());
2351        }
2352    }
2353
2354    #[must_use]
2355    pub fn contains_key(&self, key: &str) -> bool {
2356        let sym = intern(key);
2357        self.contains_key_sym(&sym)
2358    }
2359
2360    #[must_use]
2361    pub fn contains_key_sym(&self, sym: &Symbol) -> bool {
2362        match &self.0 {
2363            AttrsInner::Flat(m) => m.contains_key(sym),
2364            // Route through the cache (see get_sym) so left/right stay releasable.
2365            AttrsInner::Overlay { .. } => self.as_flat().contains_key(sym),
2366        }
2367    }
2368
2369    pub fn keys(&self) -> impl Iterator<Item = String> {
2370        self.sorted_entries().into_iter().map(|(k, _)| k)
2371    }
2372
2373    pub fn iter(&self) -> impl Iterator<Item = (String, &Value)> {
2374        self.sorted_entries().into_iter()
2375    }
2376
2377    pub fn iter_unsorted(&self) -> impl Iterator<Item = (String, &Value)> {
2378        self.as_flat().iter().map(|(sym, v)| (resolve(*sym), v)).collect::<Vec<_>>().into_iter()
2379    }
2380
2381    /// Sym-keyed unsorted iteration — ZERO interner traffic, zero allocation.
2382    ///
2383    /// This exists because live-sampling the cid marquee eval (2026-07-21,
2384    /// release-profiling binary) showed the **interner round-trip as the #1 CPU
2385    /// sink — 27–39% of the eval thread, sustained**: `iter_unsorted` above
2386    /// materializes a fresh heap `String` per key via `resolve` AND collects
2387    /// the whole map into a `Vec` on every call, and callers like
2388    /// `intersectAttrs` then re-intern each of those Strings straight back to
2389    /// the `Symbol` they started as (`contains_key(&str)` → `intern`), with a
2390    /// third intern inside `insert`. Sym→String→hash+memcmp→Sym, three times
2391    /// per key per call, at nixpkgs scale.
2392    ///
2393    /// `Symbol` is `Copy(u32)` and `as_flat()` hands back a real borrow (the
2394    /// Overlay case populates its cache), so this iterator borrows instead of
2395    /// collecting. Byte-neutral by the same argument already sealed for the
2396    /// unsorted-iteration change: the observable order of any *result* attrset
2397    /// is re-derived at observation time via `sorted_entries`.
2398    pub fn iter_syms(&self) -> impl Iterator<Item = (Symbol, &Value)> {
2399        self.as_flat().iter().map(|(sym, v)| (*sym, v))
2400    }
2401
2402    /// Sym-keyed insert — the zero-intern sibling of `insert`, for callers
2403    /// that already hold the `Symbol` (every `iter_syms` consumer).
2404    pub fn insert_sym(&mut self, sym: Symbol, value: Value) {
2405        self.ensure_flat();
2406        if let AttrsInner::Flat(ref mut m) = self.0 {
2407            m.insert(sym, value);
2408        }
2409    }
2410
2411    pub fn values(&self) -> impl Iterator<Item = &Value> {
2412        self.sorted_entries().into_iter().map(|(_, v)| v)
2413    }
2414
2415
2416    pub fn remove(&mut self, key: &str) -> Option<Value> {
2417        self.ensure_flat();
2418        if let AttrsInner::Flat(ref mut m) = self.0 {
2419            m.remove(&intern(key))
2420        } else {
2421            None
2422        }
2423    }
2424
2425    #[must_use]
2426    pub fn len(&self) -> usize {
2427        match &self.0 {
2428            AttrsInner::Flat(m) => m.len(),
2429            AttrsInner::Overlay { .. } => {
2430                // Must flatten to count unique keys, but `as_flat()` already
2431                // returns a borrow into the memoized map — cloning it just to
2432                // read `.len()` was pure O(n) waste on every overlay `len()`.
2433                self.as_flat().len()
2434            }
2435        }
2436    }
2437
2438    #[must_use]
2439    pub fn is_empty(&self) -> bool {
2440        match &self.0 {
2441            AttrsInner::Flat(m) => m.is_empty(),
2442            // Cache-first (see get_sym): a released-parent overlay is NOT empty —
2443            // its content lives in the flattened cache. Reading left/right here
2444            // (which post-release are empty) would wrongly report empty.
2445            AttrsInner::Overlay { .. } => self.as_flat().is_empty(),
2446        }
2447    }
2448
2449    /// O(1) lazy overlay: `self // other`. Does NOT merge eagerly.
2450    #[must_use]
2451    pub fn overlay(self, other: NixAttrs) -> NixAttrs {
2452        if other.is_empty() { return self; }
2453        if self.is_empty() { return other; }
2454        crate::perf::inc(crate::perf::Counter::OverlayCreated);
2455        census::made(&census::ATTRS_MADE, &census::ATTRS_LIVE);
2456        NixAttrs(AttrsInner::Overlay {
2457            left: RefCell::new(Rc::new(self)),
2458            right: RefCell::new(Rc::new(other)),
2459            cache: Rc::new(OnceCell::new()),
2460        }, None)
2461    }
2462
2463    /// Eager merge (legacy API — prefer `overlay` for `//`).
2464    #[must_use]
2465    pub fn update(&self, other: &NixAttrs) -> NixAttrs {
2466        match (&self.0, &other.0) {
2467            (AttrsInner::Flat(l), AttrsInner::Flat(r)) => {
2468                let mut result = l.clone();
2469                for (k, v) in r.iter() {
2470                    result.insert(*k, v.clone());
2471                }
2472                census::made(&census::ATTRS_MADE, &census::ATTRS_LIVE);
2473                NixAttrs(AttrsInner::Flat(result), None)
2474            }
2475            _ => {
2476                // For overlay inputs, flatten then merge
2477                let mut result = self.as_flat().clone();
2478                let other_flat = other.as_flat();
2479                for (k, v) in other_flat.iter() {
2480                    result.insert(*k, v.clone());
2481                }
2482                census::made(&census::ATTRS_MADE, &census::ATTRS_LIVE);
2483                NixAttrs(AttrsInner::Flat(result), None)
2484            }
2485        }
2486    }
2487}
2488
2489impl FromIterator<(String, Value)> for NixAttrs {
2490    fn from_iter<I: IntoIterator<Item = (String, Value)>>(iter: I) -> Self {
2491        census::made(&census::ATTRS_MADE, &census::ATTRS_LIVE);
2492        NixAttrs(AttrsInner::Flat(iter.into_iter().map(|(k, v)| (intern(&k), v)).collect()), None)
2493    }
2494}
2495
2496impl IntoIterator for NixAttrs {
2497    type Item = (String, Value);
2498    type IntoIter = Box<dyn Iterator<Item = (String, Value)>>;
2499
2500    fn into_iter(self) -> Self::IntoIter {
2501        let flat = self.as_flat().clone();
2502        Box::new(flat.into_iter().map(|(sym, v)| (resolve(sym), v)))
2503    }
2504}
2505
2506/// A closure — lambda + captured environment.
2507///
2508/// Stores rnix AST nodes so we can re-evaluate the body in the captured env.
2509///
2510/// The environment is `Rc`-wrapped so that cloning a closure (e.g., once per
2511/// element in `map`/`filter`) is a refcount bump instead of a deep copy of the
2512/// entire binding map.
2513#[derive(Debug, Clone)]
2514pub struct Closure {
2515    pub param: rnix::ast::Param,
2516    pub body: rnix::ast::Expr,
2517    pub env: Env,
2518}
2519
2520/// The function signature stored inside a [`BuiltinFn`].
2521pub type BuiltinFunc = dyn Fn(&[Value]) -> Result<Value, EvalError>;
2522
2523/// A builtin function.
2524///
2525/// Not `Send`/`Sync` because `Value` contains rnix AST nodes (rowan `SyntaxNode`)
2526/// which use `NonNull` internally. The evaluator is single-threaded.
2527#[derive(Clone)]
2528pub struct BuiltinFn {
2529    /// Name used for display and debug printing.
2530    pub name: &'static str,
2531    /// The implementation closure.
2532    pub func: Rc<BuiltinFunc>,
2533}
2534
2535impl fmt::Debug for BuiltinFn {
2536    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2537        write!(f, "<builtin {}>", self.name)
2538    }
2539}
2540
2541/// A `with` scope with optional cached forced attrset.
2542///
2543/// On first lookup, the scope value is forced and the resulting attrset
2544/// is cached.  Subsequent lookups skip forcing entirely.
2545///
2546/// The cache is wrapped in `Rc<RefCell<…>>` so that child environments
2547/// (which clone the `Vec<WithScope>`) share the same cache cell —
2548/// once any environment forces a scope, every related environment
2549/// benefits.
2550#[derive(Clone)]
2551struct WithScope {
2552    value: Value,
2553    /// Cached forced attrset.  Shared via Rc so child environments
2554    /// benefit from a parent having already forced the scope.
2555    cached: Rc<RefCell<Option<NixAttrs>>>,
2556}
2557
2558impl fmt::Debug for WithScope {
2559    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2560        f.debug_struct("WithScope")
2561            .field("value", &self.value)
2562            .field("cached", &self.cached.borrow().is_some())
2563            .finish()
2564    }
2565}
2566
2567/// Inner data for an evaluation environment.
2568///
2569/// Wrapped in `Rc` by [`Env`] so that cloning an `Env` is always a
2570/// refcount bump — never a deep copy of the binding map.
2571///
2572/// Uses a flattened `FxHashMap` for bindings: `child()` clones
2573/// the parent's map with O(1) structural sharing instead of building
2574/// a linked parent chain. Lookups are a single O(log32 n) probe
2575/// instead of walking a chain.
2576#[derive(Debug, Clone, Default)]
2577struct EnvInner {
2578    bindings: FxHashMap<Symbol, Value>,
2579    /// Dynamic `with` scopes, innermost last.
2580    with_scopes: Vec<WithScope>,
2581    /// Source file currently being evaluated, for relative path
2582    /// literals (`./foo.nix`) inside function defaults that get
2583    /// evaluated *after* control has left the file scope.
2584    eval_file: Option<std::path::PathBuf>,
2585    /// The `source_id` of the parse tree this env belongs to. Restored
2586    /// on thunk force (in lockstep with `eval_file`) so a lazily-forced
2587    /// thunk's idents key `IDENT_CACHE` against the file where the thunk
2588    /// was DEFINED, not the ambient source at force time. Without this, a
2589    /// cross-file force collides on `(source_id, text_offset)` and returns
2590    /// a wrong Symbol (the `parse.nix` `cannot select from null` bug).
2591    source_id: u32,
2592}
2593
2594/// Evaluation environment — flattened binding map with structural sharing.
2595///
2596/// Internally an `Rc<EnvInner>`, so cloning is always O(1) (refcount
2597/// bump).  `child()` clones the `FxHashMap` (O(1) structural
2598/// sharing) instead of building a parent chain.  `bind()` uses
2599/// `Rc::make_mut` for copy-on-write: if the Rc is shared, only then
2600/// does it clone the inner data.
2601#[derive(Clone, Default)]
2602pub struct Env(Rc<EnvInner>);
2603
2604impl fmt::Debug for Env {
2605    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2606        self.0.fmt(f)
2607    }
2608}
2609
2610impl Env {
2611    /// Create a root environment with no bindings.
2612    #[must_use]
2613    pub fn new() -> Self {
2614        Self(Rc::new(EnvInner {
2615            bindings: FxHashMap::default(),
2616            with_scopes: Vec::new(),
2617            eval_file: None,
2618            source_id: 0,
2619        }))
2620    }
2621
2622    /// Create a child environment that inherits from this one.
2623    ///
2624    /// O(1) — the `FxHashMap` clone is structural sharing (refcount
2625    /// bump on internal tree nodes), not a deep copy.
2626    #[must_use]
2627    pub fn child(&self) -> Self {
2628        crate::perf::inc(crate::perf::Counter::EnvClone);
2629        Self(Rc::new(EnvInner {
2630            bindings: self.0.bindings.clone(), // O(1) structural sharing
2631            with_scopes: self.0.with_scopes.clone(),
2632            // Children inherit the parent's eval file so that
2633            // path literals nested deep in let-chains still
2634            // resolve against the right directory.
2635            eval_file: self.0.eval_file.clone(),
2636            // Children inherit the parent's source_id — a child scope is
2637            // in the same parse tree as its parent (a new source_id only
2638            // arises on `eval_with_file` for an imported file).
2639            source_id: self.0.source_id,
2640        }))
2641    }
2642
2643    /// Attach a `with` scope to this environment.
2644    ///
2645    /// If the value is a thunk that's ALREADY evaluated (OnceCell cache hit),
2646    /// pre-populate the with-scope cache immediately. This avoids creating
2647    /// deferred WithIdent thunks when the fixpoint is already resolved —
2648    /// critical for the overlay chain where multiple stages access the same
2649    /// fixpoint through different `with self;` scopes.
2650    #[must_use]
2651    pub fn with_scope(mut self, value: Value) -> Self {
2652        // Pre-populate cache if the value is already resolved
2653        let pre_cached = match &value {
2654            Value::Attrs(attrs) => Some((**attrs).clone()),
2655            Value::Thunk(thunk) => thunk.peek().and_then(|v| {
2656                if let Concrete::Attrs(attrs) = v { Some((**attrs).clone()) } else { None }
2657            }),
2658            _ => None,
2659        };
2660        Rc::make_mut(&mut self.0).with_scopes.push(WithScope {
2661            value,
2662            cached: Rc::new(RefCell::new(pre_cached)),
2663        });
2664        self
2665    }
2666
2667    /// Bind a name to a value in this environment's own scope.
2668    ///
2669    /// Uses copy-on-write: if the inner `Rc` is shared, clones the
2670    /// inner data before mutating.
2671    pub fn bind(&mut self, name: String, value: Value) {
2672        Rc::make_mut(&mut self.0).bindings.insert(intern(&name), value);
2673    }
2674
2675    /// Get the eval_file for this environment.
2676    #[must_use]
2677    pub fn eval_file(&self) -> Option<&std::path::PathBuf> {
2678        self.0.eval_file.as_ref()
2679    }
2680
2681    /// Set the eval_file for this environment.
2682    pub fn set_eval_file(&mut self, file: Option<std::path::PathBuf>) {
2683        Rc::make_mut(&mut self.0).eval_file = file;
2684    }
2685
2686    /// The `source_id` of the parse tree this env belongs to (0 = top level).
2687    #[must_use]
2688    pub fn source_id(&self) -> u32 {
2689        self.0.source_id
2690    }
2691
2692    /// Set the `source_id` for this environment (called by `eval_with_file`
2693    /// for an imported parse tree).
2694    pub fn set_source_id(&mut self, id: u32) {
2695        Rc::make_mut(&mut self.0).source_id = id;
2696    }
2697
2698    /// Number of direct bindings in this environment (debug).
2699    #[must_use]
2700    pub fn binding_count(&self) -> usize {
2701        self.0.bindings.len()
2702    }
2703
2704    /// First N binding names (debug).
2705    #[must_use]
2706    pub fn binding_names_preview(&self, n: usize) -> Vec<String> {
2707        self.0.bindings.keys().take(n).map(|s| resolve(*s)).collect()
2708    }
2709
2710    /// Number of `with` scopes (debug).
2711    #[must_use]
2712    pub fn with_scope_count(&self) -> usize {
2713        self.0.with_scopes.len()
2714    }
2715
2716    /// Lookup in LEXICAL scope only (no with-scopes).
2717    /// Used by maybe_thunk to avoid forcing with-scope fixpoints during
2718    /// attrset construction.
2719    #[must_use]
2720    pub fn lookup_lexical(&self, name: &str) -> Option<Value> {
2721        let sym = intern(name);
2722        self.0.bindings.get(&sym).cloned()
2723    }
2724
2725    /// Lookup in LEXICAL scope only, by pre-interned [`Symbol`] — the
2726    /// Symbol-keyed sibling of [`lookup_lexical`](Self::lookup_lexical).
2727    ///
2728    /// Probes ONLY the lexical `bindings` map (the first thing
2729    /// [`lookup_fast`](Self::lookup_fast) does, by the same Symbol) — never
2730    /// the `with`-chain. The ENV-RESOLVE M0 fast path uses this: a
2731    /// `Resolution::Lexical{sym}` reference probes here directly with its
2732    /// precomputed Symbol; on a hit the returned value is byte-identical to
2733    /// `lookup_fast`'s (same map, same Symbol); on a miss the caller falls
2734    /// back to today's exact runtime path.
2735    #[must_use]
2736    pub fn lookup_lexical_sym(&self, sym: Symbol) -> Option<Value> {
2737        self.0.bindings.get(&sym).cloned()
2738    }
2739
2740    /// Look up a name using ONLY with-scope caches (no forcing).
2741    /// Returns Some if the name is in a cached with-scope, None otherwise.
2742    /// Used by maybe_thunk to resolve with-scope idents without forcing fixpoints.
2743    #[must_use]
2744    pub fn lookup_with_cache_only(&self, name: &str) -> Option<Value> {
2745        for scope in self.0.with_scopes.iter().rev() {
2746            let cache = scope.cached.borrow();
2747            if let Some(ref attrs) = *cache {
2748                if let Some(v) = attrs.get(name) {
2749                    return Some(v.clone());
2750                }
2751            }
2752            // Also check if the thunk is already evaluated (peek)
2753            drop(cache);
2754            if let Value::Thunk(ref thunk) = scope.value {
2755                if let Some(cached_val) = thunk.peek() {
2756                    if let Concrete::Attrs(ref attrs) = *cached_val {
2757                        // Populate the with-scope cache for future lookups
2758                        *scope.cached.borrow_mut() = Some((**attrs).clone());
2759                        if let Some(v) = attrs.get(name) {
2760                            return Some(v.clone());
2761                        }
2762                    }
2763                }
2764            } else if let Value::Attrs(ref attrs) = scope.value {
2765                *scope.cached.borrow_mut() = Some((**attrs).clone());
2766                if let Some(v) = attrs.get(name) {
2767                    return Some(v.clone());
2768                }
2769            }
2770        }
2771        None
2772    }
2773
2774    /// Get the innermost with-scope's cache and value for creating WithIdent thunks.
2775    /// Returns None if there are no with-scopes.
2776    #[must_use]
2777    pub fn innermost_with_scope(&self) -> Option<(Rc<RefCell<Option<NixAttrs>>>, Value)> {
2778        self.0.with_scopes.last().map(|scope| {
2779            (scope.cached.clone(), scope.value.clone())
2780        })
2781    }
2782
2783    /// Lookup matching Nix semantics:
2784    ///
2785    /// 1. Probe the flattened binding map (single O(log32 n) lookup).
2786    ///    Any explicit `let`/`rec`/function-arg binding wins over every
2787    ///    `with` scope.
2788    /// 2. If no lexical binding matched, iterate `with_scopes` in
2789    ///    reverse order (innermost first). So `with X; with Y; x`
2790    ///    finds `x` in Y if Y has it, otherwise in X.
2791    #[must_use]
2792    pub fn lookup(&self, name: &str) -> Option<Value> {
2793        self.lookup_fast(intern(name), name)
2794    }
2795
2796    /// Cache-BYPASSING with-scope lookup: force each `with`-scope value FRESH
2797    /// (through the full thunk chain) and check for `name`, refreshing the
2798    /// per-scope cache on the way. A force that errors (a mid-fixpoint blackhole
2799    /// or a `with (throw …); …` namespace) is caught and the scope skipped.
2800    ///
2801    /// This exists ONLY for the last-ditch retry on the about-to-throw
2802    /// `UndefinedVar` path (see the WithIdent force): the normal cache-first
2803    /// [`lookup_fast`] can trust a stale mid-fixpoint PARTIAL cached for a scope
2804    /// (e.g. `f self` before makeScope merged `callPackage` into `self`) and skip
2805    /// it; a fresh force sees the now-completed scope. Never call this on a hot
2806    /// path — it re-forces every scope.
2807    #[must_use]
2808    pub fn lookup_fresh(&self, name: &str) -> Option<Value> {
2809        let sym = intern(name);
2810        if let Some(v) = self.0.bindings.get(&sym) {
2811            return Some(v.clone());
2812        }
2813        for scope in self.0.with_scopes.iter().rev() {
2814            if let Ok(Value::Attrs(attrs)) = crate::eval::force_value(&scope.value) {
2815                if let Some(v) = attrs.get_sym(&sym) {
2816                    // Refresh the stale cache with the completed scope so a later
2817                    // lookup of a sibling name also sees it.
2818                    *scope.cached.borrow_mut() = Some((*attrs).clone());
2819                    return Some(v.clone());
2820                }
2821            }
2822        }
2823        None
2824    }
2825
2826    /// Lookup by pre-interned Symbol + string name. Avoids re-interning.
2827    #[must_use]
2828    pub fn lookup_fast(&self, sym: Symbol, name: &str) -> Option<Value> {
2829        crate::perf::inc(crate::perf::Counter::EnvLookup);
2830        if let Some(v) = self.0.bindings.get(&sym) {
2831            return Some(v.clone());
2832        }
2833        // 2. With-scope lookup — iterate innermost-first (reverse order).
2834        for scope in self.0.with_scopes.iter().rev() {
2835            // Fast path: use cached forced attrset
2836            {
2837                let cache = scope.cached.borrow();
2838                if let Some(ref attrs) = *cache {
2839                    if let Some(v) = attrs.get_sym(&sym) {
2840                        return Some(v.clone());
2841                    }
2842                    continue;
2843                }
2844            }
2845            // Slow path: force, cache, then check.
2846            // If the value is already concrete (not a thunk), use it directly.
2847            // If it's a thunk, try to force. On blackhole (fixpoint being
2848            // computed), return None so the caller can defer.
2849            let resolved = match &scope.value {
2850                Value::Attrs(attrs) => {
2851                    // Already concrete — cache and use directly
2852                    crate::perf::inc(crate::perf::Counter::WithScopeCacheClone);
2853                    *scope.cached.borrow_mut() = Some((**attrs).clone());
2854                    Some((**attrs).clone())
2855                }
2856                Value::Thunk(thunk) => {
2857                    // Check if the thunk is already evaluated (OnceCell cache)
2858                    // without entering the force state machine
2859                    if let Some(cached_val) = thunk.peek() {
2860                        if let Concrete::Attrs(ref attrs) = *cached_val {
2861                            crate::perf::inc(crate::perf::Counter::WithScopeCacheClone);
2862                            *scope.cached.borrow_mut() = Some((**attrs).clone());
2863                            Some((**attrs).clone())
2864                        } else {
2865                            None
2866                        }
2867                    } else {
2868                        // Thunk not yet evaluated — force it FULLY. Must use
2869                        // `force_value` (which chases the whole thunk chain),
2870                        // NOT `force_value_tracked` (single `force_thunk` step):
2871                        // a with-scope head like `lib.platforms` is often a
2872                        // lazy `Thunk(Thunk(Attrs))`, so one step yields a
2873                        // `Value::Thunk` whose `type_name()` peeks to "set" but
2874                        // which the `if let Value::Attrs` match REJECTS — the
2875                        // scope is then wrongly skipped and every bare-ident
2876                        // lookup through it (`with lib.platforms; unix`) fails
2877                        // with a spurious UndefinedVar.
2878                        match crate::eval::force_value(&scope.value) {
2879                            Ok(forced) => {
2880                                if let Value::Attrs(ref attrs) = forced {
2881                                    crate::perf::inc(crate::perf::Counter::WithScopeCacheClone);
2882                                    *scope.cached.borrow_mut() = Some((**attrs).clone());
2883                                    Some((**attrs).clone())
2884                                } else {
2885                                    None
2886                                }
2887                            }
2888                            Err(_) => None, // blackhole or other error — skip
2889                        }
2890                    }
2891                }
2892                _ => {
2893                    // Same full-chain force as the Thunk arm above.
2894                    match crate::eval::force_value(&scope.value) {
2895                        Ok(forced) => {
2896                            if let Value::Attrs(ref attrs) = forced {
2897                                crate::perf::inc(crate::perf::Counter::WithScopeCacheClone);
2898                                *scope.cached.borrow_mut() = Some((**attrs).clone());
2899                                Some((**attrs).clone())
2900                            } else {
2901                                None
2902                            }
2903                        }
2904                        Err(_) => None,
2905                    }
2906                }
2907            };
2908            if let Some(ref attrs) = resolved {
2909                if let Some(v) = attrs.get(name) {
2910                    return Some(v.clone());
2911                }
2912            }
2913            // If forcing fails or it's not an attrset, try next scope
2914        }
2915        None
2916    }
2917
2918    /// Look up a binding by pre-interned [`Symbol`].
2919    ///
2920    /// Same semantics as [`lookup`](Self::lookup) but skips the
2921    /// `intern()` call — for use when the caller has already cached
2922    /// the symbol (e.g. via [`intern_cached`]).
2923    #[must_use]
2924    pub fn lookup_sym(&self, sym: Symbol) -> Option<Value> {
2925        crate::perf::inc(crate::perf::Counter::EnvLookup);
2926        // 1. Flat lexical lookup — single O(1) hash + O(log32 n) probe.
2927        if let Some(v) = self.0.bindings.get(&sym) {
2928            return Some(v.clone());
2929        }
2930        // 2. With-scope lookup — iterate innermost-first (reverse order).
2931        for scope in self.0.with_scopes.iter().rev() {
2932            // Fast path: use cached forced attrset
2933            {
2934                let cache = scope.cached.borrow();
2935                if let Some(ref attrs) = *cache {
2936                    if let Some(v) = attrs.get_sym(&sym) {
2937                        return Some(v.clone());
2938                    }
2939                    continue;
2940                }
2941            }
2942            // Slow path: force, cache, then check
2943            if let Ok(forced) = crate::eval::force_value_tracked(&scope.value, "with_scope") {
2944                if let Value::Attrs(ref attrs) = forced {
2945                    let result = attrs.get_sym(&sym).cloned();
2946                    crate::perf::inc(crate::perf::Counter::WithScopeCacheClone);
2947                    *scope.cached.borrow_mut() = Some((**attrs).clone());
2948                    if result.is_some() {
2949                        return result;
2950                    }
2951                }
2952            }
2953            // If forcing fails or it's not an attrset, try next scope
2954        }
2955        None
2956    }
2957}
2958
2959/// Evaluation errors produced by the Nix evaluator.
2960#[derive(Debug, Clone, PartialEq, Eq, thiserror::Error)]
2961#[non_exhaustive]
2962pub enum EvalError {
2963    /// A variable was referenced but not bound in scope.
2964    #[error("undefined variable: {0}")]
2965    UndefinedVar(String),
2966    /// A type mismatch or coercion failure.
2967    #[error("type error: {0}")]
2968    TypeError(String),
2969    /// An attribute was selected from a set that does not contain it.
2970    #[error("attribute not found: {0}")]
2971    AttrNotFound(String),
2972    /// A type mismatch with structured expected/got information.
2973    #[error("type error: expected {expected}, got {got}")]
2974    TypeMismatch {
2975        expected: &'static str,
2976        got: &'static str,
2977    },
2978    /// An `assert` expression's condition evaluated to false.
2979    #[error("assertion failed{0}")]
2980    AssertionFailed(String),
2981    /// Integer division by zero.
2982    #[error("division by zero")]
2983    DivisionByZero,
2984    /// Infinite recursion detected (thunk blackhole or eval depth).
2985    #[error("infinite recursion ({0})")]
2986    InfiniteRecursion(String),
2987    /// An I/O error from the host filesystem.
2988    #[error("I/O error: {context}: {message}")]
2989    IoError { context: String, message: String },
2990    /// Explicit `throw` from Nix code — CATCHABLE by `builtins.tryEval`.
2991    #[error("{0}")]
2992    Throw(String),
2993    /// An `abort` from Nix code — UNCATCHABLE (CppNix's `abort`/`builtins.abort`
2994    /// is a hard error `tryEval` does NOT catch, unlike `throw`/`assert`).
2995    /// Verified: `nix eval '(builtins.tryEval (abort "x")).success'` errors.
2996    #[error("{0}")]
2997    Abort(String),
2998    /// A language feature that is not yet implemented.
2999    #[error("not yet implemented: {0}")]
3000    NotImplemented(String),
3001    /// A syntax error in the input expression.
3002    #[error("parse error: {0}")]
3003    ParseError(String),
3004    /// Maximum recursion depth exceeded.
3005    #[error("recursion limit: {0}")]
3006    RecursionLimit(String),
3007}
3008
3009impl EvalError {
3010    /// Convenience constructor for a `TypeError` variant.
3011    #[must_use]
3012    pub fn type_error(msg: impl Into<String>) -> Self {
3013        EvalError::TypeError(msg.into())
3014    }
3015
3016    /// Convenience constructor for a `TypeMismatch` variant.
3017    #[must_use]
3018    pub fn type_mismatch(expected: &'static str, got: &'static str) -> Self {
3019        EvalError::TypeMismatch { expected, got }
3020    }
3021
3022    /// Create a type error for a builtin argument type mismatch.
3023    #[must_use]
3024    pub fn builtin_type(builtin: &str, expected: &str, got: &str) -> Self {
3025        EvalError::TypeError(format!("{builtin}: expected {expected}, got {got}"))
3026    }
3027
3028    /// Create a type error for a binary operator type mismatch.
3029    ///
3030    /// CARRIES THE EVAL FILE (added 2026-07-20). Every arithmetic/comparison
3031    /// raise site routes through here, and none of them appended
3032    /// `eval_file_ctx()` — unlike the ~12 sibling raise sites in `eval.rs` that
3033    /// do — so an operator type error named no file at all. Nor could the frame
3034    /// stack help: `NixTraceGuard::drop` pops every frame during unwind, so by
3035    /// the time the error surfaces `attach_trace` has nothing left to attach.
3036    ///
3037    /// The cost of that was concrete. "cannot add string and null" was the sole
3038    /// symptom of the ident-cache aliasing bug that stopped sui evaluating
3039    /// nixpkgs, and it pointed nowhere: four parallel investigations each spent
3040    /// most of their budget just locating it, and the only tool that worked was
3041    /// `SUI_TRACE_EVAL=1` dumping 521k lines to be read backwards. One
3042    /// `format!` argument here would have named `make-derivation.nix`
3043    /// immediately.
3044    ///
3045    /// Fixing it in `op_type` rather than at the `Add` arm means every operator
3046    /// — add, sub, mul, div, comparison, update — gains the context at once,
3047    /// instead of the next one to bite us needing its own patch.
3048    #[must_use]
3049    pub fn op_type(op: &str, lhs: &str, rhs: &str) -> Self {
3050        EvalError::TypeError(format!(
3051            "cannot {op} {lhs} and {rhs}{}",
3052            crate::eval::eval_file_ctx()
3053        ))
3054    }
3055
3056    /// Whether this error was caused by `throw` or `abort`.
3057    #[must_use]
3058    pub fn is_throw(&self) -> bool {
3059        matches!(self, EvalError::Throw(_))
3060    }
3061
3062    /// Whether this error is an infinite recursion.
3063    #[must_use]
3064    pub fn is_infinite_recursion(&self) -> bool {
3065        matches!(self, EvalError::InfiniteRecursion(_))
3066    }
3067}
3068
3069impl Value {
3070    /// Convenience constructor for a context-free string.
3071    #[must_use]
3072    pub fn string(s: impl Into<SmolStr>) -> Self {
3073        Value::String(Rc::new(NixString::plain(s)))
3074    }
3075
3076    /// Convenience constructor that wraps a `Vec<Value>` in `Rc` for the
3077    /// `List` variant.
3078    #[must_use]
3079    pub fn list(items: Vec<Value>) -> Self {
3080        Value::List(Rc::new(NixList::new(items)))
3081    }
3082
3083    /// True when `self` is a `List` whose backing `Rc<Vec>` is uniquely owned
3084    /// (refcount 1). Used by [`concat_lists`] to decide the in-place fast path.
3085    #[must_use]
3086    pub fn is_uniquely_owned_list(&self) -> bool {
3087        matches!(self, Value::List(rc) if Rc::strong_count(rc) == 1)
3088    }
3089
3090    /// Convert a value to JSON for API output.
3091    #[must_use]
3092    pub fn to_json(&self) -> serde_json::Value {
3093        match self {
3094            Value::Null => serde_json::Value::Null,
3095            Value::Bool(b) => serde_json::Value::Bool(*b),
3096            Value::Int(n) => serde_json::json!(n),
3097            Value::Float(f) => serde_json::json!(f),
3098            Value::String(s) => serde_json::Value::String(s.chars.to_string()),
3099            Value::Path(p) => serde_json::Value::String(p.to_string()),
3100            Value::List(items) => {
3101                serde_json::Value::Array(items.iter().map(|v| v.to_json()).collect())
3102            }
3103            Value::Attrs(attrs) => {
3104                // nix-faithful (CppNix value-to-json.cc `tryAttrsToString`):
3105                // a derivation — an attrset carrying `__toString` or `outPath`
3106                // — serializes to THAT STRING, never its own attrs. Without
3107                // this, `to_json` recurses forever on the self-referential
3108                // derivation graph (`drv.out.drv == drv`, `drv.all`, …) and
3109                // overflows the stack. Mirrors `coerce_to_string` below.
3110                if attrs.get("__toString").is_some() || attrs.get("outPath").is_some() {
3111                    if let Ok((s, _ctx)) = self.coerce_to_string() {
3112                        return serde_json::Value::String(s);
3113                    }
3114                }
3115                let map: serde_json::Map<String, serde_json::Value> = attrs
3116                    .iter()
3117                    .map(|(k, v)| (k.clone(), v.to_json()))
3118                    .collect();
3119                serde_json::Value::Object(map)
3120            }
3121            Value::Lambda(_) => serde_json::Value::String("<lambda>".to_string()),
3122            Value::Builtin(b) => serde_json::Value::String(format!("<builtin {}>", b.name)),
3123            Value::Thunk(thunk) => {
3124                // Force the thunk for JSON conversion.
3125                match thunk.force(&|expr, env| crate::eval::eval_expr(expr, env)) {
3126                    Ok(v) => v.to_json(),
3127                    Err(_) => serde_json::Value::String("<thunk:error>".to_string()),
3128                }
3129            }
3130        }
3131    }
3132
3133    /// Like [`to_json`] but threads string context into `ctx`. Used by
3134    /// `__structuredAttrs` derivation-env building: a derivation value
3135    /// serializes to its outPath (a store-path string) and its drv reference
3136    /// must flow into the derivation's `inputDrvs`; a bare path is copy-to-store
3137    /// coerced. (`to_json` drops context, which is fine for `builtins.toJSON`
3138    /// but not for building a derivation's `__json`.)
3139    pub fn to_json_with_context(
3140        &self,
3141        ctx: &mut StringContext,
3142    ) -> Result<serde_json::Value, EvalError> {
3143        Ok(match self {
3144            Value::Null => serde_json::Value::Null,
3145            Value::Bool(b) => serde_json::Value::Bool(*b),
3146            Value::Int(n) => serde_json::json!(n),
3147            Value::Float(f) => serde_json::json!(f),
3148            Value::String(s) => {
3149                ctx.merge(&s.context);
3150                serde_json::Value::String(s.chars.to_string())
3151            }
3152            Value::Path(_) => {
3153                let (str, c) = self.coerce_to_string_copy_to_store()?;
3154                ctx.merge(&c);
3155                serde_json::Value::String(str)
3156            }
3157            Value::List(items) => {
3158                let mut arr = Vec::with_capacity(items.len());
3159                for v in items.iter() {
3160                    let fv = crate::eval::force_value(v)?;
3161                    arr.push(fv.to_json_with_context(ctx)?);
3162                }
3163                serde_json::Value::Array(arr)
3164            }
3165            Value::Attrs(attrs) => {
3166                // A derivation (attrset with `outPath`/`__toString`) serializes
3167                // to that string with its context — never its own attrs.
3168                if attrs.get("__toString").is_some() || attrs.get("outPath").is_some() {
3169                    let (s, c) = self.coerce_to_string_copy_to_store()?;
3170                    ctx.merge(&c);
3171                    return Ok(serde_json::Value::String(s));
3172                }
3173                let mut map = serde_json::Map::new();
3174                for (k, v) in attrs.iter() {
3175                    let fv = crate::eval::force_value(v)?;
3176                    map.insert(k.clone(), fv.to_json_with_context(ctx)?);
3177                }
3178                serde_json::Value::Object(map)
3179            }
3180            Value::Thunk(_) => {
3181                let forced = crate::eval::force_value(self)?;
3182                forced.to_json_with_context(ctx)?
3183            }
3184            other => {
3185                return Err(EvalError::TypeError(format!(
3186                    "cannot serialize {} to JSON (__structuredAttrs)",
3187                    other.type_name()
3188                )));
3189            }
3190        })
3191    }
3192
3193    /// Return the Nix type name for this value (e.g. `"int"`, `"set"`).
3194    #[must_use]
3195    pub fn type_name(&self) -> &'static str {
3196        match self {
3197            Value::Null => "null",
3198            Value::Bool(_) => "bool",
3199            Value::Int(_) => "int",
3200            Value::Float(_) => "float",
3201            Value::String(_) => "string",
3202            Value::Path(_) => "path",
3203            Value::List(_) => "list",
3204            Value::Attrs(_) => "set",
3205            Value::Lambda(_) => "lambda",
3206            Value::Builtin(_) => "lambda",
3207            Value::Thunk(thunk) => {
3208                // Force and delegate.
3209                match thunk.force(&|expr, env| crate::eval::eval_expr(expr, env)) {
3210                    Ok(v) => v.type_name(),
3211                    Err(_) => "thunk",
3212                }
3213            }
3214        }
3215    }
3216
3217    // ── Value coercion methods ──────────────────────────────────
3218    //
3219    // Naming conventions:
3220    //
3221    // • `as_*(&self)` — borrow. Returns a reference or Copy type.
3222    //   Primitives (`as_bool`, `as_int`) force thunks transparently
3223    //   because they return owned Copy values. Reference accessors
3224    //   (`as_string`, `as_nix_string`, `as_attrs`, `as_list`) CANNOT
3225    //   force thunks (the forced value is transient and we can't
3226    //   return a borrow into it), so they error on Thunk inputs.
3227    //
3228    // • `to_*(&self)` — clone / force. Returns an owned value and
3229    //   DOES force thunks. Use when the value may be a thunk and you
3230    //   need an owned result. Examples: `to_float`, `to_string`,
3231    //   `to_attrs`, `to_list`.
3232    //
3233    // • `coerce_to_path` — a Nix-specific coercion that accepts both
3234    //   Path and String values (many builtins accept either).
3235
3236    /// Extract a bool, forcing thunks if needed.
3237    pub fn as_bool(&self) -> Result<bool, EvalError> {
3238        match self {
3239            Value::Bool(b) => Ok(*b),
3240            Value::Thunk(thunk) => {
3241                thunk.force(&|e, env| crate::eval::eval_expr(e, env))?.as_bool()
3242            }
3243            _ => Err(EvalError::TypeMismatch { expected: "bool", got: self.type_name() }),
3244        }
3245    }
3246
3247    /// Extract an integer, forcing thunks if needed.
3248    pub fn as_int(&self) -> Result<i64, EvalError> {
3249        match self {
3250            Value::Int(n) => Ok(*n),
3251            Value::Thunk(thunk) => {
3252                thunk.force(&|e, env| crate::eval::eval_expr(e, env))?.as_int()
3253            }
3254            _ => Err(EvalError::TypeMismatch { expected: "int", got: self.type_name() }),
3255        }
3256    }
3257
3258    /// Borrow the string content without forcing thunks.
3259    pub fn as_string(&self) -> Result<&str, EvalError> {
3260        match self {
3261            Value::String(s) => Ok(&s.chars),
3262            Value::Thunk(_) => Err(EvalError::TypeError(
3263                "thunk in as_string: force first via force_value()".into(),
3264            )),
3265            _ => Err(EvalError::TypeMismatch { expected: "string", got: self.type_name() }),
3266        }
3267    }
3268
3269    /// Return a reference to the full `NixString` (with context).
3270    pub fn as_nix_string(&self) -> Result<&NixString, EvalError> {
3271        match self {
3272            Value::String(ns) => Ok(ns),
3273            Value::Thunk(_) => Err(EvalError::TypeError(
3274                "thunk in as_nix_string: force first via force_value()".into(),
3275            )),
3276            _ => Err(EvalError::TypeMismatch { expected: "string", got: self.type_name() }),
3277        }
3278    }
3279
3280    /// Force-aware string extraction. Returns an owned String by forcing
3281    /// thunks if needed. Use this instead of `as_string()` when you may
3282    /// be operating on thunked attrset values.
3283    pub fn to_str(&self) -> Result<String, EvalError> {
3284        match self {
3285            Value::String(s) => Ok(s.chars.to_string()),
3286            Value::Thunk(thunk) => {
3287                let forced = thunk.force(&|e, env| crate::eval::eval_expr(e, env))?;
3288                forced.to_str()
3289            }
3290            _ => Err(EvalError::TypeMismatch { expected: "string", got: self.type_name() }),
3291        }
3292    }
3293
3294    /// Force-aware `NixString` extraction. Returns an owned `NixString`
3295    /// (with context) by forcing thunks if needed.
3296    pub fn to_nix_string(&self) -> Result<NixString, EvalError> {
3297        match self {
3298            Value::String(s) => Ok((**s).clone()),
3299            Value::Thunk(thunk) => {
3300                let forced = thunk.force(&|e, env| crate::eval::eval_expr(e, env))?;
3301                forced.to_nix_string()
3302            }
3303            _ => Err(EvalError::TypeMismatch { expected: "string", got: self.type_name() }),
3304        }
3305    }
3306
3307    /// Borrow the inner attrs without forcing. If the value is a
3308    /// thunk, the caller should have force_value'd it first; we
3309    /// return an error rather than silently mutating the thunk
3310    /// (which would require &mut self).
3311    ///
3312    /// Most call sites should use `to_attrs()` (which forces and
3313    /// clones) unless they're certain the value is already
3314    /// concrete and want to avoid the clone.
3315    pub fn as_attrs(&self) -> Result<&NixAttrs, EvalError> {
3316        match self {
3317            Value::Attrs(a) => Ok(a),
3318            Value::Thunk(_) => Err(EvalError::TypeError(
3319                "thunk in as_attrs: force first via force_value() or use to_attrs()".into(),
3320            )),
3321            _ => Err(EvalError::TypeMismatch { expected: "set", got: self.type_name() }),
3322        }
3323    }
3324
3325    /// Borrow the list content without forcing thunks.
3326    pub fn as_list(&self) -> Result<&[Value], EvalError> {
3327        match self {
3328            Value::List(l) => Ok(l.as_slice()),
3329            Value::Thunk(_) => Err(EvalError::TypeError(
3330                "thunk in as_list: force first via force_value()".into(),
3331            )),
3332            _ => Err(crate::eval::attach_trace(
3333                EvalError::TypeMismatch { expected: "list", got: self.type_name() }
3334            )),
3335        }
3336    }
3337
3338    /// Force-aware attrs extraction. Forces the value if it is a thunk.
3339    pub fn to_attrs(&self) -> Result<NixAttrs, EvalError> {
3340        match self {
3341            Value::Attrs(a) => Ok((**a).clone()),
3342            Value::Thunk(thunk) => {
3343                let forced = thunk.force(&|e, env| crate::eval::eval_expr(e, env))?;
3344                forced.to_attrs()
3345            }
3346            _ => Err(EvalError::TypeMismatch { expected: "set", got: self.type_name() }),
3347        }
3348    }
3349
3350    /// Force-aware list extraction. Forces the value if it is a thunk.
3351    pub fn to_list(&self) -> Result<Vec<Value>, EvalError> {
3352        match self {
3353            Value::List(l) => Ok((**l).0.clone()),
3354            Value::Thunk(thunk) => {
3355                let forced = thunk.force(&|e, env| crate::eval::eval_expr(e, env))?;
3356                forced.to_list()
3357            }
3358            _ => Err(EvalError::TypeMismatch { expected: "list", got: self.type_name() }),
3359        }
3360    }
3361
3362    /// Extract a filesystem path from a `Path` or `String` value.
3363    ///
3364    /// Many builtins (`readFile`, `import`, `pathExists`, etc.) accept
3365    /// either `Path` or `String` arguments. This method centralises
3366    /// that coercion so every call-site doesn't repeat the same match.
3367    pub fn coerce_to_path(&self, context: &str) -> Result<String, EvalError> {
3368        match self {
3369            Value::Path(p) => Ok(p.to_string()),
3370            Value::String(ns) => Ok(ns.chars.to_string()),
3371            Value::Attrs(attrs) => {
3372                if let Some(out_path) = attrs.get("outPath") {
3373                    let forced = crate::eval::force_value(out_path)?;
3374                    forced.coerce_to_path(context)
3375                } else {
3376                    Err(EvalError::TypeError(format!(
3377                        "{context}: expected path or string, got set without outPath"
3378                    )))
3379                }
3380            }
3381            _ => Err(EvalError::TypeError(format!(
3382                "{context}: expected path or string, got {}",
3383                self.type_name()
3384            ))),
3385        }
3386    }
3387
3388    /// Coerce to a filesystem path AND, if this value is a **derivation**
3389    /// whose output is not yet materialized on disk, realize that output first
3390    /// (import-from-derivation).
3391    ///
3392    /// Used by the disk-read builtins (`import`, `readFile`, `readDir`,
3393    /// `pathExists`, `builtins.path`) so a read under a derivation's `outPath`
3394    /// triggers a build/substitute of that output, exactly as cppnix does.
3395    ///
3396    /// Semantics:
3397    /// - A `Path`/`String` coerces as usual — no realize (nothing to build).
3398    /// - A derivation attrset (`type == "derivation"` with `drvPath` +
3399    ///   `outPath`) whose `outPath` (after input-source materialization) does
3400    ///   **not** exist on disk invokes the realize hook with `(drvPath,
3401    ///   outPath)`. On success the returned path is the (now-present) `outPath`.
3402    /// - A non-derivation attrset with `outPath` coerces via `outPath` as usual
3403    ///   (no drv to realize).
3404    /// - If no realize hook is installed, this degrades to `coerce_to_path`
3405    ///   (the read that follows will ENOENT — a real error, never a wrong
3406    ///   value).
3407    ///
3408    /// The realize hook mutates no value the evaluator observes; it only makes
3409    /// the bytes at the already-byte-correct `outPath` present on disk (see
3410    /// [`crate::realize`]).
3411    pub fn coerce_to_realized_path(&self, context: &str) -> Result<String, EvalError> {
3412        match self {
3413            // Direct derivation attrset (`import <drv>`): drvPath + outPath are
3414            // right there.
3415            Value::Attrs(attrs) => {
3416                if let Some((drv_path, out_path)) = derivation_drv_and_out(attrs)? {
3417                    self.realize_if_absent(&drv_path, &out_path, context)?;
3418                    return Ok(out_path);
3419                }
3420            }
3421            // A string produced by interpolating a derivation
3422            // (`readFile "${drv}"`) is a store-path STRING that carries a
3423            // `ContextElement::Output { drv, output }` — the derivation-ness
3424            // survives interpolation *as string context*, which is exactly how
3425            // cppnix decides to realize. If the coerced store path is absent and
3426            // the context names the producing `.drv`, realize it.
3427            Value::String(ns) => {
3428                let out_path = ns.chars.to_string();
3429                if let Some(drv_path) = out_path_needs_realize(&out_path, &ns.context) {
3430                    self.realize_if_absent(&drv_path, &out_path, context)?;
3431                }
3432                return Ok(out_path);
3433            }
3434            _ => {}
3435        }
3436        self.coerce_to_path(context)
3437    }
3438
3439    /// If `out_path` (after input-source materialization) is not present on
3440    /// disk, invoke the realize hook to build/substitute `drv_path`. A missing
3441    /// hook is a silent fall-through (the following read ENOENTs — a real error,
3442    /// never a wrong value); a hook error is surfaced as an eval `IoError`.
3443    fn realize_if_absent(
3444        &self,
3445        drv_path: &str,
3446        out_path: &str,
3447        context: &str,
3448    ) -> Result<(), EvalError> {
3449        // The existence probe must consult the REAL tree — a fetched flake
3450        // input's `-source` prefix is redirected — so materialize first.
3451        let read_path = crate::path::materialize_str(out_path);
3452        if std::path::Path::new(&read_path).exists() {
3453            return Ok(());
3454        }
3455        match crate::realize::realize_output(drv_path, out_path) {
3456            Ok(true) | Ok(false) => Ok(()),
3457            Err(msg) => Err(EvalError::IoError {
3458                context: context.to_string(),
3459                message: format!(
3460                    "import-from-derivation: realizing {drv_path} -> {out_path}: {msg}"
3461                ),
3462            }),
3463        }
3464    }
3465
3466    /// Coerce a numeric value to float.
3467    pub fn to_float(&self) -> Result<f64, EvalError> {
3468        match self {
3469            Value::Float(f) => Ok(*f),
3470            Value::Int(n) => Ok(*n as f64),
3471            Value::Thunk(thunk) => {
3472                thunk.force(&|e, env| crate::eval::eval_expr(e, env))?.to_float()
3473            }
3474            _ => Err(EvalError::TypeMismatch { expected: "number", got: self.type_name() }),
3475        }
3476    }
3477
3478    /// Coerce this value to a string following CppNix semantics.
3479    ///
3480    /// This is the single source of truth for string coercion used by
3481    /// string interpolation, `builtins.toString`, and derivation env
3482    /// var construction.
3483    ///
3484    /// Rules (in order):
3485    /// - String → its content (with context)
3486    /// - Path → path string (adds Plain context element)
3487    /// - Int → decimal representation
3488    /// - Float → decimal representation
3489    /// - Bool → "1" for true, "" for false
3490    /// - Null → ""
3491    /// - Attrs with `__toString` → call `__toString(self)` and coerce result
3492    /// - Attrs with `outPath` → coerce outPath recursively
3493    /// - List → space-joined coerced elements
3494    /// - Lambda/Builtin/Thunk → error
3495    pub fn coerce_to_string(&self) -> Result<(String, StringContext), EvalError> {
3496        self.coerce_to_string_impl(false)
3497    }
3498
3499    /// Coerce to string in CppNix **copy-to-store** mode — the coercion used by
3500    /// string interpolation (`"${./foo}"`) and derivation-attribute population.
3501    /// A source path that isn't already in the store is absolutized,
3502    /// canonicalized, required to exist, and NAR-copied into
3503    /// `/nix/store/<hash>-<basename>`; the result string is that store path and
3504    /// it carries store-path context. This is what makes `src = ./.` reference
3505    /// the correct store path (and thus the correct drv hash) instead of a raw
3506    /// filesystem path. `builtins.toString` keeps the plain mode
3507    /// ([`coerce_to_string`]) — it does *not* copy.
3508    pub fn coerce_to_string_copy_to_store(
3509        &self,
3510    ) -> Result<(String, StringContext), EvalError> {
3511        self.coerce_to_string_impl(true)
3512    }
3513
3514    fn coerce_to_string_impl(
3515        &self,
3516        copy_to_store: bool,
3517    ) -> Result<(String, StringContext), EvalError> {
3518        let mut ctx = StringContext::new();
3519        let s = match self {
3520            Value::String(ns) => {
3521                ctx.merge(&ns.context);
3522                ns.chars.to_string()
3523            }
3524            Value::Path(p) => {
3525                let raw: &str = &**p;
3526                if copy_to_store {
3527                    // CppNix copy-to-store coercion: resolve the path to its
3528                    // canonical absolute location (relative literals resolve
3529                    // against the evaluating file's dir, matching CppNix's
3530                    // parse-time absolutization; canonicalize also yields the
3531                    // realpath, e.g. macOS /tmp → /private/tmp), require it to
3532                    // exist (CppNix errors "path '…' does not exist"), NAR-copy
3533                    // it, and reference the resulting store path.
3534                    //
3535                    // A Path VALUE is ALWAYS copied, even one already under
3536                    // /nix/store — CppNix re-NAR-copies a bare path literal
3537                    // (a store subpath like `<nixpkgs-source>/pkgs/…/default-
3538                    // builder.sh` → its own `<hash>-default-builder.sh`, or even
3539                    // a store root) to a fresh basename-named store path,
3540                    // verified against nix 2.34. Store paths that must NOT be
3541                    // re-copied (derivation outputs, fetchurl `src`, storePath)
3542                    // arrive as context-carrying *Strings*, never as Path values,
3543                    // so they never reach this arm. (The earlier `/nix/store/`
3544                    // guard kept stdenv's builder-script subpaths verbatim, which
3545                    // diverged every nixpkgs input-drv hash from nix.)
3546                    let pb = std::path::Path::new(raw);
3547                    let abs = if pb.is_absolute() {
3548                        pb.to_path_buf()
3549                    } else if let Some(dir) = crate::eval::current_eval_dir() {
3550                        dir.join(pb)
3551                    } else {
3552                        std::env::current_dir()
3553                            .map_err(|e| EvalError::IoError {
3554                                context: format!("copy-to-store coercion of {raw}"),
3555                                message: e.to_string(),
3556                            })?
3557                            .join(pb)
3558                    };
3559                    // Redirect the on-disk read to the input's real source
3560                    // tree when `abs` lies under a fetched flake input's
3561                    // `-source` store prefix (sui does not materialize that
3562                    // store path). The resulting store path is NAR-hashed
3563                    // from the tree CONTENT — byte-identical whether read from
3564                    // the store path or the cache — so no value changes.
3565                    let read_abs = crate::path::materialize(&abs);
3566                    let canon = read_abs.canonicalize().map_err(|_| {
3567                        EvalError::TypeError(format!(
3568                            "path '{}' does not exist",
3569                            abs.display()
3570                        ))
3571                    })?;
3572                    // The copied source's STORE-PATH NAME must match CppNix's
3573                    // `baseNameOf` of the input's own `-source` store path when
3574                    // the path being copied IS a fetched flake input's whole
3575                    // tree (the darwin `system-path` root): blx's `src = ./.`
3576                    // copies the blx input tree back into the store, and CppNix
3577                    // names that copy `<inner>-source` (blx's `/nix/store/<h>-
3578                    // source` basename), NOT `blx-<rev>`. sui reads the bytes
3579                    // from the fetcher cache (`canon`, basename `blx-<rev>`), so
3580                    // `canon.file_name()` gave the wrong NAME while the bytes
3581                    // (→ NAR hash) were already correct. Recover the logical
3582                    // `-source` name from the input-source map; fall back to the
3583                    // real dir's basename for a normal local `src = ./.`.
3584                    let name = crate::path::source_name_for_read_dir(&canon)
3585                        .or_else(|| {
3586                            canon
3587                                .file_name()
3588                                .map(|n| n.to_string_lossy().into_owned())
3589                        })
3590                        .unwrap_or_else(|| "source".to_string());
3591                    let src = sui_compat::source::nar_hash_source_tree(&canon, &name)
3592                        .map_err(|e| {
3593                            EvalError::TypeError(format!(
3594                                "copy-to-store coercion of '{}': {e}",
3595                                canon.display()
3596                            ))
3597                        })?;
3598                    ctx.add_plain(src.store_path.clone());
3599                    src.store_path
3600                } else {
3601                    ctx.add_plain(raw.to_string());
3602                    raw.to_string()
3603                }
3604            }
3605            Value::Int(n) => n.to_string(),
3606            // CppNix uses C printf "%f" for float → string coercion,
3607            // which always emits 6 decimal places (`1.5` → "1.500000",
3608            // `3.14159` → "3.141590"). Rust's `{}` formatter strips
3609            // trailing zeros. Match CppNix so `lib.strings.floatToString`
3610            // and module-system defaults round-trip identically.
3611            Value::Float(f) => format!("{f:.6}"),
3612            Value::Bool(true) => "1".to_string(),
3613            Value::Bool(false) => String::new(),
3614            Value::Null => String::new(),
3615            Value::Attrs(attrs) => {
3616                if let Some(to_str) = attrs.get("__toString") {
3617                    let result =
3618                        crate::eval::apply(to_str.clone(), Value::Attrs(attrs.clone()))?;
3619                    let forced = crate::eval::force_value(&result)?;
3620                    let (s, c) = forced.coerce_to_string_impl(copy_to_store)?;
3621                    ctx.merge(&c);
3622                    s
3623                } else if let Some(out_path) = attrs.get("outPath") {
3624                    let forced = crate::eval::force_value(out_path)?;
3625                    let (s, c) = forced.coerce_to_string_impl(copy_to_store)?;
3626                    ctx.merge(&c);
3627                    s
3628                } else {
3629                    return Err(EvalError::TypeError(
3630                        "cannot coerce set to string (no __toString or outPath)".into(),
3631                    ));
3632                }
3633            }
3634            Value::List(items) => {
3635                let mut parts = Vec::new();
3636                for item in items.iter() {
3637                    let forced = crate::eval::force_value(item)?;
3638                    let (s, c) = forced.coerce_to_string_impl(copy_to_store)?;
3639                    ctx.merge(&c);
3640                    parts.push(s);
3641                }
3642                parts.join(" ")
3643            }
3644            Value::Thunk(_) => {
3645                // Force thunk then coerce the result.
3646                let forced = crate::eval::force_value(self)?;
3647                let (s, c) = forced.coerce_to_string_impl(copy_to_store)?;
3648                ctx.merge(&c);
3649                s
3650            }
3651            other => {
3652                return Err(EvalError::TypeError(format!(
3653                    "cannot coerce {} to string",
3654                    other.type_name()
3655                )));
3656            }
3657        };
3658        Ok((s, ctx))
3659    }
3660}
3661
3662// ── Conversions from foreign value types ────────────────────
3663
3664impl From<&serde_json::Value> for Value {
3665    fn from(json: &serde_json::Value) -> Self {
3666        match json {
3667            serde_json::Value::Null => Value::Null,
3668            serde_json::Value::Bool(b) => Value::Bool(*b),
3669            serde_json::Value::Number(n) => {
3670                if let Some(i) = n.as_i64() {
3671                    Value::Int(i)
3672                } else {
3673                    Value::Float(n.as_f64().unwrap_or(0.0))
3674                }
3675            }
3676            serde_json::Value::String(s) => Value::string(s.clone()),
3677            serde_json::Value::Array(arr) => {
3678                Value::List(Rc::new(NixList::new(arr.iter().map(Value::from).collect())))
3679            }
3680            serde_json::Value::Object(obj) => {
3681                let mut attrs = NixAttrs::new();
3682                for (k, v) in obj {
3683                    attrs.insert(k.clone(), Value::from(v));
3684                }
3685                Value::Attrs(Rc::new(attrs))
3686            }
3687        }
3688    }
3689}
3690
3691impl From<&toml::Value> for Value {
3692    fn from(v: &toml::Value) -> Self {
3693        match v {
3694            toml::Value::String(s) => Value::string(s.clone()),
3695            toml::Value::Integer(n) => Value::Int(*n),
3696            toml::Value::Float(f) => Value::Float(*f),
3697            toml::Value::Boolean(b) => Value::Bool(*b),
3698            toml::Value::Array(arr) => {
3699                Value::List(Rc::new(NixList::new(arr.iter().map(Value::from).collect())))
3700            }
3701            toml::Value::Table(t) => {
3702                let mut attrs = NixAttrs::new();
3703                for (k, val) in t {
3704                    attrs.insert(k.clone(), Value::from(val));
3705                }
3706                Value::Attrs(Rc::new(attrs))
3707            }
3708            toml::Value::Datetime(dt) => Value::string(dt.to_string()),
3709        }
3710    }
3711}
3712
3713
3714// ── From impls for ergonomic Value construction ─────────────
3715
3716impl From<bool> for Value {
3717    fn from(b: bool) -> Self {
3718        Value::Bool(b)
3719    }
3720}
3721
3722impl From<i64> for Value {
3723    fn from(n: i64) -> Self {
3724        Value::Int(n)
3725    }
3726}
3727
3728impl From<f64> for Value {
3729    fn from(f: f64) -> Self {
3730        Value::Float(f)
3731    }
3732}
3733
3734impl From<NixString> for Value {
3735    fn from(s: NixString) -> Self {
3736        Value::String(Rc::new(s))
3737    }
3738}
3739
3740impl From<NixAttrs> for Value {
3741    fn from(attrs: NixAttrs) -> Self {
3742        Value::Attrs(Rc::new(attrs))
3743    }
3744}
3745
3746impl From<Vec<Value>> for Value {
3747    fn from(list: Vec<Value>) -> Self {
3748        Value::List(Rc::new(NixList::new(list)))
3749    }
3750}
3751
3752impl PartialEq for Value {
3753    fn eq(&self, other: &Self) -> bool {
3754        // Quick path: pointer-equal thunks are always equal.
3755        if let (Value::Thunk(a), Value::Thunk(b)) = (self, other) {
3756            if Rc::ptr_eq(&a.0, &b.0) { return true; }
3757        }
3758        // Force to Concrete, delegate to Concrete::PartialEq.
3759        // Single source of truth — no duplicated comparison logic.
3760        let l = self.demand().unwrap_or(Concrete::Null);
3761        let r = other.demand().unwrap_or(Concrete::Null);
3762        l == r
3763    }
3764}
3765
3766impl fmt::Display for Value {
3767    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
3768        match self {
3769            Value::Null => write!(f, "null"),
3770            Value::Bool(b) => write!(f, "{b}"),
3771            Value::Int(n) => write!(f, "{n}"),
3772            Value::Float(n) => write!(f, "{}", sui_compat::versions::cppnix_format_float(*n)),
3773            Value::String(s) => write!(f, "\"{}\"", s.chars.replace('\\', "\\\\").replace('"', "\\\"")),
3774            Value::Path(p) => write!(f, "{p}"),
3775            Value::List(items) => {
3776                write!(f, "[ ")?;
3777                for item in items.iter() {
3778                    write!(f, "{item} ")?;
3779                }
3780                write!(f, "]")
3781            }
3782            Value::Attrs(attrs) => {
3783                write!(f, "{{ ")?;
3784                for (k, v) in attrs.iter() {
3785                    write!(f, "{k} = {v}; ")?;
3786                }
3787                write!(f, "}}")
3788            }
3789            Value::Lambda(_) => write!(f, "<<lambda>>"),
3790            Value::Builtin(b) => write!(f, "<<builtin {}>>" , b.name),
3791            Value::Thunk(thunk) => {
3792                match thunk.force(&|e, env| crate::eval::eval_expr(e, env)) {
3793                    Ok(v) => write!(f, "{v}"),
3794                    Err(_) => write!(f, "<<thunk:error>>"),
3795                }
3796            }
3797        }
3798    }
3799}
3800
3801#[cfg(test)]
3802mod tests {
3803    use super::*;
3804    use std::rc::Rc;
3805
3806    // ── Value size assertion ──────────────────────────────
3807
3808    #[test]
3809    fn value_is_16_bytes() {
3810        assert_eq!(std::mem::size_of::<Value>(), 16);
3811    }
3812
3813    // ── Value::to_json for every variant ─────────────────
3814
3815    #[test]
3816    fn to_json_null() {
3817        assert_eq!(Value::Null.to_json(), serde_json::Value::Null);
3818    }
3819
3820    #[test]
3821    fn to_json_bool() {
3822        assert_eq!(Value::Bool(true).to_json(), serde_json::Value::Bool(true));
3823        assert_eq!(Value::Bool(false).to_json(), serde_json::Value::Bool(false));
3824    }
3825
3826    #[test]
3827    fn to_json_int() {
3828        assert_eq!(Value::Int(42).to_json(), serde_json::json!(42));
3829    }
3830
3831    #[test]
3832    fn to_json_float() {
3833        assert_eq!(Value::Float(3.14).to_json(), serde_json::json!(3.14));
3834    }
3835
3836    #[test]
3837    fn to_json_string() {
3838        assert_eq!(
3839            Value::string("hello").to_json(),
3840            serde_json::Value::String("hello".to_string()),
3841        );
3842    }
3843
3844    #[test]
3845    fn to_json_path() {
3846        assert_eq!(
3847            Value::Path(Box::new(SmolStr::from("/nix/store"))).to_json(),
3848            serde_json::Value::String("/nix/store".to_string()),
3849        );
3850    }
3851
3852    #[test]
3853    fn to_json_list() {
3854        let v = Value::list(vec![Value::Int(1), Value::Bool(true)]);
3855        assert_eq!(v.to_json(), serde_json::json!([1, true]));
3856    }
3857
3858    #[test]
3859    fn to_json_attrs() {
3860        let mut attrs = NixAttrs::new();
3861        attrs.insert("a".to_string(), Value::Int(1));
3862        let v = Value::Attrs(Rc::new(attrs));
3863        assert_eq!(v.to_json(), serde_json::json!({"a": 1}));
3864    }
3865
3866    // ── cppnix derivation-equality short-circuit (curl/git root) ─────────
3867
3868    fn mk_drv_attrs(out_path: &str, extra_key: &str, extra_val: i64) -> Value {
3869        let mut a = NixAttrs::new();
3870        a.insert("type".to_string(), Value::string("derivation"));
3871        a.insert("outPath".to_string(), Value::string(out_path));
3872        a.insert(extra_key.to_string(), Value::Int(extra_val));
3873        Value::Attrs(Rc::new(a))
3874    }
3875
3876    #[test]
3877    fn derivations_same_outpath_differing_attrs_are_equal() {
3878        // The load-bearing rule: two attrsets that are BOTH `type=="derivation"`
3879        // with an `outPath` compare by `outPath` string ONLY — differing extra
3880        // attrs must NOT make them unequal. This is what nixpkgs'
3881        // `isMismatchedPython` (`drv.pythonModule != python`) relies on; a deep
3882        // structural compare here spuriously fired the guard and dropped
3883        // `python` from flit-core's `propagatedBuildInputs` (curl/git root).
3884        let a = mk_drv_attrs("/nix/store/x-foo", "foo", 1);
3885        let b = mk_drv_attrs("/nix/store/x-foo", "bar", 2);
3886        assert!(a == b, "same-outPath derivations must compare equal");
3887        assert!(!(a != b));
3888    }
3889
3890    #[test]
3891    fn derivations_differing_outpath_are_unequal() {
3892        let a = mk_drv_attrs("/nix/store/x-foo", "foo", 1);
3893        let b = mk_drv_attrs("/nix/store/y-foo", "foo", 1);
3894        assert!(a != b, "different-outPath derivations must compare unequal");
3895    }
3896
3897    #[test]
3898    fn non_derivation_attrs_with_outpath_use_structural_eq() {
3899        // `outPath` alone (no `type == "derivation"`) does NOT trigger the
3900        // short-circuit — nix falls back to structural equality.
3901        let mut a = NixAttrs::new();
3902        a.insert("outPath".to_string(), Value::string("/nix/store/x"));
3903        a.insert("foo".to_string(), Value::Int(1));
3904        let mut b = NixAttrs::new();
3905        b.insert("outPath".to_string(), Value::string("/nix/store/x"));
3906        b.insert("foo".to_string(), Value::Int(2));
3907        assert!(
3908            Value::Attrs(Rc::new(a)) != Value::Attrs(Rc::new(b)),
3909            "non-derivation attrs with equal outPath but differing foo must be unequal",
3910        );
3911    }
3912
3913    // ── C-A: attrs-eq structural compare BY BORROW (PERF-ARSENAL) ─────────
3914    // These seal that `Concrete::eq`'s Attrs arm — now `a.as_flat() ==
3915    // b.as_flat()` instead of `a.inner() == b.inner()` — is result- and
3916    // force-identical. The clone the old path did was pure allocation waste.
3917
3918    #[test]
3919    fn attrs_eq_borrow_result_matches_multi_key() {
3920        // Structural equality over a multi-key set with a nested attrset value
3921        // must be unaffected by dropping the pre-compare clone.
3922        let mk = || {
3923            let mut inner = NixAttrs::new();
3924            inner.insert("n".to_string(), Value::Int(7));
3925            let mut a = NixAttrs::new();
3926            a.insert("a".to_string(), Value::Int(1));
3927            a.insert("b".to_string(), Value::string("two"));
3928            a.insert("c".to_string(), Value::Attrs(Rc::new(inner)));
3929            Value::Attrs(Rc::new(a))
3930        };
3931        assert!(mk() == mk(), "equal multi-key attrsets must compare equal (borrow path)");
3932
3933        // Differ in one value → unequal.
3934        let mut b = NixAttrs::new();
3935        b.insert("a".to_string(), Value::Int(1));
3936        b.insert("b".to_string(), Value::string("TWO"));
3937        let mut a2 = NixAttrs::new();
3938        a2.insert("a".to_string(), Value::Int(1));
3939        a2.insert("b".to_string(), Value::string("two"));
3940        assert!(
3941            Value::Attrs(Rc::new(a2)) != Value::Attrs(Rc::new(b)),
3942            "attrsets differing in one value must be unequal (borrow path)",
3943        );
3944
3945        // Differ in key SET → unequal.
3946        let mut a3 = NixAttrs::new();
3947        a3.insert("a".to_string(), Value::Int(1));
3948        let mut b3 = NixAttrs::new();
3949        b3.insert("a".to_string(), Value::Int(1));
3950        b3.insert("extra".to_string(), Value::Int(9));
3951        assert!(
3952            Value::Attrs(Rc::new(a3)) != Value::Attrs(Rc::new(b3)),
3953            "attrsets differing in key set must be unequal (borrow path)",
3954        );
3955    }
3956
3957    #[test]
3958    fn attrs_eq_borrow_does_not_force_or_throw_on_shared_thunk() {
3959        // The demand-order verification obligation, made concrete:
3960        // `Value::eq` swallows force errors to `Null` (unwrap_or), so the
3961        // Attrs arm NEVER throws in the map-value-compare path. Two attrsets
3962        // that carry the SAME `Rc`-shared throwing thunk under a key that is
3963        // NOT decisive for (in)equality must:
3964        //   (a) compare via the structural (borrow) path without panicking, and
3965        //   (b) never let the throw escape.
3966        // If the borrow-compare forced the thunk and propagated the error, this
3967        // test would fail — proving the clone-elision touched no `.demand()`
3968        // behaviour that the old `inner()` clone path did not already exhibit.
3969        let boom = Value::Thunk(Thunk::new_native(|| {
3970            Err(EvalError::Throw("kaboom".to_string()))
3971        }));
3972        let mut a = NixAttrs::new();
3973        a.insert("x".to_string(), Value::Int(1));
3974        a.insert("t".to_string(), boom.clone()); // same Rc-shared throwing thunk
3975        let mut b = NixAttrs::new();
3976        b.insert("x".to_string(), Value::Int(2)); // decisive differ on `x`
3977        b.insert("t".to_string(), boom);
3978        // No panic, no escaped Err: the comparison returns a bool. `x` differs,
3979        // so they are unequal — and crucially the throwing `t` thunk did not
3980        // abort the comparison.
3981        let va = Value::Attrs(Rc::new(a));
3982        let vb = Value::Attrs(Rc::new(b));
3983        assert!(va != vb, "differ on x → unequal, throwing thunk must not abort eq");
3984    }
3985
3986    #[test]
3987    fn attrs_eq_borrow_overlay_still_compares() {
3988        // The `as_flat()` borrow path must also work when one side is an
3989        // Overlay (its cache is populated by `as_flat()`), matching the old
3990        // `inner()` path which flattened via the same `as_flat()`.
3991        let mut base = NixAttrs::new();
3992        base.insert("a".to_string(), Value::Int(1));
3993        let mut over = NixAttrs::new();
3994        over.insert("b".to_string(), Value::Int(2));
3995        // Build an overlay { a = 1; } // { b = 2; } (lazy Overlay variant),
3996        // exercising the `as_flat()` cache-population path on one side.
3997        let merged = base.overlay(over);
3998        let mut flat = NixAttrs::new();
3999        flat.insert("a".to_string(), Value::Int(1));
4000        flat.insert("b".to_string(), Value::Int(2));
4001        assert!(
4002            Value::Attrs(Rc::new(merged)) == Value::Attrs(Rc::new(flat)),
4003            "overlay and equivalent flat attrset must compare equal (borrow path)",
4004        );
4005    }
4006
4007    #[test]
4008    fn to_json_lambda() {
4009        // Build a minimal rnix lambda for testing
4010        let root = rnix::Root::parse("x: x");
4011        let expr = root.tree().expr().unwrap();
4012        let lambda = match expr {
4013            rnix::ast::Expr::Lambda(l) => l,
4014            _ => panic!("expected lambda"),
4015        };
4016        let closure = Closure {
4017            param: lambda.param().unwrap(),
4018            body: lambda.body().unwrap(),
4019            env: Env::new(),
4020        };
4021        assert_eq!(
4022            Value::Lambda(Rc::new(closure)).to_json(),
4023            serde_json::Value::String("<lambda>".to_string()),
4024        );
4025    }
4026
4027    #[test]
4028    fn to_json_builtin() {
4029        let b = BuiltinFn {
4030            name: "test",
4031            func: Rc::new(|_| Ok(Value::Null)),
4032        };
4033        assert_eq!(
4034            Value::Builtin(Box::new(b)).to_json(),
4035            serde_json::Value::String("<builtin test>".to_string()),
4036        );
4037    }
4038
4039    // ── Value::type_name for every variant ───────────────
4040
4041    #[test]
4042    fn type_name_null() { assert_eq!(Value::Null.type_name(), "null"); }
4043
4044    #[test]
4045    fn type_name_bool() { assert_eq!(Value::Bool(false).type_name(), "bool"); }
4046
4047    #[test]
4048    fn type_name_int() { assert_eq!(Value::Int(0).type_name(), "int"); }
4049
4050    #[test]
4051    fn type_name_float() { assert_eq!(Value::Float(0.0).type_name(), "float"); }
4052
4053    #[test]
4054    fn type_name_string() { assert_eq!(Value::string("").type_name(), "string"); }
4055
4056    #[test]
4057    fn type_name_path() { assert_eq!(Value::Path(Box::new(SmolStr::from(""))).type_name(), "path"); }
4058
4059    #[test]
4060    fn type_name_list() { assert_eq!(Value::list(vec![]).type_name(), "list"); }
4061
4062    #[test]
4063    fn type_name_set() { assert_eq!(Value::Attrs(Rc::new(NixAttrs::new())).type_name(), "set"); }
4064
4065    #[test]
4066    fn type_name_lambda() {
4067        let root = rnix::Root::parse("x: x");
4068        let expr = root.tree().expr().unwrap();
4069        let lambda = match expr {
4070            rnix::ast::Expr::Lambda(l) => l,
4071            _ => panic!("expected lambda"),
4072        };
4073        let closure = Closure {
4074            param: lambda.param().unwrap(),
4075            body: lambda.body().unwrap(),
4076            env: Env::new(),
4077        };
4078        assert_eq!(Value::Lambda(Rc::new(closure)).type_name(), "lambda");
4079    }
4080
4081    #[test]
4082    fn type_name_builtin() {
4083        let b = BuiltinFn {
4084            name: "t",
4085            func: Rc::new(|_| Ok(Value::Null)),
4086        };
4087        assert_eq!(Value::Builtin(Box::new(b)).type_name(), "lambda");
4088    }
4089
4090    // ── as_* error on wrong type ─────────────────────────
4091
4092    #[test]
4093    fn as_bool_error_on_non_bool() {
4094        assert!(Value::Int(1).as_bool().is_err());
4095        assert!(Value::string("true").as_bool().is_err());
4096    }
4097
4098    #[test]
4099    fn as_int_error_on_non_int() {
4100        assert!(Value::Bool(true).as_int().is_err());
4101        assert!(Value::Float(1.0).as_int().is_err());
4102    }
4103
4104    #[test]
4105    fn as_string_error_on_non_string() {
4106        assert!(Value::Int(42).as_string().is_err());
4107        assert!(Value::Null.as_string().is_err());
4108    }
4109
4110    #[test]
4111    fn as_attrs_error_on_non_attrs() {
4112        assert!(Value::Int(1).as_attrs().is_err());
4113        assert!(Value::list(vec![]).as_attrs().is_err());
4114    }
4115
4116    #[test]
4117    fn as_list_error_on_non_list() {
4118        assert!(Value::Int(1).as_list().is_err());
4119        assert!(Value::Attrs(Rc::new(NixAttrs::new())).as_list().is_err());
4120    }
4121
4122    // ── concat_lists structural share (byte-neutrality) ──────────
4123
4124    #[test]
4125    fn concat_lists_uniquely_owned_reuses_and_is_correct() {
4126        // A fresh left list (Rc strong_count == 1) hits the in-place path.
4127        let left = Value::list(vec![Value::Int(1), Value::Int(2)]);
4128        assert!(left.is_uniquely_owned_list());
4129        let right = [Value::Int(3), Value::Int(4)];
4130        let out = super::concat_lists(left, &right).unwrap();
4131        assert_eq!(
4132            out.as_list().unwrap(),
4133            &[Value::Int(1), Value::Int(2), Value::Int(3), Value::Int(4)]
4134        );
4135    }
4136
4137    #[test]
4138    fn concat_lists_shared_left_is_left_untouched_and_correct() {
4139        // Keep an outstanding Rc clone so the left is NOT uniquely owned;
4140        // the clone-extend fallback fires and the shared list is unchanged.
4141        let shared = Rc::new(NixList::new(vec![Value::Int(1), Value::Int(2)]));
4142        let left = Value::List(Rc::clone(&shared));
4143        assert!(!left.is_uniquely_owned_list());
4144        let right = [Value::Int(3)];
4145        let out = super::concat_lists(left, &right).unwrap();
4146        assert_eq!(
4147            out.as_list().unwrap(),
4148            &[Value::Int(1), Value::Int(2), Value::Int(3)]
4149        );
4150        // The original shared backing Vec is untouched.
4151        assert_eq!(&*shared, &[Value::Int(1), Value::Int(2)]);
4152    }
4153
4154    #[test]
4155    fn concat_lists_empty_operands() {
4156        let out = super::concat_lists(Value::list(vec![]), &[]).unwrap();
4157        assert!(out.as_list().unwrap().is_empty());
4158        let out2 = super::concat_lists(Value::list(vec![Value::Int(9)]), &[]).unwrap();
4159        assert_eq!(out2.as_list().unwrap(), &[Value::Int(9)]);
4160        let out3 = super::concat_lists(Value::list(vec![]), &[Value::Int(9)]).unwrap();
4161        assert_eq!(out3.as_list().unwrap(), &[Value::Int(9)]);
4162    }
4163
4164    #[test]
4165    fn concat_lists_non_list_left_errors() {
4166        assert!(super::concat_lists(Value::Int(1), &[]).is_err());
4167    }
4168
4169    #[test]
4170    fn concat_lists_preserves_element_identity() {
4171        // The concatenated list must share the SAME element Rc, not deep-copy.
4172        let inner = Rc::new(NixString::plain("x"));
4173        let a = Value::String(Rc::clone(&inner));
4174        let left = Value::list(vec![a]);
4175        let out = super::concat_lists(left, &[]).unwrap();
4176        if let Value::String(rc) = &out.as_list().unwrap()[0] {
4177            assert!(Rc::ptr_eq(rc, &inner), "element Rc identity preserved");
4178        } else {
4179            panic!("expected string element");
4180        }
4181    }
4182
4183    // ── to_float int->float coercion ─────────────────────
4184
4185    #[test]
4186    fn to_float_coerces_int() {
4187        assert_eq!(Value::Int(5).to_float().unwrap(), 5.0);
4188        assert_eq!(Value::Float(2.5).to_float().unwrap(), 2.5);
4189        assert!(Value::string("x").to_float().is_err());
4190    }
4191
4192    // ── PartialEq ────────────────────────────────────────
4193
4194    #[test]
4195    fn partial_eq_int_float_cross() {
4196        assert_eq!(Value::Int(3), Value::Float(3.0));
4197        assert_eq!(Value::Float(3.0), Value::Int(3));
4198        assert_ne!(Value::Int(3), Value::Float(3.5));
4199    }
4200
4201    #[test]
4202    fn partial_eq_different_types_not_equal() {
4203        assert_ne!(Value::Int(1), Value::string("1"));
4204        assert_ne!(Value::Bool(true), Value::Int(1));
4205        assert_ne!(Value::Null, Value::Bool(false));
4206        assert_ne!(Value::list(vec![]), Value::Attrs(Rc::new(NixAttrs::new())));
4207    }
4208
4209    // ── Display for all variants ─────────────────────────
4210
4211    #[test]
4212    fn display_null() { assert_eq!(format!("{}", Value::Null), "null"); }
4213
4214    #[test]
4215    fn display_bool() {
4216        assert_eq!(format!("{}", Value::Bool(true)), "true");
4217        assert_eq!(format!("{}", Value::Bool(false)), "false");
4218    }
4219
4220    #[test]
4221    fn display_int() { assert_eq!(format!("{}", Value::Int(42)), "42"); }
4222
4223    #[test]
4224    fn display_float() {
4225        let s = format!("{}", Value::Float(3.14));
4226        assert!(s.contains("3.14"));
4227    }
4228
4229    #[test]
4230    fn display_string() {
4231        assert_eq!(format!("{}", Value::string("hi")), "\"hi\"");
4232    }
4233
4234    #[test]
4235    fn display_string_with_escapes() {
4236        let v = Value::string("a\"b\\c");
4237        let s = format!("{v}");
4238        assert!(s.contains("\\\""));
4239        assert!(s.contains("\\\\"));
4240    }
4241
4242    #[test]
4243    fn display_path() {
4244        assert_eq!(format!("{}", Value::Path(Box::new(SmolStr::from("/foo")))), "/foo");
4245    }
4246
4247    #[test]
4248    fn display_list() {
4249        let v = Value::list(vec![Value::Int(1), Value::Int(2)]);
4250        assert_eq!(format!("{v}"), "[ 1 2 ]");
4251    }
4252
4253    #[test]
4254    fn display_attrs() {
4255        let mut attrs = NixAttrs::new();
4256        attrs.insert("x".to_string(), Value::Int(1));
4257        let v = Value::Attrs(Rc::new(attrs));
4258        assert_eq!(format!("{v}"), "{ x = 1; }");
4259    }
4260
4261    #[test]
4262    fn display_lambda() {
4263        let root = rnix::Root::parse("x: x");
4264        let expr = root.tree().expr().unwrap();
4265        let lambda = match expr {
4266            rnix::ast::Expr::Lambda(l) => l,
4267            _ => panic!("expected lambda"),
4268        };
4269        let closure = Closure {
4270            param: lambda.param().unwrap(),
4271            body: lambda.body().unwrap(),
4272            env: Env::new(),
4273        };
4274        assert_eq!(format!("{}", Value::Lambda(Rc::new(closure))), "<<lambda>>");
4275    }
4276
4277    #[test]
4278    fn display_builtin() {
4279        let b = BuiltinFn {
4280            name: "add",
4281            func: Rc::new(|_| Ok(Value::Null)),
4282        };
4283        assert_eq!(format!("{}", Value::Builtin(Box::new(b))), "<<builtin add>>");
4284    }
4285
4286    // ── NixAttrs ─────────────────────────────────────────
4287
4288    #[test]
4289    fn nixattrs_update_merging() {
4290        let mut a = NixAttrs::new();
4291        a.insert("x".to_string(), Value::Int(1));
4292        a.insert("y".to_string(), Value::Int(2));
4293        let mut b = NixAttrs::new();
4294        b.insert("y".to_string(), Value::Int(99));
4295        b.insert("z".to_string(), Value::Int(3));
4296        let merged = a.update(&b);
4297        assert_eq!(merged.get("x"), Some(&Value::Int(1)));
4298        assert_eq!(merged.get("y"), Some(&Value::Int(99)));
4299        assert_eq!(merged.get("z"), Some(&Value::Int(3)));
4300        assert_eq!(merged.len(), 3);
4301    }
4302
4303    #[test]
4304    fn nixattrs_contains_key() {
4305        let mut a = NixAttrs::new();
4306        a.insert("foo".to_string(), Value::Null);
4307        assert!(a.contains_key("foo"));
4308        assert!(!a.contains_key("bar"));
4309    }
4310
4311    // ── Env ──────────────────────────────────────────────
4312
4313    #[test]
4314    fn env_lookup_through_parent_chain() {
4315        let mut root = Env::new();
4316        root.bind("a".to_string(), Value::Int(1));
4317        let mut child = root.child();
4318        child.bind("b".to_string(), Value::Int(2));
4319        let grandchild = child.child();
4320        // grandchild can see both a and b through parent chain
4321        assert_eq!(grandchild.lookup("a"), Some(Value::Int(1)));
4322        assert_eq!(grandchild.lookup("b"), Some(Value::Int(2)));
4323        assert_eq!(grandchild.lookup("c"), None);
4324    }
4325
4326    #[test]
4327    fn env_with_scope_lookup() {
4328        let mut attrs = NixAttrs::new();
4329        attrs.insert("x".to_string(), Value::Int(42));
4330        let env = Env::new().with_scope(Value::Attrs(Rc::new(attrs)));
4331        assert_eq!(env.lookup("x"), Some(Value::Int(42)));
4332        assert_eq!(env.lookup("y"), None);
4333    }
4334
4335    #[test]
4336    fn env_local_shadows_with_scope() {
4337        let mut attrs = NixAttrs::new();
4338        attrs.insert("x".to_string(), Value::Int(1));
4339        let mut env = Env::new().with_scope(Value::Attrs(Rc::new(attrs)));
4340        env.bind("x".to_string(), Value::Int(99));
4341        assert_eq!(env.lookup("x"), Some(Value::Int(99)));
4342    }
4343
4344    // ── NixString context propagation ─────────────────────
4345
4346    #[test]
4347    fn string_context_merge_combines_elements() {
4348        let mut ctx_a = StringContext::new();
4349        ctx_a.add_plain("/nix/store/aaa".to_string());
4350        let mut ctx_b = StringContext::new();
4351        ctx_b.add_plain("/nix/store/bbb".to_string());
4352        ctx_a.merge(&ctx_b);
4353        assert_eq!(ctx_a.len(), 2);
4354        assert!(ctx_a.elements().contains(&ContextElement::Plain(SmolStr::from("/nix/store/aaa"))));
4355        assert!(ctx_a.elements().contains(&ContextElement::Plain(SmolStr::from("/nix/store/bbb"))));
4356    }
4357
4358    #[test]
4359    fn string_context_merge_deduplicates() {
4360        let mut ctx = StringContext::new();
4361        ctx.add_plain("/nix/store/same".to_string());
4362        ctx.add_plain("/nix/store/same".to_string());
4363        assert_eq!(ctx.len(), 1);
4364    }
4365
4366    #[test]
4367    fn string_context_mixed_element_types() {
4368        let mut ctx = StringContext::new();
4369        ctx.add_plain("/nix/store/foo".to_string());
4370        ctx.add_output("/nix/store/bar.drv".to_string(), "out".to_string());
4371        ctx.add_drv_deep("/nix/store/baz.drv".to_string());
4372        assert_eq!(ctx.len(), 3);
4373        assert!(!ctx.is_empty());
4374    }
4375
4376    #[test]
4377    fn string_context_new_is_empty() {
4378        let ctx = StringContext::new();
4379        assert!(ctx.is_empty());
4380        assert_eq!(ctx.len(), 0);
4381    }
4382
4383    #[test]
4384    fn string_context_merge_zero_elements() {
4385        let mut ctx_a = StringContext::new();
4386        let ctx_b = StringContext::new();
4387        ctx_a.merge(&ctx_b);
4388        assert!(ctx_a.is_empty());
4389    }
4390
4391    #[test]
4392    fn string_context_merge_one_element() {
4393        let mut ctx = StringContext::new();
4394        let mut other = StringContext::new();
4395        other.add_plain("/nix/store/only".to_string());
4396        ctx.merge(&other);
4397        assert_eq!(ctx.len(), 1);
4398        assert!(ctx.elements().contains(&ContextElement::Plain(SmolStr::from("/nix/store/only"))));
4399    }
4400
4401    #[test]
4402    fn string_context_merge_two_elements() {
4403        let mut ctx = StringContext::new();
4404        ctx.add_plain("/nix/store/a".to_string());
4405        let mut other = StringContext::new();
4406        other.add_plain("/nix/store/b".to_string());
4407        ctx.merge(&other);
4408        assert_eq!(ctx.len(), 2);
4409    }
4410
4411    #[test]
4412    fn string_context_merge_five_elements() {
4413        let mut ctx = StringContext::new();
4414        for i in 0..5 {
4415            ctx.add_plain(format!("/nix/store/path-{i}"));
4416        }
4417        assert_eq!(ctx.len(), 5);
4418        for i in 0..5 {
4419            assert!(ctx.elements().contains(&ContextElement::Plain(SmolStr::from(format!("/nix/store/path-{i}").as_str()))));
4420        }
4421    }
4422
4423    #[test]
4424    fn string_context_insert_deduplicates() {
4425        let mut ctx = StringContext::new();
4426        ctx.insert(ContextElement::Plain(SmolStr::from("/nix/store/dup")));
4427        ctx.insert(ContextElement::Plain(SmolStr::from("/nix/store/dup")));
4428        ctx.insert(ContextElement::Output { drv: SmolStr::from("/nix/store/x.drv"), output: SmolStr::from("out") });
4429        ctx.insert(ContextElement::Output { drv: SmolStr::from("/nix/store/x.drv"), output: SmolStr::from("out") });
4430        assert_eq!(ctx.len(), 2);
4431    }
4432
4433    #[test]
4434    fn nix_string_plain_has_no_context() {
4435        let s = NixString::plain("hello");
4436        assert!(!s.has_context());
4437        assert_eq!(s.as_str(), "hello");
4438    }
4439
4440    #[test]
4441    fn nix_string_with_context_reports_context() {
4442        let mut ctx = StringContext::new();
4443        ctx.add_plain("/nix/store/xyz".to_string());
4444        let s = NixString::with_context("hello", ctx);
4445        assert!(s.has_context());
4446        assert_eq!(s.as_str(), "hello");
4447    }
4448
4449    #[test]
4450    fn nix_string_display_shows_chars_only() {
4451        let mut ctx = StringContext::new();
4452        ctx.add_plain("/nix/store/abc".to_string());
4453        let s = NixString::with_context("visible", ctx);
4454        assert_eq!(format!("{s}"), "visible");
4455    }
4456
4457    #[test]
4458    fn nix_string_struct_eq_includes_context() {
4459        let plain = NixString::plain("hello");
4460        let mut ctx = StringContext::new();
4461        ctx.add_plain("/nix/store/xxx".to_string());
4462        let with_ctx = NixString::with_context("hello", ctx);
4463        // NixString's derived PartialEq compares context too
4464        assert_ne!(plain, with_ctx);
4465    }
4466
4467    #[test]
4468    fn value_string_eq_ignores_context() {
4469        let plain = Value::String(Rc::new(NixString::plain("hello")));
4470        let mut ctx = StringContext::new();
4471        ctx.add_plain("/nix/store/xxx".to_string());
4472        let with_ctx = Value::String(Rc::new(NixString::with_context("hello", ctx)));
4473        // Value::PartialEq only compares .chars, ignoring context
4474        assert_eq!(plain, with_ctx);
4475    }
4476
4477    // ── Env deeply nested with-scopes ─────────────────────
4478
4479    #[test]
4480    fn env_nested_with_inner_wins() {
4481        let mut outer_attrs = NixAttrs::new();
4482        outer_attrs.insert("x".to_string(), Value::Int(1));
4483        let outer = Env::new().with_scope(Value::Attrs(Rc::new(outer_attrs)));
4484        let mut inner_attrs = NixAttrs::new();
4485        inner_attrs.insert("x".to_string(), Value::Int(2));
4486        let inner = outer.child().with_scope(Value::Attrs(Rc::new(inner_attrs)));
4487        assert_eq!(inner.lookup("x"), Some(Value::Int(2)));
4488    }
4489
4490    #[test]
4491    fn env_nested_with_fallback_to_outer() {
4492        let mut outer_attrs = NixAttrs::new();
4493        outer_attrs.insert("x".to_string(), Value::Int(1));
4494        let outer = Env::new().with_scope(Value::Attrs(Rc::new(outer_attrs)));
4495        let mut inner_attrs = NixAttrs::new();
4496        inner_attrs.insert("y".to_string(), Value::Int(2));
4497        let inner = outer.child().with_scope(Value::Attrs(Rc::new(inner_attrs)));
4498        assert_eq!(inner.lookup("x"), Some(Value::Int(1)));
4499        assert_eq!(inner.lookup("y"), Some(Value::Int(2)));
4500    }
4501
4502    #[test]
4503    fn env_lexical_binding_wins_over_all_with_scopes() {
4504        let mut outer_attrs = NixAttrs::new();
4505        outer_attrs.insert("x".to_string(), Value::Int(1));
4506        let outer = Env::new().with_scope(Value::Attrs(Rc::new(outer_attrs)));
4507        let mut inner_attrs = NixAttrs::new();
4508        inner_attrs.insert("x".to_string(), Value::Int(2));
4509        let mut inner = outer.child().with_scope(Value::Attrs(Rc::new(inner_attrs)));
4510        inner.bind("x".to_string(), Value::Int(99));
4511        assert_eq!(inner.lookup("x"), Some(Value::Int(99)));
4512    }
4513
4514    #[test]
4515    fn env_parent_lexical_wins_over_child_with_scope() {
4516        let mut root = Env::new();
4517        root.bind("x".to_string(), Value::Int(10));
4518        let mut child_attrs = NixAttrs::new();
4519        child_attrs.insert("x".to_string(), Value::Int(20));
4520        let child = root.child().with_scope(Value::Attrs(Rc::new(child_attrs)));
4521        assert_eq!(child.lookup("x"), Some(Value::Int(10)));
4522    }
4523
4524    #[test]
4525    fn env_deeply_nested_with_scopes_three_levels() {
4526        let mut a = NixAttrs::new();
4527        a.insert("x".to_string(), Value::Int(1));
4528        let env1 = Env::new().with_scope(Value::Attrs(Rc::new(a)));
4529
4530        let mut b = NixAttrs::new();
4531        b.insert("y".to_string(), Value::Int(2));
4532        let env2 = env1.child().with_scope(Value::Attrs(Rc::new(b)));
4533
4534        let mut c = NixAttrs::new();
4535        c.insert("z".to_string(), Value::Int(3));
4536        let env3 = env2.child().with_scope(Value::Attrs(Rc::new(c)));
4537
4538        assert_eq!(env3.lookup("x"), Some(Value::Int(1)));
4539        assert_eq!(env3.lookup("y"), Some(Value::Int(2)));
4540        assert_eq!(env3.lookup("z"), Some(Value::Int(3)));
4541        assert_eq!(env3.lookup("w"), None);
4542    }
4543
4544    #[test]
4545    fn env_with_scope_does_not_pollute_bindings() {
4546        // With-scope values should not appear in the flat binding map.
4547        // They should only be found via the with-scope lookup path.
4548        let mut attrs = NixAttrs::new();
4549        attrs.insert("x".to_string(), Value::Int(42));
4550        let env = Env::new().with_scope(Value::Attrs(Rc::new(attrs)));
4551        // The binding map itself should not contain "x"
4552        assert!(env.0.bindings.get(&intern("x")).is_none());
4553        // But lookup should find it via with-scope
4554        assert_eq!(env.lookup("x"), Some(Value::Int(42)));
4555    }
4556
4557    #[test]
4558    fn env_lexical_binding_not_in_with_scopes() {
4559        // Lexical bindings are in the flat binding map, not in with_scopes.
4560        let mut env = Env::new();
4561        env.bind("x".to_string(), Value::Int(42));
4562        // with_scopes should be empty
4563        assert!(env.0.with_scopes.is_empty());
4564        // But lookup finds it via the binding map
4565        assert_eq!(env.lookup("x"), Some(Value::Int(42)));
4566    }
4567
4568    #[test]
4569    fn env_child_inherits_eval_file() {
4570        let mut env = Env::new();
4571        env.set_eval_file(Some(std::path::PathBuf::from("/foo/bar.nix")));
4572        let child = env.child();
4573        assert_eq!(child.eval_file().cloned(), Some(std::path::PathBuf::from("/foo/bar.nix")));
4574    }
4575
4576    #[test]
4577    fn env_new_has_no_parent_no_with() {
4578        let env = Env::new();
4579        assert_eq!(env.lookup("anything"), None);
4580        assert!(env.eval_file().is_none());
4581    }
4582
4583    // ── Thunk state machine ───────────────────────────────
4584
4585    #[test]
4586    fn thunk_new_suspended_is_not_evaluated() {
4587        let root = rnix::Root::parse("42");
4588        let expr = root.tree().expr().unwrap();
4589        let thunk = Thunk::new_suspended(expr, Env::new());
4590        assert!(!thunk.is_evaluated());
4591    }
4592
4593    #[test]
4594    fn thunk_new_evaluated_is_evaluated() {
4595        let thunk = Thunk::new_evaluated(Value::Int(42));
4596        assert!(thunk.is_evaluated());
4597    }
4598
4599    #[test]
4600    fn thunk_force_evaluates_suspended() {
4601        let root = rnix::Root::parse("42");
4602        let expr = root.tree().expr().unwrap();
4603        let thunk = Thunk::new_suspended(expr, Env::new());
4604        let result = thunk.force(&|e, env| crate::eval::eval_expr(e, env));
4605        assert!(result.is_ok());
4606        assert_eq!(result.unwrap(), Value::Int(42));
4607        assert!(thunk.is_evaluated());
4608    }
4609
4610    #[test]
4611    fn thunk_force_memoizes_result() {
4612        let root = rnix::Root::parse("1 + 2");
4613        let expr = root.tree().expr().unwrap();
4614        let thunk = Thunk::new_suspended(expr, Env::new());
4615        let r1 = thunk.force(&|e, env| crate::eval::eval_expr(e, env)).unwrap();
4616        let r2 = thunk.force(&|e, env| crate::eval::eval_expr(e, env)).unwrap();
4617        assert_eq!(r1, Value::Int(3));
4618        assert_eq!(r2, Value::Int(3));
4619    }
4620
4621    #[test]
4622    fn thunk_force_already_evaluated_returns_value() {
4623        let thunk = Thunk::new_evaluated(Value::Bool(true));
4624        let result = thunk.force(&|_, _| panic!("should not be called"));
4625        assert_eq!(result.unwrap(), Value::Bool(true));
4626    }
4627
4628    // C-store PROVABLY-NEUTRAL seal (M2): a force whose body returns a
4629    // CONCRETE (non-Thunk) value takes the redundant-Store#2 skip path
4630    // (`!was_thunk_before_loop` early-return). It must still (a) return the
4631    // correct value, (b) be `is_evaluated()`, (c) populate the OnceCell so
4632    // the fast-path returns the identical value on re-force (proving Store#1's
4633    // guarded `cache.set` — NOT the skipped Store#2 — is what seals the cache),
4634    // and (d) `peek()` returns the value (repr holds it). If the skip dropped
4635    // the terminal state, one of these would regress.
4636    #[test]
4637    fn thunk_force_concrete_skips_redundant_store_but_caches() {
4638        // `1 + 2` evaluates directly to a concrete Int (no thunk-chain unwrap),
4639        // so it exercises the `!was_thunk_before_loop` skip branch.
4640        let root = rnix::Root::parse("1 + 2");
4641        let expr = root.tree().expr().unwrap();
4642        let thunk = Thunk::new_suspended(expr, Env::new());
4643
4644        let r1 = thunk.force(&|e, env| crate::eval::eval_expr(e, env)).unwrap();
4645        assert_eq!(r1, Value::Int(3));
4646        assert!(thunk.is_evaluated());
4647
4648        // OnceCell must be populated (peek sees the value) — proves Store#1's
4649        // guarded cache.set fired and the skipped Store#2 was truly redundant.
4650        assert_eq!(thunk.peek().map(|c| c.clone().into_value()), Some(Value::Int(3)));
4651
4652        // Re-force hits the OnceCell ultra-fast path and returns byte-identical.
4653        let r2 = thunk.force(&|_, _| panic!("re-force must hit the cache, not re-eval")).unwrap();
4654        assert_eq!(r2, Value::Int(3));
4655    }
4656
4657    #[test]
4658    fn thunk_blackhole_detects_infinite_recursion() {
4659        let root = rnix::Root::parse("42");
4660        let expr = root.tree().expr().unwrap();
4661        let thunk = Thunk::new_suspended(expr, Env::new());
4662
4663        // Manually set to blackhole to simulate re-entrance
4664        // SAFETY: Test-only, single-threaded.
4665        *unsafe { &mut *thunk.0.repr.get() } = ThunkRepr::Blackhole;
4666
4667        let result = thunk.force(&|_, _| Ok(Value::Null));
4668        assert!(result.is_err());
4669        let err_msg = format!("{}", result.unwrap_err());
4670        assert!(err_msg.contains("infinite recursion"));
4671    }
4672
4673    #[test]
4674    fn thunk_update_env_replaces_suspended_env() {
4675        let root = rnix::Root::parse("x");
4676        let expr = root.tree().expr().unwrap();
4677        let thunk = Thunk::new_suspended(expr, Env::new());
4678
4679        let mut new_env = Env::new();
4680        new_env.bind("x".to_string(), Value::Int(99));
4681        thunk.update_env(&new_env);
4682
4683        let result = thunk.force(&|e, env| crate::eval::eval_expr(e, env));
4684        assert_eq!(result.unwrap(), Value::Int(99));
4685    }
4686
4687    #[test]
4688    fn thunk_update_env_noop_when_evaluated() {
4689        let thunk = Thunk::new_evaluated(Value::Int(1));
4690        let mut new_env = Env::new();
4691        new_env.bind("x".to_string(), Value::Int(99));
4692        thunk.update_env(&new_env);
4693        assert_eq!(
4694            thunk.force(&|_, _| panic!("should not be called")).unwrap(),
4695            Value::Int(1),
4696        );
4697    }
4698
4699    #[test]
4700    fn thunk_debug_suspended() {
4701        let root = rnix::Root::parse("42");
4702        let expr = root.tree().expr().unwrap();
4703        let thunk = Thunk::new_suspended(expr, Env::new());
4704        assert_eq!(format!("{thunk:?}"), "<thunk>");
4705    }
4706
4707    #[test]
4708    fn thunk_debug_evaluated() {
4709        let thunk = Thunk::new_evaluated(Value::Int(42));
4710        let dbg = format!("{thunk:?}");
4711        assert!(dbg.contains("42"));
4712    }
4713
4714    #[test]
4715    fn thunk_error_restores_suspended_state() {
4716        let root = rnix::Root::parse("nonexistent_var");
4717        let expr = root.tree().expr().unwrap();
4718        let thunk = Thunk::new_suspended(expr, Env::new());
4719
4720        let result = thunk.force(&|e, env| crate::eval::eval_expr(e, env));
4721        assert!(result.is_err());
4722        // After error, thunk should be restored to Suspended, not stuck as Blackhole
4723        assert!(!thunk.is_evaluated());
4724        let dbg = format!("{thunk:?}");
4725        assert_eq!(dbg, "<thunk>");
4726    }
4727
4728    #[test]
4729    fn thunk_inherit_select_forces_and_selects() {
4730        let root = rnix::Root::parse(r#"{ x = 42; }"#);
4731        let expr = root.tree().expr().unwrap();
4732        let source = Thunk::new_suspended(expr, Env::new());
4733        let thunk = Thunk::new_inherit_select(source, "x".to_string());
4734        let result = thunk.force(&|e, env| crate::eval::eval_expr(e, env));
4735        assert_eq!(result.unwrap(), Value::Int(42));
4736        assert!(thunk.is_evaluated());
4737    }
4738
4739    #[test]
4740    fn thunk_inherit_select_missing_attr_errors() {
4741        let root = rnix::Root::parse(r#"{ x = 42; }"#);
4742        let expr = root.tree().expr().unwrap();
4743        let source = Thunk::new_suspended(expr, Env::new());
4744        let thunk = Thunk::new_inherit_select(source, "y".to_string());
4745        let result = thunk.force(&|e, env| crate::eval::eval_expr(e, env));
4746        assert!(result.is_err());
4747        // Thunk should restore to InheritSelect, not be stuck as Blackhole
4748        assert!(!thunk.is_evaluated());
4749    }
4750
4751    #[test]
4752    fn thunk_inherit_select_non_attrs_source_errors() {
4753        let root = rnix::Root::parse("42");
4754        let expr = root.tree().expr().unwrap();
4755        let source = Thunk::new_suspended(expr, Env::new());
4756        let thunk = Thunk::new_inherit_select(source, "x".to_string());
4757        let result = thunk.force(&|e, env| crate::eval::eval_expr(e, env));
4758        assert!(result.is_err());
4759        let msg = format!("{}", result.unwrap_err());
4760        assert!(msg.contains("not a set"));
4761    }
4762
4763    #[test]
4764    fn thunk_inherit_select_shares_source_thunk() {
4765        // Two InheritSelect thunks share the same source thunk.
4766        // Forcing one should evaluate the source; the second should
4767        // get a cache hit on the shared source thunk.
4768        let root = rnix::Root::parse(r#"{ a = 1; b = 2; }"#);
4769        let expr = root.tree().expr().unwrap();
4770        let source = Thunk::new_suspended(expr, Env::new());
4771        let thunk_a = Thunk::new_inherit_select(source.clone(), "a".to_string());
4772        let thunk_b = Thunk::new_inherit_select(source.clone(), "b".to_string());
4773        let result_a = thunk_a.force(&|e, env| crate::eval::eval_expr(e, env));
4774        assert_eq!(result_a.unwrap(), Value::Int(1));
4775        // Source thunk should now be evaluated (memoized).
4776        assert!(source.is_evaluated());
4777        // Second force should hit the source thunk's cache.
4778        let result_b = thunk_b.force(&|e, env| crate::eval::eval_expr(e, env));
4779        assert_eq!(result_b.unwrap(), Value::Int(2));
4780    }
4781
4782    // ── NixAttrs additional tests ─────────────────────────
4783
4784    #[test]
4785    fn nixattrs_empty_operations() {
4786        let a = NixAttrs::new();
4787        assert!(a.is_empty());
4788        assert_eq!(a.len(), 0);
4789        assert_eq!(a.get("x"), None);
4790        assert!(!a.contains_key("x"));
4791        assert_eq!(a.keys().count(), 0);
4792        assert_eq!(a.iter().count(), 0);
4793    }
4794
4795    #[test]
4796    fn nixattrs_update_with_empty() {
4797        let mut a = NixAttrs::new();
4798        a.insert("x".to_string(), Value::Int(1));
4799        let b = NixAttrs::new();
4800        let merged = a.update(&b);
4801        assert_eq!(merged.len(), 1);
4802        assert_eq!(merged.get("x"), Some(&Value::Int(1)));
4803    }
4804
4805    #[test]
4806    fn nixattrs_update_empty_with_nonempty() {
4807        let a = NixAttrs::new();
4808        let mut b = NixAttrs::new();
4809        b.insert("x".to_string(), Value::Int(1));
4810        let merged = a.update(&b);
4811        assert_eq!(merged.len(), 1);
4812        assert_eq!(merged.get("x"), Some(&Value::Int(1)));
4813    }
4814
4815    #[test]
4816    fn nixattrs_keys_sorted_order() {
4817        let mut a = NixAttrs::new();
4818        a.insert("c".to_string(), Value::Int(3));
4819        a.insert("a".to_string(), Value::Int(1));
4820        a.insert("b".to_string(), Value::Int(2));
4821        let keys: Vec<String> = a.keys().collect();
4822        assert_eq!(keys, vec!["a", "b", "c"]);
4823    }
4824
4825    // ── Value convenience methods ─────────────────────────
4826
4827    #[test]
4828    fn value_to_str_forces_thunks() {
4829        let root = rnix::Root::parse(r#""hello""#);
4830        let expr = root.tree().expr().unwrap();
4831        let thunk = Thunk::new_suspended(expr, Env::new());
4832        let val = Value::Thunk(thunk);
4833        assert_eq!(val.to_str().unwrap(), "hello");
4834    }
4835
4836    #[test]
4837    fn value_to_nix_string_forces_thunks() {
4838        let root = rnix::Root::parse(r#""world""#);
4839        let expr = root.tree().expr().unwrap();
4840        let thunk = Thunk::new_suspended(expr, Env::new());
4841        let val = Value::Thunk(thunk);
4842        let ns = val.to_nix_string().unwrap();
4843        assert_eq!(ns.as_str(), "world");
4844        assert!(!ns.has_context());
4845    }
4846
4847    #[test]
4848    fn value_to_attrs_forces_thunks() {
4849        let root = rnix::Root::parse("{ x = 1; }");
4850        let expr = root.tree().expr().unwrap();
4851        let thunk = Thunk::new_suspended(expr, Env::new());
4852        let val = Value::Thunk(thunk);
4853        let attrs = val.to_attrs().unwrap();
4854        assert_eq!(attrs.len(), 1);
4855    }
4856
4857    #[test]
4858    fn value_to_list_forces_thunks() {
4859        let root = rnix::Root::parse("[1 2 3]");
4860        let expr = root.tree().expr().unwrap();
4861        let thunk = Thunk::new_suspended(expr, Env::new());
4862        let val = Value::Thunk(thunk);
4863        let list = val.to_list().unwrap();
4864        assert_eq!(list.len(), 3);
4865    }
4866
4867    #[test]
4868    fn value_to_float_on_thunk() {
4869        let root = rnix::Root::parse("3.14");
4870        let expr = root.tree().expr().unwrap();
4871        let thunk = Thunk::new_suspended(expr, Env::new());
4872        let val = Value::Thunk(thunk);
4873        let f = val.to_float().unwrap();
4874        assert!((f - 3.14).abs() < f64::EPSILON);
4875    }
4876
4877    #[test]
4878    fn value_as_bool_on_thunk() {
4879        let root = rnix::Root::parse("true");
4880        let expr = root.tree().expr().unwrap();
4881        let thunk = Thunk::new_suspended(expr, Env::new());
4882        let val = Value::Thunk(thunk);
4883        assert!(val.as_bool().unwrap());
4884    }
4885
4886    #[test]
4887    fn value_as_int_on_thunk() {
4888        let root = rnix::Root::parse("42");
4889        let expr = root.tree().expr().unwrap();
4890        let thunk = Thunk::new_suspended(expr, Env::new());
4891        let val = Value::Thunk(thunk);
4892        assert_eq!(val.as_int().unwrap(), 42);
4893    }
4894
4895    #[test]
4896    fn value_string_constructor() {
4897        let v = Value::string("test");
4898        assert_eq!(v, Value::String(Rc::new(NixString::plain("test"))));
4899    }
4900
4901    #[test]
4902    fn value_partial_eq_null_null() {
4903        assert_eq!(Value::Null, Value::Null);
4904    }
4905
4906    #[test]
4907    fn value_partial_eq_lists_deep() {
4908        let a = Value::list(vec![Value::Int(1), Value::list(vec![Value::Int(2)])]);
4909        let b = Value::list(vec![Value::Int(1), Value::list(vec![Value::Int(2)])]);
4910        assert_eq!(a, b);
4911    }
4912
4913    #[test]
4914    fn value_partial_eq_attrs_deep() {
4915        let mut a = NixAttrs::new();
4916        a.insert("x".to_string(), Value::Int(1));
4917        let mut b = NixAttrs::new();
4918        b.insert("x".to_string(), Value::Int(1));
4919        assert_eq!(Value::Attrs(Rc::new(a)), Value::Attrs(Rc::new(b)));
4920    }
4921
4922    // ── EvalError variants & convenience constructors ────
4923
4924    #[test]
4925    fn eval_error_type_error_constructor() {
4926        let e = EvalError::type_error("oops");
4927        assert!(matches!(e, EvalError::TypeError(ref s) if s == "oops"));
4928    }
4929
4930    #[test]
4931    fn eval_error_type_mismatch_constructor() {
4932        let e = EvalError::type_mismatch("int", "string");
4933        match e {
4934            EvalError::TypeMismatch { expected, got } => {
4935                assert_eq!(expected, "int");
4936                assert_eq!(got, "string");
4937            }
4938            _ => panic!("expected TypeMismatch"),
4939        }
4940    }
4941
4942    #[test]
4943    fn eval_error_is_throw_yes_no() {
4944        assert!(EvalError::Throw("oops".into()).is_throw());
4945        assert!(!EvalError::TypeError("oops".into()).is_throw());
4946        assert!(!EvalError::AssertionFailed(String::new()).is_throw());
4947    }
4948
4949    #[test]
4950    fn eval_error_is_infinite_recursion_yes_no() {
4951        assert!(EvalError::InfiniteRecursion("loop".into()).is_infinite_recursion());
4952        assert!(!EvalError::DivisionByZero.is_infinite_recursion());
4953        assert!(!EvalError::Throw("x".into()).is_infinite_recursion());
4954    }
4955
4956    #[test]
4957    fn eval_error_display_undefined_var() {
4958        let s = format!("{}", EvalError::UndefinedVar("foo".into()));
4959        assert!(s.contains("undefined variable"));
4960        assert!(s.contains("foo"));
4961    }
4962
4963    #[test]
4964    fn eval_error_display_type_error() {
4965        let s = format!("{}", EvalError::TypeError("bad".into()));
4966        assert!(s.contains("type error"));
4967        assert!(s.contains("bad"));
4968    }
4969
4970    #[test]
4971    fn eval_error_display_attr_not_found() {
4972        let s = format!("{}", EvalError::AttrNotFound("x".into()));
4973        assert!(s.contains("attribute not found"));
4974        assert!(s.contains("x"));
4975    }
4976
4977    #[test]
4978    fn eval_error_display_type_mismatch() {
4979        let s = format!(
4980            "{}",
4981            EvalError::TypeMismatch { expected: "int", got: "string" }
4982        );
4983        assert!(s.contains("expected int"));
4984        assert!(s.contains("got string"));
4985    }
4986
4987    #[test]
4988    fn eval_error_display_assertion_failed() {
4989        let s = format!("{}", EvalError::AssertionFailed(String::new()));
4990        assert!(s.contains("assertion"));
4991    }
4992
4993    #[test]
4994    fn eval_error_display_division_by_zero() {
4995        let s = format!("{}", EvalError::DivisionByZero);
4996        assert!(s.contains("division by zero"));
4997    }
4998
4999    #[test]
5000    fn eval_error_display_infinite_recursion() {
5001        let s = format!("{}", EvalError::InfiniteRecursion("loop".into()));
5002        assert!(s.contains("infinite recursion"));
5003        assert!(s.contains("loop"));
5004    }
5005
5006    #[test]
5007    fn eval_error_display_io_error() {
5008        let s = format!(
5009            "{}",
5010            EvalError::IoError {
5011                context: "ctx".into(),
5012                message: "no such file".into(),
5013            }
5014        );
5015        assert!(s.contains("I/O"));
5016        assert!(s.contains("ctx"));
5017        assert!(s.contains("no such file"));
5018    }
5019
5020    #[test]
5021    fn eval_error_display_throw() {
5022        let s = format!("{}", EvalError::Throw("boom".into()));
5023        assert_eq!(s, "boom");
5024    }
5025
5026    #[test]
5027    fn eval_error_display_not_implemented() {
5028        let s = format!("{}", EvalError::NotImplemented("frob".into()));
5029        assert!(s.contains("not yet implemented"));
5030        assert!(s.contains("frob"));
5031    }
5032
5033    #[test]
5034    fn eval_error_display_parse_error() {
5035        let s = format!("{}", EvalError::ParseError("syntax".into()));
5036        assert!(s.contains("parse error"));
5037        assert!(s.contains("syntax"));
5038    }
5039
5040    #[test]
5041    fn eval_error_display_recursion_limit() {
5042        let s = format!(
5043            "{}",
5044            EvalError::RecursionLimit("max depth exceeded".into())
5045        );
5046        assert!(s.contains("recursion limit"));
5047        assert!(s.contains("max depth exceeded"));
5048    }
5049
5050    #[test]
5051    fn eval_error_partial_eq_same_variant() {
5052        assert_eq!(
5053            EvalError::UndefinedVar("x".into()),
5054            EvalError::UndefinedVar("x".into()),
5055        );
5056        assert_ne!(
5057            EvalError::UndefinedVar("x".into()),
5058            EvalError::UndefinedVar("y".into()),
5059        );
5060        assert_ne!(
5061            EvalError::UndefinedVar("x".into()),
5062            EvalError::AttrNotFound("x".into()),
5063        );
5064    }
5065
5066    // ── ContextElement display ───────────────────────────
5067
5068    #[test]
5069    fn context_element_display_plain() {
5070        let e = ContextElement::Plain("/nix/store/xyz".into());
5071        assert_eq!(format!("{e}"), "/nix/store/xyz");
5072    }
5073
5074    #[test]
5075    fn context_element_display_output() {
5076        let e = ContextElement::Output {
5077            drv: "/nix/store/abc.drv".into(),
5078            output: "out".into(),
5079        };
5080        assert_eq!(format!("{e}"), "/nix/store/abc.drv!out");
5081    }
5082
5083    #[test]
5084    fn context_element_display_drv_deep() {
5085        let e = ContextElement::DrvDeep("/nix/store/abc.drv".into());
5086        assert_eq!(format!("{e}"), "=/nix/store/abc.drv");
5087    }
5088
5089    // ── StringContext additional API ─────────────────────
5090
5091    #[test]
5092    fn string_context_iter_yields_all() {
5093        let mut ctx = StringContext::new();
5094        ctx.add_plain("/nix/store/aaa");
5095        ctx.add_plain("/nix/store/bbb");
5096        let count = ctx.iter().count();
5097        assert_eq!(count, 2);
5098    }
5099
5100    #[test]
5101    fn string_context_len_matches_set_size() {
5102        let mut ctx = StringContext::new();
5103        assert_eq!(ctx.len(), 0);
5104        ctx.add_plain("/nix/store/x");
5105        assert_eq!(ctx.len(), 1);
5106        ctx.add_output("/nix/store/y.drv", "out");
5107        assert_eq!(ctx.len(), 2);
5108    }
5109
5110    #[test]
5111    fn string_context_insert_raw_element() {
5112        let mut ctx = StringContext::new();
5113        ctx.insert(ContextElement::Plain("/nix/store/foo".into()));
5114        assert_eq!(ctx.len(), 1);
5115    }
5116
5117    #[test]
5118    fn string_context_default_is_empty() {
5119        let ctx = StringContext::default();
5120        assert!(ctx.is_empty());
5121    }
5122
5123    // ── NixString additional traits ──────────────────────
5124
5125    #[test]
5126    fn nix_string_as_ref_str() {
5127        let s = NixString::plain("hello");
5128        let r: &str = s.as_ref();
5129        assert_eq!(r, "hello");
5130    }
5131
5132    #[test]
5133    fn nix_string_deref_to_str_methods() {
5134        let s = NixString::plain("Hello World");
5135        assert_eq!(s.len(), 11);
5136        assert!(s.starts_with("Hello"));
5137        // Calling &str method via Deref proves Deref impl is wired up.
5138        assert_eq!(s.to_uppercase(), "HELLO WORLD");
5139    }
5140
5141    // ── NixAttrs additional API ──────────────────────────
5142
5143    #[test]
5144    fn nixattrs_remove_returns_value() {
5145        let mut a = NixAttrs::new();
5146        a.insert("x".into(), Value::Int(1));
5147        let removed = a.remove("x");
5148        assert_eq!(removed, Some(Value::Int(1)));
5149        assert!(!a.contains_key("x"));
5150        assert_eq!(a.remove("y"), None);
5151    }
5152
5153    #[test]
5154    fn nixattrs_values_iter() {
5155        let mut a = NixAttrs::new();
5156        a.insert("a".into(), Value::Int(1));
5157        a.insert("b".into(), Value::Int(2));
5158        let mut vs: Vec<&Value> = a.values().collect();
5159        vs.sort_by_key(|v| match v {
5160            Value::Int(n) => *n,
5161            _ => 0,
5162        });
5163        assert_eq!(vs, vec![&Value::Int(1), &Value::Int(2)]);
5164    }
5165
5166    #[test]
5167    fn nixattrs_iter_returns_sorted_pairs() {
5168        let mut a = NixAttrs::new();
5169        a.insert("zeta".into(), Value::Int(3));
5170        a.insert("alpha".into(), Value::Int(1));
5171        a.insert("mu".into(), Value::Int(2));
5172        let pairs: Vec<(String, &Value)> = a.iter().collect();
5173        assert_eq!(pairs[0].0, "alpha");
5174        assert_eq!(pairs[1].0, "mu");
5175        assert_eq!(pairs[2].0, "zeta");
5176    }
5177
5178    #[test]
5179    fn nixattrs_from_iterator() {
5180        let pairs = vec![
5181            ("a".to_string(), Value::Int(1)),
5182            ("b".to_string(), Value::Int(2)),
5183        ];
5184        let attrs: NixAttrs = pairs.into_iter().collect();
5185        assert_eq!(attrs.len(), 2);
5186        assert_eq!(attrs.get("a"), Some(&Value::Int(1)));
5187        assert_eq!(attrs.get("b"), Some(&Value::Int(2)));
5188    }
5189
5190    #[test]
5191    fn nixattrs_into_iterator_yields_owned() {
5192        let mut a = NixAttrs::new();
5193        a.insert("x".into(), Value::Int(42));
5194        let pairs: Vec<(String, Value)> = a.into_iter().collect();
5195        assert_eq!(pairs.len(), 1);
5196        assert_eq!(pairs[0].0, "x");
5197        assert_eq!(pairs[0].1, Value::Int(42));
5198    }
5199
5200    #[test]
5201    fn nixattrs_default_is_empty() {
5202        let a = NixAttrs::default();
5203        assert!(a.is_empty());
5204    }
5205
5206    // ── Value::From conversions ──────────────────────────
5207
5208    #[test]
5209    fn value_from_bool() {
5210        assert_eq!(Value::from(true), Value::Bool(true));
5211        assert_eq!(Value::from(false), Value::Bool(false));
5212    }
5213
5214    #[test]
5215    fn value_from_i64() {
5216        assert_eq!(Value::from(42_i64), Value::Int(42));
5217        assert_eq!(Value::from(-1_i64), Value::Int(-1));
5218    }
5219
5220    #[test]
5221    fn value_from_f64() {
5222        assert_eq!(Value::from(2.5_f64), Value::Float(2.5));
5223    }
5224
5225    #[test]
5226    fn value_from_nix_string() {
5227        let v: Value = NixString::plain("hi").into();
5228        assert_eq!(v, Value::string("hi"));
5229    }
5230
5231    #[test]
5232    fn value_from_nix_attrs() {
5233        let mut a = NixAttrs::new();
5234        a.insert("x".into(), Value::Int(1));
5235        let v: Value = a.into();
5236        match v {
5237            Value::Attrs(_) => {}
5238            _ => panic!("expected Attrs"),
5239        }
5240    }
5241
5242    #[test]
5243    fn value_from_vec() {
5244        let v: Value = vec![Value::Int(1), Value::Int(2)].into();
5245        assert_eq!(v, Value::list(vec![Value::Int(1), Value::Int(2)]));
5246    }
5247
5248    #[test]
5249    fn value_default_is_null() {
5250        let v: Value = Value::default();
5251        assert_eq!(v, Value::Null);
5252    }
5253
5254    // ── From<&serde_json::Value> ─────────────────────────
5255
5256    #[test]
5257    fn value_from_json_null() {
5258        let v = Value::from(&serde_json::Value::Null);
5259        assert_eq!(v, Value::Null);
5260    }
5261
5262    #[test]
5263    fn value_from_json_bool() {
5264        let v = Value::from(&serde_json::Value::Bool(true));
5265        assert_eq!(v, Value::Bool(true));
5266    }
5267
5268    #[test]
5269    fn value_from_json_int() {
5270        let v = Value::from(&serde_json::json!(42));
5271        assert_eq!(v, Value::Int(42));
5272    }
5273
5274    #[test]
5275    fn value_from_json_float() {
5276        let v = Value::from(&serde_json::json!(3.14));
5277        match v {
5278            Value::Float(f) => assert!((f - 3.14).abs() < f64::EPSILON),
5279            _ => panic!("expected Float"),
5280        }
5281    }
5282
5283    #[test]
5284    fn value_from_json_string() {
5285        let v = Value::from(&serde_json::Value::String("hi".into()));
5286        assert_eq!(v, Value::string("hi"));
5287    }
5288
5289    #[test]
5290    fn value_from_json_array() {
5291        let v = Value::from(&serde_json::json!([1, true, "x"]));
5292        match v {
5293            Value::List(items) => {
5294                assert_eq!(items.len(), 3);
5295                assert_eq!(items[0], Value::Int(1));
5296                assert_eq!(items[1], Value::Bool(true));
5297                assert_eq!(items[2], Value::string("x"));
5298            }
5299            _ => panic!("expected List"),
5300        }
5301    }
5302
5303    #[test]
5304    fn value_from_json_object() {
5305        let v = Value::from(&serde_json::json!({"a": 1, "b": "x"}));
5306        match v {
5307            Value::Attrs(attrs) => {
5308                assert_eq!(attrs.get("a"), Some(&Value::Int(1)));
5309                assert_eq!(attrs.get("b"), Some(&Value::string("x")));
5310            }
5311            _ => panic!("expected Attrs"),
5312        }
5313    }
5314
5315    #[test]
5316    fn value_from_json_nested() {
5317        let v = Value::from(&serde_json::json!({"outer": {"inner": [1, 2]}}));
5318        let json_back = v.to_json();
5319        assert_eq!(json_back, serde_json::json!({"outer": {"inner": [1, 2]}}));
5320    }
5321
5322    // ── From<&toml::Value> ──────────────────────────────
5323
5324    #[test]
5325    fn value_from_toml_string() {
5326        let t = toml::Value::String("hi".into());
5327        assert_eq!(Value::from(&t), Value::string("hi"));
5328    }
5329
5330    #[test]
5331    fn value_from_toml_int() {
5332        let t = toml::Value::Integer(42);
5333        assert_eq!(Value::from(&t), Value::Int(42));
5334    }
5335
5336    #[test]
5337    fn value_from_toml_float() {
5338        let t = toml::Value::Float(3.14);
5339        match Value::from(&t) {
5340            Value::Float(f) => assert!((f - 3.14).abs() < f64::EPSILON),
5341            _ => panic!("expected Float"),
5342        }
5343    }
5344
5345    #[test]
5346    fn value_from_toml_bool() {
5347        let t = toml::Value::Boolean(true);
5348        assert_eq!(Value::from(&t), Value::Bool(true));
5349    }
5350
5351    #[test]
5352    fn value_from_toml_array() {
5353        let t = toml::Value::Array(vec![
5354            toml::Value::Integer(1),
5355            toml::Value::Integer(2),
5356        ]);
5357        assert_eq!(
5358            Value::from(&t),
5359            Value::list(vec![Value::Int(1), Value::Int(2)]),
5360        );
5361    }
5362
5363    #[test]
5364    fn value_from_toml_table() {
5365        let mut tbl = toml::map::Map::new();
5366        tbl.insert("k".into(), toml::Value::Integer(7));
5367        let t = toml::Value::Table(tbl);
5368        match Value::from(&t) {
5369            Value::Attrs(attrs) => {
5370                assert_eq!(attrs.get("k"), Some(&Value::Int(7)));
5371            }
5372            _ => panic!("expected Attrs"),
5373        }
5374    }
5375
5376    #[test]
5377    fn value_from_toml_datetime_becomes_string() {
5378        // toml::Value::Datetime serializes via Display.
5379        let dt: toml::value::Datetime = "2024-01-01T00:00:00Z".parse().unwrap();
5380        let t = toml::Value::Datetime(dt);
5381        match Value::from(&t) {
5382            Value::String(_) => {}
5383            other => panic!("expected String, got {other:?}"),
5384        }
5385    }
5386
5387    // ── Value::coerce_to_path ────────────────────────────
5388
5389    #[test]
5390    fn coerce_to_path_from_path() {
5391        let v = Value::Path(Box::new("/foo".into()));
5392        assert_eq!(v.coerce_to_path("ctx").unwrap(), "/foo");
5393    }
5394
5395    #[test]
5396    fn coerce_to_path_from_string() {
5397        let v = Value::string("/bar");
5398        assert_eq!(v.coerce_to_path("ctx").unwrap(), "/bar");
5399    }
5400
5401    // ── Import-from-derivation (IFD) detection + realize seal ──────────
5402    //
5403    // These lock the parity-critical decision "is this coercion an
5404    // import-from-derivation that must realize?" — the same decision cppnix
5405    // makes off a string's `Output` context. Regressing them silently reopens
5406    // the marquee `import ishou.stylix-fonts` root.
5407
5408    #[test]
5409    fn out_path_needs_realize_matches_output_context() {
5410        // A store-path string carrying a derivation `Output` context IS a
5411        // derivation output → its producing `.drv` is returned for realize.
5412        let mut ctx = StringContext::new();
5413        ctx.add_output("/nix/store/aaa-thing.drv", "out");
5414        assert_eq!(
5415            super::out_path_needs_realize("/nix/store/bbb-thing", &ctx),
5416            Some("/nix/store/aaa-thing.drv".to_string()),
5417        );
5418    }
5419
5420    #[test]
5421    fn out_path_needs_realize_ignores_plain_context() {
5422        // A plain store-path reference (not a derivation output) has nothing to
5423        // build — no realize.
5424        let mut ctx = StringContext::new();
5425        ctx.add_plain("/nix/store/ccc-plain");
5426        assert_eq!(super::out_path_needs_realize("/nix/store/ccc-plain", &ctx), None);
5427    }
5428
5429    #[test]
5430    fn out_path_needs_realize_ignores_non_store_path() {
5431        // A non-store path is never a derivation output, even with an (invalid)
5432        // Output context — nothing to realize.
5433        let mut ctx = StringContext::new();
5434        ctx.add_output("/nix/store/ddd.drv", "out");
5435        assert_eq!(super::out_path_needs_realize("/etc/passwd", &ctx), None);
5436    }
5437
5438    #[test]
5439    fn out_path_needs_realize_empty_context_is_none() {
5440        // A bare store-path literal (empty context) is not a derivation output.
5441        let ctx = StringContext::new();
5442        assert_eq!(super::out_path_needs_realize("/nix/store/eee-lit", &ctx), None);
5443    }
5444
5445    #[test]
5446    fn coerce_to_realized_path_present_output_is_passthrough() {
5447        // When the output already exists on disk, no hook is needed and the
5448        // path is returned unchanged (the no-op realize path — proven live on
5449        // the already-built ifd-test derivation).
5450        let dir = std::env::temp_dir().join("sui-ifd-present-test");
5451        std::fs::create_dir_all(&dir).unwrap();
5452        let file = dir.join("out");
5453        std::fs::write(&file, b"present").unwrap();
5454        let present = file.to_string_lossy().to_string();
5455
5456        let mut ctx = StringContext::new();
5457        // Pretend it's a derivation output (Output context) — but it exists,
5458        // so realize must NOT be invoked (no hook installed → would ENOENT if
5459        // it tried). A store-prefix check would skip a temp path, so assert the
5460        // simpler invariant: an existing plain string coerces to itself.
5461        ctx.add_plain(&present);
5462        let v = Value::String(std::rc::Rc::new(NixString::with_context(
5463            present.as_str(),
5464            ctx,
5465        )));
5466        assert_eq!(v.coerce_to_realized_path("readFile").unwrap(), present);
5467    }
5468
5469    #[test]
5470    fn coerce_to_realized_path_absent_output_invokes_hook() {
5471        // A store-path string with an Output context whose output is ABSENT
5472        // invokes the realize hook with the producing drv. The mock hook
5473        // "materializes" nothing (the store path stays absent) but records the
5474        // call — proving the trigger fires end-to-end through coercion.
5475        use std::sync::{Arc, Mutex};
5476        let seen: Arc<Mutex<Vec<(String, String)>>> = Arc::new(Mutex::new(Vec::new()));
5477        let seen2 = seen.clone();
5478        let _guard = crate::realize::install_realize_hook(Box::new(move |drv, out| {
5479            seen2.lock().unwrap().push((drv.to_string(), out.to_string()));
5480            Ok(())
5481        }));
5482
5483        // An absent store path (unique per run to avoid collision with a real
5484        // build) carrying an Output context.
5485        let out = "/nix/store/zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz-ifd-absent";
5486        assert!(!std::path::Path::new(out).exists(), "test store path must be absent");
5487        let mut ctx = StringContext::new();
5488        ctx.add_output("/nix/store/qqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqq-ifd-absent.drv", "out");
5489        let v = Value::String(std::rc::Rc::new(NixString::with_context(out, ctx)));
5490
5491        // Coercion returns the outPath and fires the hook exactly once.
5492        assert_eq!(v.coerce_to_realized_path("readFile").unwrap(), out);
5493        let s = seen.lock().unwrap();
5494        assert_eq!(s.len(), 1, "realize hook should fire once for an absent output");
5495        assert_eq!(s[0].0, "/nix/store/qqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqq-ifd-absent.drv");
5496        assert_eq!(s[0].1, out);
5497    }
5498
5499    #[test]
5500    fn coerce_to_path_errors_on_int() {
5501        let v = Value::Int(1);
5502        let e = v.coerce_to_path("readFile").unwrap_err();
5503        match e {
5504            EvalError::TypeError(ref msg) => {
5505                assert!(msg.contains("readFile"));
5506                assert!(msg.contains("path or string"));
5507                assert!(msg.contains("int"));
5508            }
5509            _ => panic!("expected TypeError"),
5510        }
5511    }
5512
5513    #[test]
5514    fn coerce_to_path_errors_on_null() {
5515        let v = Value::Null;
5516        assert!(v.coerce_to_path("ctx").is_err());
5517    }
5518
5519    #[test]
5520    fn coerce_to_path_attrs_with_outpath() {
5521        let mut attrs = NixAttrs::new();
5522        attrs.insert("outPath".to_string(), Value::string("/nix/store/test"));
5523        let val = Value::Attrs(Rc::new(attrs));
5524        assert_eq!(val.coerce_to_path("test").unwrap(), "/nix/store/test");
5525    }
5526
5527    #[test]
5528    fn coerce_to_path_attrs_without_outpath_fails() {
5529        let attrs = NixAttrs::new();
5530        let val = Value::Attrs(Rc::new(attrs));
5531        assert!(val.coerce_to_path("test").is_err());
5532    }
5533
5534    // ── Value::coerce_to_string ─────────────────────────
5535
5536    #[test]
5537    fn coerce_to_string_string() {
5538        let v = Value::string("hello");
5539        let (s, _ctx) = v.coerce_to_string().unwrap();
5540        assert_eq!(s, "hello");
5541    }
5542
5543    #[test]
5544    fn coerce_to_string_path() {
5545        let v = Value::Path(Box::new("/foo".into()));
5546        let (s, ctx) = v.coerce_to_string().unwrap();
5547        assert_eq!(s, "/foo");
5548        assert!(!ctx.is_empty()); // should add a Plain context element
5549    }
5550
5551    #[test]
5552    fn coerce_to_string_int() {
5553        let v = Value::Int(42);
5554        let (s, _ctx) = v.coerce_to_string().unwrap();
5555        assert_eq!(s, "42");
5556    }
5557
5558    #[test]
5559    fn coerce_to_string_float() {
5560        // CppNix %f-format: always 6 decimal places.
5561        let v = Value::Float(3.14);
5562        let (s, _ctx) = v.coerce_to_string().unwrap();
5563        assert_eq!(s, "3.140000");
5564    }
5565
5566    #[test]
5567    fn coerce_to_string_bool_true() {
5568        let (s, _ctx) = Value::Bool(true).coerce_to_string().unwrap();
5569        assert_eq!(s, "1");
5570    }
5571
5572    #[test]
5573    fn coerce_to_string_bool_false() {
5574        let (s, _ctx) = Value::Bool(false).coerce_to_string().unwrap();
5575        assert_eq!(s, "");
5576    }
5577
5578    #[test]
5579    fn coerce_to_string_null() {
5580        let (s, _ctx) = Value::Null.coerce_to_string().unwrap();
5581        assert_eq!(s, "");
5582    }
5583
5584    #[test]
5585    fn coerce_to_string_attrs_with_outpath() {
5586        let mut attrs = NixAttrs::new();
5587        attrs.insert("outPath".to_string(), Value::string("/nix/store/abc"));
5588        let val = Value::Attrs(Rc::new(attrs));
5589        let (s, _ctx) = val.coerce_to_string().unwrap();
5590        assert_eq!(s, "/nix/store/abc");
5591    }
5592
5593    #[test]
5594    fn coerce_to_string_attrs_without_outpath_or_tostring_fails() {
5595        let attrs = NixAttrs::new();
5596        let val = Value::Attrs(Rc::new(attrs));
5597        assert!(val.coerce_to_string().is_err());
5598    }
5599
5600    #[test]
5601    fn coerce_to_string_lambda_fails() {
5602        let root = rnix::Root::parse("x: x");
5603        let expr = root.tree().expr().unwrap();
5604        let closure = Closure {
5605            param: match expr {
5606                rnix::ast::Expr::Lambda(ref l) => l.param().unwrap(),
5607                _ => panic!("expected lambda"),
5608            },
5609            body: match expr {
5610                rnix::ast::Expr::Lambda(ref l) => l.body().unwrap(),
5611                _ => panic!("expected lambda"),
5612            },
5613            env: Env::new(),
5614        };
5615        let val = Value::Lambda(Rc::new(closure));
5616        assert!(val.coerce_to_string().is_err());
5617    }
5618
5619    // ── BuiltinFn debug ──────────────────────────────────
5620
5621    #[test]
5622    fn builtin_fn_debug_includes_name() {
5623        let b = BuiltinFn {
5624            name: "myFunc",
5625            func: Rc::new(|_| Ok(Value::Null)),
5626        };
5627        let s = format!("{b:?}");
5628        assert!(s.contains("myFunc"));
5629        assert!(s.contains("builtin"));
5630    }
5631
5632    // ── Thunk additional tests ───────────────────────────
5633
5634    #[test]
5635    fn thunk_force_chains_through_inner_thunks() {
5636        // Build a thunk whose evaluator yields another thunk.
5637        let inner_root = rnix::Root::parse("99");
5638        let inner_expr = inner_root.tree().expr().unwrap();
5639        let inner_thunk = Thunk::new_suspended(inner_expr, Env::new());
5640        let outer = Thunk::new_evaluated(Value::Thunk(inner_thunk));
5641        let result = outer.force(&|e, env| crate::eval::eval_expr(e, env));
5642        // Already-evaluated outer returns the inner thunk; the chain is
5643        // collapsed by the higher-level force_value, not by force() itself
5644        // when starting from Evaluated. So we just check we got a Thunk
5645        // back unchanged.
5646        match result.unwrap() {
5647            Value::Thunk(_) | Value::Int(99) => {}
5648            other => panic!("unexpected: {other:?}"),
5649        }
5650    }
5651
5652    #[test]
5653    fn thunk_inherit_select_debug_format() {
5654        let root = rnix::Root::parse("{ x = 1; }");
5655        let expr = root.tree().expr().unwrap();
5656        let source = Thunk::new_suspended(expr, Env::new());
5657        let thunk = Thunk::new_inherit_select(source, "x");
5658        let s = format!("{thunk:?}");
5659        assert!(s.contains("inherit-select"));
5660        assert!(s.contains("x"));
5661    }
5662
5663    #[test]
5664    fn thunk_blackhole_debug_format() {
5665        let root = rnix::Root::parse("1");
5666        let expr = root.tree().expr().unwrap();
5667        let thunk = Thunk::new_suspended(expr, Env::new());
5668        // SAFETY: Test-only, single-threaded.
5669        *unsafe { &mut *thunk.0.repr.get() } = ThunkRepr::Blackhole;
5670        assert_eq!(format!("{thunk:?}"), "<blackhole>");
5671    }
5672
5673    // ── Value display for thunks ─────────────────────────
5674
5675    #[test]
5676    fn value_display_thunk_evaluates() {
5677        let root = rnix::Root::parse("42");
5678        let expr = root.tree().expr().unwrap();
5679        let thunk = Thunk::new_suspended(expr, Env::new());
5680        let val = Value::Thunk(thunk);
5681        assert_eq!(format!("{val}"), "42");
5682    }
5683
5684    #[test]
5685    fn value_to_json_thunk_forces() {
5686        let root = rnix::Root::parse(r#""world""#);
5687        let expr = root.tree().expr().unwrap();
5688        let thunk = Thunk::new_suspended(expr, Env::new());
5689        let val = Value::Thunk(thunk);
5690        assert_eq!(val.to_json(), serde_json::Value::String("world".into()));
5691    }
5692
5693    #[test]
5694    fn value_type_name_thunk_forces() {
5695        let root = rnix::Root::parse("42");
5696        let expr = root.tree().expr().unwrap();
5697        let thunk = Thunk::new_suspended(expr, Env::new());
5698        let val = Value::Thunk(thunk);
5699        assert_eq!(val.type_name(), "int");
5700    }
5701
5702    // ── as_string / as_nix_string thunk error ────────────
5703
5704    #[test]
5705    fn as_string_errors_on_thunk() {
5706        let root = rnix::Root::parse(r#""x""#);
5707        let expr = root.tree().expr().unwrap();
5708        let thunk = Thunk::new_suspended(expr, Env::new());
5709        let val = Value::Thunk(thunk);
5710        let err = val.as_string().unwrap_err();
5711        match err {
5712            EvalError::TypeError(msg) => assert!(msg.contains("thunk")),
5713            _ => panic!("expected TypeError"),
5714        }
5715    }
5716
5717    #[test]
5718    fn as_nix_string_errors_on_thunk() {
5719        let root = rnix::Root::parse(r#""x""#);
5720        let expr = root.tree().expr().unwrap();
5721        let thunk = Thunk::new_suspended(expr, Env::new());
5722        let val = Value::Thunk(thunk);
5723        assert!(val.as_nix_string().is_err());
5724    }
5725
5726    #[test]
5727    fn as_attrs_errors_on_thunk() {
5728        let root = rnix::Root::parse("{}");
5729        let expr = root.tree().expr().unwrap();
5730        let thunk = Thunk::new_suspended(expr, Env::new());
5731        let val = Value::Thunk(thunk);
5732        assert!(val.as_attrs().is_err());
5733    }
5734
5735    #[test]
5736    fn as_list_errors_on_thunk() {
5737        let root = rnix::Root::parse("[]");
5738        let expr = root.tree().expr().unwrap();
5739        let thunk = Thunk::new_suspended(expr, Env::new());
5740        let val = Value::Thunk(thunk);
5741        assert!(val.as_list().is_err());
5742    }
5743
5744    // ── as_nix_string OK on string ───────────────────────
5745
5746    #[test]
5747    fn as_nix_string_ok_on_string() {
5748        let v = Value::string("hi");
5749        let ns = v.as_nix_string().unwrap();
5750        assert_eq!(ns.as_str(), "hi");
5751    }
5752
5753    #[test]
5754    fn as_nix_string_errors_on_int() {
5755        let v = Value::Int(1);
5756        match v.as_nix_string() {
5757            Err(EvalError::TypeMismatch { expected, got }) => {
5758                assert_eq!(expected, "string");
5759                assert_eq!(got, "int");
5760            }
5761            _ => panic!("expected TypeMismatch"),
5762        }
5763    }
5764
5765    // ════════════════════════════════════════════════════════════
5766    // 1. OnceCell Thunk Cache
5767    // ════════════════════════════════════════════════════════════
5768
5769    #[test]
5770    fn oncecell_cache_populated_after_force() {
5771        let root = rnix::Root::parse("42");
5772        let expr = root.tree().expr().unwrap();
5773        let thunk = Thunk::new_suspended(expr, Env::new());
5774        // Before forcing, cache should be empty.
5775        assert!(thunk.0.cache.get().is_none());
5776        let _ = thunk.force(&|e, env| crate::eval::eval_expr(e, env)).unwrap();
5777        // After forcing, cache should be populated.
5778        assert!(thunk.0.cache.get().is_some());
5779    }
5780
5781    #[test]
5782    fn oncecell_cache_matches_force_result() {
5783        let root = rnix::Root::parse("1 + 2");
5784        let expr = root.tree().expr().unwrap();
5785        let thunk = Thunk::new_suspended(expr, Env::new());
5786        let forced = thunk.force(&|e, env| crate::eval::eval_expr(e, env)).unwrap();
5787        let cached = thunk.0.cache.get().unwrap();
5788        // Cache stores Concrete (thunk-free); force returns Value.
5789        // Compare via Concrete→Value promotion.
5790        assert_eq!((**cached).clone().into_value(), forced);
5791    }
5792
5793    #[test]
5794    fn oncecell_new_evaluated_prepopulates_cache() {
5795        let thunk = Thunk::new_evaluated(Value::Int(77));
5796        // Cache should be set immediately.
5797        let cached = thunk.0.cache.get().expect("cache should be pre-populated");
5798        assert_eq!(**cached, Concrete::Int(77));
5799    }
5800
5801    #[test]
5802    fn oncecell_is_evaluated_uses_cache() {
5803        let thunk = Thunk::new_evaluated(Value::Bool(false));
5804        // is_evaluated() checks the OnceCell cache.
5805        assert!(thunk.is_evaluated());
5806        assert!(thunk.0.cache.get().is_some());
5807    }
5808
5809    #[test]
5810    fn oncecell_already_evaluated_returns_cached_without_repr() {
5811        // Create a thunk already evaluated. Force should return
5812        // the cached value without touching repr (the evaluator
5813        // closure should never be called).
5814        let thunk = Thunk::new_evaluated(Value::Int(55));
5815        let result = thunk.force(&|_, _| panic!("evaluator should not be called"));
5816        assert_eq!(result.unwrap(), Value::Int(55));
5817    }
5818
5819    // ════════════════════════════════════════════════════════════
5820    // 2. WithScope Memoization
5821    // ════════════════════════════════════════════════════════════
5822
5823    #[test]
5824    fn with_scope_created_with_empty_cache() {
5825        // Thunk-valued scopes start with empty cache (thunk not yet forced)
5826        let thunk = Thunk::new_suspended(
5827            rnix::Root::parse("{}").tree().expr().unwrap(),
5828            Env::new(),
5829        );
5830        let env = Env::new().with_scope(Value::Thunk(thunk));
5831        let scope = &env.0.with_scopes[0];
5832        assert!(scope.cached.borrow().is_none());
5833    }
5834
5835    #[test]
5836    fn with_scope_concrete_pre_populates_cache() {
5837        // Concrete attrset scopes pre-populate cache immediately
5838        let mut attrs = NixAttrs::new();
5839        attrs.insert("x".to_string(), Value::Int(1));
5840        let env = Env::new().with_scope(Value::Attrs(Rc::new(attrs)));
5841        let scope = &env.0.with_scopes[0];
5842        assert!(scope.cached.borrow().is_some());
5843    }
5844
5845    #[test]
5846    fn with_scope_first_lookup_populates_cache() {
5847        let mut attrs = NixAttrs::new();
5848        attrs.insert("x".to_string(), Value::Int(42));
5849        let env = Env::new().with_scope(Value::Attrs(Rc::new(attrs)));
5850        // Cache is pre-populated for concrete attrsets.
5851        assert!(env.0.with_scopes[0].cached.borrow().is_some());
5852        // Lookup hits the pre-populated cache.
5853        let _ = env.lookup("x");
5854        assert!(env.0.with_scopes[0].cached.borrow().is_some());
5855    }
5856
5857    #[test]
5858    fn with_scope_second_lookup_uses_cache() {
5859        let mut attrs = NixAttrs::new();
5860        attrs.insert("x".to_string(), Value::Int(10));
5861        let env = Env::new().with_scope(Value::Attrs(Rc::new(attrs)));
5862        // First lookup populates cache.
5863        assert_eq!(env.lookup("x"), Some(Value::Int(10)));
5864        assert!(env.0.with_scopes[0].cached.borrow().is_some());
5865        // Second lookup should still work (reads from cache).
5866        assert_eq!(env.lookup("x"), Some(Value::Int(10)));
5867    }
5868
5869    #[test]
5870    fn with_scope_child_shares_cache_via_rc() {
5871        let mut attrs = NixAttrs::new();
5872        attrs.insert("shared".to_string(), Value::Int(7));
5873        let parent = Env::new().with_scope(Value::Attrs(Rc::new(attrs)));
5874        let child = parent.child();
5875        // Force via parent lookup.
5876        let _ = parent.lookup("shared");
5877        // Child's with-scope cache should share the same Rc, so
5878        // it should also show cached.
5879        assert!(child.0.with_scopes[0].cached.borrow().is_some());
5880    }
5881
5882    #[test]
5883    fn with_scope_innermost_checked_first() {
5884        let mut outer = NixAttrs::new();
5885        outer.insert("x".to_string(), Value::Int(1));
5886        outer.insert("y".to_string(), Value::Int(100));
5887        let mut inner = NixAttrs::new();
5888        inner.insert("x".to_string(), Value::Int(2));
5889        let env = Env::new()
5890            .with_scope(Value::Attrs(Rc::new(outer)))
5891            .with_scope(Value::Attrs(Rc::new(inner)));
5892        // Innermost scope has x=2, should win.
5893        assert_eq!(env.lookup("x"), Some(Value::Int(2)));
5894        // y only in outer, should fallback.
5895        assert_eq!(env.lookup("y"), Some(Value::Int(100)));
5896    }
5897
5898    // ════════════════════════════════════════════════════════════
5899    // 3. FxHashMap for NixAttrs
5900    // ════════════════════════════════════════════════════════════
5901
5902    #[test]
5903    fn fxhashmap_nixattrs_new_creates_empty() {
5904        let a = NixAttrs::new();
5905        assert!(a.is_empty());
5906        assert_eq!(a.len(), 0);
5907        // Internal map is a FxHashMap (im_rc::HashMap with FxBuildHasher).
5908        assert!(a.inner().is_empty());
5909    }
5910
5911    #[test]
5912    fn fxhashmap_insert_get_roundtrip_with_symbol_keys() {
5913        let mut a = NixAttrs::new();
5914        a.insert("mykey".to_string(), Value::Int(42));
5915        assert_eq!(a.get("mykey"), Some(&Value::Int(42)));
5916    }
5917
5918    #[test]
5919    fn fxhashmap_contains_key_with_interned_keys() {
5920        let mut a = NixAttrs::new();
5921        a.insert("alpha".to_string(), Value::Int(1));
5922        let sym = intern("alpha");
5923        assert!(a.inner().contains_key(&sym));
5924        let missing_sym = intern("beta");
5925        assert!(!a.inner().contains_key(&missing_sym));
5926    }
5927
5928    #[test]
5929    fn fxhashmap_remove_returns_value() {
5930        let mut a = NixAttrs::new();
5931        a.insert("key".to_string(), Value::Int(99));
5932        let removed = a.remove("key");
5933        assert_eq!(removed, Some(Value::Int(99)));
5934        assert!(a.is_empty());
5935    }
5936
5937    #[test]
5938    fn fxhashmap_keys_returns_sorted_strings() {
5939        let mut a = NixAttrs::new();
5940        a.insert("zulu".to_string(), Value::Int(1));
5941        a.insert("alpha".to_string(), Value::Int(2));
5942        a.insert("mike".to_string(), Value::Int(3));
5943        let keys: Vec<String> = a.keys().collect();
5944        assert_eq!(keys, vec!["alpha", "mike", "zulu"]);
5945    }
5946
5947    #[test]
5948    fn fxhashmap_iter_returns_sorted_string_value_pairs() {
5949        let mut a = NixAttrs::new();
5950        a.insert("b".to_string(), Value::Int(2));
5951        a.insert("a".to_string(), Value::Int(1));
5952        let pairs: Vec<(String, &Value)> = a.iter().collect();
5953        assert_eq!(pairs.len(), 2);
5954        assert_eq!(pairs[0].0, "a");
5955        assert_eq!(*pairs[0].1, Value::Int(1));
5956        assert_eq!(pairs[1].0, "b");
5957        assert_eq!(*pairs[1].1, Value::Int(2));
5958    }
5959
5960    #[test]
5961    fn fxhashmap_update_merges_correctly() {
5962        let mut left = NixAttrs::new();
5963        left.insert("a".to_string(), Value::Int(1));
5964        left.insert("b".to_string(), Value::Int(2));
5965        let mut right = NixAttrs::new();
5966        right.insert("b".to_string(), Value::Int(20));
5967        right.insert("c".to_string(), Value::Int(3));
5968        let merged = left.update(&right);
5969        assert_eq!(merged.get("a"), Some(&Value::Int(1)));
5970        assert_eq!(merged.get("b"), Some(&Value::Int(20))); // right overrides
5971        assert_eq!(merged.get("c"), Some(&Value::Int(3)));
5972        assert_eq!(merged.len(), 3);
5973    }
5974
5975    #[test]
5976    fn fxhashmap_from_iterator_collects_with_interning() {
5977        let pairs = vec![
5978            ("x".to_string(), Value::Int(10)),
5979            ("y".to_string(), Value::Int(20)),
5980            ("z".to_string(), Value::Int(30)),
5981        ];
5982        let attrs: NixAttrs = pairs.into_iter().collect();
5983        assert_eq!(attrs.len(), 3);
5984        assert_eq!(attrs.get("x"), Some(&Value::Int(10)));
5985        assert_eq!(attrs.get("y"), Some(&Value::Int(20)));
5986        assert_eq!(attrs.get("z"), Some(&Value::Int(30)));
5987        // Verify internal storage uses Symbol keys.
5988        let sym_x = intern("x");
5989        assert!(attrs.inner().contains_key(&sym_x));
5990    }
5991
5992    // ════════════════════════════════════════════════════════════
5993    // 4. SmallVec StringContext
5994    // ════════════════════════════════════════════════════════════
5995
5996    #[test]
5997    fn smallvec_context_empty() {
5998        let ctx = StringContext::new();
5999        assert!(ctx.is_empty());
6000        assert_eq!(ctx.len(), 0);
6001        assert_eq!(ctx.elements().len(), 0);
6002    }
6003
6004    #[test]
6005    fn smallvec_context_single_element_inline() {
6006        let mut ctx = StringContext::new();
6007        ctx.add_plain("/nix/store/single");
6008        assert_eq!(ctx.len(), 1);
6009        // SmallVec<[ContextElement; 2]> stores up to 2 inline.
6010        assert!(!ctx.is_empty());
6011    }
6012
6013    #[test]
6014    fn smallvec_context_two_elements_still_inline() {
6015        let mut ctx = StringContext::new();
6016        ctx.add_plain("/nix/store/one");
6017        ctx.add_output("/nix/store/two.drv", "out");
6018        assert_eq!(ctx.len(), 2);
6019    }
6020
6021    #[test]
6022    fn smallvec_context_three_plus_spills_to_heap() {
6023        let mut ctx = StringContext::new();
6024        ctx.add_plain("/nix/store/a");
6025        ctx.add_plain("/nix/store/b");
6026        ctx.add_drv_deep("/nix/store/c.drv");
6027        assert_eq!(ctx.len(), 3);
6028        // Verify all elements are accessible.
6029        assert!(ctx.elements().contains(&ContextElement::Plain(SmolStr::from("/nix/store/a"))));
6030        assert!(ctx.elements().contains(&ContextElement::Plain(SmolStr::from("/nix/store/b"))));
6031        assert!(ctx.elements().contains(&ContextElement::DrvDeep(SmolStr::from("/nix/store/c.drv"))));
6032    }
6033
6034    #[test]
6035    fn smallvec_context_merge_deduplicates() {
6036        let mut ctx1 = StringContext::new();
6037        ctx1.add_plain("/nix/store/dup");
6038        ctx1.add_output("/nix/store/x.drv", "out");
6039        let mut ctx2 = StringContext::new();
6040        ctx2.add_plain("/nix/store/dup");      // duplicate
6041        ctx2.add_plain("/nix/store/unique");    // new
6042        ctx1.merge(&ctx2);
6043        assert_eq!(ctx1.len(), 3); // dup not duplicated
6044    }
6045
6046    #[test]
6047    fn smallvec_context_add_plain_output_drv_deep() {
6048        let mut ctx = StringContext::new();
6049        ctx.add_plain("/nix/store/plain");
6050        assert_eq!(ctx.len(), 1);
6051        assert!(ctx.elements().contains(&ContextElement::Plain(SmolStr::from("/nix/store/plain"))));
6052
6053        ctx.add_output("/nix/store/out.drv", "lib");
6054        assert_eq!(ctx.len(), 2);
6055        assert!(ctx.elements().contains(&ContextElement::Output {
6056            drv: SmolStr::from("/nix/store/out.drv"),
6057            output: SmolStr::from("lib"),
6058        }));
6059
6060        ctx.add_drv_deep("/nix/store/deep.drv");
6061        assert_eq!(ctx.len(), 3);
6062        assert!(ctx.elements().contains(&ContextElement::DrvDeep(SmolStr::from("/nix/store/deep.drv"))));
6063    }
6064
6065    // ════════════════════════════════════════════════════════════
6066    // 5. Rc<Vec<Value>> for List
6067    // ════════════════════════════════════════════════════════════
6068
6069    #[test]
6070    fn rc_list_constructor_wraps_in_rc() {
6071        let v = Value::list(vec![Value::Int(1), Value::Int(2)]);
6072        match &v {
6073            Value::List(rc) => {
6074                assert_eq!(rc.len(), 2);
6075                assert_eq!(Rc::strong_count(rc), 1);
6076            }
6077            _ => panic!("expected List"),
6078        }
6079    }
6080
6081    #[test]
6082    fn rc_list_clone_is_refcount_bump() {
6083        let v = Value::list(vec![Value::Int(10)]);
6084        let rc1 = match &v {
6085            Value::List(rc) => rc.clone(),
6086            _ => panic!("expected List"),
6087        };
6088        let v2 = v.clone();
6089        let rc2 = match &v2 {
6090            Value::List(rc) => rc.clone(),
6091            _ => panic!("expected List"),
6092        };
6093        // Both point to the same allocation.
6094        assert!(Rc::ptr_eq(&rc1, &rc2));
6095        // Strong count should be 3: rc1, rc2, and the one inside v or v2.
6096        // Actually: v has one, v2 has one, rc1 has one, rc2 has one = 4.
6097        assert!(Rc::strong_count(&rc1) >= 2);
6098    }
6099
6100    #[test]
6101    fn rc_list_as_list_returns_slice() {
6102        let v = Value::list(vec![Value::Int(1), Value::Int(2), Value::Int(3)]);
6103        let slice = v.as_list().unwrap();
6104        assert_eq!(slice.len(), 3);
6105        assert_eq!(slice[0], Value::Int(1));
6106        assert_eq!(slice[1], Value::Int(2));
6107        assert_eq!(slice[2], Value::Int(3));
6108    }
6109
6110    #[test]
6111    fn rc_list_from_vec_wraps_in_rc() {
6112        let items = vec![Value::Bool(true), Value::Bool(false)];
6113        let v: Value = items.into();
6114        match &v {
6115            Value::List(rc) => {
6116                assert_eq!(rc.len(), 2);
6117                assert_eq!(Rc::strong_count(rc), 1);
6118            }
6119            _ => panic!("expected List"),
6120        }
6121    }
6122
6123    // ════════════════════════════════════════════════════════════
6124    // 6. String Interning
6125    // ════════════════════════════════════════════════════════════
6126
6127    #[test]
6128    fn intern_same_string_returns_same_symbol() {
6129        let s1 = intern("hello_intern_test");
6130        let s2 = intern("hello_intern_test");
6131        assert_eq!(s1, s2);
6132    }
6133
6134    #[test]
6135    fn intern_different_strings_returns_different_symbols() {
6136        let s1 = intern("unique_str_a_9182");
6137        let s2 = intern("unique_str_b_9182");
6138        assert_ne!(s1, s2);
6139    }
6140
6141    #[test]
6142    fn resolve_roundtrips_correctly() {
6143        let sym = intern("roundtrip_test_str");
6144        let resolved = resolve(sym);
6145        assert_eq!(resolved, "roundtrip_test_str");
6146    }
6147
6148    #[test]
6149    fn intern_cached_same_offset_returns_cached_symbol() {
6150        let sid = next_source_id();
6151        let sym1 = intern_cached("cached_ident_aa", sid, 100);
6152        let sym2 = intern_cached("cached_ident_aa", sid, 100);
6153        assert_eq!(sym1, sym2);
6154    }
6155
6156    #[test]
6157    fn intern_cached_different_offset_same_string_returns_same_symbol() {
6158        // Even with different offsets, the same string should intern
6159        // to the same Symbol (interning dedup at the interner level).
6160        let sid = next_source_id();
6161        let sym1 = intern_cached("dedup_test_str_77", sid, 200);
6162        let sym2 = intern_cached("dedup_test_str_77", sid, 300);
6163        // The symbols should be equal because the interner deduplicates.
6164        assert_eq!(sym1, sym2);
6165    }
6166
6167    #[test]
6168    fn clear_ident_cache_clears() {
6169        let sid = next_source_id();
6170        let _sym = intern_cached("to_be_cleared_99", sid, 500);
6171        clear_ident_cache();
6172        // After clearing, the cache is empty, but interning the same
6173        // string again should still return the same Symbol (the interner
6174        // itself is not cleared, just the offset cache).
6175        let sym2 = intern_cached("to_be_cleared_99", sid, 500);
6176        let resolved = resolve(sym2);
6177        assert_eq!(resolved, "to_be_cleared_99");
6178    }
6179
6180    #[test]
6181    fn next_source_id_increments_monotonically() {
6182        let id1 = next_source_id();
6183        let id2 = next_source_id();
6184        let id3 = next_source_id();
6185        assert_eq!(id2, id1 + 1);
6186        assert_eq!(id3, id2 + 1);
6187    }
6188
6189    // ════════════════════════════════════════════════════════════
6190    // 7. Env Operations
6191    // ════════════════════════════════════════════════════════════
6192
6193    #[test]
6194    fn env_new_creates_empty_bindings() {
6195        let env = Env::new();
6196        assert!(env.0.bindings.is_empty());
6197        assert!(env.0.with_scopes.is_empty());
6198        assert!(env.eval_file().is_none());
6199    }
6200
6201    #[test]
6202    fn env_bind_lookup_roundtrip() {
6203        let mut env = Env::new();
6204        env.bind("foo".to_string(), Value::Int(42));
6205        assert_eq!(env.lookup("foo"), Some(Value::Int(42)));
6206        assert_eq!(env.lookup("bar"), None);
6207    }
6208
6209    #[test]
6210    fn env_child_inherits_parent_bindings_flattened() {
6211        let mut parent = Env::new();
6212        parent.bind("a".to_string(), Value::Int(1));
6213        parent.bind("b".to_string(), Value::Int(2));
6214        let child = parent.child();
6215        // Child sees parent's bindings.
6216        assert_eq!(child.lookup("a"), Some(Value::Int(1)));
6217        assert_eq!(child.lookup("b"), Some(Value::Int(2)));
6218        // Verify bindings are in child's own map (flattened).
6219        let sym_a = intern("a");
6220        assert!(child.0.bindings.contains_key(&sym_a));
6221    }
6222
6223    #[test]
6224    fn env_child_inherits_with_scopes() {
6225        let mut attrs = NixAttrs::new();
6226        attrs.insert("ws".to_string(), Value::Int(10));
6227        let parent = Env::new().with_scope(Value::Attrs(Rc::new(attrs)));
6228        let child = parent.child();
6229        // Child should have the same with_scopes as parent.
6230        assert_eq!(child.0.with_scopes.len(), parent.0.with_scopes.len());
6231        assert_eq!(child.lookup("ws"), Some(Value::Int(10)));
6232    }
6233
6234    #[test]
6235    fn env_lookup_sym_fast_path_matches_lookup() {
6236        let mut env = Env::new();
6237        env.bind("target".to_string(), Value::Int(88));
6238        let sym = intern("target");
6239        let via_lookup = env.lookup("target");
6240        let via_sym = env.lookup_sym(sym);
6241        assert_eq!(via_lookup, via_sym);
6242        assert_eq!(via_sym, Some(Value::Int(88)));
6243    }
6244
6245    #[test]
6246    fn env_lookup_sym_with_scope_fallback() {
6247        let mut attrs = NixAttrs::new();
6248        attrs.insert("sym_ws".to_string(), Value::Int(33));
6249        let env = Env::new().with_scope(Value::Attrs(Rc::new(attrs)));
6250        let sym = intern("sym_ws");
6251        assert_eq!(env.lookup_sym(sym), Some(Value::Int(33)));
6252    }
6253
6254    #[test]
6255    fn env_with_scope_ordering_multiple_innermost_wins() {
6256        let mut a1 = NixAttrs::new();
6257        a1.insert("x".to_string(), Value::Int(1));
6258        let mut a2 = NixAttrs::new();
6259        a2.insert("x".to_string(), Value::Int(2));
6260        let mut a3 = NixAttrs::new();
6261        a3.insert("x".to_string(), Value::Int(3));
6262        let env = Env::new()
6263            .with_scope(Value::Attrs(Rc::new(a1)))
6264            .with_scope(Value::Attrs(Rc::new(a2)))
6265            .with_scope(Value::Attrs(Rc::new(a3)));
6266        // Innermost (a3) should win.
6267        assert_eq!(env.lookup("x"), Some(Value::Int(3)));
6268    }
6269
6270    #[test]
6271    fn env_lookup_sym_not_found_returns_none() {
6272        let env = Env::new();
6273        let sym = intern("nonexistent_sym_99");
6274        assert_eq!(env.lookup_sym(sym), None);
6275    }
6276
6277    #[test]
6278    fn env_lookup_sym_lexical_wins_over_with_scope() {
6279        let mut attrs = NixAttrs::new();
6280        attrs.insert("priority".to_string(), Value::Int(1));
6281        let mut env = Env::new().with_scope(Value::Attrs(Rc::new(attrs)));
6282        env.bind("priority".to_string(), Value::Int(99));
6283        let sym = intern("priority");
6284        assert_eq!(env.lookup_sym(sym), Some(Value::Int(99)));
6285    }
6286}