Add Pixi support to RunSpec and native executor

- Add pixi field to RunSpec and a builder to set it
- Load pixi recipe from assets and run Pixi setup in native executor
- Introduce include_asset macro to embed assets and expose it
- Extend rfdiffusion to embed its pixi recipe via include_asset
- Update Command::cd to take a Path for robust path handling

Author: lyskov

src/app.rs

@@ -66,8 +66,10 @@ pub enum MountRole {
 pub struct RunSpec {
     pub image: Image,
     pub args: Vec<String>,
-    //pub scratch: Option<PathBuf>,
+
     pub mounts: HashMap<MountRole, String>,
+
+    pub pixi: Option<String>,
 }
 
 impl RunSpec {
@@ -76,6 +78,7 @@ impl RunSpec {
             image: Image(image.into()),
             args,
             mounts: HashMap::new(),
+            pixi: None,
         }
     }
     pub fn scratch(mut self, p: impl Into<String>) -> Self {
@@ -86,6 +89,11 @@ impl RunSpec {
         self.mounts.insert(MountRole::WorkingDir, p.into());
         self
     }
+
+    pub fn pixi(mut self, p: impl Into<String>) -> Self {
+        self.pixi = Some(p.into());
+        self
+    }
 }
 
 impl App {

src/app/rfdiffusion.rs

@@ -1,4 +1,4 @@
-use crate::app::RunSpec;
+use crate::{app::RunSpec, util::include_asset};
 
 pub fn spec(mut app_args: Vec<String>) -> RunSpec {
     app_args.splice(
@@ -11,4 +11,5 @@ pub fn spec(mut app_args: Vec<String>) -> RunSpec {
     RunSpec::new("rosettacommons/rfdiffusion", app_args)
         .scratch("/app/RFdiffusion/schedules")
         .working_dir("/w")
+        .pixi(include_asset!("pixi/rfdiffusion.toml"))
 }

src/executor/native.rs

@@ -1,23 +1,40 @@
+use anyhow::Context;
 use anyhow::Result;
 use anyhow::anyhow;
+use home::home_dir;
 use yansi::Paint;
 
+use crate::util::Command;
 use crate::util::ensure_dir_signature;
 use crate::{ContainerEngine, app::RunSpec, executor::Executor};
 
 impl Executor {
-    pub(super) fn execute_native(&self, _spec: RunSpec) -> Result<()> {
+    pub(super) fn execute_native(&self, spec: RunSpec) -> Result<()> {
         assert!(matches!(self.engine, ContainerEngine::None));
 
-        Self::check_if_pixi_is_installed()?;
-
-        let pixi_evn_root = self.working_dir.join(format!("{}.pixi", self.app));
+        let recipe = spec
+            .pixi
+            .with_context(|| format!("Pixi recipe for app '{}' was not found", self.app))?;
 
-        ensure_dir_signature(&pixi_evn_root, &["qwe", &_spec.image.0], |_d| Ok(()))?;
-
-        //write_signature(&self.root, &hash.to_string())?;
+        Self::check_if_pixi_is_installed()?;
 
-        todo!("ContainerEngine::None")
+        //let pixi_evn_root = self.working_dir.join(format!("{}.pixi", self.app));
+        let pixi_evn_root = home_dir()
+            .unwrap()
+            .join(format!(".cache/rosettacommons/rc/native/{}.pixi", self.app));
+
+        ensure_dir_signature(&pixi_evn_root, &[&spec.image.0, &recipe], |d| {
+            std::fs::write(d.join("pixi.toml"), &recipe)?;
+            Command::new("pixi")
+                .cd(d)
+                .arg("run")
+                .arg("setup")
+                .live()
+                .exec()?;
+            Ok(())
+        })?;
+
+        Ok(())
     }
 
     /// Check if Pixi is installed

src/util.rs

@@ -29,3 +29,10 @@ pub fn sleep(message: impl AsRef<str>, seconds: usize) {
     print!("\r{:width$}\r", "", width = max_len);
     io::stdout().flush().unwrap();
 }
+
+macro_rules! include_asset {
+    ($path:literal) => {
+        include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/assets/", $path))
+    };
+}
+pub(crate) use include_asset;

src/util/command.rs

@@ -1,4 +1,4 @@
-use std::marker::PhantomData;
+use std::path::Path;
 use std::sync::mpsc::Sender;
 use std::{
     fmt::{self},
@@ -67,7 +67,7 @@ impl Command {
         self
     }
 
-    pub fn cd(mut self, path: impl AsRef<str>) -> Self {
+    pub fn cd(mut self, path: impl AsRef<Path>) -> Self {
         let path = std::fs::canonicalize(path.as_ref()).unwrap();
         self.cd = Some(path);
         self