Load .env files in dir of currently opened file (#63)

Author: jim-fx

lua/rest-nvim/utils/init.lua

@@ -48,13 +48,24 @@ end
 -- read_env_file Reads the environment variables found in the `.env` file and
 -- returns a table with the variables
 M.read_env_file = function()
+
   local variables = {}
-  local env_file_path = vim.fn.getcwd() .. "/.env"
+
+  -- Directories to search for env files
+  local env_file_paths = {
+    -- current working directory
+    vim.fn.getcwd() .. "/.env",
+    -- directory of the currently opened file
+    vim.fn.expand("%:p:h") .. "/.env"
+  }
+
   -- If there's an env file in the current working dir
-  if M.file_exists(env_file_path) then
-    for line in io.lines(env_file_path) do
-      local vars = M.split(line, "=", 1)
-      variables[vars[1]] = vars[2]
+  for _, env_file_path in ipairs(env_file_paths) do
+    if M.file_exists(env_file_path) then
+      for line in io.lines(env_file_path) do
+        local vars = M.split(line, "=", 1)
+        variables[vars[1]] = vars[2]
+      end
     end
   end