[ "fun" "λ" "forall" "∏" "let" "in" "end" "def" "axiom" "variable" "hypothesis" "section" "infixl" "infixr" ] @keyword

(preprocess
  (command) @keyword
  (post_command) @string)

[ "=>" "→" "⇒" "," ":=" ";" ":" ] @punctuation.delimiter

(symbol) @operator

[(star) (square)] @constant.builtin

(comment) @comment
;(block_comment) @comment

[ "(" ")" ] @punctuation.bracket

(definition
  name: (identifier) @function)

(axiom
  name: (identifier) @function)

(binding
  (identifier) @function)

(param_block
  param: (identifier)+ @variable.parameter)

(param_block
  type: (expr) @type)

(ascription
  type: (expr) @type)