conding for #362

Author: jayli

lua/easycomplete/cmdline.lua

@@ -1,34 +1,26 @@
-local Util = require "easycomplete.util"
-local log = Util.log
-local console = Util.console
+local util = require "easycomplete.util"
+local log = util.log
+local console = util.console
 local normal_chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRST0123456789#$_"
 -- cmdline_start_cmdpos 是不带偏移量的，偏移量只给 pum 定位用
 local cmdline_start_cmdpos = 0
 local zizz_flag = 0
 local zizz_timer = vim.loop.new_timer()
-local M = {}
+local this = {}
 
-function pum_complete(start_col, menu_items)
+function this.pum_complete(start_col, menu_items)
   vim.fn["easycomplete#pum#complete"](start_col, menu_items)
 end
 
-function pum_close()
+function this.pum_close()
   vim.fn["easycomplete#pum#close"]()
 end
 
-function get_typing_word()
+function this.get_typing_word()
   return vim.fn['easycomplete#util#GetTypingWord']()
-  -- -- 获取当前行的文本
-  -- local current_line = string.sub(vim.fn.getcmdline(),1,vim.fn.getcmdpos())
-  -- -- 使用 gmatch 迭代所有单词（假设单词由空格分隔）
-  -- local last_word = ""
-  -- for word in string.gmatch(current_line, "[%w%-]+") do
-  --   last_word = word
-  -- end
-  -- return last_word
 end
 
-function calculate_sign_and_linenr_width()
+function this.calculate_sign_and_linenr_width()
   local width = 0
 
   -- 检查是否有 sign 列
@@ -47,41 +39,41 @@ function calculate_sign_and_linenr_width()
   return width
 end
 
-function flush()
+function this.flush()
   vim.g.easycomplete_cmdline_pattern = ""
   vim.g.easycomplete_cmdline_typing = 0
   cmdline_start_cmdpos = 0
-  pum_close()
+  this.pum_close()
 end
 
-function pum_selected()
+function this.pum_selected()
   return vim.fn['easycomplete#pum#CompleteCursored']()
 end
 
-function pum_selected_item()
+function this.pum_selected_item()
   return vim.fn['easycomplete#pum#CursoredItem']()
 end
 
-function M.select_next()
+function this.select_next()
   vim.fn['easycomplete#pum#next']()
-  zizz()
-  local new_whole_word = get_tab_returing_opword()
+  this.zizz()
+  local new_whole_word = this.get_tab_returing_opword()
   return new_whole_word
 end
 
-function M.select_prev()
+function this.select_prev()
   vim.fn['easycomplete#pum#prev']()
-  zizz()
-  local new_whole_word = get_tab_returing_opword()
+  this.zizz()
+  local new_whole_word = this.get_tab_returing_opword()
   return new_whole_word
 end
 
-function get_tab_returing_opword()
+function this.get_tab_returing_opword()
   local backing_count = vim.fn.getcmdpos() - cmdline_start_cmdpos
   local oprator_str = string.rep("\b", backing_count)
   local new_whole_word = ""
-  if pum_selected() then
-    local item = pum_selected_item()
+  if this.pum_selected() then
+    local item = this.pum_selected_item()
     local word = item.word
     new_whole_word = oprator_str .. word
   else
@@ -90,7 +82,7 @@ function get_tab_returing_opword()
   return new_whole_word
 end
 
