feat: add telescope extension (#278)

Author: bingcoke

README.md

@@ -99,6 +99,9 @@ use {
       -- Jump to request line on run
       jump_to_request = false,
       env_file = '.env',
+      -- for telescope select
+      env_pattern = "\\.env$",
+      env_edit_command = "tabedit",
       custom_dynamic_variables = {},
       yank_dry_run = true,
     })
@@ -186,6 +189,26 @@ request method (e.g. `GET`) and run `rest.nvim`.
 Run `export DEBUG_PLENARY="debug"` before starting nvim. Logs will appear most
 likely in ~/.cache/nvim/rest.nvim.log
 
+## Telescope
+
+```lua
+
+-- first load extension
+require("telescope").load_extension("rest")
+-- then use telescope
+require("telescope").extensions.rest.select_env()
+
+```
+
+### Mappings
+
+- Enter: Select Env file
+- Ctrl+O: Edit Env file
+
+### Config
+
+- env_pattern: For env file pattern
+- env_edit_command: For env file edit command
 
 ## Contribute
 

lua/rest-nvim/config/check.lua

@@ -25,6 +25,8 @@ function check.validate(cfg)
   local ok, err = validate({
     client = { cfg.client, "string" },
     env_file = { cfg.env_file, "string" },
+    env_pattern = { cfg.env_pattern, "string" },
+    env_edit_command = { cfg.env_edit_command, "string" },
     encode_url = { cfg.encode_url, "boolean" },
     skip_ssl_verification = { cfg.skip_ssl_verification, "boolean" },
     custom_dynamic_variables = { cfg.custom_dynamic_variables, "table" },

lua/rest-nvim/config/init.lua

@@ -52,6 +52,8 @@ local logger = require("rest-nvim.logger")
 ---@class RestConfig
 ---@field client string The HTTP client to be used when running requests, default is `"curl"`
 ---@field env_file string Environment variables file to be used for the request variables in the document
+---@field env_pattern string Environment variables file pattern for telescope.nvim
+---@field env_edit_command string Neovim command to edit an environment file, default is `"tabedit"`
 ---@field encode_url boolean Encode URL before making request
 ---@field skip_ssl_verification boolean Skip SSL verification, useful for unknown certificates
 ---@field custom_dynamic_variables { [string]: fun(): string }[] Table of custom dynamic variables
@@ -67,6 +69,8 @@ local logger = require("rest-nvim.logger")
 local default_config = {
   client = "curl",
   env_file = ".env",
+  env_pattern = "\\.env$",
+  env_edit_command = "tabedit",
   encode_url = true,
   skip_ssl_verification = false,
   custom_dynamic_variables = {},

lua/telescope/_extensions/rest.lua

@@ -0,0 +1,65 @@
+local has_telescope, telescope = pcall(require, "telescope")
+
+if not has_telescope then
+  return
+end
+
+local rest = require("rest-nvim")
+
+local state = require("telescope.actions.state")
+
+local action_state = require("telescope.actions.state")
+local actions = require("telescope.actions")
+local finders = require("telescope.finders")
+local pickers = require("telescope.pickers")
+local conf = require("telescope.config").values
+
+local config = require("rest-nvim.config")
+
+local function rest_env_select(opt)
+  local pattern = config.get("env_pattern")
+  local edit = config.get("env_edit_command")
+
+  local command = string.format("fd -H '%s'", pattern)
+  local result = io.popen(command):read("*a")
+
+  local lines = {}
+  for line in result:gmatch("[^\r\n]+") do
+    table.insert(lines, line)
+  end
+
+  pickers
+    .new({}, {
+      prompt_title = "Select Evn",
+      finder = finders.new_table({
+        results = lines,
+      }),
+      attach_mappings = function(prompt_bufnr, map)
+        actions.select_default:replace(function()
+          local selection = action_state.get_selected_entry()
+          if selection == nil then
+            return
+          end
+          actions.close(prompt_bufnr)
+          rest.select_env(selection[1])
+        end)
+        map("i", "<c-o>", function()
+          actions.close(prompt_bufnr)
+          local selection = state.get_selected_entry(prompt_bufnr)
+          if selection == nil then
+            return
+          end
+          vim.api.nvim_command(edit .. " " .. selection[1])
+        end)
+        return true
+      end,
+      previewer = conf.grep_previewer({}),
+    })
+    :find()
+end
+
+return telescope.register_extension({
+  exports = {
+    select_env = rest_env_select,
+  },
+})