vim.keymap.set('', '<Leader>z', '<cmd>Goyo<CR><CR>', {})
vim.keymap.set('x', '<Leader>z', '<cmd>Goyo<CR>', {})

local goyo_enter = function()
  vim.opt.showmode = false
  vim.opt.showcmd = false
  vim.opt.scrolloff = 999
  vim.cmd('Limelight')
end

local goyo_leave = function()
  vim.opt.showmode = true
  vim.opt.showcmd = true
  vim.opt.scrolloff = 5
  vim.cmd('Limelight!')
end

local goyo_augroup = vim.api.nvim_create_augroup('Goyo', { clear = true })
vim.api.nvim_create_autocmd('User',
  {
    pattern = 'GoyoEnter',
    group = goyo_augroup,
    callback = goyo_enter,
  })
vim.api.nvim_create_autocmd('User',
  {
    pattern = 'GoyoLeave',
    group = goyo_augroup,
    callback = goyo_leave,
  })