added provider for 'vim.ui.select' (closes #314)

lua/fzf-lua/init.lua

@@ -140,6 +140,9 @@ M.lsp_code_actions = require'fzf-lua.providers.lsp'.code_actions
 M.lsp_document_diagnostics = require'fzf-lua.providers.lsp'.diagnostics
 M.lsp_workspace_diagnostics = require'fzf-lua.providers.lsp'.workspace_diagnostics
 
+M.register_ui_select = require'fzf-lua.providers.ui_select'.register
+M.deregister_ui_select = require'fzf-lua.providers.ui_select'.deregister
+
 -- API shortcuts
 M.fzf = require'fzf-lua.core'.fzf
 M.fzf_wrap = require'fzf-lua.core'.fzf_wrap

lua/fzf-lua/providers/ui_select.lua

@@ -0,0 +1,72 @@
+local core = require "fzf-lua.core"
+local utils = require "fzf-lua.utils"
+
+local M = {}
+
+local _old_ui_select = nil
+
+M.deregister = function()
+  if not _old_ui_select then
+    utils.info("vim.ui.select in not registered to fzf-lua")
+  end
+  vim.ui.select = _old_ui_select
+  _old_ui_select = nil
+end
+
+M.register = function()
+  if vim.ui.select == M.ui_select then
+    -- already registered
+    utils.info("vim.ui.select already registered to fzf-lua")
+    return
+  end
+  _old_ui_select = vim.ui.select
+  vim.ui.select = M.ui_select
+end
+
+M.ui_select = function(items, opts, on_choice)
+  --[[
+  -- Code Actions
+  opts = {
+    format_item = <function 1>,
+    kind = "codeaction",
+    prompt = "Code actions:"
+  }
+  items[1] = { 1, {
+    command = {
+      arguments = { {
+          action = "add",
+          key = "Lua.diagnostics.globals",
+          uri = "file:///home/bhagwan/.dots/.config/awesome/rc.lua",
+          value = "mymainmenu"
+        } },
+      command = "lua.setConfig",
+      title = "Mark defined global"
+    },
+    kind = "quickfix",
+    title = "Mark `mymainmenu` as defined global."
+  } } ]]
+
+  local entries = {}
+  for i, e in ipairs(items) do
+    table.insert(entries,
+      ("%s. %s"):format(utils.ansi_codes.magenta(tostring(i)),
+        opts.format_item(e)))
+  end
+
+  local o = {}
+  o.fzf_opts = {
+    ['--no-multi']        = '',
+    ['--prompt']          = opts.prompt:gsub(":$", "> "),
+    ['--preview-window']  = 'hidden:right:0',
+  }
+
+  core.fzf_wrap(o, entries, function(selected)
+
+    local idx = selected and tonumber(selected[1]:match("^(%d+).")) or nil
+    on_choice(items[idx], idx)
+
+  end)()
+
+end
+
+return M