sqlpilot.nvim

A neovim plugin for navigating database schemas and executing custom queries across multiple RDBMS and NoSQL DBs

Why?

• ❤️ KISS:
  Designed to be loose couple interaction with a variety type of DBs' CLI utility and sqlpilot.nvim.
• 🚀 Fast:
  Async execution of query without blocking editing mode.
• 🌟 Featured:
  1. Bundled with common used queries supporting Oracle, MySQL, PostgreSQL, Aurora MySQL/PostgreSQL, MongoDB, Cassandra
  2. Text expander in Insert mode for dozens of SQL keywords.
  3. Underline query and CLI command executed with actual parameter values can be easily obtained for troubleshooting.
  4. Display current connected DBs and snapshot of execution time in previous run result buffer window together with lualine.
• 💎 Extensibility:
  Fully customized scripts in language of your choice to extend support of other DBs with command line interface(CLI).

Installation

prerequisite which-key.nvim

sqlpilot.nvim is integrated with which-key.nvim to enable/disable common used queries shortcut keys dynamically for supported DBs in normal and visual modes.

with packer.nvim

1. obtain a copy of resources/sql_conn.json, change appropriately and stored in ~/projects/xxxx/sqlpilot.nvim/resources/sql_conn.json (example path)

2. use sqlpilot.nvim

   use {
     "rsdot/sqlpilot.nvim",
     requires = "folke/which-key.nvim",
     config = function()
       require("sqlpilot").setup {
         sql_conn  = vim.fn.glob("~").."/projects/xxxx/sqlpilot.nvim/resources/sql_conn.json", -- required
       }
     end
   }

with LazyVim

1. obtain a copy of resources/sql_conn.json, change appropriately and stored in ~/projects/xxxx/sqlpilot.nvim/resources/sql_conn.json (example path)

2. use sqlpilot.nvim

   return {
     "rsdot/sqlpilot.nvim",
     dependencies = "folke/which-key.nvim",
     opts = {
       sql_conn  = vim.fn.expand("~").."/projects/xxxx/sqlpilot.nvim/resources/sql_conn.json", -- required
     },
   }

Configuration

1. Integration with lualine to display current connected DBs and snapshot of execution time in previous run result buffer window

   Example of setup with lualine

   local sqlpilot = require("sqlpilot")
   
   local function db_conn_string()
     if sqlpilot.sqlpilot_dict_command_param.alias ~= nil and
       sqlpilot.sqlpilot_dict_command_param.dbname ~= nil then
       local dbmstype = sqlpilot.sqlpilot_dict_command_param.dbms
       local dbmsicon = ''
       if dbmstype == 'mssql' then
         dbmsicon = ''
       elseif dbmstype == 'mysql' then
         dbmsicon = ''
       elseif dbmstype == 'postgresql' then
         dbmsicon = ''
       elseif dbmstype == 'cassandra' then
         dbmsicon = ''
       elseif dbmstype == 'mongodb' then
         dbmsicon = ''
       elseif dbmstype == 'oracle' then
         dbmsicon = ''
       end
   
       return '⟛ ' .. dbmsicon .. sqlpilot.sqlpilot_dict_command_param.alias .. '.' .. sqlpilot.sqlpilot_dict_command_param.dbname
     else
   	return "󰆼"
     end
   end
   
   local dbconn_prod = {
     'dbconn_prod',
     cons_enabled = true,
     icon = '󰆼',
     fmt = db_conn_string,
     color = {fg = '#a14f6d', gui='italic,bold'},
     cond = function()
       return sqlpilot.sqlpilot_dict_command_param.alias ~= nil and sqlpilot.sqlpilot_dict_command_param.isprod == 1
     end,
   }
   
   local dbconn_nonprod = {
     'dbconn_nonprod',
     cons_enabled = true,
     icon = '󰆼',
     fmt = db_conn_string,
     color = {fg = '#63c259', gui='italic'},
     cond = function()
       return sqlpilot.sqlpilot_dict_command_param.alias ~= nil and sqlpilot.sqlpilot_dict_command_param.isprod == 0
     end,
   }
   
   local dbresult = {
     'dbresult',
     fmt = function()
       return vim.b.sqlpilot_display_result ~= nil and vim.b.sqlpilot_display_result or ''
     end,
     color = {fg = '#5f5f87'},
   }
   
   require("lualine").setup({
     ...
     sections = {
       ...
       lualine_c = { dbresult },
       lualine_x = { dbconn_prod, dbconn_nonprod },
       ...
     },
     inactive_sections = {
       ...
       lualine_c = { dbresult },
       lualine_x = {},
       ...
     },
     ...
   })

