feat/refacto: improve installer

1 files changed, 102 insertions, 38 deletions

diff --git a/lua/nvim-treesitter/install.lua b/lua/nvim-treesitter/install.lua
index 2f40db11c..8e39be5e7 100644
--- a/lua/nvim-treesitter/install.lua
+++ b/lua/nvim-treesitter/install.lua
@@ -1,74 +1,138 @@
 local api = vim.api
 local fn = vim.fn
+local luv = vim.loop
 
 local M = {}
-local package_path = nil
 local repositories = {
   javascript = {
     url = "https://github.com/tree-sitter/tree-sitter-javascript",
-    files = "src/parser.c src/scanner.c",
+    files = { "src/parser.c", "src/scanner.c" },
   },
   c = {
     url = "https://github.com/tree-sitter/tree-sitter-c",
-    files = "src/parser.c"
+    files = { "src/parser.c" }
+  },
+  rust = {
+    url = "https://github.com/tree-sitter/tree-sitter-rust",
+    files = { "src/parser.c", "src/scanner.c" },
   }
 }
 
--- TODO(kyazdani): there might be a better way to get the plugin path
-for _, path in pairs(api.nvim_list_runtime_paths()) do
-  if string.match(path, '.*/nvim%-treesitter') then
-    package_path = path
-    break
+local function get_package_path()
+  for _, path in pairs(api.nvim_list_runtime_paths()) do
+    if string.match(path, '.*/nvim%-treesitter') then
+      return path
+    end
   end
+
+  return nil, 'Plugin runtime path not found.'
 end
 
-local function create_compile_command(ft, path, files)
-  -- TODO(kyazdani): should download the parser in $XDG_CACHE_HOME instead of /tmp
-  return "cd /tmp/tree-sitter-"..ft..[[ && \
-    gcc -o parser.so -shared ]]..files..[[ -Os -I./src && \
-    mv parser.so ]]..path.."/parsers/"..ft..[[.so && \
-    rm -rf /tmp/tree-sitter-]]..ft
+local function get_cache_dir()
+  local home = fn.get(fn.environ(), 'HOME')
+  local xdg_cache = fn.get(fn.environ(), 'XDG_CACHE_HOME')
+
+  if xdg_cache == 0 then
+    xdg_cache = home .. '/.cache'
+  end
+
+  if luv.fs_access(xdg_cache, 'RW') then
+    return xdg_cache
+  elseif luv.fs_access('/tmp', 'RW') then
+    return '/tmp'
+  end
+
+  return nil, 'Invalid cache rights, $XDG_CACHE_HOME or /tmp should be read/write'
 end
 
-local function create_download_command(url)
-  return "cd /tmp && git clone "..url
+local function iter_cmd(cmd_list, i, ft)
+  if i == #cmd_list then return print('Treesitter parser for '..ft..' has been installed') end
+  local attr = cmd_list[i + 1]
+  print(attr.info)
+  handle = luv.spawn(attr.cmd, attr.opts, vim.schedule_wrap(function(code)
+    handle:close()
+    if code ~= 0 then return api.nvim_err_writeln(attr.err) end 
+    iter_cmd(cmd_list, i + 1, ft)
+  end))
 end
 
-function M.install_parser(lang)
+local function run_install(cache_folder, package_path, ft, files, url)
+  local project_repo = cache_folder..'/tree-sitter-'..ft
+  local parser_lib_name = package_path.."/parser/"..ft..".so"
+  local command_list = {
+    {
+      cmd = 'rm',
+      opts = {
+        args = { '-rf', project_repo },
+      }
+    },
+    {
+      cmd = 'git',
+      info = 'Downloading...',
+      err = 'Error during download, please verify your internet connection',
+      opts = {
+        args = { 'clone', url },
+        cwd = cache_folder,
+      },
+    },
+    {
+      cmd = 'cc',
+      info = 'Compiling...',
+      err = 'Error during compilation',
+      opts = {
+        args = { '-o', 'parser.so', '-shared', '-lstdc++', unpack(files), '-Os', '-I./src' },
+        cwd = project_repo
+      }
+    },
+    {
+      cmd = 'mv',
+      opts = {
+        args = { project_repo..'/parser.so', parser_lib_name }
+      }
+    },
+    {
+      cmd = 'rm',
+      opts = {
+        args = { '-rf', project_repo }
+      }
+    }
+  }
+
+  iter_cmd(command_list, 0, ft)
+end
+
+-- TODO(kyazdani): this should work on windows too
+function M.install_parser(ft)
   if fn.has('win32') == 1 then
-    -- TODO(kyazdani): this should work on windows too
-    api.nvim_err_writeln('This command is not available on windows at the moment.')
-    return
+    return api.nvim_err_writeln('This command is not available on windows at the moment.')
   end
 
-  if not lang then
-    api.nvim_err_writeln("usage: install_parser('language')")
-    return
+  if not ft then
+    return api.nvim_err_writeln("Usage: install_parser('language')")
   end
 
-  local repository = repositories[lang]
-  if not repository then
-    api.nvim_err_writeln('Parser not available for language '..lang)
-    return
+  if #api.nvim_get_runtime_file('parser/'..ft..'.so', false) > 0 then
+    local yesno = fn.input('Parser already available: would you like to reinstall ? y/n: ')
+    print('\n ') -- mandatory to avoid messing up command line
+    if not string.match(yesno, '^y.*') then return end
   end
 
-  if not package_path then
-    api.nvim_err_writeln('Plugin runtime path not found.')
-    return
+  local repository = repositories[ft]
+  if not repository then
+    return api.nvim_err_writeln('Parser not available for language '..ft)
   end
 
-  if not fn.executable('git') == 1 then
-    api.nvim_err_writeln('Please install `git` to download parsers.')
-    return
+  if fn.executable('git') == 0 then
+    return api.nvim_err_writeln('Git is required on your system to run this command')
   end
 
-  print("Downloading parser...")
-  fn.system(create_download_command(repository.url))
+  local package_path, err = get_package_path()
+  if err then return api.nvim_err_writeln(err) end
 
-  print("Compiling parser...")
-  fn.system(create_compile_command(lang, package_path, repository.files))
+  local cache_folder, err = get_cache_dir()
+  if err then return api.nvim_err_writeln(err) end
 
-  print("Install for the " ..lang.." parser was successfull")
+  run_install(cache_folder, package_path, ft, repository.files, repository.url)
 end
 
 return M