New version: created v3

config/nvim/lua/core/ui.lua

@@ -0,0 +1,143 @@
+local ui = {}
+
+function ui.nui_input()
+  -- Set up NUI for UI Input
+  -- From: https://github.com/MunifTanjim/nui.nvim/wiki/vim.ui#vimuiinput
+  local input_ui
+  vim.ui.input = function(opts, on_confirm)
+    local Input = require "nui.input"
+    local event = require("nui.utils.autocmd").event
+    if input_ui then
+      -- ensure single ui.input operation
+      vim.api.nvim_err_writeln "busy: another input is pending!"
+      return
+    end
+
+    local function on_done(value)
+      if input_ui then
+        -- if it's still mounted, unmount it
+        input_ui:unmount()
+      end
+      -- pass the input value
+      on_confirm(value)
+      -- indicate the operation is done
+      input_ui = nil
+    end
+
+    local border_top_text = opts.prompt or "[Input]"
+    local default_value = opts.default
+
+    input_ui = Input({
+      relative = "cursor",
+      position = {
+        row = 1,
+        col = 0,
+      },
+      size = {
+        -- minimum width 20
+        width = math.max(20, type(default_value) == "string" and #default_value or 0),
+      },
+      border = {
+        style = "rounded",
+        highlight = "Normal",
+        text = {
+          top = border_top_text,
+          top_align = "left",
+        },
+      },
+      win_options = {
+        winhighlight = "Normal:Normal",
+      },
+    }, {
+      default_value = default_value,
+      on_close = function()
+        on_done(nil)
+      end,
+      on_submit = function(value)
+        on_done(value)
+      end,
+    })
+
+    input_ui:mount()
+
+    -- cancel operation if cursor leaves input
+    input_ui:on(event.BufLeave, function()
+      on_done(nil)
+    end, { once = true })
+
+    -- cancel operation if <Esc> is pressed
+    input_ui:map("n", "<Esc>", function()
+      on_done(nil)
+    end, { noremap = true, nowait = true })
+  end
+end
+
+function ui.telescope_select()
+  -- Telescope UI selection
+  -- From: https://github.com/stevearc/dressing.nvim/blob/master/lua/dressing/select/telescope.lua
+  vim.ui.select = vim.schedule_wrap(function(items, opts, on_choice)
+    local themes = require "telescope.themes"
+    local actions = require "telescope.actions"
+    local state = require "telescope.actions.state"
+    local pickers = require "telescope.pickers"
+    local finders = require "telescope.finders"
+    local conf = require("telescope.config").values
+
+    if opts.format_item then
+      local format_item = opts.format_item
+      opts.format_item = function(item)
+        return tostring(format_item(item))
+      end
+    else
+      opts.format_item = tostring
+    end
+
+    local entry_maker = function(item)
+      local formatted = opts.format_item(item)
+      return {
+        display = formatted,
+        ordinal = formatted,
+        value = item,
+      }
+    end
+
+    local picker_opts = themes.get_dropdown()
+
+    pickers.new(picker_opts, {
+      prompt_title = opts.prompt,
+      previewer = false,
+      finder = finders.new_table {
+        results = items,
+        entry_maker = entry_maker,
+      },
+      sorter = conf.generic_sorter(opts),
+      attach_mappings = function(prompt_bufnr)
+        actions.select_default:replace(function()
+          local selection = state.get_selected_entry()
+          actions.close(prompt_bufnr)
+          if not selection then
+            -- User did not select anything.
+            on_choice(nil, nil)
+            return
+          end
+          local idx = nil
+          for i, item in ipairs(items) do
+            if item == selection.value then
+              idx = i
+              break
+            end
+          end
+          on_choice(selection.value, idx)
+        end)
+
+        actions.close:enhance { post = function() end }
+
+        return true
+      end,
+    }):find()
+  end)
+end
+
+for ui_addition, enabled in pairs(astronvim.user_plugin_opts("ui", { nui_input = true, telescope_select = true })) do
+  astronvim.conditional_func(ui[ui_addition], enabled)
+end