From 29f3abf2860ff7a726255ede7b5caeb58f45b656 Mon Sep 17 00:00:00 2001
From: Cian Hughes <chughes000@gmail.com>
Date: Mon, 16 Feb 2026 11:15:04 +0000
Subject: [PATCH] Minor plugin upgrades

---
 lua/config/keys.lua    | 57 +++++++++++++++++++-----------------------
 lua/plugins/editor.lua |  8 +++++-
 2 files changed, 33 insertions(+), 32 deletions(-)

diff --git a/lua/config/keys.lua b/lua/config/keys.lua
index c0ce705..875686f 100644
--- a/lua/config/keys.lua
+++ b/lua/config/keys.lua
@@ -27,6 +27,28 @@ return {
         { "<leader>p", group = "[P]recognition", icon = "󰬯" },
         { "<leader>?", group = "[?] Cheatsheet", icon = "󰧹" },
     },
+    globals = {
+        -- Window Navigation
+        { "<A-h>", "<C-w>h", desc = "Window Left" },
+        { "<A-j>", "<C-w>j", desc = "Window Down" },
+        { "<A-k>", "<C-w>k", desc = "Window Up" },
+        { "<A-l>", "<C-w>l", desc = "Window Right" },
+        -- Window Resizing
+        { "<A-=>", "<C-w>+", desc = "Resize Increase Height" },
+        { "<A-->", "<C-w>-", desc = "Resize Decrease Height" },
+        { "<A-.>", "<C-w>>", desc = "Resize Increase Width" },
+        { "<A-,>", "<C-w><", desc = "Resize Decrease Width" },
+        -- Splits
+        { "<A-n>", "<C-w>s", desc = "Split Window Horizontal" },
+        { "<A-;>", "<C-w>x", desc = "Swap Window" },
+        { "<A-q>", ":q<CR>", desc = "Close Window" },
+        -- Misc
+        { "<Esc>", "<cmd>nohlsearch<CR>", desc = "Clear Highlight" },
+        { "<leader>dd", vim.diagnostic.open_float, desc = "Show [D]iagnostics" },
+    },
+    atone = {
+        { "U", vim.api.nvim_command("Atone toggle"), desc = "[U]ndotree" },
+    },
     gitsigns = {
         {
             "<leader>gs",
@@ -89,25 +111,7 @@ return {
             expr = true,
         },
     },
-    globals = {
-        -- Window Navigation
-        { "<A-h>", "<C-w>h", desc = "Window Left" },
-        { "<A-j>", "<C-w>j", desc = "Window Down" },
-        { "<A-k>", "<C-w>k", desc = "Window Up" },
-        { "<A-l>", "<C-w>l", desc = "Window Right" },
-        -- Window Resizing
-        { "<A-=>", "<C-w>+", desc = "Resize Increase Height" },
-        { "<A-->", "<C-w>-", desc = "Resize Decrease Height" },
-        { "<A-.>", "<C-w>>", desc = "Resize Increase Width" },
-        { "<A-,>", "<C-w><", desc = "Resize Decrease Width" },
-        -- Splits
-        { "<A-n>", "<C-w>s", desc = "Split Window Horizontal" },
-        { "<A-;>", "<C-w>x", desc = "Swap Window" },
-        { "<A-q>", ":q<CR>", desc = "Close Window" },
-        -- Misc
-        { "<Esc>", "<cmd>nohlsearch<CR>", desc = "Clear Highlight" },
-        { "<leader>dd", vim.diagnostic.open_float, desc = "Show [D]iagnostics" },
-    },
+
     harpoon = {
         {
             "<leader>ha",
@@ -162,25 +166,19 @@ return {
     lazygit = {
         {
             "<leader>lg",
-            function()
-                vim.api.nvim_command("lua Snacks.lazygit()")
-            end,
+            Snacks.lazygit,
             desc = "[L]azy[G]it",
             mode = "n",
         },
         {
             "<leader>ll",
-            function()
-                vim.api.nvim_command("lua Snacks.lazygit.log()")
-            end,
+            Snacks.lazygit.log,
             desc = "[L]azyGit [L]og",
             mode = "n",
         },
         {
             "<leader>lf",
-            function()
-                vim.api.nvim_command("lua Snacks.lazygit.log_file()")
-            end,
+            Snacks.lazygit.log_file,
             desc = "[L]azyGit [F]ile Log",
             mode = "n",
         },
@@ -507,7 +505,4 @@ return {
             desc = "Toggle Rainbow Delimiters",
         },
     },
-    undotree = {
-        { "U", vim.cmd.UndotreeToggle, desc = "[U]ndotree" },
-    },
 }
diff --git a/lua/plugins/editor.lua b/lua/plugins/editor.lua
index 36549d9..3ad28d0 100644
--- a/lua/plugins/editor.lua
+++ b/lua/plugins/editor.lua
@@ -108,6 +108,12 @@ return {
     { -- Undo tree
         "mbbill/undotree",
         event = "VeryLazy",
-        keys = require("config.keys").undotree,
+    },
+    {
+        "XXiaoA/atone.nvim",
+        event = "VeryLazy",
+        cmd = "Atone",
+        keys = require("config.keys").atone,
+        opts = {},
     },
 }