From ba3167a1a4fdc9028eb1f088e823ac1dceb202f4 Mon Sep 17 00:00:00 2001
From: Thomas Vigouroux <tomvig38@gmail.com>
Date: Tue, 11 Aug 2020 14:36:13 +0200
Subject: perf: cache parser file list for performance

---
 lua/nvim-treesitter/install.lua |  2 ++
 lua/nvim-treesitter/parsers.lua | 16 +++++++++++++++-
 2 files changed, 17 insertions(+), 1 deletion(-)

(limited to 'lua')

diff --git a/lua/nvim-treesitter/install.lua b/lua/nvim-treesitter/install.lua
index e8c6cf86b..eac4af7e0 100644
--- a/lua/nvim-treesitter/install.lua
+++ b/lua/nvim-treesitter/install.lua
@@ -59,6 +59,8 @@ local function iter_cmd_sync(cmd_list)
 end
 
 local function run_install(cache_folder, package_path, lang, repo, with_sync)
+  parsers.reset_cache()
+
   local project_name = 'tree-sitter-'..lang
   local project_repo = cache_folder..'/'..project_name
   -- compile_location only needed for typescript installs.
diff --git a/lua/nvim-treesitter/parsers.lua b/lua/nvim-treesitter/parsers.lua
index 491edcb66..c1b48a00d 100644
--- a/lua/nvim-treesitter/parsers.lua
+++ b/lua/nvim-treesitter/parsers.lua
@@ -284,11 +284,25 @@ function M.get_parser_configs()
   return M.list
 end
 
+local parser_files
+
+function M.reset_cache()
+  parser_files = setmetatable({}, {
+    __index = function(tbl, key)
+      rawset(tbl, key, api.nvim_get_runtime_file('parser/' .. key .. '.*', false))
+      return rawget(tbl, key)
+    end
+  })
+end
+
+M.reset_cache()
+
 function M.has_parser(lang)
   local buf = api.nvim_get_current_buf()
   local lang = lang or M.get_buf_lang(buf)
+
   if not lang or #lang == 0 then return false end
-  return #api.nvim_get_runtime_file('parser/' .. lang .. '.*', false) > 0
+  return #parser_files[lang] > 0
 end
 
 function M.get_parser(bufnr, lang)
-- 
cgit v1.2.3-70-g09d2