refactor(tvix/glue): move builtins into separate directory
Change-Id: I25b7197458dbfbde8623545dc0a0286eb2744f10 Reviewed-on: https://cl.tvl.fyi/c/depot/+/9911 Autosubmit: flokli <flokli@flokli.de> Tested-by: BuildkiteCI Reviewed-by: raitobezarius <tvl@lahfa.xyz>
This commit is contained in:
parent
a72a1044c2
commit
c8cc31e079
7 changed files with 177 additions and 166 deletions
|
@ -3,7 +3,7 @@ use std::rc::Rc;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use std::{fs, path::PathBuf};
|
use std::{fs, path::PathBuf};
|
||||||
use tvix_glue::known_paths::KnownPaths;
|
use tvix_glue::known_paths::KnownPaths;
|
||||||
use tvix_glue::{add_derivation_builtins, configure_nix_path};
|
use tvix_glue::{builtins::add_derivation_builtins, configure_nix_path};
|
||||||
|
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use rustyline::{error::ReadlineError, Editor};
|
use rustyline::{error::ReadlineError, Editor};
|
||||||
|
|
|
@ -1,4 +1,6 @@
|
||||||
//! Implements `builtins.derivation`, the core of what makes Nix build packages.
|
//! Implements `builtins.derivation`, the core of what makes Nix build packages.
|
||||||
|
use crate::builtins::DerivationError;
|
||||||
|
use crate::known_paths::{KnownPaths, PathKind, PathName};
|
||||||
use nix_compat::derivation::{Derivation, Output};
|
use nix_compat::derivation::{Derivation, Output};
|
||||||
use nix_compat::nixhash;
|
use nix_compat::nixhash;
|
||||||
use std::cell::RefCell;
|
use std::cell::RefCell;
|
||||||
|
@ -10,9 +12,6 @@ use tvix_eval::{
|
||||||
AddContext, CatchableErrorKind, CoercionKind, ErrorKind, NixAttrs, NixList, Value, WarningKind,
|
AddContext, CatchableErrorKind, CoercionKind, ErrorKind, NixAttrs, NixList, Value, WarningKind,
|
||||||
};
|
};
|
||||||
|
|
||||||
use crate::errors::Error;
|
|
||||||
use crate::known_paths::{KnownPaths, PathKind, PathName};
|
|
||||||
|
|
||||||
// Constants used for strangely named fields in derivation inputs.
|
// Constants used for strangely named fields in derivation inputs.
|
||||||
const STRUCTURED_ATTRS: &str = "__structuredAttrs";
|
const STRUCTURED_ATTRS: &str = "__structuredAttrs";
|
||||||
const IGNORE_NULLS: &str = "__ignoreNulls";
|
const IGNORE_NULLS: &str = "__ignoreNulls";
|
||||||
|
@ -39,7 +38,7 @@ async fn populate_outputs(
|
||||||
.insert(output_name.as_str().into(), Default::default())
|
.insert(output_name.as_str().into(), Default::default())
|
||||||
.is_some()
|
.is_some()
|
||||||
{
|
{
|
||||||
return Err(Error::DuplicateOutput(output_name.as_str().into()).into());
|
return Err(DerivationError::DuplicateOutput(output_name.as_str().into()).into());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -125,7 +124,7 @@ fn handle_fixed_output(
|
||||||
|
|
||||||
// construct a NixHash.
|
// construct a NixHash.
|
||||||
let nixhash = nixhash::from_str(&hash_str, hash_algo_str.as_deref())
|
let nixhash = nixhash::from_str(&hash_str, hash_algo_str.as_deref())
|
||||||
.map_err(Error::InvalidOutputHash)?;
|
.map_err(DerivationError::InvalidOutputHash)?;
|
||||||
|
|
||||||
// construct the fixed output.
|
// construct the fixed output.
|
||||||
drv.outputs.insert(
|
drv.outputs.insert(
|
||||||
|
@ -135,7 +134,9 @@ fn handle_fixed_output(
|
||||||
ca_hash: match hash_mode_str.as_deref() {
|
ca_hash: match hash_mode_str.as_deref() {
|
||||||
None | Some("flat") => Some(nixhash::CAHash::Flat(nixhash)),
|
None | Some("flat") => Some(nixhash::CAHash::Flat(nixhash)),
|
||||||
Some("recursive") => Some(nixhash::CAHash::Nar(nixhash)),
|
Some("recursive") => Some(nixhash::CAHash::Nar(nixhash)),
|
||||||
Some(other) => return Err(Error::InvalidOutputHashMode(other.to_string()))?,
|
Some(other) => {
|
||||||
|
return Err(DerivationError::InvalidOutputHashMode(other.to_string()))?
|
||||||
|
}
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
);
|
);
|
||||||
|
@ -205,7 +206,7 @@ async fn strong_coerce_to_string(
|
||||||
}
|
}
|
||||||
|
|
||||||
#[builtins(state = "Rc<RefCell<KnownPaths>>")]
|
#[builtins(state = "Rc<RefCell<KnownPaths>>")]
|
||||||
mod derivation_builtins {
|
pub(crate) mod derivation_builtins {
|
||||||
use super::*;
|
use super::*;
|
||||||
use nix_compat::store_path::hash_placeholder;
|
use nix_compat::store_path::hash_placeholder;
|
||||||
use tvix_eval::generators::Gen;
|
use tvix_eval::generators::Gen;
|
||||||
|
@ -306,7 +307,9 @@ mod derivation_builtins {
|
||||||
.insert(name.as_str().to_string(), val_str.into())
|
.insert(name.as_str().to_string(), val_str.into())
|
||||||
.is_some()
|
.is_some()
|
||||||
{
|
{
|
||||||
return Err(Error::DuplicateEnvVar(name.as_str().to_string()).into());
|
return Err(
|
||||||
|
DerivationError::DuplicateEnvVar(name.as_str().to_string()).into()
|
||||||
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -371,7 +374,8 @@ mod derivation_builtins {
|
||||||
|
|
||||||
// At this point, derivation fields are fully populated from
|
// At this point, derivation fields are fully populated from
|
||||||
// eval data structures.
|
// eval data structures.
|
||||||
drv.validate(false).map_err(Error::InvalidDerivation)?;
|
drv.validate(false)
|
||||||
|
.map_err(DerivationError::InvalidDerivation)?;
|
||||||
|
|
||||||
// Calculate the derivation_or_fod_hash for the current derivation.
|
// Calculate the derivation_or_fod_hash for the current derivation.
|
||||||
// This one is still intermediate (so not added to known_paths)
|
// This one is still intermediate (so not added to known_paths)
|
||||||
|
@ -380,11 +384,11 @@ mod derivation_builtins {
|
||||||
|
|
||||||
// Mutate the Derivation struct and set output paths
|
// Mutate the Derivation struct and set output paths
|
||||||
drv.calculate_output_paths(&name, &derivation_or_fod_hash_tmp)
|
drv.calculate_output_paths(&name, &derivation_or_fod_hash_tmp)
|
||||||
.map_err(Error::InvalidDerivation)?;
|
.map_err(DerivationError::InvalidDerivation)?;
|
||||||
|
|
||||||
let derivation_path = drv
|
let derivation_path = drv
|
||||||
.calculate_derivation_path(&name)
|
.calculate_derivation_path(&name)
|
||||||
.map_err(Error::InvalidDerivation)?;
|
.map_err(DerivationError::InvalidDerivation)?;
|
||||||
|
|
||||||
// recompute the hash derivation modulo and add to known_paths
|
// recompute the hash derivation modulo and add to known_paths
|
||||||
let derivation_or_fod_hash_final =
|
let derivation_or_fod_hash_final =
|
||||||
|
@ -453,7 +457,7 @@ mod derivation_builtins {
|
||||||
name.as_str().to_string(),
|
name.as_str().to_string(),
|
||||||
)
|
)
|
||||||
})
|
})
|
||||||
.map_err(Error::InvalidDerivation)?
|
.map_err(DerivationError::InvalidDerivation)?
|
||||||
.to_absolute_path();
|
.to_absolute_path();
|
||||||
|
|
||||||
state.borrow_mut().plain(&path);
|
state.borrow_mut().plain(&path);
|
||||||
|
@ -465,132 +469,3 @@ mod derivation_builtins {
|
||||||
}
|
}
|
||||||
|
|
||||||
pub use derivation_builtins::builtins as derivation_builtins;
|
pub use derivation_builtins::builtins as derivation_builtins;
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use crate::{add_derivation_builtins, known_paths::KnownPaths};
|
|
||||||
use nix_compat::store_path::hash_placeholder;
|
|
||||||
use std::{cell::RefCell, rc::Rc};
|
|
||||||
use test_case::test_case;
|
|
||||||
use tvix_eval::EvaluationResult;
|
|
||||||
|
|
||||||
/// evaluates a given nix expression and returns the result.
|
|
||||||
/// Takes care of setting up the evaluator so it knows about the
|
|
||||||
// `derivation` builtin.
|
|
||||||
fn eval(str: &str) -> EvaluationResult {
|
|
||||||
let mut eval = tvix_eval::Evaluation::new_impure(str, None);
|
|
||||||
|
|
||||||
let known_paths: Rc<RefCell<KnownPaths>> = Default::default();
|
|
||||||
|
|
||||||
add_derivation_builtins(&mut eval, known_paths.clone());
|
|
||||||
|
|
||||||
// run the evaluation itself.
|
|
||||||
eval.evaluate()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn derivation() {
|
|
||||||
let result = eval(
|
|
||||||
r#"(derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux";}).outPath"#,
|
|
||||||
);
|
|
||||||
|
|
||||||
assert!(result.errors.is_empty(), "expect evaluation to succeed");
|
|
||||||
let value = result.value.expect("must be some");
|
|
||||||
|
|
||||||
match value {
|
|
||||||
tvix_eval::Value::String(s) => {
|
|
||||||
assert_eq!(
|
|
||||||
"/nix/store/xpcvxsx5sw4rbq666blz6sxqlmsqphmr-foo",
|
|
||||||
s.as_str()
|
|
||||||
);
|
|
||||||
}
|
|
||||||
_ => panic!("unexpected value type: {:?}", value),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// a derivation with an empty name is an error.
|
|
||||||
#[test]
|
|
||||||
fn derivation_empty_name_fail() {
|
|
||||||
let result = eval(
|
|
||||||
r#"(derivation { name = ""; builder = "/bin/sh"; system = "x86_64-linux";}).outPath"#,
|
|
||||||
);
|
|
||||||
|
|
||||||
assert!(!result.errors.is_empty(), "expect evaluation to fail");
|
|
||||||
}
|
|
||||||
|
|
||||||
/// construct some calls to builtins.derivation and compare produced output
|
|
||||||
/// paths.
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/17wgs52s7kcamcyin4ja58njkf91ipq8-foo"; "r:sha256")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo2"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/gi0p8vd635vpk1nq029cz3aa3jkhar5k-foo2"; "r:sha256 other name")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha1"; outputHash = "sha1-VUCRC+16gU5lcrLYHlPSUyx0Y/Q="; }).outPath"#, "/nix/store/p5sammmhpa84ama7ymkbgwwzrilva24x-foo"; "r:sha1")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "md5"; outputHash = "md5-07BzhNET7exJ6qYjitX/AA=="; }).outPath"#, "/nix/store/gmmxgpy1jrzs86r5y05wy6wiy2m15xgi-foo"; "r:md5")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha512"; outputHash = "sha512-DPkYCnZKuoY6Z7bXLwkYvBMcZ3JkLLLc5aNPCnAvlHDdwr8SXBIZixmVwjPDS0r9NGxUojNMNQqUilG26LTmtg=="; }).outPath"#, "/nix/store/lfi2bfyyap88y45mfdwi4j99gkaxaj19-foo"; "r:sha512")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "4374173a8cbe88de152b609f96f46e958bcf65762017474eec5a05ec2bd61530"; }).outPath"#, "/nix/store/17wgs52s7kcamcyin4ja58njkf91ipq8-foo"; "r:sha256 base16")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "0c0msqmyq1asxi74f5r0frjwz2wmdvs9d7v05caxx25yihx1fx23"; }).outPath"#, "/nix/store/17wgs52s7kcamcyin4ja58njkf91ipq8-foo"; "r:sha256 nixbase32")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/17wgs52s7kcamcyin4ja58njkf91ipq8-foo"; "r:sha256 base64")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "sha256-fgIr3TyFGDAXP5+qoAaiMKDg/a1MlT6Fv/S/DaA24S8="; }).outPath"#, "/nix/store/xm1l9dx4zgycv9qdhcqqvji1z88z534b-foo"; "r:sha256 base64 nopad")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "flat"; outputHashAlgo = "sha256"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/q4pkwkxdib797fhk22p0k3g1q32jmxvf-foo"; "sha256")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo2"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "flat"; outputHashAlgo = "sha256"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/znw17xlmx9r6gw8izjkqxkl6s28sza4l-foo2"; "sha256 other name")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "flat"; outputHashAlgo = "sha1"; outputHash = "sha1-VUCRC+16gU5lcrLYHlPSUyx0Y/Q="; }).outPath"#, "/nix/store/zgpnjjmga53d8srp8chh3m9fn7nnbdv6-foo"; "sha1")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "flat"; outputHashAlgo = "md5"; outputHash = "md5-07BzhNET7exJ6qYjitX/AA=="; }).outPath"#, "/nix/store/jfhcwnq1852ccy9ad9nakybp2wadngnd-foo"; "md5")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "flat"; outputHashAlgo = "sha512"; outputHash = "sha512-DPkYCnZKuoY6Z7bXLwkYvBMcZ3JkLLLc5aNPCnAvlHDdwr8SXBIZixmVwjPDS0r9NGxUojNMNQqUilG26LTmtg=="; }).outPath"#, "/nix/store/as736rr116ian9qzg457f96j52ki8bm3-foo"; "sha512")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/17wgs52s7kcamcyin4ja58njkf91ipq8-foo"; "r:sha256 outputHashAlgo omitted")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/q4pkwkxdib797fhk22p0k3g1q32jmxvf-foo"; "r:sha256 outputHashAlgo and outputHashMode omitted")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; }).outPath"#, "/nix/store/xpcvxsx5sw4rbq666blz6sxqlmsqphmr-foo"; "outputHash* omitted")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; outputs = ["foo" "bar"]; system = "x86_64-linux"; }).outPath"#, "/nix/store/hkwdinvz2jpzgnjy9lv34d2zxvclj4s3-foo-foo"; "multiple outputs")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; args = ["--foo" "42" "--bar"]; system = "x86_64-linux"; }).outPath"#, "/nix/store/365gi78n2z7vwc1bvgb98k0a9cqfp6as-foo"; "args")]
|
|
||||||
#[test_case(r#"
|
|
||||||
let
|
|
||||||
bar = builtins.derivation {
|
|
||||||
name = "bar";
|
|
||||||
builder = ":";
|
|
||||||
system = ":";
|
|
||||||
outputHash = "08813cbee9903c62be4c5027726a418a300da4500b2d369d3af9286f4815ceba";
|
|
||||||
outputHashAlgo = "sha256";
|
|
||||||
outputHashMode = "recursive";
|
|
||||||
};
|
|
||||||
in
|
|
||||||
(builtins.derivation {
|
|
||||||
name = "foo";
|
|
||||||
builder = ":";
|
|
||||||
system = ":";
|
|
||||||
inherit bar;
|
|
||||||
}).outPath
|
|
||||||
"#, "/nix/store/5vyvcwah9l9kf07d52rcgdk70g2f4y13-foo"; "full")]
|
|
||||||
fn test_outpath(code: &str, expected_path: &str) {
|
|
||||||
let value = eval(code).value.expect("must succeed");
|
|
||||||
|
|
||||||
match value {
|
|
||||||
tvix_eval::Value::String(s) => {
|
|
||||||
assert_eq!(expected_path, s.as_str());
|
|
||||||
}
|
|
||||||
_ => panic!("unexpected value type: {:?}", value),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// construct some calls to builtins.derivation that should be rejected
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "sha256-00"; }).outPath"#; "invalid outputhash")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha1"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#; "sha1 and sha256")]
|
|
||||||
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; outputs = ["foo" "foo"]; system = "x86_64-linux"; }).outPath"#; "duplicate output names")]
|
|
||||||
fn test_outpath_invalid(code: &str) {
|
|
||||||
let resp = eval(code);
|
|
||||||
assert!(resp.value.is_none(), "Value should be None");
|
|
||||||
assert!(
|
|
||||||
!resp.errors.is_empty(),
|
|
||||||
"There should have been some errors"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn builtins_placeholder_hashes() {
|
|
||||||
assert_eq!(
|
|
||||||
hash_placeholder("out").as_str(),
|
|
||||||
"/1rz4g4znpzjwh1xymhjpm42vipw92pr73vdgl6xs1hycac8kf2n9"
|
|
||||||
);
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
hash_placeholder("").as_str(),
|
|
||||||
"/171rf4jhx57xqz3p7swniwkig249cif71pa08p80mgaf0mqz5bmr"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,3 +1,4 @@
|
||||||
|
//! Contains [crate::builtins::DerivationError].
|
||||||
use nix_compat::{derivation::DerivationError, nixhash};
|
use nix_compat::{derivation::DerivationError, nixhash};
|
||||||
use std::rc::Rc;
|
use std::rc::Rc;
|
||||||
use thiserror::Error;
|
use thiserror::Error;
|
157
tvix/glue/src/builtins/mod.rs
Normal file
157
tvix/glue/src/builtins/mod.rs
Normal file
|
@ -0,0 +1,157 @@
|
||||||
|
//! Contains builtins that deal with the store or builder.
|
||||||
|
use std::{cell::RefCell, rc::Rc};
|
||||||
|
|
||||||
|
use crate::known_paths::KnownPaths;
|
||||||
|
|
||||||
|
mod derivation;
|
||||||
|
mod derivation_error;
|
||||||
|
|
||||||
|
pub use derivation_error::Error as DerivationError;
|
||||||
|
|
||||||
|
/// Adds derivation-related builtins to the passed [tvix_eval::Evaluation].
|
||||||
|
///
|
||||||
|
/// These are `derivation` and `derivationStrict`.
|
||||||
|
///
|
||||||
|
/// As they need to interact with `known_paths`, we also need to pass in
|
||||||
|
/// `known_paths`.
|
||||||
|
pub fn add_derivation_builtins(
|
||||||
|
eval: &mut tvix_eval::Evaluation,
|
||||||
|
known_paths: Rc<RefCell<KnownPaths>>,
|
||||||
|
) {
|
||||||
|
eval.builtins
|
||||||
|
.extend(derivation::derivation_builtins::builtins(known_paths));
|
||||||
|
|
||||||
|
// Add the actual `builtins.derivation` from compiled Nix code
|
||||||
|
eval.src_builtins
|
||||||
|
.push(("derivation", include_str!("derivation.nix")));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::add_derivation_builtins;
|
||||||
|
use crate::known_paths::KnownPaths;
|
||||||
|
use nix_compat::store_path::hash_placeholder;
|
||||||
|
use std::{cell::RefCell, rc::Rc};
|
||||||
|
use test_case::test_case;
|
||||||
|
use tvix_eval::EvaluationResult;
|
||||||
|
|
||||||
|
/// evaluates a given nix expression and returns the result.
|
||||||
|
/// Takes care of setting up the evaluator so it knows about the
|
||||||
|
// `derivation` builtin.
|
||||||
|
fn eval(str: &str) -> EvaluationResult {
|
||||||
|
let mut eval = tvix_eval::Evaluation::new_impure(str, None);
|
||||||
|
|
||||||
|
let known_paths: Rc<RefCell<KnownPaths>> = Default::default();
|
||||||
|
|
||||||
|
add_derivation_builtins(&mut eval, known_paths.clone());
|
||||||
|
|
||||||
|
// run the evaluation itself.
|
||||||
|
eval.evaluate()
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn derivation() {
|
||||||
|
let result = eval(
|
||||||
|
r#"(derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux";}).outPath"#,
|
||||||
|
);
|
||||||
|
|
||||||
|
assert!(result.errors.is_empty(), "expect evaluation to succeed");
|
||||||
|
let value = result.value.expect("must be some");
|
||||||
|
|
||||||
|
match value {
|
||||||
|
tvix_eval::Value::String(s) => {
|
||||||
|
assert_eq!(
|
||||||
|
"/nix/store/xpcvxsx5sw4rbq666blz6sxqlmsqphmr-foo",
|
||||||
|
s.as_str()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
_ => panic!("unexpected value type: {:?}", value),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// a derivation with an empty name is an error.
|
||||||
|
#[test]
|
||||||
|
fn derivation_empty_name_fail() {
|
||||||
|
let result = eval(
|
||||||
|
r#"(derivation { name = ""; builder = "/bin/sh"; system = "x86_64-linux";}).outPath"#,
|
||||||
|
);
|
||||||
|
|
||||||
|
assert!(!result.errors.is_empty(), "expect evaluation to fail");
|
||||||
|
}
|
||||||
|
|
||||||
|
/// construct some calls to builtins.derivation and compare produced output
|
||||||
|
/// paths.
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/17wgs52s7kcamcyin4ja58njkf91ipq8-foo"; "r:sha256")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo2"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/gi0p8vd635vpk1nq029cz3aa3jkhar5k-foo2"; "r:sha256 other name")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha1"; outputHash = "sha1-VUCRC+16gU5lcrLYHlPSUyx0Y/Q="; }).outPath"#, "/nix/store/p5sammmhpa84ama7ymkbgwwzrilva24x-foo"; "r:sha1")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "md5"; outputHash = "md5-07BzhNET7exJ6qYjitX/AA=="; }).outPath"#, "/nix/store/gmmxgpy1jrzs86r5y05wy6wiy2m15xgi-foo"; "r:md5")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha512"; outputHash = "sha512-DPkYCnZKuoY6Z7bXLwkYvBMcZ3JkLLLc5aNPCnAvlHDdwr8SXBIZixmVwjPDS0r9NGxUojNMNQqUilG26LTmtg=="; }).outPath"#, "/nix/store/lfi2bfyyap88y45mfdwi4j99gkaxaj19-foo"; "r:sha512")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "4374173a8cbe88de152b609f96f46e958bcf65762017474eec5a05ec2bd61530"; }).outPath"#, "/nix/store/17wgs52s7kcamcyin4ja58njkf91ipq8-foo"; "r:sha256 base16")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "0c0msqmyq1asxi74f5r0frjwz2wmdvs9d7v05caxx25yihx1fx23"; }).outPath"#, "/nix/store/17wgs52s7kcamcyin4ja58njkf91ipq8-foo"; "r:sha256 nixbase32")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/17wgs52s7kcamcyin4ja58njkf91ipq8-foo"; "r:sha256 base64")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "sha256-fgIr3TyFGDAXP5+qoAaiMKDg/a1MlT6Fv/S/DaA24S8="; }).outPath"#, "/nix/store/xm1l9dx4zgycv9qdhcqqvji1z88z534b-foo"; "r:sha256 base64 nopad")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "flat"; outputHashAlgo = "sha256"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/q4pkwkxdib797fhk22p0k3g1q32jmxvf-foo"; "sha256")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo2"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "flat"; outputHashAlgo = "sha256"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/znw17xlmx9r6gw8izjkqxkl6s28sza4l-foo2"; "sha256 other name")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "flat"; outputHashAlgo = "sha1"; outputHash = "sha1-VUCRC+16gU5lcrLYHlPSUyx0Y/Q="; }).outPath"#, "/nix/store/zgpnjjmga53d8srp8chh3m9fn7nnbdv6-foo"; "sha1")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "flat"; outputHashAlgo = "md5"; outputHash = "md5-07BzhNET7exJ6qYjitX/AA=="; }).outPath"#, "/nix/store/jfhcwnq1852ccy9ad9nakybp2wadngnd-foo"; "md5")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "flat"; outputHashAlgo = "sha512"; outputHash = "sha512-DPkYCnZKuoY6Z7bXLwkYvBMcZ3JkLLLc5aNPCnAvlHDdwr8SXBIZixmVwjPDS0r9NGxUojNMNQqUilG26LTmtg=="; }).outPath"#, "/nix/store/as736rr116ian9qzg457f96j52ki8bm3-foo"; "sha512")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/17wgs52s7kcamcyin4ja58njkf91ipq8-foo"; "r:sha256 outputHashAlgo omitted")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#, "/nix/store/q4pkwkxdib797fhk22p0k3g1q32jmxvf-foo"; "r:sha256 outputHashAlgo and outputHashMode omitted")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; }).outPath"#, "/nix/store/xpcvxsx5sw4rbq666blz6sxqlmsqphmr-foo"; "outputHash* omitted")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; outputs = ["foo" "bar"]; system = "x86_64-linux"; }).outPath"#, "/nix/store/hkwdinvz2jpzgnjy9lv34d2zxvclj4s3-foo-foo"; "multiple outputs")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; args = ["--foo" "42" "--bar"]; system = "x86_64-linux"; }).outPath"#, "/nix/store/365gi78n2z7vwc1bvgb98k0a9cqfp6as-foo"; "args")]
|
||||||
|
#[test_case(r#"
|
||||||
|
let
|
||||||
|
bar = builtins.derivation {
|
||||||
|
name = "bar";
|
||||||
|
builder = ":";
|
||||||
|
system = ":";
|
||||||
|
outputHash = "08813cbee9903c62be4c5027726a418a300da4500b2d369d3af9286f4815ceba";
|
||||||
|
outputHashAlgo = "sha256";
|
||||||
|
outputHashMode = "recursive";
|
||||||
|
};
|
||||||
|
in
|
||||||
|
(builtins.derivation {
|
||||||
|
name = "foo";
|
||||||
|
builder = ":";
|
||||||
|
system = ":";
|
||||||
|
inherit bar;
|
||||||
|
}).outPath
|
||||||
|
"#, "/nix/store/5vyvcwah9l9kf07d52rcgdk70g2f4y13-foo"; "full")]
|
||||||
|
fn test_outpath(code: &str, expected_path: &str) {
|
||||||
|
let value = eval(code).value.expect("must succeed");
|
||||||
|
|
||||||
|
match value {
|
||||||
|
tvix_eval::Value::String(s) => {
|
||||||
|
assert_eq!(expected_path, s.as_str());
|
||||||
|
}
|
||||||
|
_ => panic!("unexpected value type: {:?}", value),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// construct some calls to builtins.derivation that should be rejected
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha256"; outputHash = "sha256-00"; }).outPath"#; "invalid outputhash")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; system = "x86_64-linux"; outputHashMode = "recursive"; outputHashAlgo = "sha1"; outputHash = "sha256-Q3QXOoy+iN4VK2CflvRulYvPZXYgF0dO7FoF7CvWFTA="; }).outPath"#; "sha1 and sha256")]
|
||||||
|
#[test_case(r#"(builtins.derivation { name = "foo"; builder = "/bin/sh"; outputs = ["foo" "foo"]; system = "x86_64-linux"; }).outPath"#; "duplicate output names")]
|
||||||
|
fn test_outpath_invalid(code: &str) {
|
||||||
|
let resp = eval(code);
|
||||||
|
assert!(resp.value.is_none(), "Value should be None");
|
||||||
|
assert!(
|
||||||
|
!resp.errors.is_empty(),
|
||||||
|
"There should have been some errors"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn builtins_placeholder_hashes() {
|
||||||
|
assert_eq!(
|
||||||
|
hash_placeholder("out").as_str(),
|
||||||
|
"/1rz4g4znpzjwh1xymhjpm42vipw92pr73vdgl6xs1hycac8kf2n9"
|
||||||
|
);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
hash_placeholder("").as_str(),
|
||||||
|
"/171rf4jhx57xqz3p7swniwkig249cif71pa08p80mgaf0mqz5bmr"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
|
@ -1,32 +1,9 @@
|
||||||
use std::{cell::RefCell, rc::Rc};
|
pub mod builtins;
|
||||||
|
|
||||||
use known_paths::KnownPaths;
|
|
||||||
|
|
||||||
pub mod derivation;
|
|
||||||
pub mod errors;
|
|
||||||
pub mod known_paths;
|
pub mod known_paths;
|
||||||
pub mod refscan;
|
pub mod refscan;
|
||||||
pub mod tvix_io;
|
pub mod tvix_io;
|
||||||
pub mod tvix_store_io;
|
pub mod tvix_store_io;
|
||||||
|
|
||||||
/// Adds derivation-related builtins to the passed [tvix_eval::Evaluation].
|
|
||||||
///
|
|
||||||
/// These are `derivation` and `derivationStrict`.
|
|
||||||
///
|
|
||||||
/// As they need to interact with `known_paths`, we also need to pass in
|
|
||||||
/// `known_paths`.
|
|
||||||
pub fn add_derivation_builtins(
|
|
||||||
eval: &mut tvix_eval::Evaluation,
|
|
||||||
known_paths: Rc<RefCell<KnownPaths>>,
|
|
||||||
) {
|
|
||||||
eval.builtins
|
|
||||||
.extend(derivation::derivation_builtins(known_paths));
|
|
||||||
|
|
||||||
// Add the actual `builtins.derivation` from compiled Nix code
|
|
||||||
eval.src_builtins
|
|
||||||
.push(("derivation", include_str!("derivation.nix")));
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Tell the Evaluator to resolve <nix> to the path `/__corepkgs__`,
|
/// Tell the Evaluator to resolve <nix> to the path `/__corepkgs__`,
|
||||||
/// which has special handling in [tvix_io::TvixIO].
|
/// which has special handling in [tvix_io::TvixIO].
|
||||||
/// This is used in nixpkgs to import `fetchurl.nix` from `<nix>`.
|
/// This is used in nixpkgs to import `fetchurl.nix` from `<nix>`.
|
||||||
|
|
|
@ -1,3 +1,4 @@
|
||||||
|
//! Contains [DerivationError], exported as [crate::builtins::DerivationError]
|
||||||
use crate::store_path;
|
use crate::store_path;
|
||||||
use thiserror::Error;
|
use thiserror::Error;
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue