about summary refs log tree commit diff
diff options
context:
space:
mode:
authorAdam Joseph <adam@westernsemico.com>2023-12-13T04·47-0800
committertazjin <tazjin@tvl.su>2023-12-29T21·34+0000
commit11e35a77a671b3d85ea161e1799ef3f6d98201f1 (patch)
treeeac3d46a2c164445914430f9adbaf2996fe963b7
parent0c22454bb93648155b8be4b45fb5d5c8f0389673 (diff)
refactor(tvix/eval): let OpCoerceToString select the CoercionKind r/7272
Change-Id: I92d58ef216d7e0766af70f019b3dcd445284a95d
Reviewed-on: https://cl.tvl.fyi/c/depot/+/10344
Reviewed-by: tazjin <tazjin@tvl.su>
Tested-by: BuildkiteCI
-rw-r--r--tvix/eval/src/compiler/mod.rs9
-rw-r--r--tvix/eval/src/opcode.rs6
-rw-r--r--tvix/eval/src/value/mod.rs2
-rw-r--r--tvix/eval/src/vm/mod.rs11
4 files changed, 13 insertions, 15 deletions
diff --git a/tvix/eval/src/compiler/mod.rs b/tvix/eval/src/compiler/mod.rs
index f54dc8ba42ae..5e0c1899c0e5 100644
--- a/tvix/eval/src/compiler/mod.rs
+++ b/tvix/eval/src/compiler/mod.rs
@@ -34,6 +34,7 @@ use crate::spans::LightSpan;
 use crate::spans::ToSpan;
 use crate::value::{Closure, Formals, Lambda, NixAttrs, Thunk, Value};
 use crate::warnings::{EvalWarning, WarningKind};
+use crate::CoercionKind;
 use crate::SourceCode;
 
 use self::scope::{LocalIdx, LocalPosition, Scope, Upvalue, UpvalueKind};
@@ -444,7 +445,13 @@ impl Compiler<'_> {
                 ast::InterpolPart::Interpolation(ipol) => {
                     self.compile(slot, ipol.expr().unwrap());
                     // implicitly forces as well
-                    self.push_op(OpCode::OpCoerceToString, ipol);
+                    self.push_op(
+                        OpCode::OpCoerceToString(CoercionKind {
+                            strong: false,
+                            import_paths: true,
+                        }),
+                        ipol,
+                    );
                 }
 
                 ast::InterpolPart::Literal(lit) => {
diff --git a/tvix/eval/src/opcode.rs b/tvix/eval/src/opcode.rs
index b57a79a3baec..f89c1c12e7fd 100644
--- a/tvix/eval/src/opcode.rs
+++ b/tvix/eval/src/opcode.rs
@@ -201,10 +201,8 @@ pub enum OpCode {
     /// Interpolate the given number of string fragments into a single string.
     OpInterpolate(Count),
 
-    /// Force the Value on the stack and coerce it to a string, always using
-    /// `CoercionKind::Weak { import_paths: true }`. This is the behavior
-    /// necessary for path interpolation.
-    OpCoerceToString,
+    /// Force the Value on the stack and coerce it to a string
+    OpCoerceToString(crate::CoercionKind),
 
     // Paths
     /// Attempt to resolve the Value on the stack using the configured [`NixSearchPath`][]
diff --git a/tvix/eval/src/value/mod.rs b/tvix/eval/src/value/mod.rs
index 567a2f3df21b..eb54f6ae5bd8 100644
--- a/tvix/eval/src/value/mod.rs
+++ b/tvix/eval/src/value/mod.rs
@@ -161,7 +161,7 @@ macro_rules! gen_is {
 }
 
 /// Describes what input types are allowed when coercing a `Value` to a string
-#[derive(Clone, Copy, PartialEq, Debug)]
+#[derive(Clone, Copy, PartialEq, Eq, Debug)]
 pub struct CoercionKind {
     /// If false only coerce already "stringly" types like strings and paths, but
     /// also coerce sets that have a `__toString` attribute. In Tvix, this is
diff --git a/tvix/eval/src/vm/mod.rs b/tvix/eval/src/vm/mod.rs
index c3c0b259bc77..6c11997db1f5 100644
--- a/tvix/eval/src/vm/mod.rs
+++ b/tvix/eval/src/vm/mod.rs
@@ -777,20 +777,13 @@ impl<'o> VM<'o> {
                     _ => panic!("attempted to finalise a non-thunk"),
                 },
 
-                OpCode::OpCoerceToString => {
+                OpCode::OpCoerceToString(kind) => {
                     let value = self.stack_pop();
                     let gen_span = frame.current_light_span();
                     self.push_call_frame(span, frame);
 
                     self.enqueue_generator("coerce_to_string", gen_span.clone(), |co| {
-                        value.coerce_to_string(
-                            co,
-                            CoercionKind {
-                                strong: false,
-                                import_paths: true,
-                            },
-                            gen_span,
-                        )
+                        value.coerce_to_string(co, kind, gen_span)
                     });
 
                     return Ok(false);