cmdline-mode completion mvp

Author: phanen

lua/fzf-lua/actions.lua

@@ -1243,14 +1243,19 @@ end
 ---@param selected string[]
 ---@param opts fzf-lua.config.Resolved
 M.complete = function(selected, opts)
+  local mode = opts.__CTX.mode
+  local cursor = opts.__CTX.cursor
   if not selected[1] then
-    if opts.__CTX.mode == "i" then
+    if mode == "i" then
       vim.cmd [[noautocmd lua vim.api.nvim_feedkeys('i', 'n', true)]]
+    elseif mode == "c" then
+      vim.cmd [[noautocmd lua vim.api.nvim_feedkeys(':', 'n', true)]]
+      require("fzf-lua.ctx").set_cursor(mode, cursor)
     end
     return
   end
   -- cusror col is 0-based
-  local col = opts.__CTX.cursor[2] + 1
+  local col = cursor[2] + 1
   local newline, newcol
   if type(opts.complete) == "function" then
     newline, newcol = opts.complete(selected, opts, opts.__CTX.line, col)
@@ -1260,10 +1265,21 @@ M.complete = function(selected, opts)
     newline = line:sub(1, col) .. selected[1] .. after
     newcol = col + #selected[1]
   end
-  vim.api.nvim_set_current_line(newline or opts.__CTX.line)
-  vim.api.nvim_win_set_cursor(0, { opts.__CTX.cursor[1], newcol or col })
-  if opts.__CTX.mode == "i" then
+  if mode ~= "c" then
+    vim.api.nvim_set_current_line(newline or opts.__CTX.line)
+    vim.api.nvim_win_set_cursor(0, { cursor[1], newcol or col })
+  end
+  if mode == "i" then
     vim.cmd [[noautocmd lua vim.api.nvim_feedkeys('a', 'n', true)]]
+  elseif mode == "c" then
+    vim.api.nvim_create_autocmd("CmdlineEnter", {
+      once = true,
+      callback = function()
+        vim.fn.setcmdline(newline or opts.__CTX.line)
+        require("fzf-lua.ctx").set_cursor(mode, { cursor[1], cursor[2] + (newcol or col) - col })
+      end
+    })
+    vim.cmd [[noautocmd lua vim.api.nvim_feedkeys(':', 'n', true)]]
   end
 end
 

lua/fzf-lua/core.lua

@@ -365,6 +365,7 @@ M.fzf = function(contents, opts)
     opts.fzf_opts["--preview-window"] = "hidden:right:0"
   end
 
+  if opts.__CTX.mode == "c" then vim.api.nvim_feedkeys(vim.keycode("<c-c>"), "n", false) end
   -- Create the window before we calculate the preview window layout
   local fzf_bufnr = fzf_win:create()
 

lua/fzf-lua/ctx.lua

@@ -47,17 +47,18 @@ M.refresh = function(opts)
       -- not to resume or a different picker, i.e. hide files and open buffers
       or winobj and winobj:hidden()
   then
+    local mode = vim.api.nvim_get_mode().mode
     ctx = {
-      mode = vim.api.nvim_get_mode().mode,
+      mode = mode,
       bufnr = vim.api.nvim_get_current_buf(),
       bname = vim.api.nvim_buf_get_name(0),
       winid = vim.api.nvim_get_current_win(),
       last_winid = vim.fn.win_getid(vim.fn.winnr("#")),
       alt_bufnr = vim.fn.bufnr("#"),
       tabnr = vim.fn.tabpagenr(),
       tabh = vim.api.nvim_win_get_tabpage(0),
-      cursor = vim.api.nvim_win_get_cursor(0),
-      line = vim.api.nvim_get_current_line(),
+      cursor = M.get_cursor(mode),
+      line = M.get_line(mode),
       curtab_wins = (function()
         local ret = {}
         local wins = vim.api.nvim_tabpage_list_wins(0)
@@ -120,4 +121,41 @@ M.set_info = function(x)
   info = x
 end
 
+-- modified from blink.cmp
+function M.get_mode(mode)
+  mode = mode or vim.api.nvim_get_mode().mode
+  return (mode == "c" and "cmdline")
+      or (mode == "t" and "term")
+      or (vim.fn.getcmdwintype() ~= "" and "cmdwin")
+      or "default"
+end
+
+function M.get_cursor(mode)
+  mode = M.get_mode(mode)
+  return mode == "cmdline" and { 1, vim.fn.getcmdpos() - 1 } or vim.api.nvim_win_get_cursor(0)
+end
+
+function M.set_cursor(mode, cursor)
+  mode = M.get_mode(mode)
+  if vim.tbl_contains({ "default", "term", "cmdwin" }, mode) then
+    return vim.api.nvim_win_set_cursor(0, cursor)
+  end
+
+  assert(mode == "cmdline", "Unsupported mode for setting cursor: " .. mode)
+  assert(cursor[1] == 1, "Cursor must be on the first line in cmdline mode")
+  vim.fn.setcmdpos(cursor[2])
+end
+
+function M.get_line(mode, num)
+  if M.get_mode(mode) == "cmdline" then
+    assert(num == nil or num == 0,
+      "Cannot get line number " .. tostring(num) .. " in cmdline mode. Only 0 is supported")
+    return vim.fn.getcmdline()
+  end
+
+  -- This method works for normal buffers and the terminal prompt
+  if num == nil then num = M.get_cursor()[1] - 1 end
+  return vim.api.nvim_buf_get_lines(0, num, num + 1, false)[1]
+end
+
 return M