yazi: add xonsh integration

nix-community · Apr 3, 2024 · aa56eda · aa56eda
1 parent 5d688e1
commit aa56eda
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 0 deletions.
diff --git a/modules/programs/yazi.nix b/modules/programs/yazi.nix
@@ -39,6 +39,21 @@ let
       rm -fp $tmp
     }
   '';
+
+  xonshIntegration = ''
+    def __yazi_init():
+      def ya(args):
+        tmp = $(mktemp -t "yazi-cwd.XXXXX")
+        $[yazi @(args) @(f"--cwd-file={tmp}")]
+        cwd = fp"{tmp}".read_text()
+        if cwd != "" and cwd != $PWD:
+          xonsh.dirstack.cd(cwd)
+        $[rm -f -- @(tmp)]
+
+      aliases['ya'] = ya
+    __yazi_init()
+    del __yazi_init
+  '';
 in {
   meta.maintainers = [ maintainers.xyenon ];
 
@@ -60,6 +75,8 @@ in {
 
     enableNushellIntegration = mkEnableOption "Nushell integration";
 
+    enableXonshIntegration = mkEnableOption "Xonsh integration";
+
     keymap = mkOption {
       type = tomlFormat.type;
       default = { };
@@ -150,6 +167,9 @@ in {
     programs.nushell.extraConfig =
       mkIf cfg.enableNushellIntegration nushellIntegration;
 
+    programs.xonsh.rcFiles."yazi.xsh".text =
+      mkIf cfg.enableXonshIntegration xonshIntegration;
+
     xdg.configFile = {
       "yazi/keymap.toml" = mkIf (cfg.keymap != { }) {
         source = tomlFormat.generate "yazi-keymap" cfg.keymap;

diff --git a/tests/modules/programs/yazi/default.nix b/tests/modules/programs/yazi/default.nix
@@ -4,4 +4,5 @@
   yazi-zsh-integration-enabled = ./zsh-integration-enabled.nix;
   yazi-fish-integration-enabled = ./fish-integration-enabled.nix;
   yazi-nushell-integration-enabled = ./nushell-integration-enabled.nix;
+  yazi-xonsh-integration-enabled = ./xonsh-integration-enabled.nix;
 }
diff --git a/tests/modules/programs/yazi/xonsh-integration-enabled.nix b/tests/modules/programs/yazi/xonsh-integration-enabled.nix
@@ -0,0 +1,31 @@
+{ ... }:
+
+let
+  shellIntegration = ''
+    def __yazi_init():
+      def ya(args):
+        tmp = $(mktemp -t "yazi-cwd.XXXXX")
+        $[yazi @(args) @(f"--cwd-file={tmp}")]
+        cwd = fp"{tmp}".read_text()
+        if cwd != "" and cwd != $PWD:
+          xonsh.dirstack.cd(cwd)
+        $[rm -f -- @(tmp)]
+
+      aliases['ya'] = ya
+    __yazi_init()
+    del __yazi_init
+  '';
+in {
+  programs.xonsh.enable = true;
+
+  programs.yazi = {
+    enable = true;
+    enableXonshIntegration = true;
+  };
+
+  test.stubs.yazi = { };
+
+  nmt.script = ''
+    assertFileContains home-files/.config/xonsh/rc.d/yazi.xsh '${shellIntegration}'
+  '';
+}