From 1caca6285427ebd2b4f0eb10f4d1ae3956ff09c0 Mon Sep 17 00:00:00 2001
From: kylo252 <59826753+kylo252@users.noreply.github.com>
Date: Tue, 7 Jun 2022 02:09:58 +0200
Subject: [PATCH] feat(view): ability to centralize view on entering (#1310)

---
 README.md              |  1 +
 doc/nvim-tree-lua.txt  |  6 ++++++
 lua/nvim-tree.lua      | 13 +++++++++++++
 lua/nvim-tree/view.lua |  2 ++
 4 files changed, 22 insertions(+)

diff --git a/README.md b/README.md
index 1f2c4229..67c18ce2 100644
--- a/README.md
+++ b/README.md
@@ -87,6 +87,7 @@ require'nvim-tree'.setup { -- BEGIN_DEFAULT_OPTS
   respect_buf_cwd = false,
   view = {
     adaptive_size = false,
+    centralize_selection = false,
     width = 30,
     height = 30,
     hide_root_folder = false,
diff --git a/doc/nvim-tree-lua.txt b/doc/nvim-tree-lua.txt
index 154ddd49..d867b639 100644
--- a/doc/nvim-tree-lua.txt
+++ b/doc/nvim-tree-lua.txt
@@ -105,6 +105,7 @@ Values may be functions. Warning: this may result in unexpected behaviour.
       respect_buf_cwd = false,
       view = {
         adaptive_size = false,
+        centralize_selection = false,
         width = 30,
         height = 30,
         hide_root_folder = false,
@@ -454,6 +455,11 @@ Window / buffer setup.
     Only works when |nvim-tree.view.side| is `left` or `right`.
       Type: `boolean`, Default: `false`
 
+    *nvim-tree.view.centralize_selection*
+    When entering nvim-tree, reposition the view so that the current node is
+    initially centralized, see |zz|.
+      Type: `boolean`, Default: `false`
+
     *nvim-tree.view.hide_root_folder*
     Hide the path of the current working directory on top of the tree.
       Type: `boolean`, Default: `false`
diff --git a/lua/nvim-tree.lua b/lua/nvim-tree.lua
index 57708fbe..faeeb387 100644
--- a/lua/nvim-tree.lua
+++ b/lua/nvim-tree.lua
@@ -347,6 +347,18 @@ local function setup_autocommands(opts)
   if opts.reload_on_bufenter and not has_watchers then
     create_nvim_tree_autocmd("BufEnter", { pattern = "NvimTree_*", callback = reloaders.reload_explorer })
   end
+
+  if opts.view.centralize_selection then
+    create_nvim_tree_autocmd("BufEnter", {
+      pattern = "NvimTree_*",
+      callback = function()
+        vim.schedule(function()
+          local keys = api.nvim_replace_termcodes("zz", true, false, true)
+          api.nvim_feedkeys(keys, "n", true)
+        end)
+      end,
+    })
+  end
 end
 
 local DEFAULT_OPTS = { -- BEGIN_DEFAULT_OPTS
@@ -366,6 +378,7 @@ local DEFAULT_OPTS = { -- BEGIN_DEFAULT_OPTS
   respect_buf_cwd = false,
   view = {
     adaptive_size = false,
+    centralize_selection = false,
     width = 30,
     height = 30,
     hide_root_folder = false,
diff --git a/lua/nvim-tree/view.lua b/lua/nvim-tree/view.lua
index 21d674c1..e88de7f1 100644
--- a/lua/nvim-tree/view.lua
+++ b/lua/nvim-tree/view.lua
@@ -6,6 +6,7 @@ local events = require "nvim-tree.events"
 
 M.View = {
   adaptive_size = false,
+  centralize_selection = false,
   tabpages = {},
   cursors = {},
   hide_root_folder = false,
@@ -394,6 +395,7 @@ end
 function M.setup(opts)
   local options = opts.view or {}
   M.View.adaptive_size = options.adaptive_size
+  M.View.centralize_selection = options.centralize_selection
   M.View.side = options.side
   M.View.width = options.width
   M.View.height = options.height