Color attribute functions

2 files changed, 13 insertions, 1 deletions

diff --git a/editors/code/package.json b/editors/code/package.json
index d899f60e3..d078f7472 100644
--- a/editors/code/package.json
+++ b/editors/code/package.json
@@ -604,6 +604,10 @@
         ],
         "semanticTokenModifiers": [
             {
+                "id": "attribute",
+                "description": "Style for elements within attributes"
+            },
+            {
                 "id": "constant",
                 "description": "Style for compile-time constants"
             },
@@ -630,6 +634,9 @@
                     "attribute": [
                         "meta.attribute.rust"
                     ],
+                    "function.attribute": [
+                        "entity.name.function.attribute.rust"
+                    ],
                     "builtinType": [
                         "support.type.primitive.rust"
                     ],
diff --git a/editors/code/rust.tmGrammar.json b/editors/code/rust.tmGrammar.json
index aa0811326..cdcd557dc 100644
--- a/editors/code/rust.tmGrammar.json
+++ b/editors/code/rust.tmGrammar.json
@@ -75,8 +75,13 @@
                 {
                         "comment": "Attribute",
                         "name": "meta.attribute.rust",
-                        "begin": "#\\!?\\[",
+                        "begin": "#\\!?\\[(\\w*)",
                         "end": "\\]",
+                        "captures": {
+                                "1": {
+                                        "name": "entity.name.function.attribute.rust"
+                                }
+                        },
                         "patterns": [
                                 {
                                         "include": "#string_literal"