From c33d6578a8b2ce03cab58d5952d717064125ef5d Mon Sep 17 00:00:00 2001
From: Folke Lemaitre <folke.lemaitre@gmail.com>
Date: Sat, 1 Jun 2024 10:25:08 +0200
Subject: [PATCH] fix(util): make sure doc gen works

---
 lua/lazyvim/util/init.lua | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/lua/lazyvim/util/init.lua b/lua/lazyvim/util/init.lua
index 2dca92ba..05d96db5 100644
--- a/lua/lazyvim/util/init.lua
+++ b/lua/lazyvim/util/init.lua
@@ -226,11 +226,12 @@ end
 ---@param path? string
 ---@param opts? { warn?: boolean }
 function M.get_pkg_path(pkg, path, opts)
-  require("mason") -- make sure Mason is loaded
+  pcall(require, "mason") -- make sure Mason is loaded. Will fail when generating docs
+  local root = vim.env.MASON or (vim.fn.stdpath("data") .. "/mason")
   opts = opts or {}
   opts.warn = opts.warn == nil and true or opts.warn
   path = path or ""
-  local ret = vim.env.MASON .. "/packages/" .. pkg .. "/" .. path
+  local ret = root .. "/packages/" .. pkg .. "/" .. path
   if opts.warn and not vim.loop.fs_stat(ret) then
     M.warn(("Mason package path not found for **%s**:\n- `%s`"):format(pkg, path))
   end