nvim: latex snippets, treesitter was bad

1 files changed, 8 insertions, 0 deletions

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