src/runtime/src/config.rs

@@ -1,7 +1,7 @@
 use crate::sd_reader;
 use core::fmt;
 use alloc::{string::FromUtf8Error, string::String, vec::Vec};
-use core_io::{self as io, BufRead, BufReader, Read};
+use core_io::{self as io, BufRead, BufReader, Read, Write};
 
 use libboard_zynq::sdio;
 
@@ -103,4 +103,41 @@ impl Config {
     pub fn read_str<'b>(&self, key: &'b str) -> Result<'b, String> {
         Ok(String::from_utf8(self.read(key)?)?)
     }
+
+    pub fn write_str<'b>(&mut self, key: &str, data: &str) -> Result<'b, ()>{
+        if let Some(fs) = &self.fs {
+            let root_dir = fs.root_dir();
+            let mut file;
+
+            let isascii_and_short =  data.is_ascii() & (data.len() <= 100);
+            let file_path;
+            let config_key_bin = &["/CONFIG/", key, ".BIN"].concat();
+
+            if isascii_and_short {
+                file_path = "/CONFIG.TXT";
+            } else {
+                file_path = config_key_bin;
+            }
+
+            file = match root_dir.create_file(file_path) {
+                Ok(f) => f,
+                Err(_) => root_dir.open_file("/CONFIG.TXT")?
+            };
+
+            file.truncate()?;
+            file.write_all(&[key,"="].concat().as_bytes())?;
+            file.write_all(data.as_bytes())?;
+
+            if isascii_and_short {
+                match root_dir.remove(config_key_bin) {
+                    Ok(_) => {},
+                    Err(_) => {}
+                };
+            }
+
+            Ok(())
+        } else {
+            Err(Error::NoConfig)
+        }
+    }
 }