From 6fd2c88024447b7eed20ca5461be784df07a3787 Mon Sep 17 00:00:00 2001
From: Folke Lemaitre <folke.lemaitre@gmail.com>
Date: Thu, 6 Jun 2024 13:14:04 +0200
Subject: [PATCH] feat(extras): added extra for neogen. Closes #2881

---
 lua/lazyvim/plugins/extras/coding/neogen.lua | 35 ++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 lua/lazyvim/plugins/extras/coding/neogen.lua

diff --git a/lua/lazyvim/plugins/extras/coding/neogen.lua b/lua/lazyvim/plugins/extras/coding/neogen.lua
new file mode 100644
index 00000000..88f7dd54
--- /dev/null
+++ b/lua/lazyvim/plugins/extras/coding/neogen.lua
@@ -0,0 +1,35 @@
+return {
+  "danymat/neogen",
+  cmd = "Neogen",
+  keys = {
+    {
+      "<leader>cn",
+      function()
+        require("neogen").generate()
+      end,
+      desc = "Generate Annotations (Neogen)",
+    },
+  },
+  opts = function(_, opts)
+    if opts.snippet_engine ~= nil then
+      return
+    end
+
+    local map = {
+      ["LuaSnip"] = "luasnip",
+      ["nvim-snippy"] = "snippy",
+      ["vim-vsnip"] = "vsnip",
+    }
+
+    for plugin, engine in pairs(map) do
+      if LazyVim.has(plugin) then
+        opts.snippet_engine = engine
+        return
+      end
+    end
+
+    if vim.snippet then
+      opts.snippet_engine = "nvim"
+    end
+  end,
+}