1 files changed, 21 insertions, 0 deletions
diff --git a/lua/tree-sitter-wmap.lua b/lua/tree-sitter-wmap.lua
new file mode 100644
index 0000000..7c4a53d
--- /dev/null
+++ b/lua/tree-sitter-wmap.lua
@@ -0,0 +1,21 @@
+local M = {}
+
+function M.setup()
+  local config_file_path = debug.getinfo(1).source:sub(2)
+  local plugin_dir = vim.fn.fnamemodify(config_file_path, ':h:h')
+  local parser_config = require 'nvim-treesitter.parsers'.get_parser_configs()
+  parser_config.wmap = {
+    install_info = {
+      url = plugin_dir,
+      files = {'src/parser.c'},
+      branch = 'main'
+    }
+  }
+  vim.filetype.add({
+    extension = {
+      wmap = 'wmap',
+    }
+  })
+end
+
+return M