latex: add TSMath that behaves like vimtex math highlighting

CONTRIBUTING.md

@@ -167,6 +167,7 @@ Mainly for markup languages.
 @text.title
 @text.literal
 @text.uri
+@text.math (e.g. for LaTeX math enviroments)
 
 @text.note
 @text.warning

lua/nvim-treesitter/highlight.lua

@@ -84,6 +84,7 @@ hlmap["text.strike"] = "TSStrike"
 hlmap["text.title"] = "TSTitle"
 hlmap["text.literal"] = "TSLiteral"
 hlmap["text.uri"] = "TSURI"
+hlmap["text.math"] = "TSMath"
 
 hlmap["text.note"] = "TSNote"
 hlmap["text.warning"] = "TSWarning"

plugin/nvim-treesitter.vim

@@ -76,6 +76,7 @@ highlight default TSStrong term=bold cterm=bold gui=bold
 highlight default TSEmphasis term=italic cterm=italic gui=italic
 highlight default TSUnderline term=underline cterm=underline gui=underline
 highlight default TSStrike term=strikethrough cterm=strikethrough gui=strikethrough
+highlight default link TSMath Special
 highlight default link TSTitle Title
 highlight default link TSLiteral String
 highlight default link TSURI Underlined

queries/latex/highlights.scm

@@ -1,3 +1,23 @@
+;; Math
+[
+ (displayed_equation)
+ (inline_formula)
+] @text.math
+
+;; This highlights the whole environment like vimtex does
+((environment
+  (begin
+   name: (word) @_env)) @text.math
+   (#match? @_env "^(displaymath|eqn|eqnarray|align)[*]?$"))
+
+;; This at the begining of the file would be the alternative to highlight
+;; only the interior of the environment
+;((environment
+  ;(begin
+   ;name: (word) @_env) @none
+   ;(end) @none) @text.math
+   ;(#match? @_env "^(displaymath|eqn|eqnarray)[*]?$"))
+
 [
   (generic_command_name)
   "\\begin"
@@ -5,6 +25,7 @@
   "\\newcommand"
   "\\renewcommand"
   "\\DeclareRobustCommand"
+  "\\caption"
 ] @function.macro
 
 (comment) @comment
@@ -38,6 +59,11 @@
   "\\subparagraph"
 ] @type
 
+"\\item" @punctuation.special
+
+((word) @punctuation.delimiter
+(#eq? @punctuation.delimiter "&"))
+
 [
   (label_definition)
   (label_reference)
@@ -99,12 +125,11 @@
   arg: (_) @text.strong)
  (#match? @_name "^(\\textbf|\\mathbf)$"))
 
+((generic_command
+  name:(generic_command_name) @_name
+  .
+  arg: (_) @text.url)
+ (#match? @_name "^(\\url|\\href)$"))
+
 (ERROR) @error
 
-;; Controversial
-; Add @text.bullet?
-"\\item" @punctuation.special
-
-;; Math (TODO)
-;(displayed_equation)
-;(inline_formula)