From 561382a7efd73e3eb72bc890e83afc021e4993fe Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Tue, 25 Aug 2020 20:00:47 +0200
Subject: [PATCH] Add Windows support (mingw)

---
 README.md                       |  2 +-
 doc/nvim-treesitter.txt         |  2 +-
 lua/nvim-treesitter/install.lua | 18 +++++++++++++-----
 lua/nvim-treesitter/utils.lua   | 14 ++++++++------
 4 files changed, 23 insertions(+), 13 deletions(-)

diff --git a/README.md b/README.md
index 423ab8c3d..2c345da82 100644
--- a/README.md
+++ b/README.md
@@ -82,7 +82,7 @@ so you'll need to activate them by putting this in your `init.vim` file:
 ```lua
 lua <<EOF
 require'nvim-treesitter.configs'.setup {
-  ensure_installed = "all"      -- one of "all", "language", or a list of languages
+  ensure_installed = "all",     -- one of "all", "language", or a list of languages
   highlight = {
     enable = true,              -- false will disable the whole extension
     disable = { "c", "rust" },  -- list of language that will be disabled
diff --git a/doc/nvim-treesitter.txt b/doc/nvim-treesitter.txt
index 38c0378e6..284b0ba41 100644
--- a/doc/nvim-treesitter.txt
+++ b/doc/nvim-treesitter.txt
@@ -36,7 +36,7 @@ To enable supported features, put this in your `init.vim` file:
 >
   lua <<EOF
   require'nvim-treesitter.configs'.setup {
-    ensure_installed = "all"      -- one of "all", "language", or a list of languages
+    ensure_installed = "all",     -- one of "all", "language", or a list of languages
     highlight = {
       enable = true,              -- false will disable the whole extension
       disable = { "c", "rust" },  -- list of language that will be disabled
diff --git a/lua/nvim-treesitter/install.lua b/lua/nvim-treesitter/install.lua
index eac4af7e0..7d980f9b9 100644
--- a/lua/nvim-treesitter/install.lua
+++ b/lua/nvim-treesitter/install.lua
@@ -58,6 +58,10 @@ local function iter_cmd_sync(cmd_list)
   return true
 end
 
+local function select_executable(executables)
+  return vim.tbl_filter(function(c) return fn.executable(c) == 1 end, executables)[1]
+end
+
 local function run_install(cache_folder, package_path, lang, repo, with_sync)
   parsers.reset_cache()
 
@@ -66,6 +70,14 @@ local function run_install(cache_folder, package_path, lang, repo, with_sync)
   -- compile_location only needed for typescript installs.
   local compile_location = cache_folder..'/'..(repo.location or project_name)
   local parser_lib_name = package_path.."/parser/"..lang..".so"
+
+  local compilers = { "cc", "gcc" }
+  local cc = select_executable(compilers)
+  if not cc then
+    api.nvim_err_writeln('No C compiler found! "'..table.concat(compilers, '", "')..'" are not executable.')
+    return
+  end
+
   local command_list = {
     {
       cmd = 'rm',
@@ -83,7 +95,7 @@ local function run_install(cache_folder, package_path, lang, repo, with_sync)
       },
     },
     {
-      cmd = 'cc',
+      cmd = cc,
       info = 'Compiling...',
       err = 'Error during compilation',
       opts = {
@@ -148,12 +160,8 @@ local function install_lang(lang, ask_reinstall, cache_folder, package_path, wit
   run_install(cache_folder, package_path, lang, install_info, with_sync)
 end
 
--- TODO(kyazdani): this should work on windows too
 local function install(with_sync, ask_reinstall)
   return function (...)
-    if fn.has('win32') == 1 then
-      return api.nvim_err_writeln('This command is not available on windows at the moment.')
-    end
 
     if fn.executable('git') == 0 then
       return api.nvim_err_writeln('Git is required on your system to run this command')
diff --git a/lua/nvim-treesitter/utils.lua b/lua/nvim-treesitter/utils.lua
index d3d183395..4ad506e88 100644
--- a/lua/nvim-treesitter/utils.lua
+++ b/lua/nvim-treesitter/utils.lua
@@ -19,7 +19,9 @@ end
 
 function M.get_package_path()
   for _, path in pairs(api.nvim_list_runtime_paths()) do
-    if string.match(path, '.*/nvim%-treesitter') then
+    local last_segment = vim.fn.fnamemodify(path, ":t")
+    local penultimate_segment = vim.fn.fnamemodify(path, ":t:t")
+    if last_segment == "nvim-treesitter" or (last_segment == "" and penultimate_segment == "nvim-treesitter") then
       return path
     end
   end
@@ -29,14 +31,14 @@ end
 
 function M.get_cache_dir()
   local home = fn.get(fn.environ(), 'HOME')
-  local xdg_cache = fn.get(fn.environ(), 'XDG_CACHE_HOME')
+  local cache_dir = fn.stdpath('data')
 
-  if xdg_cache == 0 then
-    xdg_cache = home .. '/.cache'
+  if cache_dir == 0 then
+    cache_dir = home .. '/.cache'
   end
 
-  if luv.fs_access(xdg_cache, 'RW') then
-    return xdg_cache
+  if luv.fs_access(cache_dir, 'RW') then
+    return cache_dir
   elseif luv.fs_access('/tmp', 'RW') then
     return '/tmp'
   end