From 44a39eee24bbda4567d39411c934cee40e9161f4 Mon Sep 17 00:00:00 2001
From: Sanchayan Maity <sanchayan@sanchayanmaity.net>
Date: Sat, 19 Jun 2021 14:47:48 +0530
Subject: [PATCH] nvim: treesitter: Enable playground

Helps to see what treesitter queries are being made. Would be helpful in
tuning the color scheme.
---
 nvim/.config/nvim/lua/keymappings.lua |  1 +
 nvim/.config/nvim/lua/plugins.lua     |  1 +
 nvim/.config/nvim/lua/treesitter.lua  | 28 ++++++++++++++++++++++++++-
 3 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/nvim/.config/nvim/lua/keymappings.lua b/nvim/.config/nvim/lua/keymappings.lua
index 0b0e650..46bc0e9 100644
--- a/nvim/.config/nvim/lua/keymappings.lua
+++ b/nvim/.config/nvim/lua/keymappings.lua
@@ -116,6 +116,7 @@ remap('c', '<Tab>',   'getcmdtype() =~ \'[?/]\' ? \'<C-g>\' : feedkeys(\'<Tab>\'
 remap('c', '<S-Tab>', 'getcmdtype() =~ \'[?/]\' ? \'<C-t>\' : feedkeys(\'<S-Tab>\', \'int\')[1]', { noremap = true, expr = true })
 
 remap('n', '<F1>', ':call SyntaxAttr#SyntaxAttr()<CR>', { noremap = true })
+remap('n', '<F2>', ':TSHighlightCapturesUnderCursor<CR>', { noremap = true })
 
 -- After shifting a visual block, select it again
 remap('v', '<', '<gv', { noremap = true })
diff --git a/nvim/.config/nvim/lua/plugins.lua b/nvim/.config/nvim/lua/plugins.lua
index 3876c3b..1e42a6c 100644
--- a/nvim/.config/nvim/lua/plugins.lua
+++ b/nvim/.config/nvim/lua/plugins.lua
@@ -101,6 +101,7 @@ local init = function ()
     requires = { 'nvim-treesitter/nvim-treesitter' },
     config = "require('treesitter')",
   }
+  use 'nvim-treesitter/playground'
   -- All writing needs
   use 'lervag/vimtex'
   use 'vim-pandoc/vim-pandoc'
diff --git a/nvim/.config/nvim/lua/treesitter.lua b/nvim/.config/nvim/lua/treesitter.lua
index 0902699..2ff9f3f 100644
--- a/nvim/.config/nvim/lua/treesitter.lua
+++ b/nvim/.config/nvim/lua/treesitter.lua
@@ -1,5 +1,5 @@
 require'nvim-treesitter.configs'.setup {
-  ensure_installed = { "bash", "c", "clojure", "cpp", "fennel", "fish", "java", "javascript", "jsdoc", "json", "lua", "nix", "python", "rust", "toml" },
+  ensure_installed = { "bash", "c", "clojure", "cpp", "fennel", "fish", "java", "javascript", "jsdoc", "json", "lua", "nix", "python", "query", "rust", "toml" },
   highlight   = { enable = true, },
   textobjects = {
     select = {
@@ -25,6 +25,32 @@ require'nvim-treesitter.configs'.setup {
       },
     },
   },
+  query_linter = {
+    enable           = true,
+    use_virtual_text = true,
+    lint_events      = { "BufWrite", "CursorHold" },
+  },
 }
 
 require'nvim-treesitter.install'.compilers = { "gcc" }
+
+require "nvim-treesitter.configs".setup {
+  playground = {
+    enable          = true,
+    disable         = {},
+    updatetime      = 25,
+    persist_queries = false,
+    keybindings = {
+      toggle_query_editor       = 'o',
+      toggle_hl_groups          = 'i',
+      toggle_injected_languages = 't',
+      toggle_anonymous_nodes    = 'a',
+      toggle_language_display   = 'I',
+      focus_language            = 'f',
+      unfocus_language          = 'F',
+      update                    = 'R',
+      goto_node                 = '<CR>',
+      show_help                 = '?',
+    },
+  }
+}