summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/lib.rs14
-rw-r--r--src/plymouthd.rs43
-rw-r--r--src/theme.rs104
3 files changed, 161 insertions, 0 deletions
diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..54bee5b
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,14 @@
+mod plymouthd;
+mod theme;
+
+pub use plymouthd::Plymouthd;
+pub use theme::Theme;
+
+use std::io;
+
+fn ini_to_io_err(e: ini::ini::Error) -> io::Error {
+ match e {
+ ini::ini::Error::Io(x) => x,
+ ini::ini::Error::Parse(x) => io::Error::new(io::ErrorKind::InvalidData, x),
+ }
+} \ No newline at end of file
diff --git a/src/plymouthd.rs b/src/plymouthd.rs
new file mode 100644
index 0000000..9e991e6
--- /dev/null
+++ b/src/plymouthd.rs
@@ -0,0 +1,43 @@
+use std::path::Path;
+use std::io;
+
+use ini::Ini;
+
+pub struct Plymouthd {
+ conf: Ini,
+}
+
+impl Plymouthd {
+ pub fn from_ini(conf: Ini) -> Self {
+ Self { conf }
+ }
+
+ pub fn from_file<P: AsRef<Path>>(filename: P) -> io::Result<Self> {
+ let conf = Ini::load_from_file(filename).map_err(|e| super::ini_to_io_err(e))?;
+ Ok(Self { conf })
+ }
+
+ pub fn default() -> io::Result<Self> {
+ Self::from_file("/etc/plymouth/plymouthd.conf")
+ }
+
+ pub fn current_theme(&self) -> Option<&str> {
+ self.conf
+ .section(Some("Daemon"))
+ .and_then(|s| s.get("Theme"))
+ }
+}
+
+#[cfg(test)]
+mod tests {
+ use super::*;
+
+ #[test]
+ fn default_theme_from_ini() {
+ let mut ini = Ini::new();
+ ini.with_section(Some("Daemon")).set("Theme", "spinner");
+
+ let conf = Plymouthd::from_ini(ini);
+ assert!(conf.current_theme().unwrap() == "spinner");
+ }
+}
diff --git a/src/theme.rs b/src/theme.rs
new file mode 100644
index 0000000..2ac93f8
--- /dev/null
+++ b/src/theme.rs
@@ -0,0 +1,104 @@
+use std::fs;
+use std::io;
+use std::path::{Path, PathBuf};
+
+use ini::Ini;
+
+pub const THEMES_DIRECTORY: &str = "/usr/share/plymouth/themes";
+pub const THEME_FILE_EXTENSION: &str = "plymouth";
+
+#[derive(Debug)]
+pub struct Theme {
+ path: PathBuf,
+ name: String,
+ description: String,
+ module_name: String,
+}
+
+fn find_theme_config_file(path: &Path) -> io::Result<PathBuf> {
+ fs::read_dir(&path)?
+ .filter_map(|p| p.ok())
+ .map(|e| e.path())
+ .filter(|p| match p.extension() {
+ Some(ext) => ext == THEME_FILE_EXTENSION,
+ None => false,
+ })
+ .nth(0)
+ .map_or(
+ Err(io::Error::new(
+ io::ErrorKind::NotFound,
+ "No *.plymouth file found in directory",
+ )),
+ |e| Ok(e),
+ )
+}
+
+impl Theme {
+ pub fn from_ini<P: AsRef<Path>>(path: P, ini: Ini) -> Self {
+ let theme_section = ini
+ .section(Some("Plymouth Theme"))
+ .expect("could not find [Plymouth Theme] section");
+ let name = theme_section.get("Name").unwrap_or("").into();
+ let description = theme_section.get("Description").unwrap_or("").into();
+ let module_name = theme_section.get("ModuleName").unwrap_or("").into();
+
+ Theme {
+ path: PathBuf::from(path.as_ref()),
+ name,
+ description,
+ module_name,
+ }
+ }
+
+ pub fn from_path<P: AsRef<Path>>(path: P) -> io::Result<Self> {
+ let config_path = find_theme_config_file(path.as_ref())?;
+
+ let ini = Ini::load_from_file(&config_path).map_err(|e| super::ini_to_io_err(e))?;
+
+ Ok(Self::from_ini(path, ini))
+ }
+
+ pub fn from_name(name: &str) -> io::Result<Self> {
+ let path = Path::new(THEMES_DIRECTORY).join(name);
+ Self::from_path(path)
+ }
+
+ pub fn list() -> io::Result<impl Iterator<Item = String>> {
+ fs::read_dir(THEMES_DIRECTORY).and_then(|d| {
+ Ok(d.filter_map(|p| p.ok())
+ .filter_map(|e| e.file_name().into_string().ok()))
+ })
+ }
+}
+
+#[cfg(test)]
+mod tests {
+ use super::*;
+
+ #[test]
+ fn from_ini() {
+ let mut ini = Ini::new();
+ ini.with_section(Some("Plymouth Theme"))
+ .set("Name", "foobar")
+ .set("Description", "A foo and a bar.")
+ .set("ModuleName", "script");
+
+ let theme = Theme::from_ini("/dev/null", ini);
+ assert!(theme.name == "foobar");
+ assert!(theme.description == "A foo and a bar.");
+ assert!(theme.module_name == "script");
+ }
+
+ #[test]
+ fn from_ini_default_values() {
+ let mut ini = Ini::new();
+ // rust-ini does not support creating an empty section, so we add
+ // a fake foo=bar entry to create the section
+ ini.with_section(Some("Plymouth Theme")).set("foo", "bar");
+
+ let theme = Theme::from_ini("/dev/null", ini);
+ assert!(theme.name == "");
+ assert!(theme.description == "");
+ assert!(theme.module_name == "");
+ }
+}