[
    "import"
    "define"
    "type"
    "enum"
    "service"
    "notification"
] @keyword

(type) @type

(import_declaration source: (string) @string)

(define_declaration key: (identifier) @type.definition)

(type_declaration name: (identifier) @type.definition)

(attribute (pair key: (identifier) @attribute))

(enum_declaration name: (identifier) @type.definition)
(enum_value_declaration) @enum
(enum_value_assignment (identifier) @enum)
(enum_value_assignment (number) @number)

(service_declaration name: (identifier) @type.definition)
(function_decorator name: (identifier) @comment)
(service_function name: (identifier) @function)
(function_argument name: (identifier) @attribute)
(service_notification name: (identifier) @function)


(string) @string
(number) @number