From 3b7a14f62b2915790dc155f58ab9404acce3df6b Mon Sep 17 00:00:00 2001
From: Robert Hensing <robert@roberthensing.nl>
Date: Wed, 2 Apr 2025 13:30:57 +0200
Subject: [PATCH] feat: EvalState.require_bool

(cherry picked from commit e0d451f22cedae383fa2f18809f3494e674371b4)
---
 rust/nix-expr/src/eval_state.rs | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/rust/nix-expr/src/eval_state.rs b/rust/nix-expr/src/eval_state.rs
index af1718c..af24045 100644
--- a/rust/nix-expr/src/eval_state.rs
+++ b/rust/nix-expr/src/eval_state.rs
@@ -255,6 +255,14 @@ impl EvalState {
         unsafe { check_call!(raw::get_int(&mut self.context, v.raw_ptr())) }
     }
 
+    pub fn require_bool(&mut self, v: &Value) -> Result<bool> {
+        let t = self.value_type(v)?;
+        if t != ValueType::Bool {
+            bail!("expected a bool, but got a {:?}", t);
+        }
+        unsafe { check_call!(raw::get_bool(&mut self.context, v.raw_ptr())) }
+    }
+
     /// Evaluate, and require that the value is an attrset.
     /// Returns a list of the keys in the attrset.
     ///
@@ -813,6 +821,13 @@ mod tests {
             es.force(&v).unwrap();
             let t = es.value_type_unforced(&v);
             assert!(t == Some(ValueType::Bool));
+            let b = es.require_bool(&v).unwrap();
+            assert!(b);
+
+            let v = es.eval_from_string("false", "<test>").unwrap();
+            es.require_bool(&v).unwrap();
+            let b = es.require_bool(&v).unwrap();
+            assert!(!b);
         })
         .unwrap();
     }
@@ -848,6 +863,22 @@ mod tests {
         .unwrap();
     }
 
+    #[test]
+    fn eval_state_require_bool_forces_thunk() {
+        gc_registering_current_thread(|| {
+            let store = Store::open(None, HashMap::new()).unwrap();
+            let mut es = EvalState::new(store, []).unwrap();
+            let f = es.eval_from_string("x: !x", "<test>").unwrap();
+            let a = es.eval_from_string("true", "<test>").unwrap();
+            let v = es.new_value_apply(&f, &a).unwrap();
+            let t = es.value_type_unforced(&v);
+            assert!(t == None);
+            let i = es.require_bool(&v).unwrap();
+            assert!(i == false);
+        })
+        .unwrap();
+    }
+
     /// A helper that turns an expression into a thunk.
     fn make_thunk(es: &mut EvalState, expr: &str) -> Value {
         // This would be silly in real code, but it works for the current Nix implementation.