SpaceVim/bundle/flygrep.nvim/lua/flygrep.lua

local M = {}
local conf = require('flygrep.config')
local job = require('spacevim.api.job')
local ok, cmp = pcall(require, 'cmp')
if not ok then
  vim.cmd('doautocmd InsertEnter')
  ok, cmp = pcall(require, 'cmp')
end

local grep_timer_id = -1
local grep_input = ''
local search_jobid = -1
local search_hi_id = -1
local fix_string = false

-- all buffers
local result_bufid = -1
local result_winid = -1
local prompt_bufid = -1
local prompt_winid = -1

local prompt_count_id
local extns = vim.api.nvim_create_namespace('floatgrep_ext')

local function update_result_count()
  local count = vim.api.nvim_buf_line_count(result_bufid)
  local line = vim.api.nvim_win_get_cursor(result_winid)[1]
  prompt_count_id = vim.api.nvim_buf_set_extmark(prompt_bufid, extns, 0, 0, {
    id = prompt_count_id,
    virt_text = { { string.format('%d/%d', line, count), 'Comment' } },
    virt_text_pos = 'right_align',
  })
end

local function build_grep_command()
  local cmd = {conf.command.execute}
  for _, v in ipairs(conf.command.default_opts) do
    table.insert(cmd, v)
  end
  if fix_string then table.insert(cmd, conf.command.fixed_string_opt) else table.insert(cmd, conf.command.expr_opt) end
  table.insert(cmd, grep_input)
  table.insert(cmd, '.')
  return cmd
end

local function grep_timer(t)
  vim.api.nvim_buf_set_lines(result_bufid, 0, -1, false, {})
  if prompt_count_id then
    pcall(vim.api.nvim_buf_del_extmark, prompt_bufid, extns, prompt_count_id)
    prompt_count_id = nil
  end
  search_jobid = job.start(build_grep_command(), {
    on_stdout = function(id, data)
      if id == search_jobid and vim.api.nvim_buf_is_valid(prompt_bufid) and vim.api.nvim_win_is_valid(prompt_winid) then
        if vim.fn.getbufline(result_bufid, 1)[1] == '' then
          vim.api.nvim_buf_set_lines(result_bufid, 0, -1, false, data)
        else
          vim.api.nvim_buf_set_lines(result_bufid, -1, -1, false, data)
        end
        update_result_count()
      end
    end,
  })
end

local function build_prompt_title()
  local t = {}
  table.insert(t, { ' FlyGrep ', 'FlyGrep_a' })
  table.insert(t, { '', 'FlyGrep_a_FlyGrep_b' })
  if not fix_string then
    table.insert(t, { ' expr ', 'FlyGrep_b' })
  else
    table.insert(t, { ' string ', 'FlyGrep_b' })
  end
  table.insert(t, { '', 'FlyGrep_b' })
  table.insert(t, { ' ' .. vim.fn.getcwd() .. ' ', 'FlyGrep_b' })
  table.insert(t, { '', 'FlyGrep_b_Normal' })
  -- return {{}, {}, {}}
  return t
end

local function toggle_fix_string()
  fix_string = not fix_string
  vim.cmd('doautocmd TextChangedI')
  local conf = vim.api.nvim_win_get_config(prompt_winid)
  conf.title = build_prompt_title()
  vim.api.nvim_win_set_config(prompt_winid, conf)
end

