1 files changed, 21 insertions, 8 deletions
diff --git a/queries/proto/highlights.scm b/queries/proto/highlights.scm
index a24314834..e869b87e0 100644
--- a/queries/proto/highlights.scm
+++ b/queries/proto/highlights.scm
@@ -1,19 +1,31 @@
 [
-  "syntax"
-  "option"
-  "service"
-  "rpc"
-  "returns"
-  "message"
   "enum"
+  "extend"
+  "extensions"
+  "message"
   "oneof"
-  "optional"
-  "repeated"
+  "option"
   "reserved"
+  "service"
+  "syntax"
   "to"
 ] @keyword
 
 [
+ "rpc"
+] @keyword.function
+
+[
+ "returns"
+] @keyword.return
+
+[
+  "optional"
+  "repeated"
+  "required"
+] @type.qualifier
+
+[
   "package"
   "import"
 ] @include
@@ -34,6 +46,7 @@
 [
   (string)
   "\"proto3\""
+  "\"proto2\""
 ] @string
 
 (int_lit) @number