better

ftdetect/perga.vim

@@ -1 +0,0 @@
-autocmd VimEnter,BufWinEnter,BufRead,BufNewFile *.pg setlocal filetype=perga

lua/perga/init.lua

@@ -1,6 +1,10 @@
 local M = {}
 
 function M.setup(opts)
+    vim.filetype.add({
+        extension = { pg = 'perga' }
+    })
+
     local parser_config = require "nvim-treesitter.parsers".get_parser_configs()
     parser_config.perga = {
         install_info = {

queries/perga/highlights.scm

@@ -1,8 +1,17 @@
 [ "fun" "λ" "forall" "∏" ] @keyword
 [ "->" "=>" "→" "⇒" "," ":=" ";" ":" ] @punctuation.delimiter
-[ "(" ")" ] @punctuation.bracket
-;[ "*" "□" "[]" ] @constant.builtin
+
+[(star) (square)] @constant.builtin
+
 (comment) @comment
-(identifier) @variable
-;(param) @variable.parameter
-;(type) @type
+
+[ "(" ")" ] @punctuation.bracket
+
+(definition
+  name: (identifier) @function)
+
+(param_block
+  param: (identifier)+ @variable.parameter)
+
+(param_block
+  type: (expr) @type)

queries/rainbow-delimiters.scm

@@ -0,0 +1 @@
+["(" ")"] @delimiter