--- your custom UI

local function on_ui_update(name, data)
  -- logic
end


require('plug').setup({
  bundle_dir = 'D:/bundle_dir',
  max_processes = 5, -- max number of processes used for nvim-plug job
  base_url = 'https://github.com',
  ui = on_ui_update, -- default ui is notify, use `default` for split window UI
})

vim.o.number = false         -- 禁用行号，启用行号可以设置为 true
vim.o.relativenumber = false -- 禁用相对行号

vim.o = setmetatable({}, {
  __index = function(_, k)
    return api.nvim_get_option_value(k, {})
  end,
  __newindex = function(_, k, v)
    return api.nvim_set_option_value(k, v, {})
  end,
})

local function set_local(winid, bufnr, opts)
  for k, v in pairs(opts) do
    if vim.api.nvim_get_option_info2(k, {}).scope == 'win' then
      vim.api.nvim_set_option_value(k, v, { win = winid })
    elseif vim.api.nvim_get_option_info2(k, {}).scope == 'buf' then
      vim.api.nvim_set_option_value(k, v, { buf = bufnr })
    else
      -- skip global opt
    end
  end
end
local winid, bufnr = open_plugin_float_win()
set_local(winid, bufnr, {
  number = false,
  relativenumber = false,
  filetype = 'java',
  bufhidden = 'wipe',
})

local function test1()
  print(1)
end
local function test2()
  print(2)
end
local group =
  vim.api.nvim_create_augroup('test_augroup_neovim', { clear = true })
local autocmd_id1 = vim.api.nvim_create_autocmd({ 'WinEnter' }, {
  group = group,
  pattern = { '*' },
  callback = test1,
})
local autocmd_id2 = vim.api.nvim_create_autocmd({ 'WinEnter' }, {
  group = group,
  pattern = { '*' },
  callback = test2,
})

vim.api.nvim_del_autocmd(autocmd_id2)

vim.api.nvim_create_user_command('SayHello', function(opt)
  print('hello')
end, {
  nargs = '*',
  bang = true,
})

vim.api.nvim_create_user_command('Test', function(opt)
  print(opt.bang)
end, {
  nargs = '*',
  bang = true,
})

vim.api.nvim_create_user_command('Gina', function(opt)
  if not opt.bang then
    print('it is Gina')
  elseif opt.bang and opt.fargs[1] == '!' then
    print('it is Gina!!')
  else
    print('it is Gina!')
  end
end, {
  nargs = '*',
  bang = true,
})

vim.api.nvim_create_user_command('Test1', function(opt)
end, {
  nargs = 0,
})
vim.api.nvim_create_user_command('Test2', function(opt)
  vim.print(opt.args)
  vim.print(opt.fargs)
end, {
  nargs = 1,
})
vim.api.nvim_create_user_command('Test3', function(opt)
  vim.print(opt.args)
  vim.print(opt.fargs)
end, {
  nargs = '*',
})
vim.api.nvim_create_user_command('Test4', function(opt)
  vim.print(opt.args)
  vim.print(opt.fargs)
end, {
  nargs = '+',
})
vim.api.nvim_create_user_command('Test5', function(opt)
  vim.print(opt.args)
  vim.print(opt.fargs)
end, {
  nargs = '?',
})

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 a = {1, 2}
vim.fn.add(a, 3)
vim.print(a) --- 输出是 {1, 2}

-- vim.fn.{func}(...)
---@nodoc
vim.fn = setmetatable({}, {
  --- @param t table<string,function>
  --- @param key string
  --- @return function
  __index = function(t, key)
    local _fn --- @type function
    if vim.api[key] ~= nil then
      _fn = function()
        error(string.format('Tried to call API function with vim.fn: use vim.api.%s instead', key))
      end
    else
      _fn = function(...)
        return vim.call(key, ...)
      end
    end
    t[key] = _fn
    return _fn
  end,
})

local a = { 1, 2 }
table.insert(a, 3)
vim.print(a) --- 输出是 {1, 2, 3}

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