nvim: latex snippets, treesitter was bad

2 files changed, 9 insertions, 0 deletions

diff --git a/.config/nvim/lua/plugin_settings.lua b/.config/nvim/lua/plugin_settings.lua
index 5b999fe..56612f5 100644
--- a/.config/nvim/lua/plugin_settings.lua
+++ b/.config/nvim/lua/plugin_settings.lua
@@ -254,6 +254,7 @@ require 'nvim-treesitter.configs'.setup {
     ensure_installed = "maintained",
     highlight = {
         enable = true,
+        disable = { "latex" },
         additional_vim_regex_highlighting = false,
     },
     incremental_selection = {
diff --git a/.config/nvim/my_snippets/tex.snippets b/.config/nvim/my_snippets/tex.snippets
index 503221f..ea34781 100644
--- a/.config/nvim/my_snippets/tex.snippets
+++ b/.config/nvim/my_snippets/tex.snippets
@@ -1,6 +1,11 @@
 # vim:ft=snippets:
 # https://castel.dev/post/lecture-notes-1/
 
+global !p
+def math():
+    return vim.eval('vimtex#syntax#in_mathzone()') == '1'
+endglobal
+
 snippet template "Basic template" b
 \documentclass[a4paper]{article}
 
@@ -96,14 +101,17 @@ ${1:${VISUAL}}
 .\] $0
 endsnippet
 
+context "math()"
 snippet '([A-Za-z])(\d)' "auto subscript" wrA
 `!p snip.rv = match.group(1)`_`!p snip.rv = match.group(2)`
 endsnippet
 
+context "math()"
 snippet '([A-Za-z])_(\d\d)' "auto subscript2" wrA
 `!p snip.rv = match.group(1)`_{`!p snip.rv = match.group(2)`}
 endsnippet
 
+context "math()"
 snippet // "Fraction" iA
 \\frac{$1}{$2}$0
 endsnippet