feat(shell): add dynamic disk display and image size for neofetch

zakuciael · Jan 2, 2025 · dc73bb1 · dc73bb1
1 parent c8685b6
commit dc73bb1
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/modules/shell/neofetch.nix b/modules/shell/neofetch.nix
@@ -7,15 +7,25 @@
   ...
 }:
 let
-  inherit (lib) mkIf mkEnableOption;
+  inherit (lib)
+    mkIf
+    mkEnableOption
+    attrNames
+    concatMapStringsSep
+    length
+    ;
 
   cfg = config.modules.shell.neofetch;
   imageFile = dotfiles.neofetch."nixos.png".source;
   templateFile = dotfiles.neofetch."config.conf".source;
+
+  disks = attrNames config.fileSystems;
+  lines = 20 + (length disks);
+  lineHeight = 20;
   configFile = pkgs.runCommand "config.conf" { } ''
     substitute "${templateFile}" "$out" \
-      --replace-fail "{{% DISK_SHOW %}}" "'/'" \
-      --replace-fail "{{% IMAGE_SIZE %}}" "400px" \
+      --replace-fail "{{% DISK_SHOW %}}" "${concatMapStringsSep " " (disk: "'${disk}'") disks}" \
+      --replace-fail "{{% IMAGE_SIZE %}}" "${builtins.toString (lines * lineHeight)}px" \
       --replace-fail "{{% IMAGE_SOURCE %}}" "${imageFile}"
   '';
   neofetchPkg = pkgs.neofetch.overrideAttrs {