local function augroup(name)
    return vim.api.nvim_create_augroup('nvim' .. name, { clear = true })
end

-- Strip trailing spaces before write
vim.api.nvim_create_autocmd({ 'BufWritePre' }, {
    group = augroup('strip_space'),
    pattern = { '*' },
    callback = function()
        vim.cmd([[ %s/\s\+$//e ]])
    end,
})

-- Check if we need to reload the file when it changed
vim.api.nvim_create_autocmd({ 'FocusGained', 'TermClose', 'TermLeave' }, {
    group = augroup('checktime'),
    command = 'checktime',
})

-- Highlight on yank
vim.api.nvim_create_autocmd('TextYankPost', {
    group = augroup('highlight_yank'),
    callback = function()
        vim.highlight.on_yank()
    end,
})

-- resize splits if window got resized
vim.api.nvim_create_autocmd({ 'VimResized' }, {
    group = augroup('resize_splits'),
    callback = function()
        vim.cmd('tabdo wincmd =')
    end,
})

-- go to last loc when opening a buffer
vim.api.nvim_create_autocmd('BufReadPost', {
    group = augroup('last_loc'),
    callback = function()
        local mark = vim.api.nvim_buf_get_mark(0, '"')
        local lcount = vim.api.nvim_buf_line_count(0)
        if mark[1] > 0 and mark[1] <= lcount then
            pcall(vim.api.nvim_win_set_cursor, 0, mark)
        end
    end,
})

-- close some filetypes with <q>
vim.api.nvim_create_autocmd('FileType', {
    group = augroup('close_with_q'),
    pattern = {
        'Jaq',
        'PlenaryTestPopup',
        'Avante',
        'AvanteInput',
        'AvanteSelectedFiles',
        'fugitive',
        'git',
        'help',
        'lir',
        'lspinfo',
        'man',
        'netrw',
        'notify',
        'qf',
        'spectre_panel',
        'startuptime',
        'tsplayground',
    },
    callback = function(event)
        vim.bo[event.buf].buflisted = false
        vim.keymap.set('n', 'q', '<cmd>close<cr>', { buffer = event.buf, silent = true })
    end,
})

-- wrap and check for spell in text filetypes
vim.api.nvim_create_autocmd('FileType', {
    group = augroup('wrap_spell'),
    pattern = { 'gitcommit', 'markdown' },
    callback = function()
        vim.opt_local.wrap = true
        vim.opt_local.spell = true
    end,
})

-- Auto create dir when saving a file, in case some intermediate directory does not exist
vim.api.nvim_create_autocmd({ 'BufWritePre' }, {
    group = augroup('auto_create_dir'),
    callback = function(event)
        local file = vim.loop.fs_realpath(event.match) or event.match
        vim.fn.mkdir(vim.fn.fnamemodify(file, ':p:h'), 'p')
    end,
})

-- Set arb filetype
vim.api.nvim_create_autocmd({ 'BufEnter' }, {
    group = augroup('set_file_type'),
    pattern = { '*.arb' },
    command = 'setfiletype arb',
})

-- Disable format options
vim.api.nvim_create_autocmd('FileType', {
    group = augroup('disable_formatoptions'),
    pattern = '*',
    callback = function()
        vim.opt_local.formatoptions:remove({ 'c', 'r', 'o' })
    end,
})

local function enable_autoformat()
    vim.api.nvim_create_autocmd({ 'BufWritePre' }, {
        group = augroup('autoformat'),
        pattern = { '*' },
        callback = function()
            vim.lsp.buf.format()
        end,
    })
end

enable_autoformat()

vim.api.nvim_create_user_command('WriteNoFormat', function()
    -- Temporarily disable the autoformat autocmd
    vim.api.nvim_del_augroup_by_name('keycodex_autoformat')
    vim.cmd('write')
    -- Re-enable the autoformat autocmd
    enable_autoformat()
end, {})

-- Macro recording notifications
vim.api.nvim_create_autocmd('RecordingEnter', {
    callback = function()
        vim.notify('Macro recording started', vim.log.levels.INFO, {
            title = 'Macro',
            timeout = 2000,
        })
        -- Print to command line for debugging
        vim.cmd("echo 'Recording macro...'")
    end,
})

vim.api.nvim_create_autocmd('RecordingLeave', {
    callback = function()
        vim.notify('Macro recording stopped', vim.log.levels.INFO, {
            title = 'Macro',
            timeout = 2000,
        })
        -- Print to command line for debugging
        vim.cmd("echo 'Macro recording stopped'")
    end,
})

-- Function to show macro replay notification
local function show_macro_replay(count)
    local message = count > 1 and string.format('Replaying macro %d times', count) or 'Replaying macro'
    vim.notify(message, vim.log.levels.INFO, {
        title = 'Macro',
        timeout = 2000,
    })
end

-- Function to clear all macros
local function clear_all_macros()
    -- Clear all registers from a to z
    for i = 97, 122 do -- ASCII values for 'a' to 'z'
        vim.fn.setreg(string.char(i), '')
    end
    -- Clear all registers from 0 to 9
    for i = 48, 57 do -- ASCII values for '0' to '9'
        vim.fn.setreg(string.char(i), '')
    end
    vim.notify('All macros cleared', vim.log.levels.INFO, {
        title = 'Macro',
        timeout = 2000,
    })
end

-- Create keymaps for macro replay with notifications
vim.keymap.set('n', '@', function()
    local count = vim.v.count1
    show_macro_replay(count)
    return '@'
end, { expr = true })

-- Add keybinding to clear all macros
vim.keymap.set('n', '<leader>oc', clear_all_macros, { desc = 'Clear all macros' })