Usage examples

Text expander

In Insert mode

type	substituted with
;s	SELECT
;f	FROM
;w	WHERE
;c	COUNT(1)
;o	ORDER BY
;l	LIMIT 1
;g	GROUP BY
;h	HAVING
;i	INSERT INTO
;v	VALUES(
;i	INNER JOIN
;u	UPDATE
;N	IS NOT NULL
;U	IS NULL
;b	BEGIN
;e	END

Executing ad-hoc queries

<space>f; to setup DB connection, in Visual Line mode, select lines, then <space>ff to run query async. Result would show up in the buffer window once they are ready, not blocking current editing

mysql example

SELECT now();
SELECT SLEEP(10);
SELECT now();

View table index definition

In Normal mode, move cursor on top of a table name, then <space>f4 to view index defintion of the table under the cursor

Extensibility

Tweak CLI run commands

obtain a copy of resources/sql_run.json, change appropriately and stored in ~/projects/xxxx/sqlpilot.nvim/resources/sql_conn.json (example path)

for each dbms, program/scripts (with full path and excutable) referenced by following commands attributes can be tweaked.

postgresql as example

  ...
  "postgresql": {
    "sql_csv": {
      "command": "PGCONNECT_TIMEOUT=3 PGPASSWORD='{password}' psql -P pager=off -v ON_ERROR_STOP=1 -h {dbserver} -U {loginname} -p {port} -d {dbname} -f {infile} > {outfile} 2>&1",
      "header": [
        "COPY",
        "("
      ],
      "footer": [
        ")",
        "TO STDOUT",
        "WITH CSV HEADER"
      ]
    },
    "sql_cli": {
      "command": "PGCONNECT_TIMEOUT=3 PGPASSWORD='{password}' psql -P pager=off -v ON_ERROR_STOP=1 -h {dbserver} -U {loginname} -p {port} -d {dbname} -f {infile} > {outfile} 2>&1"
    },
    "sql_ddl": {
      "command": "{scriptpath}/ddl_postgresql.sh -h {dbserver} -p {port} -U {loginname} -P '{password}' -d {dbname} -o {objectname} > {objectfile} 2>&1"
    }
  },
  ...

Modify queries

obtain a copy of resources/sql_query.json, change appropriately and stored in ~/projects/xxxx/sqlpilot.nvim/resources/sql_query.json (example path)

Put all together

use sqlpilot.nvim in packer

use {
  "rsdot/sqlpilot.nvim",
  config = function()
    require("sqlpilot").setup {
      sql_conn  = vim.fn.glob("~").."/projects/xxxx/sqlpilot.nvim/resources/sql_conn.json",  -- required
      sql_query = vim.fn.glob("~").."/projects/xxxx/sqlpilot.nvim/resources/sql_query.json", -- overwrite default
      sql_run   = vim.fn.glob("~").."/projects/xxxx/sqlpilot.nvim/resources/sql_run.json",   -- overwrite default
      registers = {
        cmd   = "y", -- default vim register to store last used cmd                          -- change to other register if needed
        query = "z"  -- default vim register to store last used query                        -- change to other register if needed
      },
      which_key_registers = {
        normal = "D", -- default which-key normal mode key register                          -- change to avoid conflicting with existing
        visual = "D", -- default which-key visual mode key register                          -- change to avoid conflicting with existing
      }
    }
  end
}

use sqlpilot.nvim in LazyVim

   return {
     "rsdot/sqlpilot.nvim",
     dependencies = "folke/which-key.nvim",
     opts = {
       sql_conn  = vim.fn.glob("~").."/projects/xxxx/sqlpilot.nvim/resources/sql_conn.json", -- required
       sql_query = vim.fn.glob("~").."/projects/xxxx/sqlpilot.nvim/resources/sql_query.json", -- overwrite default
       sql_run   = vim.fn.glob("~").."/projects/xxxx/sqlpilot.nvim/resources/sql_run.json",   -- overwrite default
       registers = {
         cmd   = "y", -- default vim register to store last used cmd                          -- change to other register if needed
         query = "z"  -- default vim register to store last used query                        -- change to other register if needed
       },
       which_key_registers = {
         normal = "D", -- default which-key normal mode key register                          -- change to avoid conflicting with existing
         visual = "D", -- default which-key visual mode key register                          -- change to avoid conflicting with existing
       }
     },
   }

• In Normal mode, "yp will paste content from register y, which stores last used cmd with actual parameter values
• In Normal mode, "zp will paste content from register z, which stores last used query