From 036c73070504c462c499099747cf8b86d4bf915c Mon Sep 17 00:00:00 2001
From: Marcelo Jacobus <marcelo.jacobus@gmail.com>
Date: Thu, 20 Jun 2024 14:17:21 -0300
Subject: [PATCH] Replace treesitter-context plugin

---
 lua/config/lazy.lua                |  1 +
 lua/plugins/treesiter-context.lua  | 22 ----------------------
 lua/plugins/treesitter-context.lua | 20 ++++++++++++++++++++
 3 files changed, 21 insertions(+), 22 deletions(-)
 delete mode 100644 lua/plugins/treesiter-context.lua
 create mode 100644 lua/plugins/treesitter-context.lua

diff --git a/lua/config/lazy.lua b/lua/config/lazy.lua
index 813ce7e..3052cae 100644
--- a/lua/config/lazy.lua
+++ b/lua/config/lazy.lua
@@ -22,6 +22,7 @@ require("lazy").setup({
     { import = "lazyvim.plugins.extras.editor.aerial" }, -- <leader>ss list constants
     { import = "lazyvim.plugins.extras.editor.dial" }, -- like speadating
     { import = "lazyvim.plugins.extras.editor.leap" }, -- like easy motion
+    { import = "lazyvim.plugins.extras.ui.treesitter-context" },
     -- import/override with your plugins
     { import = "plugins" },
   },
diff --git a/lua/plugins/treesiter-context.lua b/lua/plugins/treesiter-context.lua
deleted file mode 100644
index 70c9462..0000000
--- a/lua/plugins/treesiter-context.lua
+++ /dev/null
@@ -1,22 +0,0 @@
--- keep the function name at the top of the file
-return {
-  {
-    "nvim-treesitter/nvim-treesitter-context",
-    config = function()
-      require("nvim-treesitter.configs").setup({
-        enable = true, -- Enable this plugin (Can be enabled/disabled later via commands)
-        max_lines = 3, -- How many lines the window should span. Values <= 0 mean no limit.
-        min_window_height = 0, -- Minimum editor window height to enable context. Values <= 0 mean no limit.
-        line_numbers = true,
-        multiline_threshold = 20, -- Maximum number of lines to show for a single context
-        trim_scope = "outer", -- Which context lines to discard if `max_lines` is exceeded. Choices: 'inner', 'outer'
-        mode = "cursor", -- Line used to calculate context. Choices: 'cursor', 'topline'
-        -- Separator between context and content. Should be a single character string, like '-'.
-        -- When separator is set, the context will only show up when there are at least 2 lines above cursorline.
-        separator = nil,
-        zindex = 20, -- The Z-index of the context window
-        on_attach = nil, -- (fun(buf: integer): boolean) return false to disable attaching
-      })
-    end,
-  },
-}
diff --git a/lua/plugins/treesitter-context.lua b/lua/plugins/treesitter-context.lua
new file mode 100644
index 0000000..df1f36c
--- /dev/null
+++ b/lua/plugins/treesitter-context.lua
@@ -0,0 +1,20 @@
+return {
+  "nvim-treesitter/nvim-treesitter-context",
+  event = "LazyFile",
+  opts = { mode = "cursor", max_lines = 3 },
+  keys = {
+    {
+      "<leader>ut",
+      function()
+        local tsc = require("treesitter-context")
+        tsc.toggle()
+        if LazyVim.inject.get_upvalue(tsc.toggle, "enabled") then
+          LazyVim.info("Enabled Treesitter Context", { title = "Option" })
+        else
+          LazyVim.warn("Disabled Treesitter Context", { title = "Option" })
+        end
+      end,
+      desc = "Toggle Treesitter Context",
+    },
+  },
+}