Add in-memory context variables (#241)

Author: aPisC

lua/rest-nvim/curl/init.lua

@@ -114,6 +114,7 @@ local function create_callback(curl_cmd, opts)
         pretty_print = vim.pretty_print,
         json_decode = vim.fn.json_decode,
         set_env = utils.set_env,
+        set = utils.set_context,
       }
       local env = { context = context }
       setmetatable(env, { __index = _G })

lua/rest-nvim/utils/init.lua

@@ -4,6 +4,7 @@ local random = math.random
 math.randomseed(os.time())
 
 local M = {}
+local contexts = {}
 
 M.binary_content_types = {
   "octet-stream",
@@ -30,6 +31,16 @@ M.set_env = function(key, value)
   M.write_env_file(variables)
 end
 
+-- set_context sets a context variable for the current file
+-- @param key The key to set
+-- @param value The value to set
+M.set_context = function(key, value)
+  local env_file = "/" .. (config.get("env_file") or ".env")
+  local context = contexts[env_file] or {}
+  context[key] = value
+  contexts[env_file] = context
+end
+
 M.write_env_file = function(variables)
   local env_file = "/" .. (config.get("env_file") or ".env")
 
@@ -142,6 +153,11 @@ M.get_env_variables = function()
   return variables
 end
 
+M.get_context_variables = function()
+  local env_file = "/" .. (config.get("env_file") or ".env")
+  return contexts[env_file] or {}
+end
+
 -- get_variables Reads the environment variables found in the env_file option
 -- (default: .env) specified in configuration or from the files being read
 -- with variables beginning with @ and returns a table with the variables
@@ -229,10 +245,11 @@ end
 
 M.read_variables = function()
   local first = M.get_variables()
-  local second = M.read_dynamic_variables()
-  local third = M.read_document_variables()
+  local second = M.get_context_variables()
+  local third = M.read_dynamic_variables()
+  local fourth = M.read_document_variables()
 
-  return vim.tbl_extend("force", first, second, third)
+  return vim.tbl_extend("force", first, second, third, fourth)
 end
 
 -- replace_vars replaces the env variables fields in the provided string