From b48f00b6aa5c58bad02faf99ac7bf3bc934147f4 Mon Sep 17 00:00:00 2001
From: Janis Hutz <development@janishutz.com>
Date: Mon, 20 Oct 2025 09:53:06 +0200
Subject: [PATCH] [Neogen] Fix

---
 nvim/lua/plugins/util/neogen.lua | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/nvim/lua/plugins/util/neogen.lua b/nvim/lua/plugins/util/neogen.lua
index 1737bf9..ad98297 100755
--- a/nvim/lua/plugins/util/neogen.lua
+++ b/nvim/lua/plugins/util/neogen.lua
@@ -2,7 +2,16 @@ return {
     "danymat/neogen",
     config = function()
         local opts = { noremap = true, silent = true, desc = "Generate docs" }
-        vim.api.nvim_set_keymap("n", "<leader>gd", ":lua require('neogen').generate()<CR>", opts)
+        vim.keymap.set("n", "<leader>gd", ":lua require('neogen').generate()<CR>", opts)
+
+        opts.desc = "Generate function docs"
+        vim.keymap.set("n", "<leader>gg", ":lua require('neogen').generate( { type = 'func' } )<CR>", opts)
+
+        opts.desc = "Generate type docs"
+        vim.keymap.set("n", "<leader>gt", ":lua require('neogen').generate( { type = 'type' } )<CR>", opts)
+
+        opts.desc = "Generate class docs"
+        vim.keymap.set("n", "<leader>gc", ":lua require('neogen').generate( { type = 'class' } )<CR>", opts)
         require("neogen").setup({
             snippet_engine = "luasnip",
         })