Fixed some path issues with windows

2 files changed, 16 insertions, 1 deletions

diff --git a/lua/talha/functions.lua b/lua/talha/functions.lua
index 946bfd8..e17cada 100644
--- a/lua/talha/functions.lua
+++ b/lua/talha/functions.lua
@@ -60,3 +60,18 @@ function ToggleSpellCheck()
     vim.opt.spell = not(vim.opt.spell:get())
     vim.opt.spelllang = 'en_us'
 end
+
+function GetHomeDir()
+  -- for linux/mac
+  local home = os.getenv("HOME")
+  -- for windows
+  local user_profile = os.getenv("UserProfile")
+
+  if home ~= nil then
+    return home
+  elseif user_profile ~= nil then
+    return user_profile
+  end
+
+  return nil
+end
diff --git a/lua/talha/set.lua b/lua/talha/set.lua
index 0366b03..404b1d8 100644
--- a/lua/talha/set.lua
+++ b/lua/talha/set.lua
@@ -13,7 +13,7 @@ op.wrap = false
 
 op.swapfile = false
 op.backup = false
-op.undodir = os.getenv("HOME") .. "/.vim/undodir"
+op.undodir = GetHomeDir() .. "/.vim/undodir"
 op.undofile = true
 
 op.hlsearch = false