om health: Shell dotfiles are Nix-managed

Cargo.toml

@@ -62,6 +62,7 @@ uuid = { version = "1.3.0", features = ["serde", "v4", "js"] }
 which = { version = "4.4.2" }
 clap_complete = "4.5.0"
 whoami = "1.5.2"
+dirs = "5.0.1"
 
 [profile.release]
 strip = true    # Automatically strip symbols from the binary.

crates/nix_health/Cargo.toml

@@ -33,3 +33,4 @@ which = { workspace = true }
 bytesize = { workspace = true }
 semver = { workspace = true }
 serde_repr = { workspace = true }
+dirs = { workspace = true }

crates/nix_health/src/check/mod.rs

@@ -5,5 +5,6 @@ pub mod flake_enabled;
 pub mod max_jobs;
 pub mod min_nix_version;
 pub mod rosetta;
+pub mod shell_configurations;
 pub mod system;
 pub mod trusted_users;

crates/nix_health/src/check/shell_configurations.rs

@@ -0,0 +1,97 @@
+use nix_rs::info;
+use serde::{Deserialize, Serialize};
+use std::collections::HashMap;
+use std::path::Path;
+
+use crate::traits::{Check, CheckResult, Checkable};
+
+#[derive(Debug, Serialize, Deserialize, Clone, Hash, Eq, PartialEq)]
+#[serde(rename_all = "lowercase")]
+pub enum Shell {
+    Zsh,
+    Bash,
+    #[serde(other)]
+    Unknown,
+}
+
+impl Shell {
+    fn from_path(path: &str) -> Self {
+        let shell_name = Path::new(path)
+            .file_name()
+            .and_then(|name| name.to_str())
+            .unwrap_or("");
+
+        match shell_name {
+            "zsh" => Shell::Zsh,
+            "bash" => Shell::Bash,
+            _ => Shell::Unknown,
+        }
+    }
+}
+
+#[derive(Debug, Serialize, Deserialize, Clone)]
+#[serde(default)]
+pub struct ShellConfigurations {
+    dotfiles: HashMap<Shell, Vec<String>>,
+}
+
+impl Default for ShellConfigurations {
+    fn default() -> Self {
+        let mut dotfiles = HashMap::new();
+        dotfiles.insert(Shell::Zsh, vec![".zshrc".to_string()]);
+        dotfiles.insert(
+            Shell::Bash,
+            vec![
+                ".bashrc".to_string(),
+                ".bash_profile".to_string(),
+                ".profile".to_string(),
+            ],
+        );
+        Self { dotfiles }
+    }
+}
+
+impl ShellConfigurations {
+    fn check_shell_configuration(&self) -> bool {
+        std::env::var("SHELL")
+            .ok()
+            .map(|shell| {
+                let shell_name = Shell::from_path(&shell);
+                self.are_shell_dotfiles_nix_managed(&shell_name)
+            })
+            .unwrap_or(false)
+    }
+
+    fn are_shell_dotfiles_nix_managed(&self, shell: &Shell) -> bool {
+        dirs::home_dir()
+            .map(|home| {
+                self.dotfiles.get(shell).map_or(false, |shell_files| {
+                    shell_files.iter().all(|dotfile| {
+                        std::fs::canonicalize(home.join(dotfile))
+                            .map(|canonical_path| canonical_path.starts_with("/nix/store/"))
+                            .unwrap_or(false)
+                    })
+                })
+            })
+            .unwrap_or(false)
+    }
+}
+
+impl Checkable for ShellConfigurations {
+    fn check(&self, _: &info::NixInfo, _: Option<&nix_rs::flake::url::FlakeUrl>) -> Vec<Check> {
+        let check = Check {
+            title: "Shell Configurations".to_string(),
+            info: "Dotfiles managed by Nix".to_string(),
+            result: if self.check_shell_configuration() {
+                CheckResult::Green
+            } else {
+                CheckResult::Red {
+                    msg: "Shell Configurations are not managed by Nix".into(),
+                    suggestion: "Manage shell (zsh, bash) configurations through https://github.com/juspay/nixos-unified-template".into(),
+                }
+            },
+            required: false,
+        };
+        vec![check]
+    }
+}

crates/nix_health/src/lib.rs

@@ -18,7 +18,7 @@ use traits::Check;
 
 use self::check::{
     caches::Caches, flake_enabled::FlakeEnabled, max_jobs::MaxJobs, min_nix_version::MinNixVersion,
-    rosetta::Rosetta, trusted_users::TrustedUsers,
+    rosetta::Rosetta, shell_configurations::ShellConfigurations, trusted_users::TrustedUsers,
 };
 
 /// Nix Health check information for user's install
@@ -35,6 +35,7 @@ pub struct NixHealth {
     pub trusted_users: TrustedUsers,
     pub rosetta: Rosetta,
     pub direnv: Direnv,
+    pub shell_configurations: ShellConfigurations,
 }
 
 impl<'a> IntoIterator for &'a NixHealth {
@@ -52,6 +53,7 @@ impl<'a> IntoIterator for &'a NixHealth {
             &self.caches,
             &self.trusted_users,
             &self.direnv,
+            &self.shell_configurations,
         ];
         items.into_iter()
     }