add db actions

mosheavni · Dec 28, 2023 · c2b0628 · c2b0628
1 parent ee0a607
commit c2b0628
Showing 1 changed file with 37 additions and 1 deletion.
diff --git a/.config/nvim/lua/plugins/databases.lua b/.config/nvim/lua/plugins/databases.lua
@@ -1,7 +1,43 @@
 local M = {
   {
     'tpope/vim-dadbod',
-    cmd = { 'DB', 'DBUI' },
+    cmd = {
+      'DB',
+      'DBUI',
+      'DBUIClose',
+      'DBUIToggle',
+      'DBUIFindBuffer',
+      'DBUIRenameBuffer',
+      'DBUIAddConnection',
+      'DBUILastQueryInfo',
+      'DBUIHideNotifications',
+      'DBCompletionClearCache',
+    },
+    init = function()
+      require('user.menu').add_actions('Database', {
+        ['Add Connection'] = function()
+          vim.cmd 'DBUIAddConnection'
+        end,
+        ['Rename Buffer'] = function()
+          vim.cmd 'DBUIRenameBuffer'
+        end,
+        ['Find Buffer'] = function()
+          vim.cmd 'DBUIFindBuffer'
+        end,
+        ['Last Query Info'] = function()
+          vim.cmd 'DBUILastQueryInfo'
+        end,
+        ['Toggle UI'] = function()
+          vim.cmd 'DBUIToggle'
+        end,
+        ['Close UI'] = function()
+          vim.cmd 'DBUIClose'
+        end,
+        ['Hide UI Notifications'] = function()
+          vim.cmd 'DBUIHideNotifications'
+        end,
+      })
+    end,
     dependencies = {
       {
         'kristijanhusak/vim-dadbod-ui',