From 6b3f3d9140699c87b6f6b64471333812e361dc3b Mon Sep 17 00:00:00 2001
From: Myzel394 <50424412+Myzel394@users.noreply.github.com>
Date: Sat, 7 Oct 2023 11:58:53 +0200
Subject: [PATCH] feat: Add set mapping

---
 lua/easytables/options.lua | 78 +++++++++++++++++++++++++++++++++++++-
 lua/easytables/window.lua  |  2 +
 2 files changed, 79 insertions(+), 1 deletion(-)

diff --git a/lua/easytables/options.lua b/lua/easytables/options.lua
index 3981d44..1f01f55 100644
--- a/lua/easytables/options.lua
+++ b/lua/easytables/options.lua
@@ -50,7 +50,83 @@ local options = {
                 vertical = "|",
             }
         }
-    }
+    },
+    set_mappings = function(buf)
+        vim.api.nvim_buf_set_keymap(
+            buf,
+            "n",
+            "<Left>",
+            ":JumpLeft<CR>",
+            {}
+        )
+        vim.api.nvim_buf_set_keymap(
+            buf,
+            "n",
+            "<S-Left>",
+            ":SwapWithLeftCell<CR>",
+            {}
+        )
+
+        vim.api.nvim_buf_set_keymap(
+            buf,
+            "n",
+            "<Right>",
+            ":JumpRight<CR>",
+            {}
+        )
+        vim.api.nvim_buf_set_keymap(
+            buf,
+            "n",
+            "<S-Right>",
+            ":SwapWithRightCell<CR>",
+            {}
+        )
+
+        vim.api.nvim_buf_set_keymap(
+            buf,
+            "n",
+            "<Up>",
+            ":JumpUp<CR>",
+            {}
+        )
+        vim.api.nvim_buf_set_keymap(
+            buf,
+            "n",
+            "<S-Up>",
+            ":SwapWithUpperCell<CR>",
+            {}
+        )
+
+        vim.api.nvim_buf_set_keymap(
+            buf,
+            "n",
+            "<Down>",
+            ":JumpDown<CR>",
+            {}
+        )
+        vim.api.nvim_buf_set_keymap(
+            buf,
+            "n",
+            "<S-Down>",
+            ":SwapWithLowerCell<CR>",
+            {}
+        )
+
+        vim.api.nvim_buf_set_keymap(
+            buf,
+            "n",
+            "<Tab>",
+            ":JumpToNextCell<CR>",
+            {}
+        )
+        vim.api.nvim_buf_set_keymap(
+            buf,
+            "n",
+            "<S-Tab>",
+            ":JumpToPrevCell<CR>",
+            {}
+        )
+    end
 }
 
 -- You can ignore everything below this line
diff --git a/lua/easytables/window.lua b/lua/easytables/window.lua
index 86dca53..588e54e 100644
--- a/lua/easytables/window.lua
+++ b/lua/easytables/window.lua
@@ -382,6 +382,8 @@ function M:register_listeners()
         end,
         {}
     )
+
+    o.options.set_mappings(self.prompt_buffer)
 end
 
 return M