From fdcd0edf24ea23ebd2e82899a1401ef1f9f05872 Mon Sep 17 00:00:00 2001
From: Łukasz Niemier <lukasz@niemier.pl>
Date: Tue, 2 Nov 2021 16:54:34 +0100
Subject: fix: make comment doc highlighting work for any doc tag

---
 queries/elixir/highlights.scm | 25 ++++++++++++++-----------
 1 file changed, 14 insertions(+), 11 deletions(-)

diff --git a/queries/elixir/highlights.scm b/queries/elixir/highlights.scm
index b69bdbcdd..1a93a7687 100644
--- a/queries/elixir/highlights.scm
+++ b/queries/elixir/highlights.scm
@@ -180,13 +180,6 @@
       ] operator: "/" right: (integer) @operator)
   ])
 
-; Module attributes
-(unary_operator
-  operator: "@" @constant
-  operand: [
-    (identifier) @constant
-    (call target: (identifier) @constant)])
-
 ; Sigils
 (sigil
   "~" @string.special
@@ -205,10 +198,20 @@
   ((sigil_modifiers) @string)?
   (#any-of? @_sigil_name "s" "S"))
 
+; Module attributes
+(unary_operator
+  operator: "@"
+  operand: [
+    (identifier)
+    (call target: (identifier) @constant)]) @constant
+
 ; Documentation
 (unary_operator
-  operator: "@" @comment
+  operator: "@"
   operand: (call
-    target: ((identifier) @_identifier)
-    _) @comment
-  (#any-of? @_identifier "moduledoc" "typedoc" "shortdoc" "doc"))
+    target: ((identifier) @_identifier (#match? @_identifier "doc$")) @comment
+    (arguments [
+      (string)
+      (boolean)
+      (charlist)
+    ] @comment))) @comment
-- 
cgit v1.2.3-70-g09d2