Update readme

[rbdr/tree-sitter-api-notation] / lua / tree-sitter-api-notation.lua

diff --git a/lua/tree-sitter-api-notation.lua b/lua/tree-sitter-api-notation.lua
index 184cb573548ab36bd7b02db27fc496bf9514985e..95d5bd558eba0568130fef05c0276136bb9a1f49 100644 (file)
--- a/lua/tree-sitter-api-notation.lua
+++ b/lua/tree-sitter-api-notation.lua
@@ -12,9 +12,11 @@ function M.setup()
       branch = 'main'
     }
   }
-  vim.api.nvim_create_autocmd({'BufNewFile', 'BufRead'}, {
-    pattern = {'*.api', '*.apinotation'},
-    callback = function () vim.opt.filetype = 'api_notation' end
+  vim.filetype.add({
+    extension = {
+      api = 'api_notation',
+      apinotation = 'api_notation'
+    }
   })
 end