From feef58f427d54ffebeec8f09b4d8c31dbea9b1c3 Mon Sep 17 00:00:00 2001
From: Folke Lemaitre <folke.lemaitre@gmail.com>
Date: Sat, 1 Jun 2024 20:11:20 +0200
Subject: [PATCH] fix: automatically add `/lua` when needed

---
 lua/lazydev/buf.lua | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/lua/lazydev/buf.lua b/lua/lazydev/buf.lua
index f8e2fdd..e771471 100644
--- a/lua/lazydev/buf.lua
+++ b/lua/lazydev/buf.lua
@@ -16,8 +16,10 @@ M.library = {}
 M.query = nil
 
 function M.setup()
-  table.insert(M.library, vim.fs.normalize(Config.runtime) .. "/lua")
-  vim.list_extend(M.library, Config.library)
+  M.add(Config.runtime)
+  for _, lib in ipairs(Config.library) do
+    M.add(lib)
+  end
 
   M.query = vim.treesitter.query.parse(
     "lua",
@@ -53,6 +55,17 @@ function M.setup()
   M.on_change()
 end
 
+---@param path string
+function M.add(path)
+  path = vim.fs.normalize(path)
+  if not path:find("/lua/?$") and vim.uv.fs_stat(path .. "/lua") then
+    path = path .. "/lua"
+  end
+  if not vim.tbl_contains(M.library, path) then
+    table.insert(M.library, path)
+  end
+end
+
 function M.get_clients()
   ---@param client vim.lsp.Client
   return vim.tbl_filter(function(client)