From 2dcb61d396b45f10d9e0621a7358b361f94323ff Mon Sep 17 00:00:00 2001
From: happysalada <raphael@megzari.com>
Date: Mon, 3 Apr 2023 09:48:47 -0400
Subject: [PATCH] atuin: enable nushell integration

---
 modules/programs/atuin.nix | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/modules/programs/atuin.nix b/modules/programs/atuin.nix
index c427866f8..b5e3c6ca6 100644
--- a/modules/programs/atuin.nix
+++ b/modules/programs/atuin.nix
@@ -85,6 +85,14 @@ in {
         of options.
       '';
     };
+
+    enableNushellIntegration = mkOption {
+      default = true;
+      type = types.bool;
+      description = ''
+        Whether to enable Nushell integration.
+      '';
+    };
   };
 
   config = let flagsStr = escapeShellArgs cfg.flags;
@@ -114,5 +122,18 @@ in {
     programs.fish.interactiveShellInit = mkIf cfg.enableFishIntegration ''
       ${cfg.package}/bin/atuin init fish ${flagsStr} | source
     '';
+
+    programs.nushell = mkIf cfg.enableNushellIntegration {
+      extraEnv = ''
+        let atuin_cache = "${config.xdg.cacheHome}/atuin"
+        if not ($atuin_cache | path exists) {
+          mkdir $atuin_cache
+        }
+        ${cfg.package}/bin/atuin init nu | save --force ${config.xdg.cacheHome}/atuin/init.nu
+      '';
+      extraConfig = ''
+        source ${config.xdg.cacheHome}/atuin/init.nu
+      '';
+    };
   };
 }