Trying to figure out blkid thing. Add --probe flag.

Cargo.lock

@@ -121,9 +121,9 @@ checksum = "7cd812cc2bc1d69d4764bd80df88b4317eaef9e773c75226407d9bc0876b211c"
 
 [[package]]
 name = "either"
-version = "1.16.0"
+version = "1.17.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "91622ff5e7162018101f2fea40d6ebf4a78bbe5a49736a2020649edf9693679e"
+checksum = "9e5e8f6c15a24b9a3ee5efec809ccd006d3b30e8b3bb63c39af737c7f87daa1d"
 
 [[package]]
 name = "equivalent"
@@ -227,9 +227,9 @@ dependencies = [
 
 [[package]]
 name = "libc"
-version = "0.2.186"
+version = "0.2.189"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "68ab91017fe16c622486840e4c83c9a37afeff978bd239b5293d61ece587de66"
+checksum = "3eaf3ede3fee6db1a4c2ee091bf8a8b4dccdc6d17f656fb07896ee72867612f2"
 
 [[package]]
 name = "libm"
@@ -305,7 +305,7 @@ checksum = "6aa2c4e539b869820a2b82e1aef6ff40aa85e65decdd5185e83fb4b1249cd00f"
 
 [[package]]
 name = "nyanit"
-version = "0.1.5"
+version = "0.1.7"
 dependencies = [
  "nix",
  "ratatui",
@@ -514,7 +514,7 @@ checksum = "e7a5d71263a5a7d47b41f6b3f06ba276f10cc18b0931f1799f710578e2309348"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn 3.0.2",
+ "syn 3.0.3",
 ]
 
 [[package]]
@@ -563,9 +563,9 @@ dependencies = [
 
 [[package]]
 name = "syn"
-version = "3.0.2"
+version = "3.0.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "a207d6d6a2b7fc470b80443726053f18a2481b7e1eee970597051596567987a3"
+checksum = "53e9bae58849f64dfa4f5d5ae372c8341f7305f82a3868709269343628b659a3"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -599,7 +599,7 @@ checksum = "43cbfe0cf76104d42a574802844187e84a305e531ed54455f11fbde0f10541cd"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn 3.0.2",
+ "syn 3.0.3",
 ]
 
 [[package]]

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "nyanit"
-version = "0.1.5"
+version = "0.1.7"
 edition = "2024"
 
 [dependencies]
@@ -10,7 +10,6 @@ ratatui-textarea = { version = "0.9", default-features = false, features = ["ter
 regex = "1.1"
 termion = "4"
 
-
 [profile.release]
 lto = true
 debug = false

src/main.rs

@@ -39,9 +39,9 @@ use nyan_system::ExecCommandError;
 const BEE_CAT: &str = include_str!("bee_cat.txt");
 
 //const ROOT_PARTLABEL: &str = "bee-root";
-const ROOT_DEV: &str = "/dev/disk/by-partlabel/bee-root";
+const ROOT_PARTLABEL: &str = "bee-root";
 const TMP_ROOT: &str = "/mnt/bee_root";
-const TMP_ROOT_SNAPS: &str = "/mnt/bee-root/bzz_snaps";
+const TMP_ROOT_SNAPS: &str = "/mnt/bee_root/bzz_snaps";
 const SYS_INIT: &str = "/sbin/init";
 
 // fn bee_root_is_present() -> bool {
@@ -74,9 +74,9 @@ enum NyanitState {
 }
 
 /// Mounting bee-root and chiking if there any snapshots
-fn mount_and_check_snaps() -> Result<NyanitState, ExecCommandError> {
+fn mount_and_check_snaps(bee_root_dev: &str) -> Result<NyanitState, ExecCommandError> {
     // mount bee-root
-    nyan_system::mount_bee_root(TMP_ROOT, ROOT_DEV)?;
+    nyan_system::mount_bee_root(TMP_ROOT, bee_root_dev)?;
 
     // list snaps if any
     let mut snaps_list = nyan_system::list_bzzpss_snaps(TMP_ROOT_SNAPS)?;
@@ -90,23 +90,32 @@ fn mount_and_check_snaps() -> Result<NyanitState, ExecCommandError> {
     Ok(NyanitState::HandOff(TMP_ROOT.into()))
 }
 
-fn init_stage() -> Result<NyanitState, ExecCommandError> {
+fn init_stage() -> Result<(String, NyanitState), ExecCommandError> {
     // mount sysfss
     nyan_system::prepare_env().map_err(|ex| format!("prepare_env error: {}", ex))?;
 
+    // get bee-root device name aka /dev/sda1
+    let bee_root_dev = match nyan_system::find_partlabel(ROOT_PARTLABEL) {
+        Err(_ex) => {
+            return Err(ExecCommandError::Other(nyan_system::list_partlabel().unwrap()))
+        },
+        Ok(v)=>v
+    };
     // check if device is encrypted
-    if nyan_system::bzzpss_dev_encrypted(ROOT_DEV).map_err(|ex| format!("Error checking if device is encrypted: {}", ex))? {
+    if nyan_system::bzzpss_dev_encrypted(&bee_root_dev).map_err(|ex| format!("Error checking if device is encrypted: {}", ex))? {
         // if encrypted return EnterPassword so used get promted for pass
-        return Ok(NyanitState::EnterPassword);
+        return Ok((bee_root_dev, NyanitState::EnterPassword));
     }
 
     // if unencrypted
-    mount_and_check_snaps()
+    let check_res = mount_and_check_snaps(&bee_root_dev)?;
+    Ok((bee_root_dev, check_res))
 }
 
 struct NyanitTUI<'a> {
     exit: bool,
     state: NyanitState,
+    bee_root_dev: String,
     list_state: ListState,
     time_counter: u32,
     textarea: Option<TextArea<'a>>,
@@ -117,6 +126,7 @@ impl NyanitTUI<'_> {
         Ok(Self {
             exit: false,
             state: NyanitState::Init,
+            bee_root_dev: String::new(),
             list_state: ListState::default().with_selected(Some(0)),
             time_counter: 0,
             textarea: None,
@@ -126,7 +136,10 @@ impl NyanitTUI<'_> {
     /// runs the application's main loop until the user quits
     pub fn run(&mut self) -> io::Result<()> {
         self.state = match init_stage() {
-            Ok(st) => st,
+            Ok(res) => {
+                self.bee_root_dev = res.0;
+                res.1
+            },
             Err(ex) => {
                 NyanitState::Error(format!("Error while trying to bzzz-pss things:\n{}", ex))
             },
@@ -276,11 +289,11 @@ impl NyanitTUI<'_> {
                         // try to unlock if enter pressed
                         Event::Key(event::Key::Char('\n')) => {
                             let password = self.textarea.as_ref().unwrap().lines()[0].clone();
-                            match nyan_system::bcachefs_unlock(password, ROOT_DEV) {
+                            match nyan_system::bcachefs_unlock(password, &self.bee_root_dev) {
                                 Ok(unlocked) => {
                                     if unlocked {
                                         // need to check for subvolumes and stuff
-                                        self.state = match mount_and_check_snaps() {
+                                        self.state = match mount_and_check_snaps(&self.bee_root_dev) {
                                             Ok(st) => {
                                                 self.time_counter = 0;
                                                 st

src/nyan_system.rs

@@ -110,6 +110,48 @@ pub fn exec_command(cmd: &str, args: &Vec<&str>) -> Result<(), ExecCommandError>
     }
 }
 
+/// Gets device node aka /dev/sda1 from gpr partition label
+/// blkid -s PARTLABEL --match-token "PARTLABEL=primary"
+pub fn find_partlabel(partlabel: &str) -> Result<String, ExecCommandError> {
+    let out = Command::new("/usr/bin/blkid")
+        .args([
+            "--probe",
+            "-s",
+            "PARTLABEL",
+            "--match-token",
+            format!("\"PARTLABEL={}\"", partlabel).as_str(),
+        ])
+        .output()?;
+    if out.status.success() {
+        let out_text = String::from_utf8(out.stdout)?;
+
+        let split_res = out_text.split_once(':').ok_or_else(|| {
+            ExecCommandError::Other(format!("Unexpected blkid output format: \"{}\"", out_text))
+        })?;
+
+        Ok(split_res.0.to_string())
+    } else {
+        Err(ExecCommandError::Other(format!(
+            "Connot find device with partlabel={} or something other is wrong: \"{}\"",
+            partlabel,
+            String::from_utf8(out.stderr).unwrap_or_else(|_| String::new())
+        )))
+    }
+}
+
+
+pub fn list_partlabel() -> Result<String, ExecCommandError> {
+    let out = Command::new("/usr/bin/blkid")
+        .args(["--probe", "-s", "PARTLABEL"])
+        .output()?;
+    if out.status.success() {
+        let out_text = String::from_utf8(out.stdout)?;
+        Ok(out_text)
+    } else {
+        Err(ExecCommandError::Other("Oh no".to_string()))
+    }
+}
+
 /// exec given command with arguments, returning stdout parsed to UTF-8 string
 /// capturing stderr in case or failure
 // pub fn exec_command_out(cmd: &str, args: &Vec<&str>) -> Result<String, ExecCommandError> {
@@ -224,7 +266,7 @@ pub fn list_bzzpss_snaps(snaps_dir: &str) -> Result<Vec<String>, ExecCommandErro
 // mount -t sysfs none /sys
 // modprobe bcachefs
 pub fn prepare_env() -> Result<(), String> {
-    use nix::mount::{mount, MsFlags};
+    use nix::mount::{MsFlags, mount};
 
     use std::path::Path;