local function open_win()
  require('flygrep.highlight').def_higroup()
  -- 窗口位置
  -- 宽度： columns 的 80%
  local screen_width = math.floor(vim.o.columns * 0.8)
  -- 起始位位置： lines * 10%, columns * 10%
  local start_col = math.floor(vim.o.columns * 0.1)
  local start_row = math.floor(vim.o.lines * 0.1)
  -- 整体高度：lines 的 80%
  local screen_height = math.floor(vim.o.lines * 0.8)

  prompt_bufid = vim.api.nvim_create_buf(false, true)
  prompt_winid = vim.api.nvim_open_win(prompt_bufid, true, {
    relative = 'editor',
    width = screen_width,
    height = 1,
    col = start_col,
    row = start_row + screen_height - 3,
    focusable = true,
    border = 'rounded',
    title = build_prompt_title(),
    title_pos = 'left',
    -- noautocmd = true,
  })

  vim.api.nvim_set_option_value(
    'winhighlight',
    'NormalFloat:Normal,FloatBorder:WinSeparator',
    { win = prompt_winid }
  )
  vim.api.nvim_set_option_value('number', false, { win = prompt_winid })
  vim.api.nvim_set_option_value('relativenumber', false, { win = prompt_winid })
  vim.api.nvim_set_option_value('cursorline', false, { win = prompt_winid })
  vim.api.nvim_set_option_value('signcolumn', 'yes', { win = prompt_winid })
  vim.api.nvim_buf_set_extmark(prompt_bufid, extns, 0, 0, {
    sign_text = '>',
    sign_hl_group = 'Error',
  })

  result_bufid = vim.api.nvim_create_buf(false, true)
  result_winid = vim.api.nvim_open_win(result_bufid, false, {
    relative = 'editor',
    width = screen_width,
    height = screen_height - 5,
    col = start_col,
    row = start_row,
    focusable = false,
    border = 'rounded',
    -- title = 'Result',
    -- title_pos = 'center',
    -- noautocmd = true,
  })
  vim.api.nvim_set_option_value(
    'winhighlight',
    'NormalFloat:Normal,FloatBorder:WinSeparator',
    { win = result_winid }
  )
  vim.api.nvim_set_option_value('cursorline', true, { win = result_winid })
  cmp.setup.buffer({
    completion = {
      autocomplete = false,
    },
  })
  vim.cmd('noautocmd startinsert')

  local augroup = vim.api.nvim_create_augroup('floatgrep', {
    clear = true,
  })

  vim.api.nvim_create_autocmd({ 'TextChangedI' }, {
    group = augroup,
    buffer = prompt_bufid,
    callback = function(ev)
      grep_input = vim.api.nvim_buf_get_lines(prompt_bufid, 0, 1, false)[1]
      if grep_input ~= '' then
        pcall(vim.fn.matchdelete, search_hi_id, result_winid)
        pcall(vim.fn.timer_stop, grep_timer_id)
        search_hi_id = vim.fn.matchadd(conf.matched_higroup, grep_input, 10, -1, { window = result_winid })
        grep_timer_id = vim.fn.timer_start(conf.timeout, grep_timer, { ['repeat'] = 1 })
      else
        pcall(vim.fn.matchdelete, search_hi_id, result_winid)
        pcall(vim.fn.timer_stop, grep_timer_id)
        job.stop(search_jobid)
        search_jobid = 0
        vim.api.nvim_buf_set_lines(result_bufid, 0, -1, false, {})
      end
    end,
  })

  -- 使用 Esc/C-c 关闭整个界面
  for _, k in ipairs({ '<Esc>', '<C-c>' }) do
    vim.keymap.set('i', k, function()
      vim.cmd('noautocmd stopinsert')
      vim.api.nvim_win_close(prompt_winid, true)
      vim.api.nvim_win_close(result_winid, true)
    end, { buffer = prompt_bufid })
  end

  -- 搜索结果行转换成文件名、光标位置
  local function get_file_pos(line)
    local filename = vim.fn.fnameescape(vim.fn.split(line, [[:\d\+:]])[1])
    local linenr = vim.fn.str2nr(string.sub(vim.fn.matchstr(line, [[:\d\+:]]), 2, -2))
    local colum = vim.fn.str2nr(string.sub(vim.fn.matchstr(line, [[\(:\d\+\)\@<=:\d\+:]]), 2, -2))
    return filename, linenr, colum
  end
  -- 使用回车键打开光标所在的搜索结果，同时关闭界面
  local function open_item(cmd)
    vim.cmd('noautocmd stopinsert')
    -- 获取搜索结果光表行
    local line_number = vim.api.nvim_win_get_cursor(result_winid)[1]
    local filename, linenr, colum =
      get_file_pos(vim.api.nvim_buf_get_lines(result_bufid, line_number - 1, line_number, false)[1])
    vim.api.nvim_win_close(prompt_winid, true)
    vim.api.nvim_win_close(result_winid, true)
    vim.cmd(cmd .. ' ' .. filename)
    vim.api.nvim_win_set_cursor(0, { linenr, colum })
  end
  vim.keymap.set('i', '<Enter>', function()
    open_item('edit')
  end, { buffer = prompt_bufid })
  vim.keymap.set('i', '<C-v>', function()
    open_item('vsplit')
  end, { buffer = prompt_bufid })
  vim.keymap.set('i', '<C-s>', function()
    open_item('split')
  end, { buffer = prompt_bufid })
  vim.keymap.set('i', '<C-t>', function()
    open_item('tabedit')
  end, { buffer = prompt_bufid })

  -- 避免使用 jk 切换到 normal 模式
  -- https://github.com/neovim/neovim/discussions/32208
  -- vim.keymap.del('i', 'jk', {buffer = prompt_bufid})
  if vim.fn.hasmapto('j', 'i') == 1 then
    vim.keymap.set('i', 'j', 'j', {
      nowait = true,
      buffer = prompt_bufid,
    })
  end

  -- 使用 Tab/Shift-Tab 上下移动搜素结果
  vim.keymap.set('i', '<Tab>', function()
    local line_number = vim.api.nvim_win_get_cursor(result_winid)[1]
    pcall(vim.api.nvim_win_set_cursor, result_winid, { line_number + 1, 0 })
    update_result_count()
  end, { buffer = prompt_bufid })

  vim.keymap.set('i', '<S-Tab>', function()
    local line_number = vim.api.nvim_win_get_cursor(result_winid)[1]
    pcall(vim.api.nvim_win_set_cursor, result_winid, { line_number - 1, 0 })
    update_result_count()
  end, { buffer = prompt_bufid })
  vim.keymap.set('i', '<C-e>', function()
    toggle_fix_string()
    update_result_count()
  end, { buffer = prompt_bufid })

  -- 高亮文件名及位置
  vim.fn.matchadd(
    'Comment',
    [[\([A-Z]:\)\?[^:]*:\d\+:\(\d\+:\)\?]],
    11,
    -1,
    { window = result_winid }
  )
end

function M.open()
  open_win()
end

function M.setup(conf)
  require('flygrep.config').setup(conf)
end

return M