From 6a8c4ae3afa3b3dec3261be806e4777f9897ec8e Mon Sep 17 00:00:00 2001
From: danymat <d.danymat@gmail.com>
Date: Sun, 9 Jan 2022 19:01:29 +0100
Subject: [PATCH] feat(rust) Add annotations for traits

---
 lua/neogen/configurations/rust.lua | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/lua/neogen/configurations/rust.lua b/lua/neogen/configurations/rust.lua
index 99d7c63..fdee5bb 100644
--- a/lua/neogen/configurations/rust.lua
+++ b/lua/neogen/configurations/rust.lua
@@ -1,12 +1,12 @@
 return {
     parent = {
-        func = { "function_item" },
-        class = { "struct_item" },
+        func = { "function_item", "function_signature_item" },
+        class = { "struct_item", "trait_item" },
         file = { "source_file" },
     },
     data = {
         func = {
-            ["function_item"] = {
+            ["function_item|function_signature_item"] = {
                 ["0"] = {
                     extract = function(node)
                         local tree = {
@@ -46,7 +46,7 @@ return {
             },
         },
         class = {
-            ["struct_item"] = {
+            ["struct_item|trait_item"] = {
                 ["0"] = {
                     extract = function(node)
                         local tree = {