From 6e08d7952362f3294e5b0983e0c1f7505a05a594 Mon Sep 17 00:00:00 2001
From: Daniil Tsivinsky <daniil@tsivinsky.com>
Date: Thu, 21 Dec 2023 15:38:19 +0300
Subject: [PATCH] neovim: focus undotree window when opening

---
 neovim/.config/nvim/after/plugin/undotree.lua | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/neovim/.config/nvim/after/plugin/undotree.lua b/neovim/.config/nvim/after/plugin/undotree.lua
index e63a278..cd31f6c 100644
--- a/neovim/.config/nvim/after/plugin/undotree.lua
+++ b/neovim/.config/nvim/after/plugin/undotree.lua
@@ -1,4 +1,7 @@
 vim.opt.undodir = os.getenv("HOME") .. "/.vim/undodir"
 vim.opt.undofile = true
 
-vim.keymap.set("n", "<leader>u", vim.cmd.UndotreeToggle)
+vim.keymap.set("n", "<leader>u", function()
+  vim.cmd.UndotreeToggle()
+  vim.cmd.UndotreeFocus()
+end)