feat: Major improvements in API. Particularly relating to entries.

scratch/batched_finder_and_sorter.lua

@@ -0,0 +1,55 @@
+local actions = require('telescope.actions')
+local finders = require('telescope.finders')
+local previewers = require('telescope.previewers')
+local pickers = require('telescope.pickers')
+local sorters = require('telescope.sorters')
+local utils = require('telescope.utils')
+
+local Job = require('plenary.job')
+
+-- local live_grepper = finders.new {
+--   fn_command = function(_, prompt)
+--     -- TODO: Make it so that we can start searching on the first character.
+--     if not prompt or prompt == "" then
+--       return nil
+--     end
+
+--     return {
+--       command = 'rg',
+--       args = {"--vimgrep", prompt},
+--     }
+--   end
+-- }
+
+local f = function(prompt, process_result, process_complete)
+  local fzf = Job:new {
+    command = 'fzf';
+
+    writer = Job:new {
+      command = "fdfind",
+      args = nil,
+      cwd = "/home/tj/build/neovim",
+
+      enable_handlers = false,
+    },
+
+    -- Still doesn't work if you don't pass these args and just run `fzf`
+    args = {'--no-sort', '--filter', prompt};
+  }
+
+
+  local start = vim.fn.reltime()
+  print(vim.inspect(fzf:sync()), vim.fn.reltimestr(vim.fn.reltime(start)))
+end
+
+
+-- Process all the files
+-- f("", nil, nil)
+-- Filter on nvimexec
+f("nvim/executor", nil, nil)
+
+-- pickers.new({}, {
+--   prompt    = 'Live Grep',
+--   finder    = f,
+--   previewer = previewers.vimgrep,
+-- }):find()

scratch/clason_finders.lua

@@ -0,0 +1,130 @@
+vim.cmd [[packadd! plenary.nvim]]
+vim.cmd [[packadd! popup.nvim]]
+vim.cmd [[packadd! telescope.nvim]]
+
+local finders = require('telescope.finders')
+local previewers = require('telescope.previewers')
+local pickers = require('telescope.pickers')
+local sorters = require('telescope.sorters')
+local utils = require('telescope.utils')
+
+local rgargs = {'--color=never', '--no-heading', '--with-filename', '--line-number', '--column', '--smart-case'}
+-- grep typed string in current directory (live, not fuzzy!)
+finders.rg_live = function(opts)
+  local live_grepper = finders.new {
+    fn_command = function(_, prompt)
+      if not prompt or prompt == "" then
+        return nil
+      end
+
+      return {
+        command = 'rg',
+        args = vim.tbl_flatten{rgargs, prompt},
+      }
+    end
+  }
+
+  pickers.new(opts, {
+    prompt    = 'Live Grep',
+    finder    = live_grepper,
+    previewer = previewers.vimgrep,
+  }):find()
+end
+
+-- fuzzy grep string in current directory (slow!)
+finders.rg = function(opts)
+  opts = opts or {}
+
+  local search = opts.search or ''
+
+  pickers.new(opts, {
+    prompt = 'Find Word',
+    finder = finders.new_oneshot_job(vim.tbl_flatten{'rg', rgargs, search}),
+    previewer = previewers.vimgrep,
+    sorter = sorters.get_norcalli_sorter(),
+  }):find()
+end
+
+-- fuzzy find files in current directory (may be slow in root dir)
+finders.fd = function(opts)
+  pickers.new(opts, {
+    prompt = 'Find Files',
+    finder = finders.new_oneshot_job {"fd"},
+    previewer = previewers.bat,
+    sorter = sorters.get_fuzzy_file(),
+  }):find()
+end
+
+-- fuzzy find in references to symbol under cursor
+finders.lsp_references = function(opts)
+  local params = vim.lsp.util.make_position_params()
+  params.context = { includeDeclaration = false }
+
+  local results_lsp = vim.lsp.buf_request_sync(0, "textDocument/references", params)
+  local locations = {}
+  for _, server_results in pairs(results_lsp) do
+    vim.list_extend(locations, vim.lsp.util.locations_to_items(server_results.result) or {})
+  end
+
+  local results = utils.quickfix_items_to_entries(locations)
+
+  if vim.tbl_isempty(results) then
+    return
+  end
+
+  pickers.new(opts, {
+    prompt    = 'LSP References',
+    finder    = finders.new_table(results),
+    previewer = previewers.qflist,
+    sorter    = sorters.get_norcalli_sorter(),
+  }):find()
+end
+
+-- fuzzy find in document symbols
+finders.lsp_document_symbols = function(opts)
+  local params = vim.lsp.util.make_position_params()
+  local results_lsp = vim.lsp.buf_request_sync(0, "textDocument/documentSymbol", params)
+  local locations = {}
+  for _, server_results in pairs(results_lsp) do
+    vim.list_extend(locations, vim.lsp.util.symbols_to_items(server_results.result, 0) or {})
+  end
+
+  local results = utils.quickfix_items_to_entries(locations)
+
+  if vim.tbl_isempty(results) then
+    return
+  end
+
+  pickers.new(opts, {
+    prompt    = 'LSP Document Symbols',
+    finder    = finders.new_table(results),
+    previewer = previewers.qflist,
+    sorter    = sorters.get_norcalli_sorter(),
+  }):find()
+end
+
+-- fuzzy find in all workspace symbols (may need longer timeout!)
+finders.lsp_workspace_symbols = function(opts)
+  local params = {query = ''}
+  local results_lsp = vim.lsp.buf_request_sync(0, "workspace/symbol", params, 1000)
+  local locations = {}
+  for _, server_results in pairs(results_lsp) do
+    vim.list_extend(locations, vim.lsp.util.symbols_to_items(server_results.result, 0) or {})
+  end
+
+  local results = utils.quickfix_items_to_entries(locations)
+
+  if vim.tbl_isempty(results) then
+    return
+  end
+
+  pickers.new(opts, {
+    prompt    = 'LSP Workspace Symbols',
+    finder    = finders.new_table(results),
+    previewer = previewers.qflist,
+    sorter    = sorters.get_norcalli_sorter(),
+  }):find()
+end
+
+
+return finders