-local function bind_cmdline_event()
+function this.bind_cmdline_event()
   local augroup = vim.api.nvim_create_augroup('CustomCmdlineComplete', { clear = true })
 
   vim.api.nvim_create_autocmd("CmdlineEnter", {
@@ -109,17 +101,17 @@ local function bind_cmdline_event()
     })
 
   vim.keymap.set("c", "<Tab>", function()
-    return M.select_next()
+    return this.select_next()
   end, { expr = true, noremap = true })
 
   vim.keymap.set("c", "<S-Tab>", function()
-    return M.select_prev()
+    return this.select_prev()
   end, { expr = true, noremap = true })
 
   vim.api.nvim_create_autocmd("CmdlineLeave", {
       group = augroup,
       callback = function()
-        flush()
+        this.flush()
       end
     })
   vim.on_key(function(keys, _)
@@ -130,7 +122,7 @@ local function bind_cmdline_event()
     vim.g.easycomplete_cmdline_typing = 1
     vim.defer_fn(function()
       vim.schedule(function()
-        local ok, ret = pcall(cmdline_handler, keys, key_str)
+        local ok, ret = pcall(this.cmdline_handler, keys, key_str)
         if not ok then
           print(ret)
         end
@@ -139,7 +131,7 @@ local function bind_cmdline_event()
   end)
 end
 
-function normalize_list(arr, word)
+function this.normalize_list(arr, word)
   if #arr == 0 then return arr end
   local ret = {}
   for index, value in ipairs(arr) do
@@ -153,36 +145,97 @@ function normalize_list(arr, word)
   return vim.fn['easycomplete#util#CompleteMenuFilter'](ret, word, 500)
 end
 
-function cmdline_handler(keys, key_str)
+function this.cmdline_handler(keys, key_str)
   if vim.g.easycomplete_cmdline_pattern == "" then
     return
   end
-  if zizzing() then return end
+  if this.zizzing() then return end
   local cmdline = vim.fn.getcmdline()
   cmdline_start_cmdpos = 0
   if string.byte(key_str) == 9 then
     -- console("Tab 键被按下")
   elseif string.byte(key_str) == 32 then
     -- console("空格键被按下")
-    pum_close()
+    this.pum_close()
   elseif string.byte(key_str) == 8 or string.byte(key_str) == 128 then
     -- console("退格键被按下")
-    pum_close()
+    this.pum_close()
   elseif string.byte(key_str) == 13 then
     -- console("回车键被按下")
-    pum_close()
+    this.pum_close()
   else
     -- console("其他键被按下: " .. keys)
-    local word = get_typing_word()
-    local menu_items = vim.fn.getcompletion(word, "function")
-    local start_col = vim.fn.getcmdpos() - calculate_sign_and_linenr_width() - #word
+    local word = this.get_typing_word()
+    local start_col = vim.fn.getcmdpos() - this.calculate_sign_and_linenr_width() - #word
     cmdline_start_cmdpos = vim.fn.getcmdpos() - #word
-    pum_complete(start_col, normalize_list(menu_items, word))
+    local menu_items = this.get_cmp_items()
+    if menu_items == nil or #menu_items == 0 then
+      this.pum_close()
+    else
+      this.pum_complete(start_col, this.normalize_list(menu_items, word))
+    end
   end
   vim.cmd("redraw")
 end
 
-function zizz()
+function this.get_all_commands()
+  local all_commands = {}
+  local tmp_items = vim.fn.getcompletion("", "command")
+  for index, value in ipairs(tmp_items) do
+    if string.match(value, "^[_a-zA-Z0-9]") then
+      table.insert(all_commands, value)
+    end
+  end
+  return all_commands
+end
+
+-- 路由
+function this.get_cmp_items()
+  if this.typing_cmd() then
+    return this.get_all_commands()
+  end
+end
+
+function this.trim_before(str)
+  if str == "" then return "" end
+  return string.gsub(str, "^%s*(.-)$", "%1")
+end
+
+-- 正在输入命令
+function this.typing_cmd()
+  local cmdline_all = vim.fn.getcmdline()
+  local cmdline_typed = this.trim_before(string.sub(cmdline_all, 1, vim.fn.getcmdpos()))
+  if string.find(cmdline_typed, "%s") then
+    return false
+  else
+    return true
+  end
+end
+
+-- 正在输入buffer
+-- 正在输入路径
+-- 正在输入函数
+-- 正在输入文本
+-- 正在输入color
+-- 正在输入...
+
+function this.get_completion_type()
+  local cmdline = vim.fn.getcmdline()
+  local cmd_name = string.match(cmdline, "([^%s]+)")
+  if cmd_name == nil then
+    return nil
+  end
+  local cmd_type = ""
+  for key, value in pairs(this.commands_type) do
+    if cmd_name == key then
+      cmd_type = value
+      break
+    end
+  end
+  return cmd_type
+end
+
+function this.zizz()
   if zizz_flag > 0 then
     zizz_timer:stop()
     zizz_flag = 0
@@ -193,24 +246,130 @@ function zizz()
   zizz_flag = 1
 end
 
-function zizzing()
+function this.zizzing()
   if zizz_flag == 1 then
     return true
   else
     return false
   end
 end
 
+this.commands_type = {
+  -- File completion
+  edit = 'file',
+  read = 'file',
+  write = 'file',
+  saveas = 'file',
+  source = 'file',
+  split = 'file',
+  vsplit = 'file',
+  tabedit = 'file',
+  diffsplit = 'file',
+  diffpatch = 'file',
+  explore = 'file',
+  lexplore = 'file',
+  sexplore = 'file',
+  vexplore = 'file',
+  find = 'file_in_path',
+  sfind = 'file_in_path',
+  tabfind = 'file_in_path',
+  -- Directory completion
+  cd = 'dir',
+  lcd = 'dir',
+  tcd = 'dir',
+  chdir = 'dir',
+  -- Buffer completion
+  buffer = 'buffer',
+  bdelete = 'buffer',
+  bwipeout = 'buffer',
+  bnext = 'buffer',
+  bprevious = 'buffer',
+  bfirst = 'buffer',
+  blast = 'buffer',
+  sbuffer = 'buffer',
+  sball = 'buffer',
+  diffthis = 'diff_buffer',
+  diffoff = 'diff_buffer',
+  diffupdate = 'diff_buffer',
+  -- Command completion
+  command = 'command',
+  delcommand = 'command',
+  -- Option completion
+  set = 'option',
+  setlocal = 'option',
+  setglobal = 'option',
+  -- Help completion
+  help = 'help',
+  -- Expression completion
+  substitute = 'expression',
+  global = 'expression',
+  vglobal = 'expression',
+  let = 'expression',
+  echo = 'expression',
+  -- Tag completion
+  tag = 'tag',
+  stag = 'tag',
+  tselect = 'tag',
+  tjump = 'tag',
+  tlast = 'tag',
+  tnext = 'tag',
+  tprev = 'tag',
+  tunmenu = 'tag',
+  -- Argument completion
+  args = 'arglist',
+  argadd = 'file',
+  argdelete = 'file',
+  argdo = 'file',
+  -- User completion (for user-defined functions/commands)
+  ['function'] = 'function',
+  delfunction = 'function',
+  -- Mapping completion
+  map = 'mapping',
+  noremap = 'mapping',
+  unmap = 'mapping',
+  nmap = 'mapping',
+  vmap = 'mapping',
+  imap = 'mapping',
+  cmap = 'mapping',
+  nunmap = 'mapping',
+  vunmap = 'mapping',
+  iunmap = 'mapping',
+  cunmap = 'mapping',
+  -- Autocmd completion
+  autocmd = 'event',
+  augroup = 'augroup',
+  doautocmd = 'event',
+  doautoall = 'event',
+  -- Shell command completion
+  terminal = 'shellcmd',
+  ['!'] = 'shellcmd',
+  -- Misc
+  ['='] = 'lua',
+  colorscheme = 'color',
+  compiler = 'compiler',
+  filetype = 'filetype',
+  highlight = 'highlight',
+  history = 'history',
+  lua = 'lua',
+  messages = 'messages',
+  packadd = 'packadd',
+  register = 'register',
+  runtime = 'runtime',
+  sign = 'sign',
+  syntax = 'syntax',
+  user = 'user',
+}
+
 
-function M.init_once()
+function this.init_once()
   -- TODO here -----------------------------
   if true then return end
   console(1)
   -- TODO here -----------------------------
   vim.g.easycomplete_cmdline_pattern = ""
   vim.g.easycomplete_cmdline_typing = 0
-  bind_cmdline_event()
+  this.bind_cmdline_event()
 end
 
 
-return M
+return this