123456use std::borrow::Cow;7use std::cell::RefCell;8use std::ffi::{CStr, CString, c_char, c_int, c_uint, c_void};9use std::fmt;10use std::ptr::null_mut;11use std::sync::LazyLock;12use std::{collections::HashMap, path::PathBuf};1314use anyhow::{Context, bail};15use serde::Serialize;16use serde::de::DeserializeOwned;1718pub use anyhow::Result;1920use self::logging::nix_logging_cxx;21use self::nix_cxx::set_fetcher_setting;22use self::nix_raw::{23 alloc_value, c_context, c_context_create, err_code, err_info_msg, eval_state_build,24 eval_state_builder_new, expr_eval_from_string, fetchers_settings, fetchers_settings_free,25 fetchers_settings_new, flake_lock, flake_lock_flags, flake_lock_flags_free,26 flake_lock_flags_new, flake_reference_parse_flags, flake_reference_parse_flags_free,27 flake_reference_parse_flags_new, flake_reference_parse_flags_set_base_directory,28 flake_settings, flake_settings_free, flake_settings_new, init_bool, init_int, init_string,29 locked_flake_free, locked_flake_get_output_attrs, set_err_msg, setting_set, state_free,30 value_decref, value_force, value_incref,31};3233mod value;3435pub mod logging;36#[doc(hidden)]37pub mod macros;38pub mod util;3940#[allow(non_upper_case_globals, non_camel_case_types, non_snake_case)]41mod nix_raw {42 include!(concat!(env!("OUT_DIR"), "/bindings.rs"));43}44#[cxx::bridge]45pub mod nix_cxx {46 unsafe extern "C++" {47 type nix_fetchers_settings;48 include!("nix-eval/src/lib.hh");4950 unsafe fn set_fetcher_setting(51 settings: *mut nix_fetchers_settings,52 setting: *const c_char,53 value: *const c_char,54 );55 }56}5758#[derive(Debug, PartialEq, Eq)]59pub enum NixType {60 Thunk,61 Int,62 Float,63 Bool,64 String,65 Path,66 Null,67 Attrs,68 List,69 Function,70 External,71}72impl NixType {73 fn from_int(c: c_uint) -> Self {74 match c {75 0 => Self::Thunk,76 1 => Self::Int,77 2 => Self::Float,78 3 => Self::Bool,79 4 => Self::String,80 5 => Self::Path,81 6 => Self::Null,82 7 => Self::Attrs,83 8 => Self::List,84 9 => Self::Function,85 10 => Self::External,86 _ => unreachable!("unknown nix type: {c}"),87 }88 }89}9091#[derive(Debug)]92#[repr(i32)]93enum NixErrorKind {94 Unknown = 1,95 Overflow = 2,96 Key = 3,97 Generic = 4,98}99impl NixErrorKind {100 fn from_int(v: c_int) -> Option<Self> {101 Some(match v {102 0 => return None,103 -1 => Self::Unknown,104 -2 => Self::Overflow,105 -3 => Self::Key,106 -4 => Self::Generic,107 _ => {108 debug_assert!(false, "unexpected nix error kind: {v}");109 Self::Unknown110 }111 })112 }113}114115pub fn gc_register_my_thread() {116 assert_eq!(unsafe { nix_raw::GC_thread_is_registered() }, 0);117118 let mut sb = nix_raw::GC_stack_base {119 mem_base: null_mut(),120 };121 let r = unsafe { nix_raw::GC_get_stack_base(&mut sb) };122 if r as u32 != nix_raw::GC_SUCCESS {123 panic!("failed to get thread stack base");124 }125 unsafe { nix_raw::GC_register_my_thread(&sb) };126}127pub fn gc_unregister_my_thread() {128 assert_eq!(unsafe { nix_raw::GC_thread_is_registered() }, 1);129130 unsafe { nix_raw::GC_unregister_my_thread() };131}132133struct ThreadRegisterGuard {}134impl ThreadRegisterGuard {135 fn new() -> Self {136 gc_register_my_thread();137 Self {}138 }139}140impl Drop for ThreadRegisterGuard {141 fn drop(&mut self) {142 gc_unregister_my_thread();143 }144}145146struct NixContext(*mut c_context);147impl NixContext {148 fn set_err(&mut self, err: NixErrorKind, msg: &CStr) {149 unsafe { set_err_msg(self.0, err as c_int, msg.as_ptr()) };150 }151 fn new() -> Self {152 let ctx = unsafe { c_context_create() };153 Self(ctx)154 }155 fn error_kind(&self) -> Option<NixErrorKind> {156 let code = unsafe { err_code(self.0) };157 NixErrorKind::from_int(code)158 }159 fn error<'t>(&self) -> Option<Cow<'t, str>> {160 if let NixErrorKind::Generic = self.error_kind()? {161 let mut err_out = String::new();162 unsafe {163 err_info_msg(164 null_mut(),165 self.0,166 Some(copy_nix_str),167 (&raw mut err_out).cast(),168 )169 };170 return Some(Cow::Owned(err_out));171 };172173 174 175 let str = unsafe { nix_raw::err_msg(null_mut(), self.0, null_mut()) };176 Some(unsafe { CStr::from_ptr(str) }.to_string_lossy())177178 179 180 181 182 183 }184 fn clean_err(&mut self) {185 unsafe {186 nix_raw::clear_err(self.0);187 }188 }189190 fn bail_if_error(&self) -> Result<()> {191 if let Some(err) = self.error() {192 bail!("{err}");193 };194 Ok(())195 }196197 fn run_in_context<T>(&mut self, f: impl FnOnce(*mut c_context) -> T) -> Result<T> {198 self.clean_err();199 let o = f(self.0);200 self.bail_if_error()?;201 self.clean_err();202 Ok(o)203 }204}205impl Drop for NixContext {206 fn drop(&mut self) {207 unsafe {208 nix_raw::c_context_free(self.0);209 }210 }211}212struct GlobalState {213 store: Store,214 state: EvalState,215}216impl GlobalState {217 fn new() -> Result<Self> {218 let mut ctx = NixContext::new();219 let store = ctx220 .run_in_context(|c| unsafe { nix_raw::store_open(c, c"daemon".as_ptr(), null_mut()) })221 .map(Store)?;222223 let builder = ctx.run_in_context(|c| unsafe { eval_state_builder_new(c, store.0) })?;224 ctx.run_in_context(|c| {225 unsafe {226 nix_raw::eval_state_builder_set_eval_setting(227 c,228 builder,229 c"lazy-trees".as_ptr(),230 c"true".as_ptr(),231 )232 }233 234 })?;235 let state = ctx236 .run_in_context(|c| unsafe { eval_state_build(c, builder) })237 .map(EvalState)?;238239 Ok(Self { store, state })240 }241}242243struct ThreadState {244 ctx: NixContext,245}246impl ThreadState {247 fn new() -> Result<Self> {248 let ctx = NixContext::new();249250 Ok(Self { ctx })251 }252}253254static GLOBAL_STATE: LazyLock<GlobalState> =255 LazyLock::new(|| GlobalState::new().expect("global state init shouldn't fail"));256257thread_local! {258 static THREAD_STATE: RefCell<ThreadState> = RefCell::new(ThreadState::new().expect("thread state init shouldn't fail"));259}260fn with_default_context<T>(261 f: impl FnOnce(*mut c_context, *mut nix_raw::EvalState) -> T,262) -> Result<T> {263 let global = &GLOBAL_STATE.state;264 let (ctx, state) = THREAD_STATE.with_borrow_mut(|w| (w.ctx.0, global.0));265 let mut ctx = NixContext(ctx);266 let v = ctx.run_in_context(|c| f(c, state));267 268 std::mem::forget(ctx);269 v270}271272fn set_setting(s: &CStr, v: &CStr) -> Result<()> {273 with_default_context(|c, _| unsafe { setting_set(c, s.as_ptr(), v.as_ptr()) }).map(|_| ())274}275276pub struct FetchSettings(*mut fetchers_settings);277impl FetchSettings {278 pub fn new() -> Self {279 Self::try_new().expect("allocation should not fail")280 }281 fn try_new() -> Result<Self> {282 with_default_context(|c, _| unsafe { fetchers_settings_new(c) }).map(Self)283 }284 pub fn set(&mut self, setting: &CStr, value: &CStr) {285 unsafe {286 set_fetcher_setting(self.0.cast(), setting.as_ptr(), value.as_ptr());287 };288 }289}290unsafe impl Send for FetchSettings {}291unsafe impl Sync for FetchSettings {}292impl Drop for FetchSettings {293 fn drop(&mut self) {294 unsafe { fetchers_settings_free(self.0) };295 }296}297pub struct FlakeSettings(*mut flake_settings);298impl FlakeSettings {299 pub fn new() -> Result<Self> {300 with_default_context(|c, _| unsafe { flake_settings_new(c) }).map(Self)301 }302}303unsafe impl Send for FlakeSettings {}304unsafe impl Sync for FlakeSettings {}305impl Drop for FlakeSettings {306 fn drop(&mut self) {307 unsafe {308 flake_settings_free(self.0);309 }310 }311}312313struct FlakeReferenceParseFlags(*mut flake_reference_parse_flags);314impl FlakeReferenceParseFlags {315 fn new(settings: &mut FlakeSettings) -> Result<Self> {316 with_default_context(|c, _| unsafe { flake_reference_parse_flags_new(c, settings.0) })317 .map(Self)318 }319 fn set_base_dir(&mut self, dir: &str) -> Result<()> {320 with_default_context(|c, _| {321 unsafe {322 flake_reference_parse_flags_set_base_directory(323 c,324 self.0,325 dir.as_ptr().cast(),326 dir.len(),327 )328 };329 })330 }331}332impl Drop for FlakeReferenceParseFlags {333 fn drop(&mut self) {334 unsafe {335 flake_reference_parse_flags_free(self.0);336 }337 }338}339struct FlakeLockFlags(*mut flake_lock_flags);340impl FlakeLockFlags {341 fn new(settings: &mut FlakeSettings) -> Result<Self> {342 with_default_context(|c, _| unsafe { flake_lock_flags_new(c, settings.0) }).map(Self)343 }344}345impl Drop for FlakeLockFlags {346 fn drop(&mut self) {347 unsafe {348 flake_lock_flags_free(self.0);349 }350 }351}352353unsafe extern "C" fn copy_nix_str(start: *const c_char, n: c_uint, user_data: *mut c_void) {354 let s = unsafe { std::slice::from_raw_parts(start.cast::<u8>(), n as usize) };355 let s = std::str::from_utf8(s).expect("c string has invalid utf-8");356 unsafe { *user_data.cast::<String>() = s.to_owned() };357}358359struct Store(*mut nix_raw::Store);360unsafe impl Send for Store {}361unsafe impl Sync for Store {}362363struct EvalState(*mut nix_raw::EvalState);364impl EvalState {365 366 fn new_raw(store: *mut nix_raw::Store) -> Result<Self> {367 let builder =368 with_default_context(|c, _| unsafe { nix_raw::eval_state_builder_new(c, store) })?;369370 with_default_context(|c, _| unsafe { eval_state_build(c, builder) }).map(Self)371372 373 }374}375unsafe impl Send for EvalState {}376unsafe impl Sync for EvalState {}377impl Drop for EvalState {378 fn drop(&mut self) {379 unsafe {380 state_free(self.0);381 }382 }383}384385pub struct FlakeReference(*mut nix_raw::flake_reference);386impl FlakeReference {387 pub fn new(s: &str, fetch: &FetchSettings) -> Result<(Self, String)> {388 let mut flake_settings = FlakeSettings::new()?;389 let mut parse_flags = FlakeReferenceParseFlags::new(&mut flake_settings)?;390391 392393 let mut out = null_mut();394 let mut fragment = String::new();395 396 with_default_context(|c, _| unsafe {397 nix_raw::flake_reference_and_fragment_from_string(398 c,399 fetch.0,400 flake_settings.0,401 parse_flags.0,402 s.as_ptr().cast(),403 s.len(),404 &mut out,405 Some(copy_nix_str),406 (&raw mut fragment).cast(),407 )408 })?;409 assert!(!out.is_null());410411 Ok((Self(out), fragment))412 }413 pub fn lock(&mut self, fetch: &FetchSettings) -> Result<LockedFlake> {414 let mut settings = FlakeSettings::new()?;415 let lock_flags = FlakeLockFlags::new(&mut settings)?;416 with_default_context(|c, es| unsafe {417 flake_lock(c, fetch.0, settings.0, es, lock_flags.0, self.0)418 })419 .map(LockedFlake)420 }421}422unsafe impl Send for FlakeReference {}423unsafe impl Sync for FlakeReference {}424425pub struct LockedFlake(*mut nix_raw::locked_flake);426impl LockedFlake {427 pub fn get_attrs(&self, settings: &mut FlakeSettings) -> Result<Value> {428 with_default_context(|c, es| unsafe {429 locked_flake_get_output_attrs(c, settings.0, es, self.0)430 })431 .map(Value)432 }433}434unsafe impl Send for LockedFlake {}435unsafe impl Sync for LockedFlake {}436impl Drop for LockedFlake {437 fn drop(&mut self) {438 unsafe {439 locked_flake_free(self.0);440 };441 }442}443444type FieldName = [u8; 32];445fn init_field_name(v: &str) -> FieldName {446 let mut f = [0; 32];447 assert!(v.len() < 32, "max field name is 31 char");448 assert!(449 v.bytes().all(|v| v != 0),450 "nul bytes are unsupported in field name"451 );452 f[0..v.len()].copy_from_slice(v.as_bytes());453 f454}455456pub struct RealisedString(*mut nix_raw::realised_string);457impl fmt::Debug for RealisedString {458 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {459 self.as_str().fmt(f)460 }461}462463impl RealisedString {464 fn as_str(&self) -> &str {465 let len = unsafe { nix_raw::realised_string_get_buffer_size(self.0) };466 let data: *const u8 = unsafe { nix_raw::realised_string_get_buffer_start(self.0) }.cast();467 let data = unsafe { std::slice::from_raw_parts(data, len) };468 std::str::from_utf8(data).expect("non-utf8 strings not supported")469 }470 fn path_count(&self) -> usize {471 unsafe { nix_raw::realised_string_get_store_path_count(self.0) }472 }473 fn path(&self, i: usize) -> String {474 assert!(i < self.path_count());475 let path = unsafe { nix_raw::realised_string_get_store_path(self.0, i) };476 let mut err_out = String::new();477 unsafe { nix_raw::store_path_name(path, Some(copy_nix_str), (&raw mut err_out).cast()) };478 err_out479 }480}481482unsafe impl Send for RealisedString {}483impl Drop for RealisedString {484 fn drop(&mut self) {485 with_default_context(|c, _| unsafe { nix_raw::realised_string_free(self.0) })486 .expect("string free should not fail")487 }488}489490pub struct Value(*mut nix_raw::value);491492unsafe impl Send for Value {}493unsafe impl Sync for Value {}494495pub trait AsFieldName {496 fn as_field_name<T>(&self, v: impl FnOnce(FieldName) -> Result<T>) -> Result<T>;497 fn to_field_name(&self) -> Result<String>;498}499impl AsFieldName for Value {500 fn as_field_name<T>(&self, v: impl FnOnce(FieldName) -> Result<T>) -> Result<T> {501 let f = self.to_string()?;502 v(init_field_name(&f))503 }504 fn to_field_name(&self) -> Result<String> {505 self.to_string()506 }507}508impl<E> AsFieldName for E509where510 E: AsRef<str>,511{512 fn as_field_name<T>(&self, v: impl FnOnce(FieldName) -> Result<T>) -> Result<T> {513 let f = self.as_ref();514 v(init_field_name(f))515 }516 fn to_field_name(&self) -> Result<String> {517 Ok(self.as_ref().to_owned())518 }519}520521struct AttrsBuilder(*mut nix_raw::BindingsBuilder);522impl AttrsBuilder {523 fn new(capacity: usize) -> Self {524 with_default_context(|c, es| unsafe { nix_raw::make_bindings_builder(c, es, capacity) })525 .map(Self)526 .expect("alloc should not fail")527 }528 fn insert(&mut self, k: &impl AsFieldName, v: Value) {529 k.as_field_name(|name| {530 with_default_context(|c, _| unsafe {531 nix_raw::bindings_builder_insert(c, self.0, name.as_ptr().cast(), v.0)532 })533 })534 .expect("builder insert shouldn't fail");535 }536}537impl Drop for AttrsBuilder {538 fn drop(&mut self) {539 unsafe { nix_raw::bindings_builder_free(self.0) };540 }541}542543impl Value {544 pub fn new_attrs(v: HashMap<&str, Value>) -> Result<Self> {545 let out = Self::new_uninit()?;546 let mut b = AttrsBuilder::new(v.len());547 for (k, v) in v {548 b.insert(&k, v);549 }550 with_default_context(|c, _| unsafe { nix_raw::make_attrs(c, out.0, b.0) })?;551 Ok(out)552 }553 fn new_list<T: Into<Self>>(v: Vec<T>) -> Result<Self> {554 todo!()555 }556 fn new_uninit() -> Result<Self> {557 let out = with_default_context(|c, es| unsafe { alloc_value(c, es) })?;558 Ok(Self(out))559 }560 fn new_str(v: &str) -> Result<Self> {561 let s = CString::new(v).expect("string should not contain NULs");562 let uninit = Self::new_uninit()?;563 564 with_default_context(|c, _| unsafe { init_string(c, uninit.0, s.as_ptr()) })?;565 Ok(uninit)566 }567 fn new_int(i: i64) -> Result<Self> {568 let uninit = Self::new_uninit()?;569 with_default_context(|c, _| unsafe { init_int(c, uninit.0, i) })?;570 Ok(uninit)571 }572 fn new_bool(v: bool) -> Result<Self> {573 let uninit = Self::new_uninit()?;574 with_default_context(|c, _| unsafe { init_bool(c, uninit.0, v) })?;575 Ok(uninit)576 }577 fn force(&mut self, st: &mut EvalState) -> Result<()> {578 with_default_context(|c, _| unsafe { value_force(c, st.0, self.0) })?;579 Ok(())580 }581 pub fn type_of(&self) -> Result<NixType> {582 let ty = with_default_context(|c, _| unsafe { nix_raw::get_type(c, self.0) })?;583 Ok(NixType::from_int(ty))584 }585 pub fn to_string(&self) -> Result<String> {586 Ok(self.to_realised_string()?.as_str().to_owned())587 }588 pub fn to_realised_string(&self) -> Result<RealisedString> {589 with_default_context(|c, es| unsafe { nix_raw::string_realise(c, es, self.0, false) })590 .map(RealisedString)591592 593 594 595 596 597 598 599 }600601 pub fn has_field(&self, field: &str) -> Result<bool> {602 let f = init_field_name(field);603 with_default_context(|c, es| unsafe {604 nix_raw::has_attr_byname(c, self.0, es, f.as_ptr().cast())605 })606 }607 608 609 610 pub fn list_fields(&self) -> Result<Vec<String>> {611 if !matches!(self.type_of()?, NixType::Attrs) {612 bail!("invalid type: expected attrs");613 }614615 let len = with_default_context(|c, _| unsafe { nix_raw::get_attrs_size(c, self.0) })?;616 let mut out = Vec::with_capacity(len as usize);617618 for i in 0..len {619 let name = with_default_context(|c, es| unsafe {620 nix_raw::get_attr_name_byidx(c, self.0, es, i)621 })?;622 let c = unsafe { CStr::from_ptr(name) };623 out.push(c.to_str().expect("nix field names are utf-8").to_owned());624 }625 Ok(out)626 }627 pub fn get_elem(&self, v: usize) -> Result<Self> {628 if !matches!(self.type_of()?, NixType::List) {629 bail!("invalid type: expected list");630 }631 let len =632 with_default_context(|c, _| unsafe { nix_raw::get_list_size(c, self.0) })? as usize;633 if v >= len {634 bail!("oob list get: {v} >= {len}");635 }636637 with_default_context(|c, es| unsafe { nix_raw::get_list_byidx(c, self.0, es, v as u32) })638 .map(Self)639 }640 pub fn attrs_update(self, other: Value) -> Result<Self> {641 let a_fields = self.list_fields()?;642 let b_fields = other.list_fields()?;643 match (a_fields.len(), b_fields.len()) {644 (_, 0) => return Ok(self),645 (0, _) => return Ok(other),646 _ => {}647 }648 let mut out = HashMap::new();649 for f in a_fields.iter() {650 if b_fields.contains(f) {651 break;652 }653 out.insert(f.as_str(), self.get_field(f)?);654 }655 if out.is_empty() {656 657 return Ok(other);658 }659 for f in b_fields.iter() {660 out.insert(f.as_str(), other.get_field(f)?);661 }662 Self::new_attrs(out)663 }664 pub fn get_field(&self, name: impl AsFieldName) -> Result<Self> {665 if !matches!(self.type_of()?, NixType::Attrs) {666 bail!("invalid type: expected attrs");667 }668669 name.as_field_name(|name| {670 with_default_context(|c, es| unsafe {671 nix_raw::get_attr_byname(c, self.0, es, name.as_ptr().cast())672 })673 .map(Self)674 })675 .with_context(|| format!("getting field {:?}", name.to_field_name()))676 }677 pub fn call(&self, v: Value) -> Result<Self> {678 if !matches!(self.type_of()?, NixType::Function) {679 680 bail!("invalid type: expected function");681 }682683 let out = Value::new_uninit()?;684 with_default_context(|c, es| unsafe { nix_raw::value_call(c, es, self.0, v.0, out.0) })?;685686 Ok(out)687 }688 pub fn eval(v: &str) -> Result<Self> {689 let s = CString::new(v).expect("expression shouldn't have internal NULs");690 let out = Self::new_uninit()?;691 with_default_context(|c, es| unsafe {692 expr_eval_from_string(c, es, s.as_ptr(), c"/homeless-shelter".as_ptr(), out.0)693 })?;694 Ok(out)695 }696 pub async fn build(&self, output: &str) -> Result<PathBuf> {697 if !self.is_derivation() {698 bail!("expected derivation to build")699 }700 let output_name = self.get_field("outputName")?.to_string()?;701 let v = if output_name != output {702 let out = self.get_field(output)?;703 if !out.is_derivation() {704 bail!("unknown output: {output}");705 }706 out707 } else {708 self.clone()709 };710 711 let drv_path = tokio::task::spawn_blocking(move || v.get_field("outPath")?.to_string())712 .await713 .expect("should not fail")?;714 Ok(PathBuf::from(drv_path))715 }716 pub fn as_json<T: DeserializeOwned>(&self) -> Result<T> {717 let to_json = Self::eval("builtins.toJSON")?;718 let s = to_json.call(self.clone())?.to_string()?;719 Ok(serde_json::from_str(&s)?)720 }721 pub fn serialized<T: Serialize>(v: &T) -> Result<Self> {722 Self::eval(&nixlike::serialize(v)?)723 }724725 726 fn to_string_weak(&self) -> Result<String> {727 728 self.to_string()729 }730731 fn is_derivation(&self) -> bool {732 if !matches!(self.type_of(), Ok(NixType::Attrs)) {733 return false;734 }735 let Some(ty) = self.get_field("type").ok() else {736 return false;737 };738 matches!(ty.to_string().as_deref(), Ok("derivation"))739 }740}741742impl From<String> for Value {743 fn from(value: String) -> Self {744 Value::new_str(&value).expect("todo: TryFrom")745 }746}747impl From<bool> for Value {748 fn from(value: bool) -> Self {749 Value::new_bool(value).expect("todo: TryFrom")750 }751}752impl From<&str> for Value {753 fn from(value: &str) -> Self {754 Value::new_str(&value).expect("todo: TryFrom")755 }756}757impl<T> From<Vec<T>> for Value758where759 T: Into<Value>,760{761 fn from(value: Vec<T>) -> Self {762 Value::new_list(value).expect("todo: TryFrom")763 }764}765766impl Clone for Value {767 fn clone(&self) -> Self {768 with_default_context(|c, _| unsafe { value_incref(c, self.0) })769 .expect("value incref should not fail");770 Self(self.0)771 }772}773impl Drop for Value {774 fn drop(&mut self) {775 with_default_context(|c, _| unsafe { value_decref(c, self.0) })776 .expect("value drop should not fail");777 }778}779780pub fn init_libraries() {781 unsafe { nix_raw::GC_allow_register_threads() };782783 let mut ctx = NixContext::new();784 ctx.run_in_context(|c| unsafe { nix_raw::libutil_init(c) })785 .expect("util init should not fail");786 ctx.run_in_context(|c| unsafe { nix_raw::libstore_init(c) })787 .expect("store init should not fail");788 ctx.run_in_context(|c| unsafe { nix_raw::libexpr_init(c) })789 .expect("expr init should not fail");790791 nix_logging_cxx::apply_tracing_logger();792}793794#[test_log::test]795fn test_native() -> Result<()> {796 let mut fetch_settings = FetchSettings::new();797 fetch_settings.set(c"warn-dirty", c"false");798 799800 let (mut r, _) = FlakeReference::new("/home/lach/build/fleet", &fetch_settings)?;801 let locked = r.lock(&fetch_settings)?;802 let attrs = locked.get_attrs(&mut FlakeSettings::new()?)?;803804 let builtins = Value::eval("builtins")?;805 dbg!(builtins.type_of()?);806807 dbg!(attrs.type_of()?);808 dbg!(attrs.list_fields()?);809 dbg!(810 attrs811 .get_field("packages")?812 .get_field("x86_64-linux")?813 .get_field("fleet")?814 .get_field("outPath")?815 .to_string()816 );817818 Ok(())819}820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894
1use std::borrow::Cow;2use std::cell::RefCell;3use std::ffi::{CStr, CString, c_char, c_int, c_uint, c_void};4use std::fmt;5use std::ptr::null_mut;6use std::sync::LazyLock;7use std::{collections::HashMap, path::PathBuf};89use anyhow::{Context, anyhow, bail};10use serde::Serialize;11use serde::de::DeserializeOwned;1213pub use anyhow::Result;1415use self::logging::nix_logging_cxx;16use self::nix_cxx::set_fetcher_setting;17use self::nix_raw::{18 alloc_value, c_context, c_context_create, err_code, err_info_msg, eval_state_build,19 eval_state_builder_new, expr_eval_from_string, fetchers_settings, fetchers_settings_free,20 fetchers_settings_new, flake_lock, flake_lock_flags, flake_lock_flags_free,21 flake_lock_flags_new, flake_reference_parse_flags, flake_reference_parse_flags_free,22 flake_reference_parse_flags_new, flake_reference_parse_flags_set_base_directory,23 flake_settings, flake_settings_free, flake_settings_new, init_bool, init_int, init_string,24 locked_flake_free, locked_flake_get_output_attrs, set_err_msg, setting_set, state_free,25 value_decref, value_incref,26};272829pub mod logging;30#[doc(hidden)]31pub mod macros;32pub mod util;3334#[allow(35 non_upper_case_globals,36 non_camel_case_types,37 non_snake_case,38 dead_code39)]40mod nix_raw {41 include!(concat!(env!("OUT_DIR"), "/bindings.rs"));42}43#[cxx::bridge]44pub mod nix_cxx {45 unsafe extern "C++" {46 type nix_fetchers_settings;47 include!("nix-eval/src/lib.hh");4849 unsafe fn set_fetcher_setting(50 settings: *mut nix_fetchers_settings,51 setting: *const c_char,52 value: *const c_char,53 );54 }55}5657#[derive(Debug, PartialEq, Eq)]58pub enum NixType {59 Thunk,60 Int,61 Float,62 Bool,63 String,64 Path,65 Null,66 Attrs,67 List,68 Function,69 External,70}71impl NixType {72 fn from_int(c: c_uint) -> Self {73 match c {74 0 => Self::Thunk,75 1 => Self::Int,76 2 => Self::Float,77 3 => Self::Bool,78 4 => Self::String,79 5 => Self::Path,80 6 => Self::Null,81 7 => Self::Attrs,82 8 => Self::List,83 9 => Self::Function,84 10 => Self::External,85 _ => unreachable!("unknown nix type: {c}"),86 }87 }88}8990enum FunctorKind {91 Function,92 Functor,93}9495#[derive(Debug)]96#[repr(i32)]97enum NixErrorKind {98 Unknown = 1,99 Overflow = 2,100 Key = 3,101 Generic = 4,102}103impl NixErrorKind {104 fn from_int(v: c_int) -> Option<Self> {105 Some(match v {106 0 => return None,107 -1 => Self::Unknown,108 -2 => Self::Overflow,109 -3 => Self::Key,110 -4 => Self::Generic,111 _ => {112 debug_assert!(false, "unexpected nix error kind: {v}");113 Self::Unknown114 }115 })116 }117}118119pub fn gc_register_my_thread() {120 assert_eq!(unsafe { nix_raw::GC_thread_is_registered() }, 0);121122 let mut sb = nix_raw::GC_stack_base {123 mem_base: null_mut(),124 };125 let r = unsafe { nix_raw::GC_get_stack_base(&mut sb) };126 if r as u32 != nix_raw::GC_SUCCESS {127 panic!("failed to get thread stack base");128 }129 unsafe { nix_raw::GC_register_my_thread(&sb) };130}131pub fn gc_unregister_my_thread() {132 assert_eq!(unsafe { nix_raw::GC_thread_is_registered() }, 1);133134 unsafe { nix_raw::GC_unregister_my_thread() };135}136137pub struct ThreadRegisterGuard {}138impl ThreadRegisterGuard {139 pub fn new() -> Self {140 gc_register_my_thread();141 Self {}142 }143}144impl Drop for ThreadRegisterGuard {145 fn drop(&mut self) {146 gc_unregister_my_thread();147 }148}149150pub struct NixContext(*mut c_context);151impl NixContext {152 pub fn set_err(&mut self, err: NixErrorKind, msg: &CStr) {153 unsafe { set_err_msg(self.0, err as c_int, msg.as_ptr()) };154 }155 pub fn new() -> Self {156 let ctx = unsafe { c_context_create() };157 Self(ctx)158 }159 fn error_kind(&self) -> Option<NixErrorKind> {160 let code = unsafe { err_code(self.0) };161 NixErrorKind::from_int(code)162 }163 fn error<'t>(&self) -> Option<Cow<'t, str>> {164 if let NixErrorKind::Generic = self.error_kind()? {165 let mut err_out = String::new();166 unsafe {167 err_info_msg(168 null_mut(),169 self.0,170 Some(copy_nix_str),171 (&raw mut err_out).cast(),172 )173 };174 return Some(Cow::Owned(err_out));175 };176177 178 179 let str = unsafe { nix_raw::err_msg(null_mut(), self.0, null_mut()) };180 Some(unsafe { CStr::from_ptr(str) }.to_string_lossy())181 }182 fn clean_err(&mut self) {183 unsafe {184 nix_raw::clear_err(self.0);185 }186 }187188 fn bail_if_error(&self) -> Result<()> {189 if let Some(err) = self.error() {190 bail!("{err}");191 };192 Ok(())193 }194195 fn run_in_context<T>(&mut self, f: impl FnOnce(*mut c_context) -> T) -> Result<T> {196 self.clean_err();197 let o = f(self.0);198 self.bail_if_error()?;199 self.clean_err();200 Ok(o)201 }202}203impl Drop for NixContext {204 fn drop(&mut self) {205 unsafe {206 nix_raw::c_context_free(self.0);207 }208 }209}210struct GlobalState {211 212 #[allow(dead_code)]213 store: Store,214 state: EvalState,215}216impl GlobalState {217 fn new() -> Result<Self> {218 let mut ctx = NixContext::new();219 let store = ctx220 .run_in_context(|c| unsafe { nix_raw::store_open(c, c"daemon".as_ptr(), null_mut()) })221 .map(Store)?;222223 let builder = ctx.run_in_context(|c| unsafe { eval_state_builder_new(c, store.0) })?;224 ctx.run_in_context(|c| {225 unsafe {226 nix_raw::eval_state_builder_set_eval_setting(227 c,228 builder,229 c"lazy-trees".as_ptr(),230 c"true".as_ptr(),231 )232 }233 234 })?;235 let state = ctx236 .run_in_context(|c| unsafe { eval_state_build(c, builder) })237 .map(EvalState)?;238239 Ok(Self { store, state })240 }241}242243struct ThreadState {244 ctx: NixContext,245}246impl ThreadState {247 fn new() -> Result<Self> {248 let ctx = NixContext::new();249250 Ok(Self { ctx })251 }252}253254static GLOBAL_STATE: LazyLock<GlobalState> =255 LazyLock::new(|| GlobalState::new().expect("global state init shouldn't fail"));256257thread_local! {258 static THREAD_STATE: RefCell<ThreadState> = RefCell::new(ThreadState::new().expect("thread state init shouldn't fail"));259}260fn with_default_context<T>(261 f: impl FnOnce(*mut c_context, *mut nix_raw::EvalState) -> T,262) -> Result<T> {263 let global = &GLOBAL_STATE.state;264 let (ctx, state) = THREAD_STATE.with_borrow_mut(|w| (w.ctx.0, global.0));265 let mut ctx = NixContext(ctx);266 let v = ctx.run_in_context(|c| f(c, state));267 268 std::mem::forget(ctx);269 v270}271272pub fn set_setting(s: &CStr, v: &CStr) -> Result<()> {273 with_default_context(|c, _| unsafe { setting_set(c, s.as_ptr(), v.as_ptr()) }).map(|_| ())274}275276pub struct FetchSettings(*mut fetchers_settings);277impl FetchSettings {278 pub fn new() -> Self {279 Self::try_new().expect("allocation should not fail")280 }281 fn try_new() -> Result<Self> {282 with_default_context(|c, _| unsafe { fetchers_settings_new(c) }).map(Self)283 }284 pub fn set(&mut self, setting: &CStr, value: &CStr) {285 unsafe {286 set_fetcher_setting(self.0.cast(), setting.as_ptr(), value.as_ptr());287 };288 }289}290unsafe impl Send for FetchSettings {}291unsafe impl Sync for FetchSettings {}292293impl Default for FetchSettings {294 fn default() -> Self {295 Self::new()296 }297}298299impl Drop for FetchSettings {300 fn drop(&mut self) {301 unsafe { fetchers_settings_free(self.0) };302 }303}304pub struct FlakeSettings(*mut flake_settings);305impl FlakeSettings {306 pub fn new() -> Result<Self> {307 with_default_context(|c, _| unsafe { flake_settings_new(c) }).map(Self)308 }309}310unsafe impl Send for FlakeSettings {}311unsafe impl Sync for FlakeSettings {}312impl Drop for FlakeSettings {313 fn drop(&mut self) {314 unsafe {315 flake_settings_free(self.0);316 }317 }318}319320pub struct FlakeReferenceParseFlags(*mut flake_reference_parse_flags);321impl FlakeReferenceParseFlags {322 pub fn new(settings: &mut FlakeSettings) -> Result<Self> {323 with_default_context(|c, _| unsafe { flake_reference_parse_flags_new(c, settings.0) })324 .map(Self)325 }326 pub fn set_base_dir(&mut self, dir: &str) -> Result<()> {327 with_default_context(|c, _| {328 unsafe {329 flake_reference_parse_flags_set_base_directory(330 c,331 self.0,332 dir.as_ptr().cast(),333 dir.len(),334 )335 };336 })337 }338}339impl Drop for FlakeReferenceParseFlags {340 fn drop(&mut self) {341 unsafe {342 flake_reference_parse_flags_free(self.0);343 }344 }345}346struct FlakeLockFlags(*mut flake_lock_flags);347impl FlakeLockFlags {348 fn new(settings: &mut FlakeSettings) -> Result<Self> {349 with_default_context(|c, _| unsafe { flake_lock_flags_new(c, settings.0) }).map(Self)350 }351}352impl Drop for FlakeLockFlags {353 fn drop(&mut self) {354 unsafe {355 flake_lock_flags_free(self.0);356 }357 }358}359360unsafe extern "C" fn copy_nix_str(start: *const c_char, n: c_uint, user_data: *mut c_void) {361 let s = unsafe { std::slice::from_raw_parts(start.cast::<u8>(), n as usize) };362 let s = std::str::from_utf8(s).expect("c string has invalid utf-8");363 unsafe { *user_data.cast::<String>() = s.to_owned() };364}365366struct Store(*mut nix_raw::Store);367unsafe impl Send for Store {}368unsafe impl Sync for Store {}369370struct EvalState(*mut nix_raw::EvalState);371unsafe impl Send for EvalState {}372unsafe impl Sync for EvalState {}373374impl Drop for EvalState {375 fn drop(&mut self) {376 unsafe {377 state_free(self.0);378 }379 }380}381382pub struct FlakeReference(*mut nix_raw::flake_reference);383impl FlakeReference {384 pub fn new(s: &str, fetch: &FetchSettings) -> Result<(Self, String)> {385 let mut flake_settings = FlakeSettings::new()?;386 let parse_flags = FlakeReferenceParseFlags::new(&mut flake_settings)?;387388 let mut out = null_mut();389 let mut fragment = String::new();390 391 with_default_context(|c, _| unsafe {392 nix_raw::flake_reference_and_fragment_from_string(393 c,394 fetch.0,395 flake_settings.0,396 parse_flags.0,397 s.as_ptr().cast(),398 s.len(),399 &mut out,400 Some(copy_nix_str),401 (&raw mut fragment).cast(),402 )403 })?;404 assert!(!out.is_null());405406 Ok((Self(out), fragment))407 }408 pub fn lock(&mut self, fetch: &FetchSettings) -> Result<LockedFlake> {409 let mut settings = FlakeSettings::new()?;410 let lock_flags = FlakeLockFlags::new(&mut settings)?;411 with_default_context(|c, es| unsafe {412 flake_lock(c, fetch.0, settings.0, es, lock_flags.0, self.0)413 })414 .map(LockedFlake)415 }416}417unsafe impl Send for FlakeReference {}418unsafe impl Sync for FlakeReference {}419420pub struct LockedFlake(*mut nix_raw::locked_flake);421impl LockedFlake {422 pub fn get_attrs(&self, settings: &mut FlakeSettings) -> Result<Value> {423 with_default_context(|c, es| unsafe {424 locked_flake_get_output_attrs(c, settings.0, es, self.0)425 })426 .map(Value)427 }428}429unsafe impl Send for LockedFlake {}430unsafe impl Sync for LockedFlake {}431impl Drop for LockedFlake {432 fn drop(&mut self) {433 unsafe {434 locked_flake_free(self.0);435 };436 }437}438439type FieldName = [u8; 32];440fn init_field_name(v: &str) -> FieldName {441 let mut f = [0; 32];442 assert!(v.len() < 32, "max field name is 31 char");443 assert!(444 v.bytes().all(|v| v != 0),445 "nul bytes are unsupported in field name"446 );447 f[0..v.len()].copy_from_slice(v.as_bytes());448 f449}450451pub struct RealisedString(*mut nix_raw::realised_string);452impl fmt::Debug for RealisedString {453 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {454 self.as_str().fmt(f)455 }456}457458impl RealisedString {459 pub fn as_str(&self) -> &str {460 let len = unsafe { nix_raw::realised_string_get_buffer_size(self.0) };461 let data: *const u8 = unsafe { nix_raw::realised_string_get_buffer_start(self.0) }.cast();462 let data = unsafe { std::slice::from_raw_parts(data, len) };463 std::str::from_utf8(data).expect("non-utf8 strings not supported")464 }465 pub fn path_count(&self) -> usize {466 unsafe { nix_raw::realised_string_get_store_path_count(self.0) }467 }468 pub fn path(&self, i: usize) -> String {469 assert!(i < self.path_count());470 let path = unsafe { nix_raw::realised_string_get_store_path(self.0, i) };471 let mut err_out = String::new();472 unsafe { nix_raw::store_path_name(path, Some(copy_nix_str), (&raw mut err_out).cast()) };473 err_out474 }475}476477unsafe impl Send for RealisedString {}478impl Drop for RealisedString {479 fn drop(&mut self) {480 unsafe { nix_raw::realised_string_free(self.0) }481 }482}483484pub struct Value(*mut nix_raw::value);485486unsafe impl Send for Value {}487unsafe impl Sync for Value {}488489pub trait AsFieldName {490 fn as_field_name<T>(&self, v: impl FnOnce(FieldName) -> Result<T>) -> Result<T>;491 fn to_field_name(&self) -> Result<String>;492}493impl AsFieldName for Value {494 fn as_field_name<T>(&self, v: impl FnOnce(FieldName) -> Result<T>) -> Result<T> {495 let f = self.to_string()?;496 v(init_field_name(&f))497 }498 fn to_field_name(&self) -> Result<String> {499 self.to_string()500 }501}502impl<E> AsFieldName for E503where504 E: AsRef<str>,505{506 fn as_field_name<T>(&self, v: impl FnOnce(FieldName) -> Result<T>) -> Result<T> {507 let f = self.as_ref();508 v(init_field_name(f))509 }510 fn to_field_name(&self) -> Result<String> {511 Ok(self.as_ref().to_owned())512 }513}514515struct AttrsBuilder(*mut nix_raw::BindingsBuilder);516impl AttrsBuilder {517 fn new(capacity: usize) -> Self {518 with_default_context(|c, es| unsafe { nix_raw::make_bindings_builder(c, es, capacity) })519 .map(Self)520 .expect("alloc should not fail")521 }522 fn insert(&mut self, k: &impl AsFieldName, v: Value) {523 k.as_field_name(|name| {524 with_default_context(|c, _| unsafe {525 nix_raw::bindings_builder_insert(c, self.0, name.as_ptr().cast(), v.0)526 })527 })528 .expect("builder insert shouldn't fail");529 }530}531impl Drop for AttrsBuilder {532 fn drop(&mut self) {533 unsafe { nix_raw::bindings_builder_free(self.0) };534 }535}536537impl Value {538 pub fn new_attrs(v: HashMap<&str, Value>) -> Self {539 let out = Self::new_uninit();540 let mut b = AttrsBuilder::new(v.len());541 for (k, v) in v {542 b.insert(&k, v);543 }544 with_default_context(|c, _| unsafe { nix_raw::make_attrs(c, out.0, b.0) })545 .expect("attrs initialization should not fail");546 out547 }548 fn new_list<T: Into<Self>>(v: Vec<T>) -> Self {549 todo!()550 }551 fn new_uninit() -> Self {552 let out = with_default_context(|c, es| unsafe { alloc_value(c, es) })553 .expect("value allocation should not fail");554 Self(out)555 }556 pub fn new_str(v: &str) -> Self {557 let s = CString::new(v).expect("string should not contain NULs");558 let out = Self::new_uninit();559 560 with_default_context(|c, _| unsafe { init_string(c, out.0, s.as_ptr()) })561 .expect("string initialization should not fail");562 out563 }564 pub fn new_int(i: i64) -> Self {565 let out = Self::new_uninit();566 with_default_context(|c, _| unsafe { init_int(c, out.0, i) })567 .expect("int initialization should not fail");568 out569 }570 pub fn new_bool(v: bool) -> Self {571 let out = Self::new_uninit();572 with_default_context(|c, _| unsafe { init_bool(c, out.0, v) })573 .expect("bool initialization should not fail");574 out575 }576 577 578 579 580 581 pub fn type_of(&self) -> NixType {582 let ty = with_default_context(|c, _| unsafe { nix_raw::get_type(c, self.0) })583 .expect("get_type should not fail");584 NixType::from_int(ty)585 }586 pub fn to_string(&self) -> Result<String> {587 Ok(self.to_realised_string()?.as_str().to_owned())588 }589 pub fn to_realised_string(&self) -> Result<RealisedString> {590 with_default_context(|c, es| unsafe { nix_raw::string_realise(c, es, self.0, false) })591 .map(RealisedString)592593 594 595 596 597 598 599 600 }601602 pub fn has_field(&self, field: &str) -> Result<bool> {603 let f = init_field_name(field);604 with_default_context(|c, es| unsafe {605 nix_raw::has_attr_byname(c, self.0, es, f.as_ptr().cast())606 })607 }608 609 610 611 pub fn list_fields(&self) -> Result<Vec<String>> {612 if !matches!(self.type_of(), NixType::Attrs) {613 bail!("invalid type: expected attrs");614 }615616 let len = with_default_context(|c, _| unsafe { nix_raw::get_attrs_size(c, self.0) })?;617 let mut out = Vec::with_capacity(len as usize);618619 for i in 0..len {620 let name = with_default_context(|c, es| unsafe {621 nix_raw::get_attr_name_byidx(c, self.0, es, i)622 })?;623 let c = unsafe { CStr::from_ptr(name) };624 out.push(c.to_str().expect("nix field names are utf-8").to_owned());625 }626 Ok(out)627 }628 pub fn get_elem(&self, v: usize) -> Result<Self> {629 if !matches!(self.type_of(), NixType::List) {630 bail!("invalid type: expected list");631 }632 let len =633 with_default_context(|c, _| unsafe { nix_raw::get_list_size(c, self.0) })? as usize;634 if v >= len {635 bail!("oob list get: {v} >= {len}");636 }637638 with_default_context(|c, es| unsafe { nix_raw::get_list_byidx(c, self.0, es, v as u32) })639 .map(Self)640 }641 pub fn attrs_update(self, other: Value) -> Result<Self> {642 let a_fields = self.list_fields()?;643 let b_fields = other.list_fields()?;644 match (a_fields.len(), b_fields.len()) {645 (_, 0) => return Ok(self),646 (0, _) => return Ok(other),647 _ => {}648 }649 let mut out = HashMap::new();650 for f in a_fields.iter() {651 if b_fields.contains(f) {652 break;653 }654 out.insert(f.as_str(), self.get_field(f)?);655 }656 if out.is_empty() {657 658 return Ok(other);659 }660 for f in b_fields.iter() {661 out.insert(f.as_str(), other.get_field(f)?);662 }663 Ok(Self::new_attrs(out))664 }665 pub fn get_field(&self, name: impl AsFieldName) -> Result<Self> {666 if !matches!(self.type_of(), NixType::Attrs) {667 bail!("invalid type: expected attrs");668 }669670 name.as_field_name(|name| {671 with_default_context(|c, es| unsafe {672 nix_raw::get_attr_byname(c, self.0, es, name.as_ptr().cast())673 })674 .map(Self)675 })676 .with_context(|| format!("getting field {:?}", name.to_field_name()))677 }678 pub fn call(&self, v: Value) -> Result<Self> {679 let kind = self680 .functor_kind()681 .ok_or_else(|| anyhow!("can only call function or functor"))?;682683 let function = match kind {684 FunctorKind::Function => self.clone(),685 FunctorKind::Functor => {686 let f = self.get_field("__functor")?;687 assert_eq!(688 f.type_of(),689 NixType::Function,690 "invalid functor encountered"691 );692 f693 }694 };695696 let out = Value::new_uninit();697 with_default_context(|c, es| unsafe {698 nix_raw::value_call(c, es, function.0, v.0, out.0)699 })?;700701 Ok(out)702 }703 pub fn eval(v: &str) -> Result<Self> {704 let s = CString::new(v).expect("expression shouldn't have internal NULs");705 let out = Self::new_uninit();706 with_default_context(|c, es| unsafe {707 expr_eval_from_string(c, es, s.as_ptr(), c"/homeless-shelter".as_ptr(), out.0)708 })?;709 Ok(out)710 }711 pub async fn build(&self, output: &str) -> Result<PathBuf> {712 if !self.is_derivation() {713 bail!("expected derivation to build")714 }715 let output_name = self.get_field("outputName")?.to_string()?;716 let v = if output_name != output {717 let out = self.get_field(output)?;718 if !out.is_derivation() {719 bail!("unknown output: {output}");720 }721 out722 } else {723 self.clone()724 };725 726 let drv_path = tokio::task::spawn_blocking(move || v.get_field("outPath")?.to_string())727 .await728 .expect("should not fail")?;729 Ok(PathBuf::from(drv_path))730 }731 pub fn as_json<T: DeserializeOwned>(&self) -> Result<T> {732 let to_json = Self::eval("builtins.toJSON")?;733 let s = to_json.call(self.clone())?.to_string()?;734 Ok(serde_json::from_str(&s)?)735 }736 pub fn serialized<T: Serialize>(v: &T) -> Result<Self> {737 Self::eval(&nixlike::serialize(v)?)738 }739740 741 742 743 744 745746 fn is_derivation(&self) -> bool {747 if !matches!(self.type_of(), NixType::Attrs) {748 return false;749 }750 let Some(ty) = self.get_field("type").ok() else {751 return false;752 };753 matches!(ty.to_string().as_deref(), Ok("derivation"))754 }755 fn functor_kind(&self) -> Option<FunctorKind> {756 match self.type_of() {757 NixType::Attrs => self758 .has_field("__functor")759 .expect("has_field shouldn't fail for attrs")760 .then_some(FunctorKind::Functor),761 NixType::Function => Some(FunctorKind::Function),762 _ => None,763 }764 }765 pub fn is_function(&self) -> bool {766 self.functor_kind().is_some()767 }768}769770impl From<String> for Value {771 fn from(value: String) -> Self {772 Value::new_str(&value)773 }774}775impl From<bool> for Value {776 fn from(value: bool) -> Self {777 Value::new_bool(value)778 }779}780impl From<&str> for Value {781 fn from(value: &str) -> Self {782 Value::new_str(value)783 }784}785impl<T> From<Vec<T>> for Value786where787 T: Into<Value>,788{789 fn from(value: Vec<T>) -> Self {790 Value::new_list(value)791 }792}793794impl Clone for Value {795 fn clone(&self) -> Self {796 with_default_context(|c, _| unsafe { value_incref(c, self.0) })797 .expect("value incref should not fail");798 Self(self.0)799 }800}801impl Drop for Value {802 fn drop(&mut self) {803 with_default_context(|c, _| unsafe { value_decref(c, self.0) })804 .expect("value drop should not fail");805 }806}807808pub fn init_libraries() {809 unsafe { nix_raw::GC_allow_register_threads() };810811 let mut ctx = NixContext::new();812 ctx.run_in_context(|c| unsafe { nix_raw::libutil_init(c) })813 .expect("util init should not fail");814 ctx.run_in_context(|c| unsafe { nix_raw::libstore_init(c) })815 .expect("store init should not fail");816 ctx.run_in_context(|c| unsafe { nix_raw::libexpr_init(c) })817 .expect("expr init should not fail");818819 nix_logging_cxx::apply_tracing_logger();820}821822#[test_log::test]823fn test_native() -> Result<()> {824 init_libraries();825826 let mut fetch_settings = FetchSettings::new();827 fetch_settings.set(c"warn-dirty", c"false");828829 let manifest = format!("{}/../../", env!("CARGO_MANIFEST_DIR"));830 let (mut r, _) = FlakeReference::new(&manifest, &fetch_settings)?;831 let locked = r.lock(&fetch_settings)?;832 let attrs = locked.get_attrs(&mut FlakeSettings::new()?)?;833834 let builtins = Value::eval("builtins")?;835 assert_eq!(builtins.type_of(), NixType::Attrs);836837 assert_eq!(attrs.type_of(), NixType::Attrs);838 let test_data = nix_go!(attrs.testData);839840 let test_string: String = nix_go_json!(test_data.testString);841 assert_eq!(test_string, "hello");842843 Ok(())844}