From 334b18e3092dba9121c2ad9b979a6a175cc80c48 Mon Sep 17 00:00:00 2001
From: Martin Larsson <larssonmartin1998@gmail.com>
Date: Sun, 23 Feb 2025 02:09:55 +0000
Subject: [PATCH] Fix so zen mode doesnt exit when leaping in normal mode.

---
 nvim/lua/leap_keymap_handler.lua | 10 +++++++++-
 nvim/lua/plugs/snacks.lua        |  8 +++++++-
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/nvim/lua/leap_keymap_handler.lua b/nvim/lua/leap_keymap_handler.lua
index 55b791c..609d77f 100644
--- a/nvim/lua/leap_keymap_handler.lua
+++ b/nvim/lua/leap_keymap_handler.lua
@@ -12,9 +12,17 @@ local function leap_in_current_buffer()
     })
 end
 
+local function normal_mode_leap()
+    if _G["snacks_zen_mode"] then
+        leap_in_current_buffer()
+    else
+        leap_across_windows()
+    end
+end
+
 function M.set_leap_keymapping()
     require("utils").foreach({
-        { "n", "m", leap_across_windows },
+        { "n", "m", normal_mode_leap },
         { "v", "m", leap_in_current_buffer },
         { "o", "m", leap_in_current_buffer }
     }, function(mapping)
diff --git a/nvim/lua/plugs/snacks.lua b/nvim/lua/plugs/snacks.lua
index 1247d8a..418b207 100644
--- a/nvim/lua/plugs/snacks.lua
+++ b/nvim/lua/plugs/snacks.lua
@@ -37,7 +37,13 @@ return {
         },
         zen = {
             enabled = true,
-            toggles = { dim = false, }
+            toggles = { dim = false, },
+            on_open = function()
+                _G["snacks_zen_mode"] = true
+            end,
+            on_close = function()
+                _G["snacks_zen_mode"] = false
+            end,
         },
     },
     keys = {