local M = {}

function M.setup()
    local wk = require("which-key")

    wk.setup({
        preset = "helix",
    })

    wk.add({
        { mode = { "n", "v" } },
        { "<leader>c",        group = "code" },
        { "<leader>f",        group = "file/find" },
        { "<leader>g",        group = "git" },
        { "<leader>q",        group = "quit/session" },
        { "<leader>s",        group = "search" },
        { "<leader>u",        group = "ui" },
        { "<leader>x",        group = "diagnostics/quickfix" },
        { "[",                group = "prev" },
        { "]",                group = "next" },
        { "g",                group = "goto" },
        { "gs",               group = "surround" },
        { "z",                group = "fold" },
        { "gx",               desc = "Open with system app" },
    })

    vim.keymap.set("n", "<leader>?", function()
        wk.show({ global = false })
    end, { desc = "Buffer Keymaps (which-key)" })

    vim.keymap.set("n", "<c-w><space>", function()
        wk.show({ keys = "<c-w>", loop = true })
    end, { desc = "Window Hydra Mode (which-key)" })
end

return M