From 0957a4853c731697aa61be3db25094b2fe88d33c Mon Sep 17 00:00:00 2001
From: "Kenneth Benzie (Benie)" <benie@infektor.net>
Date: Fri, 12 Apr 2024 00:43:20 +0100
Subject: [PATCH] Add nvim-treesitter-context plugin

---
 lua/plugins/colorscheme.lua | 1 +
 lua/plugins/treesitter.lua  | 1 +
 2 files changed, 2 insertions(+)

diff --git a/lua/plugins/colorscheme.lua b/lua/plugins/colorscheme.lua
index e274159..896d191 100644
--- a/lua/plugins/colorscheme.lua
+++ b/lua/plugins/colorscheme.lua
@@ -23,6 +23,7 @@ return {
     -- Override highlight groups
     vim.cmd.highlight('WinSeparator', 'guifg=' .. dragon.ui.bg_m3, 'guibg=' .. dragon.ui.bg_m3)
     vim.cmd.highlight('MsgSeparator', 'guifg=' .. dragon.ui.bg_m3, 'guibg=' .. dragon.ui.bg_m3)
+    vim.cmd.highligh('TreesitterContext', 'guibg=' .. dragon.ui.bg_gutter)
     vim.cmd.highlight('link zshTodo Comment')
   end
 }
diff --git a/lua/plugins/treesitter.lua b/lua/plugins/treesitter.lua
index 0cc5f7a..d614591 100644
--- a/lua/plugins/treesitter.lua
+++ b/lua/plugins/treesitter.lua
@@ -2,6 +2,7 @@ return {
   'nvim-treesitter/nvim-treesitter',
   dependencies = {
     'nvim-treesitter/nvim-treesitter-textobjects',
+    { 'nvim-treesitter/nvim-treesitter-context', opts = {} },
     -- TODO: Fork this and add CMake support
     -- TODO: Add shell support (sh/bash/zsh if/for/while/etc)
     'RRethy/nvim-treesitter-endwise',