feat: add shell task

silitics · Nov 29, 2023 · 3306eec · 3306eec
1 parent 33374b6
commit 3306eec
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/rugpi-bakery/Cargo.toml b/crates/rugpi-bakery/Cargo.toml
@@ -12,6 +12,7 @@ anyhow = "1.0.71"
 camino = { version = "1.1.4", features = ["serde"] }
 clap = { version = "4.3.8", features = ["derive"] }
 hex = "0.4.3"
+nix = { version = "0.27.1", features = ["process"] }
 rugpi-common = { path = "../rugpi-common" }
 serde = { version = "1.0.171", features = ["derive", "rc"] }
 sha1 = "0.10.5"

diff --git a/crates/rugpi-bakery/src/main.rs b/crates/rugpi-bakery/src/main.rs
@@ -1,4 +1,7 @@
-use std::path::PathBuf;
+use std::{
+    ffi::{CStr, CString},
+    path::PathBuf,
+};
 
 use clap::Parser;
 use rugpi_common::Anyhow;
@@ -31,6 +34,8 @@ pub enum Task {
     Customize(CustomizeTask),
     /// Bake a final image for distribution.
     Bake(BakeTask),
+    /// Spawn a shell in the Rugpi Bakery Docker container.
+    Shell,
 }
 
 fn main() -> Anyhow<()> {
@@ -45,6 +50,10 @@ fn main() -> Anyhow<()> {
         Task::Bake(task) => {
             bake::run(&args, task)?;
         }
+        Task::Shell => {
+            let zsh_prog = CString::new("/bin/zsh")?;
+            nix::unistd::execv::<&CStr>(&zsh_prog, &[])?;
+        }
     }
     Ok(())
 }