From f4ad775e44e4febe8d80d43ed858e7850f379acb Mon Sep 17 00:00:00 2001
From: Andres Monge <aemonge@users.noreply.github.com>
Date: Fri, 29 Dec 2023 18:47:01 +0000
Subject: [PATCH] Ask what to map

Only map normal or insert if configured
---
 lua/llm/keymaps.lua | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/lua/llm/keymaps.lua b/lua/llm/keymaps.lua
index 1356aed..443a4fe 100644
--- a/lua/llm/keymaps.lua
+++ b/lua/llm/keymaps.lua
@@ -29,15 +29,19 @@ function M.setup()
 
   local accept_keymap = config.get().accept_keymap
   local dismiss_keymap = config.get().dismiss_keymap
+  local map_normal = config.get().normal_mappings
+  local map_insert = config.get().insert_mappings
 
-  vim.keymap.set("i", accept_keymap, accept_suggestion, { expr = true })
-
-  vim.keymap.set("n", accept_keymap, accept_suggestion, { expr = true })
-
-  vim.keymap.set("i", dismiss_keymap, dismiss_suggestion, { expr = true })
-
-  vim.keymap.set("n", dismiss_keymap, dismiss_suggestion, { expr = true })
+  if map_insert then
+    vim.keymap.set("i", accept_keymap, accept_suggestion, { expr = true })
+    vim.keymap.set("i", dismiss_keymap, dismiss_suggestion, { expr = true })
+  end
 
+  if map_normal then
+    vim.keymap.set("n", accept_keymap, accept_suggestion, { expr = true })
+    vim.keymap.set("n", dismiss_keymap, dismiss_suggestion, { expr = true })
+  end
+  
   M.setup_done = true
 end