1 files changed, 28 insertions, 13 deletions

diff --git a/src/dirs.rs b/src/dirs.rs
index 5ebeaa2..2fa6e50 100644
--- a/src/dirs.rs
+++ b/src/dirs.rs
@@ -49,7 +49,7 @@ pub fn db_file(server: &str, email: &str) -> std::path::PathBuf {
     let server =
         percent_encoding::percent_encode(server.as_bytes(), INVALID_PATH)
             .to_string();
-    cache_dir().join(format!("{}:{}.json", server, email))
+    cache_dir().join(format!("{server}:{email}.json"))
 }
 
 #[must_use]
@@ -79,32 +79,47 @@ pub fn socket_file() -> std::path::PathBuf {
 
 #[must_use]
 fn config_dir() -> std::path::PathBuf {
-    let project_dirs = directories::ProjectDirs::from("", "", "rbw").unwrap();
+    let project_dirs =
+        directories::ProjectDirs::from("", "", &profile()).unwrap();
     project_dirs.config_dir().to_path_buf()
 }
 
 #[must_use]
 fn cache_dir() -> std::path::PathBuf {
-    let project_dirs = directories::ProjectDirs::from("", "", "rbw").unwrap();
+    let project_dirs =
+        directories::ProjectDirs::from("", "", &profile()).unwrap();
     project_dirs.cache_dir().to_path_buf()
 }
 
 #[must_use]
 fn data_dir() -> std::path::PathBuf {
-    let project_dirs = directories::ProjectDirs::from("", "", "rbw").unwrap();
+    let project_dirs =
+        directories::ProjectDirs::from("", "", &profile()).unwrap();
     project_dirs.data_dir().to_path_buf()
 }
 
 #[must_use]
 fn runtime_dir() -> std::path::PathBuf {
-    let project_dirs = directories::ProjectDirs::from("", "", "rbw").unwrap();
-    match project_dirs.runtime_dir() {
-        Some(dir) => dir.to_path_buf(),
-        None => format!(
-            "{}/rbw-{}",
-            std::env::temp_dir().to_string_lossy(),
-            nix::unistd::getuid().as_raw()
-        )
-        .into(),
+    let project_dirs =
+        directories::ProjectDirs::from("", "", &profile()).unwrap();
+    project_dirs.runtime_dir().map_or_else(
+        || {
+            format!(
+                "{}/{}-{}",
+                std::env::temp_dir().to_string_lossy(),
+                &profile(),
+                rustix::process::getuid().as_raw()
+            )
+            .into()
+        },
+        std::path::Path::to_path_buf,
+    )
+}
+
+#[must_use]
+pub fn profile() -> String {
+    match std::env::var("RBW_PROFILE") {
+        Ok(profile) if !profile.is_empty() => format!("rbw-{profile}"),
+        _ => "rbw".to_string(),
     }
 }