-- 窗口位置
-- 宽度： 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)

local prompt_bufid = vim.api.nvim_create_buf(false, true)
local 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 = 'Input',
  title_pos = 'center',
  -- noautocmd = true,
})

local result_bufid = vim.api.nvim_create_buf(false, true)
local 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,
})

local job = require('spacevim.api.job')

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

vim.api.nvim_create_autocmd({ 'TextChangedI' }, {
  group = augroup,
  buffer = prompt_bufid,
  callback = function(ev)
    local text = vim.api.nvim_buf_get_lines(prompt_bufid, 0, 1, false)[1]
    if text ~= '' then
      local grep_cmd = {
        'rg',
        '--no-heading',
        '--color=never',
        '--with-filename',
        '--line-number',
        '--column',
        '-g',
        '!.git',
        '-e',
        text,
        '.',
      }

      job.start(grep_cmd, {
        on_stdout = function(id, data)
          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
        end,
      })
    else
      vim.api.nvim_buf_set_lines(result_bufid, 0, -1, false, {})
    end
  end,
})

vim.api.nvim_create_autocmd({ 'TextChangedI' }, {
  group = augroup,
  buffer = prompt_bufid,
  callback = function(ev)
    local text = vim.api.nvim_buf_get_lines(prompt_bufid, 0, 1, false)[1]
    if text ~= '' then
      local grep_cmd = {
        'rg',
        '--no-heading',
        '--color=never',
        '--with-filename',
        '--line-number',
        '--column',
        '-g',
        '!.git',
        '-e',
        text,
        '.',
      }

      vim.system(grep_cmd, {
        stdout = function(err, data)
          vim.api.nvim_buf_set_lines(result_bufid, 0, -1, false, vim.split(data, '\n'))
        end,
      })
    end
  end,
})

-- 使用 Esc 关闭整个界面
vim.keymap.set('i', '<Esc>', 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 })

-- 搜索结果行转换成文件名、光标位置
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
-- 使用回车键打开光标所在的搜索结果，同时关闭界面
vim.keymap.set('i', '<Enter>', function()
  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('edit ' .. filename)
  vim.api.nvim_win_set_cursor(0, { linenr, colum })
end, { buffer = prompt_bufid })

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

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

-- 高亮文件名及位置

vim.fn.matchadd(
  'Comment',
  [[\([A-Z]:\)\?[^:]*:\d\+:\(\d\+:\)\?]],
  10,
  -1,
  { window = result_winid }
)

vim.api.nvim_set_option_value('number', false, { win = prompt_winid })
vim.api.nvim_set_option_value('relativenumber', false, { win = prompt_winid })
local extns = vim.api.nvim_create_namespace('floatgrep_ext')
vim.api.nvim_buf_set_extmark(prompt_bufid, extns, 0, 0, {
  sign_text = '>',
  sign_hl_group = 'Error',
})

vim.api.nvim_buf_set_extmark(
  0,
  vim.api.nvim_create_namespace('test_extmark'),
  28,
  3,
  { end_row = 28, end_col = 5, hl_group = 'TODO' }
)

vim.api.nvim_buf_set_extmark(4, vim.api.nvim_create_namespace("test_extmark"), 47, 3, {
	end_row = 47,
	end_col = 5,
	hl_group = "TODO",
	virt_text = { { "This is ", "Comment" }, { "hello", "Number" }, { " extmarks", "TODO" } },
})

local augroup = vim.api.nvim_create_augroup('winbar.nvim', {
  clear = true,
})
vim.api.nvim_create_autocmd({ 'BufWinEnter' }, {
  group = augroup,
  pattern = '*',
  callback = function(e)
    redraw_winbar()
  end,
})

function M.setup(opts)
    config.set_options(opts)
    local winbar = require('winbar.winbar')

    winbar.init()
    if opts.enabled == true then
        vim.api.nvim_create_autocmd({ 'DirChanged', 'CursorMoved', 'BufWinEnter', 'BufFilePost', 'InsertEnter', 'BufWritePost' }, {
            callback = function()
                winbar.show_winbar()
            end
        })
    end
end

local function redraw_winbar()
  local file_name = vim.fn.expand('%:t')

  if file_name == '' then
    return
  end

  local value = '%#SpaceVim_winbar#' .. file_name .. ' %#SpaceVim_winbar_Normal#' .. default_conf.winbar_seperator .. '%#Normal#'

  vim.api.nvim_set_option_value('winbar', value, { scope = 'local' })
end

local file_name = vim.fn.expand('%:t')
local ft = vim.filetype.match({ filename = file_name })
for _, v in ipairs(excluded_filetypes) do
  if v == ft then
    return
  end
end

local excluded_regex = [[\.txt$]]
local re = vim.regex(excluded_regex)
local file_name = vim.fn.expand('%:t')
for _, v in ipairs(excluded_filetypes) do
  if re:match(file_name) then
    return
  end
end