From 44d4526e186a1f7168cf38fedb29e6fba81858a7 Mon Sep 17 00:00:00 2001
From: derekstride <derek.stride@shopify.com>
Date: Wed, 21 Apr 2021 12:10:48 -0230
Subject: Ensure there is a nested table allocated for #set!

Fixes the following bug:
```
test/functional/helpers.lua:107: Error executing lua:
vim/treesitter/query.lua:256: attempt to index a nil value
```
---
 runtime/lua/vim/treesitter/query.lua | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'runtime/lua/vim')

diff --git a/runtime/lua/vim/treesitter/query.lua b/runtime/lua/vim/treesitter/query.lua
index ed5146be44..51d538c0ff 100644
--- a/runtime/lua/vim/treesitter/query.lua
+++ b/runtime/lua/vim/treesitter/query.lua
@@ -253,7 +253,11 @@ local directive_handlers = {
   ["set!"] = function(_, _, _, pred, metadata)
     if #pred == 4 then
       -- (#set! @capture "key" "value")
-      metadata[pred[2]][pred[3]] = pred[4]
+      local capture = pred[2]
+      if not metadata[capture] then
+        metadata[capture] = {}
+      end
+      metadata[capture][pred[3]] = pred[4]
     else
       -- (#set! "key" "value")
       metadata[pred[2]] = pred[3]
-- 
cgit