maurice/snix
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(tvix/cli): add known_paths module

Browse source 
This module implements types used to track the set of known paths in
the context of an evaluation.

These are used to determine the build references of a derivation.

Change-Id: I81e15ae33632784e699128916485751613b231a3
Reviewed-on: https://cl.tvl.fyi/c/depot/+/7816
Tested-by: BuildkiteCI
Reviewed-by: flokli <flokli@flokli.de>
This commit is contained in:
Vincent Ambo 2023-01-12 13:59:58 +03:00 • committed by tazjin
parent d644ed389a
commit f27f5ef0c9
3 changed files with 143 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

33
tvix/cli/src/refscan.rs
View file

@ -13,17 +13,34 @@ use std::io;
/// Represents a "primed" reference scanner with an automaton that knows the set
/// of store paths to scan for.
pub struct ReferenceScanner<'c, 's> {
candidates: &'c [&'s str],
pub struct ReferenceScanner<'s> {
candidates: Vec<&'s str>,
searcher: AhoCorasick,
matches: BTreeSet<&'s str>,
}
impl<'c, 's> ReferenceScanner<'c, 's> {
pub trait ToOwnedVec<T> {
fn to_owned_vec(self) -> Vec<T>;
}
impl<T: Clone> ToOwnedVec<T> for &[T] {
fn to_owned_vec(self) -> Vec<T> {
self.to_vec()
}
}
impl<T> ToOwnedVec<T> for Vec<T> {
fn to_owned_vec(self) -> Vec<T> {
self
}
}
impl<'s> ReferenceScanner<'s> {
/// Construct a new `ReferenceScanner` that knows how to scan for the given
/// candidate store paths.
pub fn new(candidates: &'c [&'s str]) -> Self {
let searcher = AhoCorasick::new_auto_configured(candidates);
pub fn new<V: ToOwnedVec<&'s str>>(candidates: V) -> Self {
let candidates = candidates.to_owned_vec();
let searcher = AhoCorasick::new_auto_configured(&candidates);
ReferenceScanner {
searcher,
@ -70,6 +87,12 @@ mod tests {
// The actual derivation of `nixpkgs.hello`.
const HELLO_DRV: &'static str = r#"Derive([("out","/nix/store/33l4p0pn0mybmqzaxfkpppyh7vx1c74p-hello-2.12.1","","")],[("/nix/store/6z1jfnqqgyqr221zgbpm30v91yfj3r45-bash-5.1-p16.drv",["out"]),("/nix/store/ap9g09fxbicj836zm88d56dn3ff4clxl-stdenv-linux.drv",["out"]),("/nix/store/pf80kikyxr63wrw56k00i1kw6ba76qik-hello-2.12.1.tar.gz.drv",["out"])],["/nix/store/9krlzvny65gdc8s7kpb6lkx8cd02c25b-default-builder.sh"],"x86_64-linux","/nix/store/4xw8n979xpivdc46a9ndcvyhwgif00hz-bash-5.1-p16/bin/bash",["-e","/nix/store/9krlzvny65gdc8s7kpb6lkx8cd02c25b-default-builder.sh"],[("buildInputs",""),("builder","/nix/store/4xw8n979xpivdc46a9ndcvyhwgif00hz-bash-5.1-p16/bin/bash"),("cmakeFlags",""),("configureFlags",""),("depsBuildBuild",""),("depsBuildBuildPropagated",""),("depsBuildTarget",""),("depsBuildTargetPropagated",""),("depsHostHost",""),("depsHostHostPropagated",""),("depsTargetTarget",""),("depsTargetTargetPropagated",""),("doCheck","1"),("doInstallCheck",""),("mesonFlags",""),("name","hello-2.12.1"),("nativeBuildInputs",""),("out","/nix/store/33l4p0pn0mybmqzaxfkpppyh7vx1c74p-hello-2.12.1"),("outputs","out"),("patches",""),("pname","hello"),("propagatedBuildInputs",""),("propagatedNativeBuildInputs",""),("src","/nix/store/pa10z4ngm0g83kx9mssrqzz30s84vq7k-hello-2.12.1.tar.gz"),("stdenv","/nix/store/cp65c8nk29qq5cl1wyy5qyw103cwmax7-stdenv-linux"),("strictDeps",""),("system","x86_64-linux"),("version","2.12.1")])"#;
impl<T: Clone, const N: usize> ToOwnedVec<T> for &[T; N] {
fn to_owned_vec(self) -> Vec<T> {
self.to_vec()
}
}
#[test]
fn test_empty() {
let mut scanner = ReferenceScanner::new(&[]);