From e98e2eafa29958a414e5785f58a9cbda117c1bf1 Mon Sep 17 00:00:00 2001
From: derekstride <derek.stride@shopify.com>
Date: Thu, 15 Apr 2021 15:48:48 -0230
Subject: downcase! directive to use with ruby injections

```
(#downcase! "language")
```

downcase! will ensure the metadata value for the specified key will be
downcased. If the value is a node, it will downcase the text specified
by the node.

```
(#downcase! @node "key")
```

You can also namespace the key with a specific capture, similar to how
you can call `(#set! @node "key" "value")`
---
 lua/nvim-treesitter/query_predicates.lua | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

(limited to 'lua')

diff --git a/lua/nvim-treesitter/query_predicates.lua b/lua/nvim-treesitter/query_predicates.lua
index a41ad287b..1748250e5 100644
--- a/lua/nvim-treesitter/query_predicates.lua
+++ b/lua/nvim-treesitter/query_predicates.lua
@@ -87,3 +87,30 @@ end)
 
 -- Just avoid some anoying warnings for this directive
 query.add_directive('make-range!', function() end)
+
+query.add_directive('downcase!', function(match, _, bufnr, pred, metadata)
+  local text, key, value
+
+  if #pred == 3 then
+    -- (#downcase! @capture "key")
+    key = pred[3]
+    value = metadata[pred[2]][key]
+  else
+    -- (#downcase! "key")
+    key = pred[2]
+    value = metadata[key]
+  end
+
+  if type(value) == "string" then
+    text = value
+  else
+    local node = match[value]
+    text = query.get_node_text(node, bufnr)
+  end
+
+  if #pred == 3 then
+    metadata[pred[2]][key] = string.lower(text)
+  else
+    metadata[key] = string.lower(text)
+  end
+end)
-- 
cgit v1.2.3-70-g09d2