fix(treesitter): markdown injection

kiyoon · Dec 16, 2024 · b0b7157 · b0b7157
1 parent 92b559f
commit b0b7157
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/after/queries/python/injections.scm b/after/queries/python/injections.scm
@@ -4,11 +4,27 @@
 ((
   (comment) @_mdcomment
   . (expression_statement 
-      (string (string_content) @markdown @markdown_inline)))
-  (#lua-match? @_mdcomment "^# %%%% %[markdown%]"))
+      (string (string_content) @injection.content)))
+  (#lua-match? @_mdcomment "^# %%%% %[markdown%]")
+  (#set! injection.language "markdown"))
 
 ((
   (comment) @_mdcomment
   . (expression_statement 
-      (string (string_content) @markdown @markdown_inline)))
-  (#lua-match? @_mdcomment "^# %%%% %[md%]"))
+      (string (string_content) @injection.content)))
+  (#lua-match? @_mdcomment "^# %%%% %[markdown%]")
+  (#set! injection.language "markdown_inline"))
+
+((
+  (comment) @_mdcomment
+  . (expression_statement 
+      (string (string_content) @injection.content)))
+  (#lua-match? @_mdcomment "^# %%%% %[md%]")
+  (#set! injection.language "markdown"))
+
+((
+  (comment) @_mdcomment
+  . (expression_statement 
+      (string (string_content) @injection.content)))
+  (#lua-match? @_mdcomment "^# %%%% %[md%]")
+  (#set! injection.language "markdown_inline"))