-- ShaDa registers saving/reading support
local t = require('test.testutil')
local n = require('test.functional.testnvim')()
local t_shada = require('test.functional.shada.testutil')

local nvim_command, fn, eq = n.command, n.fn, t.eq
local reset, clear = t_shada.reset, t_shada.clear
local expect_exit = n.expect_exit

local setreg = function(name, contents, typ)
  if type(contents) == 'string' then
    contents = { contents }
  end
  fn.setreg(name, contents, typ)
end

local getreg = function(name)
  return {
    fn.getreg(name, 1, 1),
    fn.getregtype(name),
  }
end

describe('ShaDa support code', function()
  before_each(reset)
  after_each(clear)

  it('is able to dump and restore registers and their type', function()
    setreg('c', { 'd', 'e', '' }, 'c')
    setreg('l', { 'a', 'b', 'cde' }, 'l')
    setreg('b', { 'bca', 'abc', 'cba' }, 'b3')
    expect_exit(nvim_command, 'qall')
    reset()
    eq({ { 'd', 'e', '' }, 'v' }, getreg('c'))
    eq({ { 'a', 'b', 'cde' }, 'V' }, getreg('l'))
    eq({ { 'bca', 'abc', 'cba' }, '\0223' }, getreg('b'))
  end)

  it('does not dump registers with zero <', function()
    nvim_command("set shada='0,<0")
    setreg('c', { 'd', 'e', '' }, 'c')
    setreg('l', { 'a', 'b', 'cde' }, 'l')
    setreg('b', { 'bca', 'abc', 'cba' }, 'b3')
    expect_exit(nvim_command, 'qall')
    reset()
    eq({ {}, '' }, getreg('c'))
    eq({ {}, '' }, getreg('l'))
    eq({ {}, '' }, getreg('b'))
  end)

  it('does restore registers with zero <', function()
    setreg('c', { 'd', 'e', '' }, 'c')
    setreg('l', { 'a', 'b', 'cde' }, 'l')
    setreg('b', { 'bca', 'abc', 'cba' }, 'b3')
    expect_exit(nvim_command, 'qall')
    reset("set shada='0,<0")
    eq({ { 'd', 'e', '' }, 'v' }, getreg('c'))
    eq({ { 'a', 'b', 'cde' }, 'V' }, getreg('l'))
    eq({ { 'bca', 'abc', 'cba' }, '\0223' }, getreg('b'))
  end)

  it('does not dump registers with zero "', function()
    nvim_command('set shada=\'0,\\"0')
    setreg('c', { 'd', 'e', '' }, 'c')
    setreg('l', { 'a', 'b', 'cde' }, 'l')
    setreg('b', { 'bca', 'abc', 'cba' }, 'b3')
    expect_exit(nvim_command, 'qall')
    reset()
    eq({ {}, '' }, getreg('c'))
    eq({ {}, '' }, getreg('l'))
    eq({ {}, '' }, getreg('b'))
  end)

  it('does restore registers with zero "', function()
    setreg('c', { 'd', 'e', '' }, 'c')
    setreg('l', { 'a', 'b', 'cde' }, 'l')
    setreg('b', { 'bca', 'abc', 'cba' }, 'b3')
    expect_exit(nvim_command, 'qall')
    reset('set shada=\'0,\\"0')
    eq({ { 'd', 'e', '' }, 'v' }, getreg('c'))
    eq({ { 'a', 'b', 'cde' }, 'V' }, getreg('l'))
    eq({ { 'bca', 'abc', 'cba' }, '\0223' }, getreg('b'))
  end)

  it('does dump registers with zero ", but non-zero <', function()
    nvim_command('set shada=\'0,\\"0,<50')
    setreg('c', { 'd', 'e', '' }, 'c')
    setreg('l', { 'a', 'b', 'cde' }, 'l')
    setreg('b', { 'bca', 'abc', 'cba' }, 'b3')
    expect_exit(nvim_command, 'qall')
    reset()
    eq({ { 'd', 'e', '' }, 'v' }, getreg('c'))
    eq({ { 'a', 'b', 'cde' }, 'V' }, getreg('l'))
    eq({ { 'bca', 'abc', 'cba' }, '\0223' }, getreg('b'))
  end)

  it('does limit number of lines according to <', function()
    nvim_command("set shada='0,<2")
    setreg('o', { 'd' }, 'c')
    setreg('t', { 'a', 'b', 'cde' }, 'l')
    expect_exit(nvim_command, 'qall')
    reset()
    eq({ { 'd' }, 'v' }, getreg('o'))
    eq({ {}, '' }, getreg('t'))
  end)

  it('does limit number of lines according to "', function()
    nvim_command('set shada=\'0,\\"2')
    setreg('o', { 'd' }, 'c')
    setreg('t', { 'a', 'b', 'cde' }, 'l')
    expect_exit(nvim_command, 'qall')
    reset()
    eq({ { 'd' }, 'v' }, getreg('o'))
    eq({ {}, '' }, getreg('t'))
  end)

  it('does limit number of lines according to < rather then "', function()
    nvim_command('set shada=\'0,\\"2,<3')
    setreg('o', { 'd' }, 'c')
    setreg('t', { 'a', 'b', 'cde' }, 'l')
    setreg('h', { 'abc', 'acb', 'bac', 'bca', 'cab', 'cba' }, 'b3')
    expect_exit(nvim_command, 'qall')
    reset()
    eq({ { 'd' }, 'v' }, getreg('o'))
    eq({ { 'a', 'b', 'cde' }, 'V' }, getreg('t'))
    eq({ {}, '' }, getreg('h'))
  end)

  it('dumps and loads register correctly with utf-8 contents', function()
    reset()
    setreg('e', { '«' }, 'c')
    expect_exit(nvim_command, 'qall')
    reset()
    eq({ { '«' }, 'v' }, getreg('e'))
  end)

  it('dumps and loads history correctly with 8-bit single-byte', function()
    reset()
    -- \171 is U+00AB LEFT-POINTING DOUBLE ANGLE QUOTATION MARK in latin1
    setreg('e', { '\171«' }, 'c')
    expect_exit(nvim_command, 'qall')
    reset()
    eq({ { '\171«' }, 'v' }, getreg('e'))
  end)

  it("has a blank unnamed register if it wasn't set and register 0 is empty", function()
    setreg('1', { 'one' }, 'c')
    setreg('2', { 'two' }, 'c')
    setreg('a', { 'a' }, 'c')
    expect_exit(nvim_command, 'qall')
    reset()
    eq({ {}, '' }, getreg('0'))
    eq({ { 'one' }, 'v' }, getreg('1'))
    eq({ {}, '' }, getreg('"'))
    eq({ { 'a' }, 'v' }, getreg('a'))
  end)

  it("defaults the unnamed register to register 0 if it wasn't set", function()
    setreg('0', { 'zero' }, 'c')
    setreg('1', { 'one' }, 'c')
    setreg('2', { 'two' }, 'c')
    expect_exit(nvim_command, 'qall')
    reset()
    eq({ { 'zero' }, 'v' }, getreg('0'))
    eq({ { 'one' }, 'v' }, getreg('1'))
    eq({ { 'zero' }, 'v' }, getreg('"'))
  end)

  it('remembers which register was the unnamed register when loading', function()
    setreg('0', { 'zero' }, 'c')
    setreg('1', { 'one' }, 'cu')
    setreg('2', { 'two' }, 'c')
    expect_exit(nvim_command, 'qall')
    reset()
    eq({ { 'zero' }, 'v' }, getreg('0'))
    eq({ { 'one' }, 'v' }, getreg('1'))
    eq({ { 'one' }, 'v' }, getreg('"'))
  end)
end)