close_if_last_window

[joaomendoncaa]

I was getting frustrated when I tried to exit neovim with a codecompanion buffer opened — so I mimicked the neo-tree implementation of closing the editor in case a codecompanion buffer is the last window. This might be useful to enshrine in the plugin as a setup opt-in (or a default). Thoughts?

local close_if_last_window = function()
    local current_win = vim.api.nvim_get_current_win()
    if vim.api.nvim_win_get_config(current_win).relative ~= '' then
        return
    end

    local current_tab = vim.api.nvim_get_current_tabpage()
    local normal_windows = vim.tbl_filter(function(win)
        return vim.api.nvim_win_get_config(win).relative == ''
    end, vim.api.nvim_tabpage_list_wins(current_tab))

    local window_count = #normal_windows
    if window_count ~= 1 and window_count ~= 2 then
        return
    end

    local function get_window_filetypes(windows)
        local types = {}
        for _, win in ipairs(windows) do
            local bufnr = vim.api.nvim_win_get_buf(win)
            table.insert(types, vim.bo[bufnr].filetype)
        end
        return types
    end

    local filetypes = get_window_filetypes(normal_windows)

    local function should_close()
        if window_count == 1 then
            return filetypes[1] == 'codecompanion'
        end

        local has_companion = filetypes[1] == 'codecompanion' or filetypes[2] == 'codecompanion'
        local has_neotree = filetypes[1] == 'neo-tree' or filetypes[2] == 'neo-tree'

        return has_companion and has_neotree
    end

    if should_close() then
        vim.cmd 'qa!'
    end
end

vim.api.nvim_create_autocmd('WinEnter', {
    callback = close_if_last_window,
})